snapshot of https://github.com/cortezaproject/corteza.git (2024.9.x, 2026-09-18; blobs > 131072B omitted)

server/app/cli.go

@@ -0,0 +1,158 @@
+package app
+
+import (
+	"context"
+	"fmt"
+	"sync"
+
+	composeCommands "github.com/cortezaproject/corteza/server/compose/commands"
+
+	authCommands "github.com/cortezaproject/corteza/server/auth/commands"
+	federationCommands "github.com/cortezaproject/corteza/server/federation/commands"
+	"github.com/cortezaproject/corteza/server/pkg/actionlog"
+	"github.com/cortezaproject/corteza/server/pkg/api/server"
+	"github.com/cortezaproject/corteza/server/pkg/cli"
+	"github.com/cortezaproject/corteza/server/pkg/dal"
+	"github.com/cortezaproject/corteza/server/pkg/envoyx"
+	"github.com/cortezaproject/corteza/server/pkg/id"
+	"github.com/cortezaproject/corteza/server/pkg/options"
+	"github.com/cortezaproject/corteza/server/store"
+	systemCommands "github.com/cortezaproject/corteza/server/system/commands"
+	"go.uber.org/zap"
+)
+
+// InitCLI function initializes basic Corteza subsystems
+// and sets-up the command line interface
+func (app *CortezaApp) InitCLI() {
+	var (
+		ctx = cli.Context()
+
+		// path to all environmental files (or locations with .env file)
+		// filled from flag values
+		envs []string
+	)
+
+	id.Init(ctx)
+
+	app.Command = cli.RootCommand(func() error {
+		return nil
+	})
+
+	// Environmental variables (from the env, files, see cli.LoadEnv) MUST be
+	// loaded at this point!
+	if len(envs) == 0 {
+		envs = []string{"."}
+	}
+
+	// Loading env after the rootcommand is added, so as not to break
+	// the Execute() if there is an error
+	if err := cli.LoadEnv(envs...); err != nil {
+		app.Log.Error("failed to load environmental variables", zap.Error(err))
+		return
+	}
+
+	app.Opt = options.Init()
+
+	app.Command.Flags().StringSliceVar(&envs, "env-file", nil,
+		"Load environmental variables from files and directories containing .env file.\n"+
+			"Values from loaded files DO NOT override existing variables from the environment.\n"+
+			"This flag can be used multiple times, values are loaded from all provided locations.\n"+
+			"If no paths are provided, corteza loads .env file from the current directory (equivalent to --env-file .)")
+
+	serveCmd := cli.ServeCommand(func() (err error) {
+		wg := &sync.WaitGroup{}
+
+		{ // @todo refactor wait-for out of HTTP API server.
+
+			// test http server options first
+			if err = server.Test(app.Opt); err != nil {
+				return err
+			}
+
+			app.HttpServer = server.New(app.Log, app.Opt)
+
+			wg.Add(1)
+			go func() {
+				app.HttpServer.Serve(actionlog.RequestOriginToContext(ctx, actionlog.RequestOrigin_API_REST))
+				wg.Done()
+			}()
+		}
+
+		{
+			// @todo add other server/listeners here...
+			//wg.Add(1)
+			//go func(ctx context.Context) {
+			//	grpcApi.Serve(actionlog.RequestOriginToContext(ctx, actionlog.RequestOrigin_API_GRPC))
+			//	wg.Done()
+			//}(ctx)
+		}
+
+		if err = app.Activate(ctx); err != nil {
+			cli.HandleError(fmt.Errorf("could not start Corteza: %w", err))
+			return nil
+		}
+
+		app.HttpServer.Activate(app.mountHttpRoutes)
+
+		// Wait for all servers to be done
+		wg.Wait()
+
+		app.HttpServer.Shutdown()
+
+		return nil
+	})
+
+	upgradeCmd := cli.UpgradeCommand(func() (err error) {
+		if err = app.InitStore(ctx); err != nil {
+			return
+		}
+
+		return
+	})
+
+	provisionCmd := cli.ProvisionCommand(func() (err error) {
+		if err = app.Provision(ctx); err != nil {
+			return
+		}
+
+		return
+	})
+
+	storeInit := func(ctx context.Context) (store.Storer, error) {
+		err := app.InitStore(ctx)
+		return app.Store, err
+	}
+
+	dalInit := func(ctx context.Context) (dal.FullService, error) {
+		err := app.initDAL(ctx, app.Log)
+		return dal.Service(), err
+	}
+
+	envoyInit := func(ctx context.Context) (svc *envoyx.Service, err error) {
+		err = app.initEnvoy(ctx, app.Log)
+		return envoyx.Global(), err
+	}
+
+	app.Command.AddCommand(
+		systemCommands.Users(ctx, app),
+		systemCommands.Roles(ctx, app),
+		systemCommands.RBAC(ctx, storeInit),
+		systemCommands.Sink(ctx, app),
+		systemCommands.Settings(ctx, app),
+		systemCommands.Import(ctx, storeInit, dalInit, envoyInit),
+		systemCommands.Export(ctx, storeInit, dalInit, envoyInit),
+		serveCmd,
+		upgradeCmd,
+		provisionCmd,
+		authCommands.Command(ctx, app, storeInit),
+		federationCommands.Sync(ctx, app),
+		composeCommands.Base(ctx, app),
+		cli.EnvCommand(),
+		cli.VersionCommand(),
+	)
+
+}
+
+func (app *CortezaApp) Execute() error {
+	return app.Command.Execute()
+}