Refactor elemental-agent to use subcommands

Signed-off-by: Andrea Mazzotti <[email protected]>

Dockerfile.agent

@@ -8,7 +8,8 @@ WORKDIR /workspace
 # Copy the go source
 COPY go.mod go.mod
 COPY go.sum go.sum
-COPY cmd/agent/main.go cmd/agent/main.go
+COPY main.go main.go
+COPY cmd/agent/ cmd/agent/
 COPY api/ api/
 COPY internal/ internal/
 COPY pkg/ pkg/
@@ -24,7 +25,7 @@ RUN CGO_ENABLED=1 GOOS=${TARGETOS:-linux} GOARCH=${TARGETARCH} go build \
     -X github.com/rancher-sandbox/cluster-api-provider-elemental/internal/version.Version=$TAG  \
     -X github.com/rancher-sandbox/cluster-api-provider-elemental/internal/version.Commit=$COMMIT  \
     -X github.com/rancher-sandbox/cluster-api-provider-elemental/internal/version.CommitDate=$COMMITDATE"  \
-    -a -o elemental-agent cmd/agent/main.go
+    -a -o elemental-agent main.go
 
 # Build elemental-toolkit plugin
 RUN CGO_ENABLED=1 go build \

Makefile

@@ -116,13 +116,13 @@ test: manifests generate fmt vet envtest generate-mocks $(GINKGO) ## Run tests.
 ##@ Build
 .PHONY: build-agent
 build-agent: fmt vet ## Build agent binary for local architecture.
-	CGO_ENABLED=1 go build -ldflags '$(LDFLAGS)' -o bin/elemental-agent cmd/agent/main.go
+	CGO_ENABLED=1 go build -ldflags '$(LDFLAGS)' -o bin/elemental-agent main.go
 
 # This does depend on cross compilation library, for example: cross-aarch64-gcc13
 .PHONY: build-agent-all
 build-agent-all: fmt vet ## Build agent binary for all architectures.
-	CGO_ENABLED=1 GOOS=linux GOARCH=amd64 go build -ldflags '$(LDFLAGS)' -o bin/elemental_agent_linux_amd64 cmd/agent/main.go
-	CC=$(CROSS_COMPILER) CGO_ENABLED=1 GOOS=linux GOARCH=arm64 go build -ldflags '$(LDFLAGS)' -o bin/elemental_agent_linux_arm64 cmd/agent/main.go
+	CGO_ENABLED=1 GOOS=linux GOARCH=amd64 go build -ldflags '$(LDFLAGS)' -o bin/elemental_agent_linux_amd64 main.go
+	CC=$(CROSS_COMPILER) CGO_ENABLED=1 GOOS=linux GOARCH=arm64 go build -ldflags '$(LDFLAGS)' -o bin/elemental_agent_linux_arm64 main.go
 
 .PHONY: build-manager
 build-manager: manifests generate fmt vet ## Build manager binary.

cmd/agent/install.go

@@ -0,0 +1,29 @@
+package agent
+
+import (
+	"github.com/rancher-sandbox/cluster-api-provider-elemental/internal/agent/log"
+	"github.com/rancher-sandbox/cluster-api-provider-elemental/internal/agent/phase"
+	"github.com/spf13/cobra"
+)
+
+// installCmd represents the install command.
+var installCmd = &cobra.Command{
+	Use:   "install",
+	Short: "Installs the OS on this Elemental host",
+	Long:  "Installs the OS on this Elemental host",
+	Run: func(cmd *cobra.Command, args []string) {
+		log.Info("Initializing agent")
+		agentContext, err := InitAgent()
+		if err != nil {
+			log.Fatal(err, "Could not initialize agent")
+		}
+
+		log.Info("Installing host")
+		installationHandler := phase.NewInstallHandler(*agentContext)
+		installationHandler.Install()
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(installCmd)
+}