service.go

// Copyright 2022 The kubegems.io Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package apps

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

	"github.com/spf13/cobra"
	"golang.org/x/sync/errgroup"
	_ "kubegems.io/kubegems/docs/swagger"
	"kubegems.io/kubegems/pkg/log"
	"kubegems.io/kubegems/pkg/service"
	"kubegems.io/kubegems/pkg/service/models"
	"kubegems.io/kubegems/pkg/service/options"
	"kubegems.io/kubegems/pkg/utils/config"
	"kubegems.io/kubegems/pkg/utils/database"
	"kubegems.io/kubegems/pkg/utils/debug"
	"kubegems.io/kubegems/pkg/utils/redis"
	"kubegems.io/kubegems/pkg/version"
)

func NewServiceCmd() *cobra.Command {
	options := options.DefaultOptions()
	cmd := &cobra.Command{
		Use:          "service",
		Short:        "run service",
		SilenceUsage: true,
		Version:      version.Get().String(),
		RunE: func(cmd *cobra.Command, args []string) error {
			if err := config.Parse(cmd.Flags()); err != nil {
				return err
			}

			ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
			defer cancel()

			if err := debug.ApplyPortForwardingOptions(ctx, options.DebugMode, options.Mysql, options.Redis, nil, options.Appstore, nil); err != nil {
				return err
			}
			return service.Run(ctx, options)
		},
	}
	cmd.AddCommand(
		newGenServiceCfgCmd(),
		newServiceMigrateCmd(),
	)
	config.AutoRegisterFlags(cmd.Flags(), "", options)
	return cmd
}

func newGenServiceCfgCmd() *cobra.Command {
	return &cobra.Command{
		Use:   "gencfg",
		Short: "generate config template",
		Run: func(_ *cobra.Command, _ []string) {
			config.GenerateConfig(options.DefaultOptions())
		},
	}
}

type MigratOptions struct {
	Mysql         *database.Options `json:"mysql,omitempty"`
	Redis         *redis.Options    `json:"redis,omitempty"`
	MigrateModels bool              `json:"migrateModels,omitempty" description:"migrate models"`
	InitData      bool              `json:"initData,omitempty" description:"insert init data into database"`
	Wait          bool              `json:"wait,omitempty" description:"wait util database server ready"`
	GlobalValues  string            `json:"globalValues,omitempty" description:"global values json string"`
}

func newServiceMigrateCmd() *cobra.Command {
	options := &MigratOptions{
		Mysql:         database.NewDefaultOptions(),
		Redis:         redis.NewDefaultOptions(),
		MigrateModels: false,
		InitData:      false,
		Wait:          true,
		GlobalValues:  "",
	}

	cmd := &cobra.Command{
		Use:   "migrate",
		Short: "execute migrate, init datbases and base data (use server config)",
		RunE: func(cmd *cobra.Command, _ []string) error {
			if err := config.Parse(cmd.Flags()); err != nil {
				return err
			}

			ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, syscall.SIGTERM)
			defer cancel()

			ctx = log.NewContext(ctx, log.LogrLogger)

			eg := errgroup.Group{}
			eg.Go(func() error {
				if options.Wait {
					if err := models.WaitDatabaseServer(ctx, options.Mysql); err != nil {
						return err
					}
				}
				return models.MigrateDatabaseAndInitData(ctx, options.Mysql, options.MigrateModels, options.InitData, options.GlobalValues)
			})
			eg.Go(func() error {
				if options.Wait {
					return models.WaitRedis(ctx, *options.Redis)
				}
				return nil
			})
			return eg.Wait()
		},
	}
	config.AutoRegisterFlags(cmd.Flags(), "", options)
	return cmd
}