ziggi/gmbase
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

working commit

Browse Source
This commit is contained in:
Олег Бородин
2026-06-07 15:06:37 +02:00
parent 52dd2cef20
commit c656b562d8
17 changed files with 291 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cmd/mbaseadmin/account/util.go → cmd/mbaseadmin/account/accutil.go
+26 -7
View File
@@ -4,6 +4,9 @@
package account
import (
"context"
"fmt"
"mbase/app/operator"
"github.com/spf13/cobra"
@@ -17,11 +20,12 @@ const (
type Util struct {
oper *operator.Logic
subCmd *cobra.Command
createAccountParams createAccountParams
listAccountsParams listAccountsParams
updateAccountParams updateAccountParams
deleteAccountParams deleteAccountParams
createAccountParams CreateAccountParams
listAccountsParams ListAccountsParams
updateAccountParams UpdateAccountParams
deleteAccountParams DeleteAccountParams
commonParams CommonParams
operID string
}
func NewUtil(oper *operator.Logic) *Util {
@@ -41,9 +45,10 @@ func (util *Util) GetRooCmd() *cobra.Command {
func (util *Util) BuildCmds() *cobra.Command {
subCmd := &cobra.Command{
Use: "account",
Short: "\nAccount operations",
SilenceUsage: true,
Use: "account",
Short: "\nAccount operations",
SilenceUsage: true,
PersistentPreRunE: util.ValidateOper,
}
subCmd.CompletionOptions.DisableDefaultCmd = true
@@ -79,6 +84,8 @@ func (util *Util) BuildCmds() *cobra.Command {
Args: cobra.ExactArgs(1),
Run: util.UpdateAccount,
}
updateAccountsCmd.Flags().StringVarP(&util.updateAccountParams.Password, "pass", "P", util.updateAccountParams.Password, "New Password")
//updateAccountsCmd.Flags().StringVarP(&util.updateAccountParams.Username, "user", "U", util.updateAccountParams.Username, "New Username")
subCmd.AddCommand(updateAccountsCmd)
var deleteAccountCmd = &cobra.Command{
@@ -92,3 +99,15 @@ func (util *Util) BuildCmds() *cobra.Command {
util.subCmd = subCmd
return util.subCmd
}
func (util *Util) ValidateOper(cmd *cobra.Command, args []string) error {
var err error
ctx := context.Background()
authOk, operID, err := util.oper.ValidateAccount(ctx, util.commonParams.Username, util.commonParams.Password)
if !authOk {
err := fmt.Errorf("Authentification error")
return err
}
util.operID = operID
return err
}
cmd/mbaseadmin/account/create.go
+18 -28
View File
@@ -4,50 +4,40 @@
package account
import (
"context"
"mbase/pkg/mbctl"
"github.com/spf13/cobra"
)
func (util *Util) CreateAccount(cmd *cobra.Command, args []string) {
//util.createAccountParams.Filename = args[0]
res, err := util.createAccount(util.createAccountParams)
util.createAccountParams.Username = args[0]
util.createAccountParams.Password = args[1]
res, err := util.createAccount(&util.createAccountParams)
printResponse(res, err)
}
type createAccountParams struct {
type CreateAccountParams struct {
Username string
Password string
}
type createAccountResult struct{}
func (util *Util) createAccount(params createAccountParams) (createAccountResult, error) {
var err error
res := createAccountResult{}
/*
operParams := &mbctl.CreateAccountParams{
Username: params.Username,
Password: params.Password,
}
res, err = util.lg.CreateAccount(ctx, operID, params)
if err != nil {
return res, err
}
*/
return res, err
type CreateAccountResult struct {
AccountID string `json:"accountId"`
}
/*
func (util *Util) CreateAccount(ctx context.Context, operID string) (*mbctl.CreateAccountResult, error) {
func (util *Util) createAccount(params *CreateAccountParams) (*CreateAccountResult, error) {
var err error
res := &mbctl.CreateAccountResult{}
params := &mbctl.CreateAccountParams{
Username: util.username,
Password: util.password,
res := &CreateAccountResult{}
ctx := context.Background()
operParams := &mbctl.CreateAccountParams{
Username: params.Username,
Password: params.Password,
}
res, err = util.lg.CreateAccount(ctx, operID, params)
operRes, err := util.oper.CreateAccount(ctx, util.operID, operParams)
if err != nil {
return res, err
}
res.AccountID = operRes.AccountID
return res, err
}
*/
cmd/mbaseadmin/account/delete.go
+20 -6
View File
@@ -4,20 +4,34 @@
package account
import (
"context"
"mbase/pkg/mbctl"
"github.com/spf13/cobra"
)
func (util *Util) DeleteAccount(cmd *cobra.Command, args []string) {
//util.deleteAccountParams.Filename = args[0]
res, err := util.deleteAccount(util.deleteAccountParams)
util.deleteAccountParams.Username = args[0]
res, err := util.deleteAccount(&util.deleteAccountParams)
printResponse(res, err)
}
type deleteAccountParams struct{}
type deleteAccountResult struct{}
type DeleteAccountParams struct {
Username string
}
type DeleteAccountResult struct{}
func (util *Util) deleteAccount(params deleteAccountParams) (deleteAccountResult, error) {
func (util *Util) deleteAccount(params *DeleteAccountParams) (*DeleteAccountResult, error) {
var err error
res := deleteAccountResult{}
res := &DeleteAccountResult{}
ctx := context.Background()
operParams := &mbctl.DeleteAccountParams{
Username: params.Username,
}
_, err = util.oper.DeleteAccount(ctx, util.operID, operParams)
if err != nil {
return res, err
}
return res, err
}
cmd/mbaseadmin/account/list.go
+18 -6
View File
@@ -4,20 +4,32 @@
package account
import (
"context"
"mbase/pkg/mbctl"
"github.com/spf13/cobra"
)
func (util *Util) ListAccounts(cmd *cobra.Command, args []string) {
//util.listAccountsParams.Filename = args[0]
res, err := util.listAccounts(util.listAccountsParams)
res, err := util.listAccounts(&util.listAccountsParams)
printResponse(res, err)
}
type listAccountsParams struct{}
type listAccountsResult struct{}
type ListAccountsParams struct{}
type ListAccountsResult struct {
Accounts []*mbctl.AccountShortDescr `json:"accounts"`
}
func (util *Util) listAccounts(params listAccountsParams) (listAccountsResult, error) {
func (util *Util) listAccounts(params *ListAccountsParams) (*ListAccountsResult, error) {
var err error
res := listAccountsResult{}
ctx := context.Background()
res := &ListAccountsResult{}
operParams := &mbctl.ListAccountsParams{}
operRes, err := util.oper.ListAccounts(ctx, util.operID, operParams)
if err != nil {
return res, err
}
res.Accounts = operRes.Accounts
return res, err
}
cmd/mbaseadmin/account/update.go
+24 -6
View File
@@ -4,20 +4,38 @@
package account
import (
"context"
"mbase/pkg/mbctl"
"github.com/spf13/cobra"
)
func (util *Util) UpdateAccount(cmd *cobra.Command, args []string) {
//util.updateAccountParams.Filename = args[0]
res, err := util.updateAccount(util.updateAccountParams)
util.updateAccountParams.Username = args[0]
res, err := util.updateAccount(&util.updateAccountParams)
printResponse(res, err)
}
type updateAccountParams struct{}
type updateAccountResult struct{}
type UpdateAccountParams struct {
Username string
Password string
}
type UpdateAccountResult struct {
AccountID string
}
func (util *Util) updateAccount(params updateAccountParams) (updateAccountResult, error) {
func (util *Util) updateAccount(params *UpdateAccountParams) (*UpdateAccountResult, error) {
var err error
res := updateAccountResult{}
res := &UpdateAccountResult{}
ctx := context.Background()
operParams := &mbctl.UpdateAccountParams{
Username: params.Username,
//Password: params.Password,
}
_, err = util.oper.UpdateAccount(ctx, util.operID, operParams)
if err != nil {
return res, err
}
return res, err
}
cmd/mbaseadmin/grant/util.go → cmd/mbaseadmin/grant/grantutil.go
+22 -8
View File
@@ -4,6 +4,9 @@
package grant
import (
"context"
"fmt"
"github.com/spf13/cobra"
"github.com/spf13/viper"
"mbase/app/operator"
@@ -11,10 +14,11 @@ import (
type Util struct {
oper *operator.Logic
subCmd *cobra.Command
operID string
subCmd *cobra.Command
createGrantParams createGrantParams
deleteGrantParams deleteGrantParams
commonParams CommonParams
commonParams CommonParams
}
func NewUtil(oper *operator.Logic) *Util {
@@ -34,9 +38,10 @@ func (util *Util) GetRooCmd() *cobra.Command {
func (util *Util) BuildCmds() *cobra.Command {
subCmd := &cobra.Command{
Use: "grant",
Short: "\nGrant operations",
SilenceUsage: true,
Use: "grant",
Short: "\nGrant operations",
SilenceUsage: true,
PersistentPreRunE: util.ValidateOper,
}
subCmd.CompletionOptions.DisableDefaultCmd = true
@@ -51,9 +56,6 @@ func (util *Util) BuildCmds() *cobra.Command {
util.commonParams.Username = vi.GetString("user")
util.commonParams.Password = vi.GetString("pass")
var createGrantCmd = &cobra.Command{
Use: "create name|id password",
Short: "Create grant",
@@ -73,3 +75,15 @@ func (util *Util) BuildCmds() *cobra.Command {
util.subCmd = subCmd
return util.subCmd
}
func (util *Util) ValidateOper(cmd *cobra.Command, args []string) error {
var err error
ctx := context.Background()
authOk, operID, err := util.oper.ValidateAccount(ctx, util.commonParams.Username, util.commonParams.Password)
if !authOk {
err := fmt.Errorf("Authentification error")
return err
}
util.operID = operID
return err
}
cmd/mbaseadmin/main.go
+4
View File
@@ -4,18 +4,22 @@
package main
import (
"mbase/pkg/logger"
"os"
)
func main() {
var err error
log := logger.NewLogger("main")
util := NewUtil()
err = util.Build()
if err != nil {
log.Errorf("Build error: %v", err)
os.Exit(1)
}
err = util.Exec(os.Args[1:])
if err != nil {
log.Errorf("Exec error: %v", err)
os.Exit(1)
}
}
cmd/mbaseadmin/printr.go
+27
View File
@@ -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))
}
cmd/mbaseadmin/util.go
+119 -6
View File
@@ -4,6 +4,8 @@
package main
import (
"context"
"io/ioutil"
"os"
"path/filepath"
@@ -12,18 +14,27 @@ import (
"mbase/app/operator"
"mbase/cmd/mbaseadmin/account"
"mbase/cmd/mbaseadmin/grant"
"mbase/pkg/auxtool"
"mbase/pkg/descr"
"mbase/pkg/logger"
"github.com/spf13/cobra"
"go.yaml.in/yaml/v4"
)
type Util struct {
lg *operator.Logic
log *logger.Logger
oper *operator.Logic
db *maindb.Database
state descr.Server
sfile string
rootCmd *cobra.Command
}
func NewUtil() *Util {
return &Util{}
return &Util{
log: logger.NewLogger("util"),
}
}
func (util *Util) GetRooCmd() *cobra.Command {
@@ -34,6 +45,9 @@ func (util *Util) Build() error {
var err error
conf := config.NewConfig()
util.sfile = filepath.Join(conf.DataDir, "mbase.yaml")
err = conf.ReadFile()
if err != nil {
return err
@@ -42,11 +56,29 @@ func (util *Util) Build() error {
if err != nil {
return err
}
err = conf.Normalize()
if err != nil {
return err
}
err = conf.Validate()
if err != nil {
return err
}
//util.log.Infof("Create %s dir", conf.DataDir)
err = os.MkdirAll(conf.DataDir, 0750)
if err != nil {
return err
}
// Load state
err = util.LoadState()
if err != nil {
//return err
}
db, err := maindb.NewDatabase(conf.DataDir)
if err != nil {
return err
}
err = db.OpenDatabase()
if err != nil {
return err
@@ -55,11 +87,24 @@ func (util *Util) Build() error {
logicConfig := &operator.LogicConfig{
Database: util.db,
}
util.lg, err = operator.NewLogic(logicConfig)
util.oper, err = operator.NewLogic(logicConfig)
if err != nil {
return err
}
if !util.state.DatabaseInitialized {
// Create schema
util.log.Infof("Init database")
err = util.db.InitDatabase()
if err != nil {
return err
}
util.state.DatabaseInitialized = true
err = util.SaveState()
if err != nil {
return err
}
}
execName := filepath.Base(os.Args[0])
rootCmd := &cobra.Command{
Use: execName,
@@ -68,12 +113,19 @@ func (util *Util) Build() error {
}
rootCmd.CompletionOptions.DisableDefaultCmd = true
accountUtil := account.NewUtil(util.lg)
accountUtil := account.NewUtil(util.oper)
rootCmd.AddCommand(accountUtil.BuildCmds())
grantUtil := grant.NewUtil(util.lg)
grantUtil := grant.NewUtil(util.oper)
rootCmd.AddCommand(grantUtil.BuildCmds())
var deleteGrantCmd = &cobra.Command{
Use: "seed",
Short: "Create initial account",
Run: util.SeedAccount,
}
rootCmd.AddCommand(deleteGrantCmd)
util.rootCmd = rootCmd
return err
}
@@ -84,3 +136,64 @@ func (util *Util) Exec(args []string) error {
err = util.rootCmd.Execute()
return err
}
func (util *Util) LoadState() error {
var err error
_, err = os.Stat(util.sfile)
if os.IsNotExist(err) {
err = nil
return err
}
file, err := os.Open(util.sfile)
if err != nil {
return err
}
defer file.Close()
stateBytes, err := ioutil.ReadAll(file)
if err != nil {
return err
}
err = yaml.Unmarshal(stateBytes, &util.state)
if err != nil {
return err
}
return err
}
func (util *Util) SaveState() error {
var err error
file, err := os.OpenFile(util.sfile, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0640)
if err != nil {
return err
}
defer file.Close()
util.state.UpdatedAt = auxtool.TimeNow()
if util.state.CreatedAt == "" {
util.state.CreatedAt = util.state.UpdatedAt
}
stateBytes, err := yaml.Marshal(util.state)
if err != nil {
return err
}
_, err = file.Write(stateBytes)
if err != nil {
return err
}
return err
}
func (util *Util) SeedAccount(cmd *cobra.Command, args []string) {
err := util.seedAccount()
printResponse(nil, err)
}
func (util *Util) seedAccount() error {
// Seed accounts
ctx := context.Background()
_, err := util.oper.SeedAccount(ctx)
if err != nil {
return err
}
return err
}