serve.go

package cmd

import (
	"context"
	"crypto/tls"
	"fmt"

	"github.com/kong/koko/internal/config"
	"github.com/kong/koko/internal/log"
	"github.com/spf13/cobra"
	"go.uber.org/zap"
)

var cfgFile string

// serveCmd is 'koko serve' command.
var serveCmd = &cobra.Command{
	Use:   "serve",
	Short: "Control plane software for Kong Gateway",
	RunE: func(cmd *cobra.Command, args []string) error {
		err := serveMain(cmd.Context())
		if err != nil {
			return err
		}
		return nil
	},
}

func init() {
	rootCmd.AddCommand(serveCmd)
}

type initOpts struct {
	Config config.Config
	Logger *zap.Logger
}

func serveMain(ctx context.Context) error {
	opts, err := setup()
	if err != nil {
		return err
	}
	logger := opts.Logger
	logger.Debug("setup successful")

	cert, err := tls.LoadX509KeyPair(opts.Config.Control.TLSCertPath,
		opts.Config.Control.TLSKeyPath)
	if err != nil {
		return fmt.Errorf("unable to load tls/cert/key: %w", err)
	}

	return Run(ctx, ServerConfig{
		DPAuthCert:              cert,
		KongCPCert:              cert,
		Logger:                  logger,
		Database:                opts.Config.Database,
		Metrics:                 opts.Config.Metrics,
		DisableAnonymousReports: opts.Config.DisableAnonymousReports,
	})
}

func setup() (initOpts, error) {
	cfg, err := config.Get(cfgFile)
	if err != nil {
		return initOpts{}, err
	}

	logger, err := log.SetupLogging(cfg.Log.Level, cfg.Log.Format)
	if err != nil {
		return initOpts{}, err
	}
	return initOpts{Config: cfg, Logger: logger}, nil
}