Rakefile

require "./gorake.rb"

def os
    case RUBY_PLATFORM
    when /linux/
      "linux"
    when /darwin/
      "darwin"
    else
      fail 'Unsupported OS'
    end
  end

desc "Setup dependencies"
task :deps do
  system("go mod download")
end

task :default => [:ci]

desc "Build Datadog Process agent"
task :build do
  case os
  when "windows"
    bin = "process-agent.exe"
  else
    bin = "process-agent"
  end
  if ENV["STS_DEV_CONTAINER"] != "true"
    sh "./prebuild-datadog-agent.sh -i"
  end
  go_build("github.com/StackVista/stackstate-process-agent/cmd/agent", {
    :cmd => "go build -o #{bin}",
    :race => ENV['GO_RACE'] == 'true',
    :add_build_vars => ENV['PROCESS_AGENT_ADD_BUILD_VARS'] != 'false',
    :embed_path => ENV['STACKSTATE_EMBEDDED_PATH'],
    :bpf => true
  })
end


desc "Run goderive to generate necessary go code"
task :derive do
  sh "go run github.com/awalterschulze/goderive@886b66b111a4 ./..."
end

desc "Run prebuild steps"
task :prebuild do
  sh "./prebuild-datadog-agent.sh -g"
end

desc "Run goderive to generate necessary go code (Windows)"
task :derive_win do
  system("go install github.com/awalterschulze/goderive@886b66b111a4")
  system("go generate ./...")
end

desc "Install Datadog Process agent"
task :install do
  case os
  when "windows"
    bin = "process-agent.exe"
  else
    bin = "process-agent"
  end    
  go_build("github.com/StackVista/stackstate-process-agent/agent", :cmd=> "go build -i -o $GOPATH/bin/#{bin}")
end

desc "Test Datadog Process agent"
task :test do
  if ENV["STS_DEV_CONTAINER"] != "true"
    sh "./prebuild-datadog-agent.sh -i"
  end
  go_test("$(go list ./...)", {
   :bpf => true,
   :embed_path => ENV['STACKSTATE_EMBEDDED_PATH'],
  })
end

desc "Test Datadog Process agent -- cmd"
task :cmdtest do
  cmd = "for /f %f in ('go list ./... ^| find /V \"vendor\"') do go test %f"
  sh cmd
end

task :vet do
  if ENV["STS_DEV_CONTAINER"] != "true"
    sh "./prebuild-datadog-agent.sh -i"
  end
  go_vet("$(go list ./...)", {
    :bpf => true,
    :embed_path => ENV['STACKSTATE_EMBEDDED_PATH'],
  })
end

task :imports do
  sh "go install golang.org/x/tools/cmd/goimports@latest"
  # goimports want file paths instead of package import paths. 
  # We want to use the exact file name because we want to exclude file generated by protobuf (`pb.go`).
  # packages = `go list -f '{{range .GoFiles}}{{$.Dir}}/{{.}}{{"\n"}}{{end}}' ./... | grep -v '/vendor/' | grep -v 'pb.go$'`.split("\n")
  packages = `go list -f '{{range .GoFiles}}{{$.Dir}}/{{.}}{{"\\n"}}{{end}}' ./... | grep -v '/vendor/' | grep -v 'pb.go$'`.split("\n")
  failure = false
  packages.each do |pkg|
    out = `goimports -d -w #{pkg}`
    errors = out.split("\n")
    if errors.length > 0
      # if there is at least one error we will fail at the end.
      failure = true
      errors.each do |error|
        $stderr.puts error
      end
    end
  end

  if failure
    fail
  end
end

task :lint do
  sh "go install github.com/mgechev/revive@v1.3.4"
  packages = `go list ./... | grep -v vendor`.split("\n")
  packages.each do |pkg|
    puts "revive -formatter stylish -config revive-recommended.toml #{pkg}"
    output = `revive -formatter stylish -config revive-recommended.toml #{pkg}`
    puts output
    if output != ""
      fail "Error during linting"
    end
  end
end

desc "Compile the protobuf files for the Process Agent"
task :protobuf do
  protocv = `bash -c "protoc --version"`.strip
  if protocv != 'libprotoc 3.6.1'
    fail "Requires protoc version 3.6.1"
  end

  gogo_path = get_go_module_path("github.com/gogo/protobuf")
  sketched_path = get_go_module_path("github.com/DataDog/sketches-go")

  sh "protoc proto/agent_payload.proto --proto_path=#{gogo_path} -I proto --gogofaster_out model/"
  sh "protoc proto/agent.proto --proto_path=#{gogo_path} --proto_path=#{sketched_path} -I proto --gogofaster_out model/"
end

desc "Process Agent CI script (imports, vet, etc)"
task :ci => [:deps, :imports, :vet, :lint, :test, :build]

desc "Process Agent local build"
task :local_build => [:prebuild, :build]

task :err do
  system("go install github.com/kisielk/errcheck")
  sh "errcheck github.com/StackVista/stackstate-process-agent"
end