update-datadog-dependency.sh

#!/bin/bash

# Okay, here it goes: The process agent depends on the datadog agent, which:
# - for some modules (EBPF) uses go:generate which needs to be run to even build the dependency.
# - for some other modules generate EBPF code, which we need but is not part of the regular go build
# This file makes sure those prebuild steps (go:generate and ebpf files) get built and installed.

set -e

DIR="${BASH_SOURCE%/*}"
if [ ! -d "$DIR" ]; then DIR="$PWD"; fi
if [ "$DIR" = "." ]; then DIR="$PWD"; fi

printUsage() {
  cat << USAGE

Usage: ./update-datadog-dependency.sh --local <path>|--branch <branch>|--help

This script will update go.mod to point to a different remote
  -l/--local           -- Use a local directory as dependency
  -b/--branch [branch] -- Use the given branch of github.com/StackVista/datadog-agent-upstream-for-process-agent as a dependency, defaults to stackstate-7.49.1
  -h/--help            -- Print this help page

USAGE
}

ACTION=""
BRANCH="stackstate-7.49.1"
while [ $# -gt 0 ]; do
  case $1 in
    -l|--local)
      shift
      go mod edit -replace "github.com/DataDog/datadog-agent=$1"
      go mod tidy
      exit 0
    ;;
    -b|--branch)
      shift
      BRANCH="${1}"
      echo "Using branch ${BRANCH}"
      go mod edit -replace "github.com/DataDog/datadog-agent=github.com/StackVista/datadog-agent-upstream-for-process-agent@${BRANCH}"
      go mod tidy
      exit 0
    ;;
    -h|--help)
      printUsage
      exit 0
    ;;
    *)
      echo "Unknown flag"
      printUsage
      exit 1
    ;;
  esac
done