working commit

2026-06-05 18:25:03 +02:00
commit 7d8abba003
82 changed files with 21863 additions and 0 deletions
@@ -0,0 +1,21 @@
+/*
+ * Copyright 2026 Oleg Borodin <onborodin@gmail.com>
+ */
+package main
+
+import (
+	"os"
+)
+
+func main() {
+	var err error
+	util := NewUtil()
+	err = util.Build()
+	if err != nil {
+		os.Exit(1)
+	}
+	err = util.Exec(os.Args[1:])
+	if err != nil {
+		os.Exit(1)
+	}
+}
@@ -0,0 +1,27 @@
+/*
+ * Copyright 2026 Oleg Borodin <onborodin@gmail.com>
+ */
+package main
+
+import (
+	"fmt"
+
+	"sigs.k8s.io/yaml"
+)
+
+func printResponse(res any, err error) {
+	type Response struct {
+		Error   bool   `json:"error"             yaml:"error"`
+		Message string `json:"message,omitempty" yaml:"message,omitempty"`
+		Result  any    `json:"result,omitempty"  yaml:"result,omitempty"`
+	}
+	resp := Response{}
+	if err != nil {
+		resp.Error = true
+		resp.Message = err.Error()
+	} else {
+		resp.Result = res
+	}
+	respBytes, _ := yaml.Marshal(resp)
+	fmt.Printf("---\n%s\n", string(respBytes))
+}
@@ -0,0 +1,220 @@
+/*
+ * Copyright 2026 Oleg Borodin <onborodin@gmail.com>
+ */
+package main
+
+import (
+	"os"
+	"path/filepath"
+        "context"
+        "io"
+        "bytes"
+
+	"mbase/app/config"
+	"mbase/app/database"
+        "mbase/app/descriptor"
+        "mbase/pkg/auxtool"
+        "mbase/pkg/logger"
+
+        "go.yaml.in/yaml/v4"
+	"github.com/spf13/cobra"
+)
+
+type Util struct {
+	rootCmd *cobra.Command
+        dumpDatabaseParams dumpDatabaseParams
+	restoreDatabaseParams restoreDatabaseParams
+}
+
+func NewUtil() *Util {
+	return &Util{}
+}
+
+func (util *Util) GetRooCmd() *cobra.Command {
+	return util.rootCmd
+}
+
+func (util *Util) Build() error {
+	var err error
+	execName := filepath.Base(os.Args[0])
+	rootCmd := &cobra.Command{
+		Use:          execName,
+		Short:        "\nDump application database",
+		SilenceUsage: true,
+	}
+	rootCmd.CompletionOptions.DisableDefaultCmd = true
+
+	var dumpDatabaseCmd = &cobra.Command{
+		Use:   "dump [filename|-]",
+		Short: "Dump application database",
+		Args:  cobra.ExactArgs(1),
+		Run:   util.DumpDatabase,
+	}
+	rootCmd.AddCommand(dumpDatabaseCmd)
+
+	var restoreDatabaseCmd = &cobra.Command{
+		Use:   "restore [] [filename|-]",
+		Short: "Restore application database",
+		Args:  cobra.ExactArgs(1),
+		Run:   util.RestoreDatabase,
+	}
+	restoreDatabaseCmd.Flags().BoolVarP(&util.restoreDatabaseParams.DeleteAllRecords, "clean", "C", false, "Clean all record")
+	rootCmd.AddCommand(restoreDatabaseCmd)
+
+	util.rootCmd = rootCmd
+	return err
+}
+
+func (util *Util) Exec(args []string) error {
+	var err error
+	util.rootCmd.SetArgs(args)
+	err = util.rootCmd.Execute()
+	return err
+}
+
+func (util *Util) DumpDatabase(cmd *cobra.Command, args []string) {
+        util.dumpDatabaseParams.Filename = args[0]
+        res, err := util.dumpDatabase(util.dumpDatabaseParams)
+        printResponse(res, err)
+}
+
+type dumpDatabaseParams struct {
+        Filename string
+}
+type dumpDatabaseResult struct {}
+
+func (util *Util) dumpDatabase(params dumpDatabaseParams) (dumpDatabaseResult, error) {
+	var err error
+        res := dumpDatabaseResult{}
+        ctx := context.Background()
+
+	conf := config.NewConfig()
+	err = conf.ReadFile()
+	if err != nil {
+		return res, err
+	}
+	err = conf.ReadEnv()
+	if err != nil {
+		return res, err
+	}
+	db, err := database.NewDatabase(conf.DataDir)
+	if err != nil {
+		return res, err
+	}
+	err = db.OpenDatabase()
+	if err != nil {
+		return res, err
+	}
+	file := os.Stdout
+	if params.Filename != "-" {
+		file, err = os.OpenFile(params.Filename, os.O_CREATE|os.O_WRONLY, 0640)
+		if err != nil {
+			return res, err
+		}
+		defer file.Close()
+	}
+	listAccounts, err := db.CompletedListAccounts(ctx)
+	if err != nil {
+		return res, err
+	}
+	listGrants, err := db.ListGrants(ctx)
+	if err != nil {
+		return res, err
+	}
+	dump := descriptor.Dump{
+		Timestamp: auxtool.TimeNow(),
+		Accounts:  listAccounts,
+		Grants:    listGrants,
+	}
+	dumpBytes, err := yaml.Marshal(dump)
+	if err != nil {
+		return res, err
+	}
+	_, err = file.Write(dumpBytes)
+	if err != nil {
+		return res, err
+	}
+	return res, err
+}
+
+
+func (util *Util) RestoreDatabase(cmd *cobra.Command, args []string) {
+        util.restoreDatabaseParams.Filename = args[0]
+        res, err := util.restoreDatabase(util.restoreDatabaseParams)
+        printResponse(res, err)
+}
+
+type restoreDatabaseParams struct {
+        Filename string
+        DeleteAllRecords bool
+}
+type restoreDatabaseResult struct {}
+
+func (util *Util) restoreDatabase(params restoreDatabaseParams) (restoreDatabaseResult, error) {
+	var err error
+        res := restoreDatabaseResult{}
+        ctx := context.Background()
+        log := logger.NewLogger("restore")
+
+	conf := config.NewConfig()
+	err = conf.ReadFile()
+	if err != nil {
+		return res, err
+	}
+	err = conf.ReadEnv()
+	if err != nil {
+		return res, err
+	}
+
+	db, err := database.NewDatabase(conf.DataDir)
+	if err != nil {
+		return res, err
+	}
+	err = db.OpenDatabase()
+	if err != nil {
+		return res, err
+	}
+	file := os.Stdin
+	if params.Filename != "-" {
+		file, err = os.Open(params.Filename)
+		if err != nil {
+			return res, err
+		}
+		defer file.Close()
+	}
+	buffer := bytes.NewBuffer(nil)
+	_, err = io.Copy(buffer, file)
+	if err != nil {
+		return res, err
+	}
+	dump := descriptor.Dump{}
+	err = yaml.Unmarshal(buffer.Bytes(), &dump)
+	if err != nil {
+		return res, err
+	}
+	if params.DeleteAllRecords {
+		err = db.CleanDatabase(ctx)
+		if err != nil {
+			return res, err
+		}
+	}
+	for _, account := range dump.Accounts {
+		log.Infof("Insert account %s", account.Username)
+		err = db.InsertAccount(ctx, &account)
+		if err != nil {
+			log.Errorf("Insert account error: %v", err)
+		}
+	}
+	for _, grant := range dump.Grants {
+		log.Infof("Insert grant %s for account %d", grant.Operation, grant.AccountID)
+		err = db.InsertGrant(ctx, &grant)
+		if err != nil {
+			log.Errorf("Insert account error: %v", err)
+		}
+	}
+	return res, err
+}
+
+
+
+