feat: architecture detection (#106)

Signed-off-by: Leonardo Cecchi <[email protected]>
Signed-off-by: Marco Nenciarini <[email protected]>
Co-authored-by: Marco Nenciarini <[email protected]>

Author: leonardoce

protoc-gen-go-grpc/dagger/go.mod

@@ -3,7 +3,7 @@ module dagger/protoc-gen-go-grpc
 go 1.24.0
 
 require (
-	github.com/99designs/gqlgen v0.17.79
+	github.com/99designs/gqlgen v0.17.80
 	github.com/Khan/genqlient v0.8.1
 	github.com/vektah/gqlparser/v2 v2.5.30
 	golang.org/x/sync v0.17.0

protoc-gen-go-grpc/dagger/go.sum

@@ -1,5 +1,5 @@
-github.com/99designs/gqlgen v0.17.79 h1:RTsJZtdzcfROeWdt42NGMIabIbiBn69YyVmLEAuxtnA=
-github.com/99designs/gqlgen v0.17.79/go.mod h1:vgNcZlLwemsUhYim4dC1pvFP5FX0pr2Y+uYUoHFb1ig=
+github.com/99designs/gqlgen v0.17.80 h1:S64VF9SK+q3JjQbilgdrM0o4iFQgB54mVQ3QvXEO4Ek=
+github.com/99designs/gqlgen v0.17.80/go.mod h1:vgNcZlLwemsUhYim4dC1pvFP5FX0pr2Y+uYUoHFb1ig=
 github.com/Khan/genqlient v0.8.1 h1:wtOCc8N9rNynRLXN3k3CnfzheCUNKBcvXmVv5zt6WCs=
 github.com/Khan/genqlient v0.8.1/go.mod h1:R2G6DzjBvCbhjsEajfRjbWdVglSH/73kSivC9TLWVjU=
 github.com/andreyvit/diff v0.0.0-20170406064948-c7f18ee00883 h1:bvNMNQO63//z+xNgfBlViaCIJKLlCJ6/fmUseuG0wVQ=

protoc-gen-go-grpc/dagger/main.go

@@ -6,6 +6,7 @@ import (
 	"context"
 	"fmt"
 	"path"
+	"runtime"
 	"strings"
 
 	"dagger/protoc-gen-go-grpc/internal/dagger"
@@ -17,29 +18,43 @@ type ProtocGenGoGRPC struct {
 }
 
 func New(
-	// Custom image to use to run protoc.
-	// +optional
-	// renovate image: datasource=docker depName=golang versioning=docker
-	// +default="golang:1.24.5-bookworm"
+	ctx context.Context,
+// Custom image to use to run protoc.
+// +optional
+// renovate image: datasource=docker depName=golang versioning=docker
+// +default="golang:1.24.5-bookworm"
 	goImage string,
-	// +optional
-	// renovate: datasource=github-tags depName=protocolbuffers/protobuf versioning="regex:^v?(?<major>\\d+)\\.(?<minor>\\d+)$"
-	// +default="v30.2"
+// +optional
+// renovate: datasource=github-tags depName=protocolbuffers/protobuf versioning="regex:^v?(?<major>\\d+)\\.(?<minor>\\d+)$"
+// +default="v32.1"
 	protobufVersion string,
-	// +optional
-	// renovate: datasource=go depName=google.golang.org/protobuf/cmd/protoc-gen-go versioning=semver
-	// +default="v1.36.6"
+// +optional
+// renovate: datasource=go depName=google.golang.org/protobuf/cmd/protoc-gen-go versioning=semver
+// +default="v1.36.6"
 	protocGenGoVersion string,
-	// +optional
-	// renovate: datasource=go depName=google.golang.org/grpc/cmd/protoc-gen-go-grpc versioning=semver
-	// +default="v1.5.1"
+// +optional
+// renovate: datasource=go depName=google.golang.org/grpc/cmd/protoc-gen-go-grpc versioning=semver
+// +default="v1.5.1"
 	protocGenGoGRPCVersion string,
-) *ProtocGenGoGRPC {
-	protobufRelURL := fmt.Sprintf("https://github.com/protocolbuffers/protobuf/releases/download/%v/protoc-%v-linux-x86_64.zip",
-		protobufVersion, strings.TrimPrefix(protobufVersion, "v"))
-
+) (*ProtocGenGoGRPC, error) {
+	architecturesMap := map[string]string{
+		"arm64": "aarch_64",
+		"amd64": "x86_64",
+		"ppc":   "ppcle_64",
+		"s390x": "s390_64",
+	}
 	protobuf := dag.Container().
-		From(goImage).
+		From(goImage)
+
+	architecture := runtime.GOARCH
+	if remappedArchitecture, ok := architecturesMap[architecture]; ok {
+		architecture = remappedArchitecture
+	}
+
+	protobufRelURL := fmt.Sprintf("https://github.com/protocolbuffers/protobuf/releases/download/%v/protoc-%v-linux-%v.zip",
+		protobufVersion, strings.TrimPrefix(protobufVersion, "v"), architecture)
+
+	protobuf = protobuf.
 		WithExec([]string{"apt", "update"}).
 		WithExec([]string{"apt", "install", "-y", "unzip"}).
 		WithExec([]string{"curl", "-LO", protobufRelURL}).
@@ -54,7 +69,7 @@ func New(
 
 	return &ProtocGenGoGRPC{
 		Ctr: protobuf,
-	}
+	}, nil
 }
 
 // Container get the current container