main.go

// Copyright Louis Royer and the NextMN contributors. All rights reserved.
// Use of this source code is governed by a MIT-style license that can be
// found in the LICENSE file.
// SPDX-License-Identifier: MIT

package main

import (
	"context"
	"os"
	"os/signal"
	"runtime/debug"
	"syscall"

	"github.com/nextmn/cli-xdg"
	"github.com/nextmn/json-api/healthcheck"
	"github.com/nextmn/logrus-formatter/logger"

	"github.com/nextmn/srv6-ctrl/internal/app"
	"github.com/nextmn/srv6-ctrl/internal/config"

	"github.com/sirupsen/logrus"
	"github.com/urfave/cli/v3"
)

func main() {
	logger.Init("Nextm-SRv6-ctrl")
	ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGTERM, syscall.SIGINT)
	defer cancel()
	version := "Unknown version"
	if info, ok := debug.ReadBuildInfo(); ok {
		version = info.Main.Version
	}
	app := &cli.Command{
		Name:                  "srv6-ctrl",
		Usage:                 "NextMN-SRv6-ctrl - Controller for NextMN-SRv6",
		EnableShellCompletion: true,
		Authors: []any{
			"Louis Royer",
		},
		Version: version,
		Flags: []cli.Flag{
			&cli.StringFlag{
				Name:      "config",
				TakesFile: true,
				Aliases:   []string{"c"},
				Usage:     "load configuration from `FILE`",
				// XXX: https://github.com/urfave/cli/issues/2244
				// Required:    true,
				DefaultText: "${XDG_CONFIG_DIRS}/nextmn-srv6-ctrl/config.yaml",
				Sources: cli.NewValueSourceChain(
					cli.EnvVar("CONFIG_FILE"),
					clixdg.ConfigFile("nextmn-srv6-ctrl/config.yaml"),
				),
			},
		},
		DefaultCommand: "run",
		Commands: []*cli.Command{
			{
				Name:  "run",
				Usage: "Runs the SRv6 controller",
				Action: func(ctx context.Context, cmd *cli.Command) error {
					// XXX: https://github.com/urfave/cli/issues/2244
					if cmd.String("config") == "" {
						logrus.Fatal("Required flag \"config\" not set")
					}

					conf, err := config.ParseConf(cmd.String("config"))
					if err != nil {
						logrus.WithContext(ctx).WithError(err).Fatal("Error loading config, exiting…")
					}
					if conf.Logger != nil {
						logrus.SetLevel(conf.Logger.Level)
					}
					if err := app.NewSetup(conf).Run(ctx); err != nil {
						logrus.WithError(err).Fatal("Error while running, exiting…")
					}
					return nil
				},
			},
			{
				Name:  "healthcheck",
				Usage: "Checks status of the node",
				Action: func(ctx context.Context, cmd *cli.Command) error {
					// XXX: https://github.com/urfave/cli/issues/2244
					if cmd.String("config") == "" {
						logrus.Fatal("Required flag \"config\" not set")
					}

					conf, err := config.ParseConf(cmd.String("config"))
					if err != nil {
						logrus.WithContext(ctx).WithError(err).Fatal("Error loading config, exiting…")
					}
					if conf.Logger != nil {
						logrus.SetLevel(conf.Logger.Level)
					}
					if err := healthcheck.NewHealthcheck(*conf.Control.Uri.JoinPath("status"), "go-github-nextmn-srv6-ctrl").Run(ctx); err != nil {
						os.Exit(1)
					}
					return nil
				},
			},
		},
	}
	if err := app.Run(ctx, os.Args); err != nil {
		logrus.WithError(err).Fatal("Fatal error while running the application")
	}
}