```ruby
name "datadog-upgrade-helper"
default_version "master"
source github: "DataDog/dd-agent-windows-upgrade-helper"
env = {
  "GOPATH" => "#{Omnibus::Config.cache_dir}/src/#{name}",
}

if ohai["platform"] == "windows"
  gobin = "C:/Go/bin/go"
end

build do
  command "#{gobin} build && mv datadog-upgrade-helper #{install_dir}/bin/upgrade-helper", :env => env
end
```

Add ability to specify UPGRADE_HELPER_BRANCH in environment
user
```ruby
name "datadog-upgrade-helper"

source github: "DataDog/dd-agent-windows-upgrade-helper"
env = {
  "GOPATH" => "#{Omnibus::Config.cache_dir}/src/#{name}",
}
helper_branch = ENV["UPGRADE_HELPER_BRANCH"]
if helper_branch.nil? || helper_branch.empty?
  default_version "master"
else
  default_version helper_branch
end

if ohai["platform"] == "windows"
  gobin = "C:/Go/bin/go"
end

build do
  command "#{gobin} build && mv datadog-upgrade-helper #{install_dir}/bin/upgrade-helper", :env => env
end
```
assistant