agent.go

package agent

import (
	"os"

	"github.com/sirupsen/logrus"
	"github.com/skevetter/devpod/cmd/agent/container"
	"github.com/skevetter/devpod/cmd/agent/workspace"
	"github.com/skevetter/devpod/cmd/flags"
	"github.com/skevetter/devpod/pkg/config"
	"github.com/skevetter/devpod/pkg/envfile"
	"github.com/skevetter/log"
	"github.com/spf13/cobra"
)

var AgentExecutedAnnotation = "loft.sh/agent-executed"

// NewAgentCmd returns a new root command.
func NewAgentCmd(globalFlags *flags.GlobalFlags) *cobra.Command {
	agentCmd := &cobra.Command{
		Use:   "agent",
		Short: "DevPod Agent",
		PersistentPreRunE: func(cobraCmd *cobra.Command, args []string) error {
			return AgentPersistentPreRunE(cobraCmd, args, globalFlags)
		},
		Hidden: true,
	}

	agentCmd.AddCommand(workspace.NewWorkspaceCmd(globalFlags))
	agentCmd.AddCommand(container.NewContainerCmd(globalFlags))
	agentCmd.AddCommand(NewDaemonCmd(globalFlags))
	agentCmd.AddCommand(NewContainerTunnelCmd(globalFlags))
	agentCmd.AddCommand(NewGitCredentialsCmd(globalFlags))
	agentCmd.AddCommand(NewGitSSHSignatureCmd(globalFlags))
	agentCmd.AddCommand(NewGitSSHSignatureHelperCmd(globalFlags))
	agentCmd.AddCommand(NewDockerCredentialsCmd(globalFlags))
	return agentCmd
}

func AgentPersistentPreRunE(
	cobraCmd *cobra.Command,
	args []string,
	globalFlags *flags.GlobalFlags,
) error {
	// get top level parent
	parent := cobraCmd
	for parent.Parent() != nil {
		parent = parent.Parent()
	}
	if parent.Annotations == nil {
		parent.Annotations = map[string]string{}
	}
	parent.Annotations[AgentExecutedAnnotation] = "true"

	if globalFlags.LogOutput == "json" {
		log.Default.SetFormat(log.JSONFormat)
	} else {
		log.Default.MakeRaw()
	}

	if globalFlags.Silent {
		log.Default.SetLevel(logrus.FatalLevel)
	} else if globalFlags.Debug {
		log.Default.SetLevel(logrus.DebugLevel)
	} else if os.Getenv(config.EnvDebug) == config.BoolTrue {
		log.Default.SetLevel(logrus.DebugLevel)
	}

	if globalFlags.DevPodHome != "" {
		_ = os.Setenv(config.EnvHome, globalFlags.DevPodHome)
	}

	// apply environment
	envfile.Apply(log.Default.ErrorStreamOnly())
	return nil
}