feat(tools): support for arbitrary version download of libddwaf (#84)

Signed-off-by: Eliott Bouhana <[email protected]>

Author: eliottness

_tools/libddwaf-updater/go.mod

@@ -1,6 +1,6 @@
 module github.com/DataDog/go-libddwaf/libddwaf-updater
 
-go 1.18
+go 1.20
 
 require (
 	github.com/bitfield/script v0.22.0

_tools/libddwaf-updater/update.go

@@ -11,6 +11,7 @@ import (
 	"context"
 	_ "embed"
 	"errors"
+	"flag"
 	"fmt"
 	"io"
 	"os"
@@ -34,24 +35,35 @@ const (
 	goVersionUnsupported = "go1.23"
 )
 
+var (
+	forceFlag   *bool
+	versionFlag *string
+)
+
 func main() {
-	force := false
 
-	if len(os.Args) >= 2 {
-		force = os.Args[1] == "--force"
-	}
+	flag.Parse()
+
+	var (
+		release *github.RepositoryRelease
+		err     error
+		gh      = github.NewClient(nil)
+	)
 
-	gh := github.NewClient(nil)
+	if *versionFlag == "latest" {
+		release, _, err = gh.Repositories.GetLatestRelease(context.Background(), "DataDog", "libddwaf")
+	} else {
+		release, _, err = gh.Repositories.GetReleaseByTag(context.Background(), "DataDog", "libddwaf", *versionFlag)
+	}
 
-	release, _, err := gh.Repositories.GetLatestRelease(context.Background(), "DataDog", "libddwaf")
 	if err != nil {
 		panic(err)
 	}
 
 	version := *release.TagName
 	if version == currentVersion {
 		fmt.Printf("Already up-to-date with v%s\n", version)
-		if force {
+		if *forceFlag {
 			fmt.Println("--force is set, re-downloading assets anyway!")
 		} else {
 			return
@@ -71,7 +83,7 @@ func main() {
 	wg := sync.WaitGroup{}
 	wg.Add(len(targets))
 	for _, tgt := range targets {
-		go handleTarget(&wg, version, tgt, assets, force)
+		go handleTarget(&wg, version, tgt, assets, *forceFlag)
 	}
 
 	wg.Wait()
@@ -298,6 +310,14 @@ func (t target) embedSourceDirective() string {
 }
 
 func init() {
+
+	forceFlag = flag.Bool("force", false, "Force the download of assets even if the version is the same")
+	versionFlag = flag.String("version", "latest", "Force the download of assets for a specific version (by git tag), or 'latest' for the latest release")
+
+	if forceFlag == nil || versionFlag == nil {
+		panic("unexpected nil flag")
+	}
+
 	_, filename, _, _ := runtime.Caller(0)
 	dir := path.Dir(filename)
 	rootDir = path.Join(dir, "..", "..")