main.go

package main

import (
	"context"
	"fmt"
	"os"
	"os/signal"
	"syscall"

	"github.com/noderax/noderax-agent/internal/agent"
	"github.com/noderax/noderax-agent/internal/agentctl"
	"github.com/noderax/noderax-agent/internal/api"
	"github.com/noderax/noderax-agent/internal/brand"
	"github.com/noderax/noderax-agent/internal/config"
	"github.com/noderax/noderax-agent/internal/logger"
)

var (
	version   = "dev"
	commit    = "unknown"
	buildDate = "unknown"
)

func main() {
	baseLog := logger.New("info")
	cli := agentctl.CLI{
		Logger:    baseLog,
		Version:   version,
		Commit:    commit,
		BuildDate: buildDate,
		Stdin:     os.Stdin,
		Stdout:    os.Stdout,
		Stderr:    os.Stderr,
	}

	cliCtx, cliStop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
	defer cliStop()

	if handled, err := cli.Handle(cliCtx, os.Args[1:]); handled {
		if err != nil {
			baseLog.Error("command failed", "error", err)
			os.Exit(1)
		}
		return
	}

	cfg, err := config.Load()
	if err != nil {
		fmt.Fprintf(os.Stderr, "load config: %v\n", err)
		os.Exit(1)
	}

	log := logger.New(cfg.LogLevel)
	log.Info("starting noderax agent", "version", version, "commit", commit, "build_date", buildDate)

	client, err := api.NewClient(cfg.APIURL, cfg.RequestTimeout, cfg.APITLSCAFile)
	if err != nil {
		log.Error("configure API client", "error", err)
		os.Exit(1)
	}
	if cfg.AgentToken != "" {
		client.SetAgentToken(cfg.AgentToken)
	}

	if len(os.Args) > 1 && os.Args[1] == "enroll" {
		ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
		defer stop()

		brand.PrintLogo(os.Stdout)

		if err := agent.RunInteractiveEnrollment(ctx, cfg, client, log, version, os.Stdin, os.Stdout); err != nil {
			log.Error("interactive enrollment failed", "error", err)
			os.Exit(1)
		}

		log.Info("interactive enrollment completed")
		return
	}

	svc := agent.NewService(cfg, client, log, version)
	ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
	defer stop()

	if err := svc.Run(ctx); err != nil {
		log.Error("agent exited with error", "error", err)
		os.Exit(1)
	}

	log.Info("agent stopped")
}