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

update

Browse Source
This commit is contained in:
Олег Бородин
2024-09-14 07:49:45 +02:00
parent f25197e714
commit 7a267cdc4d
22 changed files with 1026 additions and 342 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
cmd/certmanagerctl/main.go
View File

@@ -83,6 +83,7 @@ type Util struct {
ipAdressesList string
serviceID int64
serviceName string
encodingKey string
accountID int64
username string
@@ -199,6 +200,7 @@ func (util *Util) GetOpt() error {
flagSet.StringVar(&util.issuerCommonName, "cn", util.issuerCommonName, "new issuer canonic name")
flagSet.Int64Var(&util.signerID, "signerID", util.signerID, "optional issuer ID for sign")
flagSet.StringVar(&util.signerName, "signerName", util.signerName, "optional issuer name for sign")
flagSet.StringVar(&util.encodingKey, "encodingKey", util.encodingKey, "key for encoding private key")
flagSet.Usage = func() {
fmt.Printf("\n")
@@ -216,6 +218,7 @@ func (util *Util) GetOpt() error {
flagSet.StringVar(&util.issuerName, "issuerName", util.issuerName, "issuer name")
flagSet.Int64Var(&util.issuerID, "issuerID", util.issuerID, "issuer ID")
flagSet.StringVar(&util.encodingKey, "encodingKey", util.encodingKey, "key for encoding private key")
flagSet.Usage = func() {
fmt.Printf("\n")

163
cmd/certmanagerdump/maindump.go Normal file
View File

@@ -0,0 +1,163 @@
/*
* Copyright 2022 Oleg Borodin <borodin@unix7.org>
*/
package main
import (
"context"
"flag"
"fmt"
"os"
"path/filepath"
"time"
"certmanager/internal/config"
"certmanager/internal/database"
"certmanager/internal/descriptor"
"gopkg.in/yaml.v3"
)
func main() {
var err error
util := NewUtil()
err = util.Exec()
if err != nil {
fmt.Printf("Exec error: %s\n", err)
}
}
type Util struct {
conf *config.Config
db *database.Database
filename string
}
func NewUtil() *Util {
var util Util
return &util
}
func (util *Util) GetOpt() error {
var err error
exeName := filepath.Base(os.Args[0])
help := func() {
fmt.Println("")
fmt.Printf("Usage: %s [option]\n", exeName)
fmt.Printf("\n")
flag.PrintDefaults()
fmt.Printf("\n")
}
flag.Usage = help
flag.StringVar(&util.filename, "file", util.filename, "dump file name")
flag.Parse()
return err
}
func (util *Util) Exec() error {
var err error
err = util.GetOpt()
if err != nil {
return err
}
const timeout = 30 * time.Second
ctx, _ := context.WithTimeout(context.Background(), timeout)
err = util.DumpRecords(ctx)
type ErrorDescr struct {
Error bool `json:"error,omitempty"`
Message string `json:"errorMessage,omitempty" yaml:"errorMessage,omitempty"`
}
errDescr := ErrorDescr{}
if err != nil {
errDescr.Error = true
errDescr.Message = fmt.Sprintf("%v", err)
}
errBytes, _ := yaml.Marshal(errDescr)
fmt.Printf("%s\n", string(errBytes))
return err
}
func (util *Util) DumpRecords(ctx context.Context) error {
var err error
util.conf = config.NewConfig()
err = util.conf.ReadFile()
if err != nil {
return err
}
err = util.conf.ReadEnv()
if err != nil {
return err
}
db, err := database.NewDatabase(util.conf.DataDir)
if err != nil {
return err
}
util.db = db
err = util.db.InitDatabase(ctx)
if err != nil {
return err
}
file := os.Stdout
if util.filename != "" {
file, err = os.OpenFile(util.filename, os.O_CREATE|os.O_WRONLY, 0640)
if err != nil {
return err
}
defer file.Close()
}
listIssuers, err := util.db.ListIssuers(ctx)
if err != nil {
return err
}
listServices, err := util.db.ListServices(ctx)
if err != nil {
return err
}
listAccounts, err := util.db.ListAccounts(ctx)
if err != nil {
return err
}
listGrants, err := util.db.ListGrants(ctx)
if err != nil {
return err
}
dump := descriptor.Dump{
Timestamp: time.Now().Format(time.RFC3339),
Issuers: listIssuers,
Services: listServices,
Accounts: listAccounts,
Grants: listGrants,
}
dumpBytes, err := yaml.Marshal(dump)
if err != nil {
return err
}
_, err = file.Write(dumpBytes)
if err != nil {
return err
}
return err
}

183
cmd/certmanagerrestore/mainrestore.go Normal file
View File

@@ -0,0 +1,183 @@
/*
* Copyright 2022 Oleg Borodin <borodin@unix7.org>
*/
package main
import (
"bytes"
"context"
"flag"
"fmt"
"io"
"os"
"path/filepath"
"time"
"certmanager/internal/config"
"certmanager/internal/database"
"certmanager/internal/descriptor"
"certmanager/pkg/logger"
"gopkg.in/yaml.v3"
)
func main() {
var err error
util := NewUtil()
err = util.Exec()
if err != nil {
fmt.Printf("Exec error: %s\n", err)
}
}
type Util struct {
conf *config.Config
db *database.Database
log *logger.Logger
filename string
deleteAllRecords bool
}
func NewUtil() *Util {
var util Util
util.log = logger.NewLogger("logic")
return &util
}
func (util *Util) GetOpt() error {
var err error
exeName := filepath.Base(os.Args[0])
help := func() {
fmt.Println("")
fmt.Printf("Usage: %s [option]\n", exeName)
fmt.Printf("\n")
flag.PrintDefaults()
fmt.Printf("\n")
}
flag.Usage = help
flag.StringVar(&util.filename, "file", util.filename, "dump file name")
flag.BoolVar(&util.deleteAllRecords, "deleteAllRecords", util.deleteAllRecords, "delete all existing zones before restoring")
flag.Parse()
return err
}
func (util *Util) Exec() error {
var err error
err = util.GetOpt()
if err != nil {
return err
}
const timeout = 30 * time.Second
ctx, _ := context.WithTimeout(context.Background(), timeout)
err = util.RestoreRecords(ctx)
type ErrorDescr struct {
Error bool `json:"error,omitempty"`
Message string `json:"errorMessage,omitempty" yaml:"errorMessage,omitempty"`
}
errDescr := ErrorDescr{}
if err != nil {
errDescr.Error = true
errDescr.Message = fmt.Sprintf("%v", err)
}
errBytes, _ := yaml.Marshal(errDescr)
fmt.Printf("%s\n", string(errBytes))
return err
}
func (util *Util) RestoreRecords(ctx context.Context) error {
var err error
util.conf = config.NewConfig()
err = util.conf.ReadFile()
if err != nil {
return err
}
err = util.conf.ReadEnv()
if err != nil {
return err
}
db, err := database.NewDatabase(util.conf.DataDir)
if err != nil {
return err
}
util.db = db
err = util.db.InitDatabase(ctx)
if err != nil {
return err
}
file := os.Stdin
if util.filename != "" {
file, err = os.Open(util.filename)
if err != nil {
return err
}
defer file.Close()
}
buffer := bytes.NewBuffer(nil)
_, err = io.Copy(buffer, file)
if err != nil {
return err
}
dump := descriptor.Dump{}
err = yaml.Unmarshal(buffer.Bytes(), &dump)
if err != nil {
return err
}
if util.deleteAllRecords {
err = util.db.CleanDatabase(ctx)
if err != nil {
return err
}
}
for _, issuer := range dump.Issuers {
util.log.Infof("Insert issuer %s", issuer.Name)
err = util.db.InsertIssuer(ctx, &issuer)
if err != nil {
util.log.Errorf("Insert issuer error: %v", err)
}
}
for _, service := range dump.Services {
util.log.Infof("Insert service %s", service.Name)
err = util.db.InsertService(ctx, &service)
if err != nil {
util.log.Errorf("Insert service error: %v", err)
}
}
for _, account := range dump.Accounts {
util.log.Infof("Insert account %s", account.Username)
err = util.db.InsertAccount(ctx, &account)
if err != nil {
util.log.Errorf("Insert account error: %v", err)
}
}
for _, grant := range dump.Grants {
util.log.Infof("Insert grant %s for account %d", grant.Operation, grant.AccountID)
err = util.db.InsertGrant(ctx, &grant)
if err != nil {
util.log.Errorf("Insert account error: %v", err)
}
}
return err
}