emissary/internal/command/agent/run.go

package agent

import (
	"time"

	"forge.cadoles.com/Cadoles/emissary/internal/agent"
	"forge.cadoles.com/Cadoles/emissary/internal/agent/controller/app"
	"forge.cadoles.com/Cadoles/emissary/internal/agent/controller/openwrt"
	"forge.cadoles.com/Cadoles/emissary/internal/agent/controller/persistence"
	"forge.cadoles.com/Cadoles/emissary/internal/agent/controller/proxy"
	"forge.cadoles.com/Cadoles/emissary/internal/agent/controller/spec"
	"forge.cadoles.com/Cadoles/emissary/internal/agent/metadata"
	"forge.cadoles.com/Cadoles/emissary/internal/agent/metadata/collector/buildinfo"
	"forge.cadoles.com/Cadoles/emissary/internal/agent/metadata/collector/shell"
	"forge.cadoles.com/Cadoles/emissary/internal/command/common"
	"forge.cadoles.com/Cadoles/emissary/internal/config"
	"forge.cadoles.com/Cadoles/emissary/internal/jwk"
	"forge.cadoles.com/Cadoles/emissary/internal/machineid"
	"github.com/pkg/errors"
	"github.com/urfave/cli/v2"
	"gitlab.com/wpetit/goweb/logger"
)

func RunCommand() *cli.Command {
	flags := common.Flags()

	return &cli.Command{
		Name:  "run",
		Usage: "Run the emissary agent",
		Flags: flags,
		Action: func(ctx *cli.Context) error {
			conf, err := common.LoadConfig(ctx)
			if err != nil {
				return errors.Wrap(err, "Could not load configuration")
			}

			logger.SetFormat(logger.Format(conf.Logger.Format))
			logger.SetLevel(logger.Level(conf.Logger.Level))

			controllers := make([]agent.Controller, 0)

			ctrlConf := conf.Agent.Controllers

			if ctrlConf.Persistence.Enabled {
				controllers = append(controllers, persistence.NewController(string(ctrlConf.Persistence.StateFile)))
			}

			if ctrlConf.Spec.Enabled {
				controllers = append(controllers, spec.NewController())
			}

			if ctrlConf.Proxy.Enabled {
				controllers = append(controllers, proxy.NewController())
			}

			if ctrlConf.UCI.Enabled {
				controllers = append(controllers, openwrt.NewUCIController(
					string(ctrlConf.UCI.BinPath),
				))
			}

			if ctrlConf.App.Enabled {
				controllers = append(controllers, app.NewController(
					app.WithDataDir(string(ctrlConf.App.DataDir)),
					app.WithDownloadDir(string(ctrlConf.App.DownloadDir)),
				))
			}

			if ctrlConf.SysUpgrade.Enabled {
				sysUpgradeArgs := make([]string, 0)
				if len(ctrlConf.SysUpgrade.SysUpgradeCommand) > 1 {
					sysUpgradeArgs = ctrlConf.SysUpgrade.SysUpgradeCommand[1:]
				}

				firmwareVersionArgs := make([]string, 0)
				if len(ctrlConf.SysUpgrade.FirmwareVersionCommand) > 1 {
					firmwareVersionArgs = ctrlConf.SysUpgrade.FirmwareVersionCommand[1:]
				}

				controllers = append(controllers, openwrt.NewSysUpgradeController(
					openwrt.WithSysUpgradeCommand(
						ctrlConf.SysUpgrade.SysUpgradeCommand[0],
						sysUpgradeArgs...,
					),
					openwrt.WithSysUpgradeShellFirmwareVersion(
						ctrlConf.SysUpgrade.FirmwareVersionCommand[0],
						firmwareVersionArgs...,
					),
				))
			}

			key, err := jwk.LoadOrGenerate(string(conf.Agent.PrivateKeyPath), jwk.DefaultKeySize)
			if err != nil {
				return errors.WithStack(err)
			}

			thumbprint, err := machineid.Get()
			if err != nil {
				return errors.WithStack(err)
			}

			collectors := createShellCollectors(&conf.Agent)
			collectors = append(collectors, buildinfo.NewCollector())

			agent := agent.New(
				string(conf.Agent.ServerURL),
				key,
				thumbprint,
				agent.WithInterval(time.Duration(conf.Agent.ReconciliationInterval)*time.Second),
				agent.WithControllers(controllers...),
				agent.WithCollectors(collectors...),
			)

			if err := agent.Run(ctx.Context); err != nil {
				return errors.WithStack(err)
			}

			return nil
		},
	}
}

func createShellCollectors(conf *config.AgentConfig) []metadata.Collector {
	collectors := make([]metadata.Collector, 0)

	for _, c := range conf.Collectors {
		collector := shell.NewCollector(string(c.Name), string(c.Command), c.Args...)

		collectors = append(collectors, collector)
	}

	return collectors
}