courgette/main.go

package main

import (
	"context"
	commands "courgette/internal/commands"
	daemon "courgette/internal/daemon"
	logger "courgette/internal/logging"
	"github.com/spf13/cobra"
	"os"
)

var cli = &cobra.Command{
	Use: "runner",
	PersistentPreRun: func(cmd *cobra.Command, args []string) {
		logger.ConfigureLogger()
		configPath, err := cmd.Flags().GetString("config")

		ctx := cmd.Context()

		if err == nil {
			configuration, err := commands.NewConfigFromFile(configPath)

			if err != nil {
				logger.Error(logger.Red("Failed to parse configuration (%s)!"), configPath)
				os.Exit(1)
			}

			ctx = context.WithValue(ctx, "config", configuration)
		}

		cmd.SetContext(ctx)
	},
}

var execute = &cobra.Command{
	Use:   "execute [workflow-file]",
	Short: "Executes the provided workflow.",
	Args:  cobra.ExactArgs(1),
	Run: func(cmd *cobra.Command, args []string) {
		config := cmd.Context().Value("config").(*commands.Configuration)

		if err := commands.ExecuteWorkflow(*config, args[0]); err != nil {
			logger.Error(logger.Red("%s"), err)
			os.Exit(1)
		}
	},
}

var validate = &cobra.Command{
	Use:   "validate [workflow-file]",
	Short: "Validates the structure of the provided workflow.",
	Args:  cobra.ExactArgs(1),
	Run: func(cmd *cobra.Command, args []string) {
		config := cmd.Context().Value("config").(*commands.Configuration)

		if err := commands.ValidateWorkflow(*config, args[0]); err != nil {
			logger.Error(logger.Red("%s"), err)
			os.Exit(1)
		}
	},
}

var daemonCmd = &cobra.Command{
	Use:   "daemon",
	Short: "Starts the daemon process",
	Run: func(cmd *cobra.Command, args []string) {
		// FIXME: Sample daemon setup.
		d := daemon.NewDaemon()
		d.Start(":8080")
	},
}

var knownCommands = []*cobra.Command{
	execute,
	validate,
	daemonCmd,
}

func main() {
	cli.PersistentFlags().StringP("config", "c", "./config.yml", "Declarative runner configuration.")

	for _, command := range knownCommands {
		cli.AddCommand(command)
	}

	cli.Execute()
}