working commit

cmd/mbasedump/rest.go

@@ -0,0 +1,96 @@
+/*
+ * Copyright 2026 Oleg Borodin <onborodin@gmail.com>
+ */
+package main
+
+import (
+	"bytes"
+	"context"
+	"io"
+	"os"
+
+	"mbase/app/config"
+	"mbase/app/maindb"
+	"mbase/pkg/descr"
+	"mbase/pkg/logger"
+
+	"github.com/spf13/cobra"
+	"go.yaml.in/yaml/v4"
+)
+
+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 := maindb.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 := descr.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
+}