forked from TrueCloudLab/frostfs-node
d189d60925
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
124 lines
3 KiB
Go
124 lines
3 KiB
Go
package config
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"text/template"
|
|
|
|
"github.com/nspcc-dev/neofs-node/pkg/innerring"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
type configTemplate struct {
|
|
Endpoint string
|
|
AlphabetDir string
|
|
MaxObjectSize int
|
|
EpochDuration int
|
|
Glagolitics []string
|
|
}
|
|
|
|
const configTxtTemplate = `rpc-endpoint: {{ .Endpoint}}
|
|
alphabet-wallets: {{ .AlphabetDir}}
|
|
network:
|
|
max_object_size: {{ .MaxObjectSize}}
|
|
epoch_duration: {{ .EpochDuration}}
|
|
# if credentials section is omitted, then neofs-adm will require manual password input
|
|
credentials:{{ range.Glagolitics}}
|
|
{{.}}: password{{end}}
|
|
`
|
|
|
|
func initConfig(cmd *cobra.Command, args []string) error {
|
|
configPath, err := readConfigPathFromArgs(cmd)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
pathDir := path.Dir(configPath)
|
|
err = os.MkdirAll(pathDir, 0700)
|
|
if err != nil {
|
|
return fmt.Errorf("create dir %s: %w", pathDir, err)
|
|
}
|
|
|
|
f, err := os.OpenFile(configPath, os.O_RDWR|os.O_CREATE|os.O_TRUNC|os.O_SYNC, 0600)
|
|
if err != nil {
|
|
return fmt.Errorf("open %s: %w", configPath, err)
|
|
}
|
|
defer f.Close()
|
|
|
|
configText, err := generateConfigExample(pathDir, 7)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = f.WriteString(configText)
|
|
if err != nil {
|
|
return fmt.Errorf("writing to %s: %w", configPath, err)
|
|
}
|
|
|
|
cmd.Printf("Initial config file saved to %s\n", configPath)
|
|
|
|
return nil
|
|
}
|
|
|
|
func readConfigPathFromArgs(cmd *cobra.Command) (string, error) {
|
|
configPath, err := cmd.Flags().GetString(configPathFlag)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if configPath != "" {
|
|
return configPath, nil
|
|
}
|
|
|
|
return defaultConfigPath()
|
|
}
|
|
|
|
func defaultConfigPath() (string, error) {
|
|
home, err := os.UserHomeDir()
|
|
if err != nil {
|
|
return "", fmt.Errorf("getting home dir path: %w", err)
|
|
}
|
|
|
|
return path.Join(home, ".neofs", "adm", "config.yml"), nil
|
|
}
|
|
|
|
// generateConfigExample builds .yml representation of the config file. It is
|
|
// easier to build it manually with template instead of using viper, because we
|
|
// want to order records in specific order in file and, probably, provide
|
|
// some comments as well.
|
|
func generateConfigExample(appDir string, credSize int) (string, error) {
|
|
input := configTemplate{
|
|
Endpoint: "https://neo.rpc.node:30333",
|
|
MaxObjectSize: 67108864, // 64 MiB
|
|
EpochDuration: 240, // 1 hour with 15s per block
|
|
Glagolitics: make([]string, 0, credSize),
|
|
}
|
|
|
|
appDir, err := filepath.Abs(appDir)
|
|
if err != nil {
|
|
return "", fmt.Errorf("making absolute path for %s: %w", appDir, err)
|
|
}
|
|
input.AlphabetDir = path.Join(appDir, "alphabet-wallets")
|
|
|
|
var i innerring.GlagoliticLetter
|
|
for i = 0; i < innerring.GlagoliticLetter(credSize); i++ {
|
|
input.Glagolitics = append(input.Glagolitics, i.String())
|
|
}
|
|
|
|
t, err := template.New("config.yml").Parse(configTxtTemplate)
|
|
if err != nil {
|
|
return "", fmt.Errorf("parsing config template: %w", err)
|
|
}
|
|
|
|
buf := bytes.NewBuffer(nil)
|
|
|
|
err = t.Execute(buf, input)
|
|
if err != nil {
|
|
return "", fmt.Errorf("generating config from tempalte: %w", err)
|
|
}
|
|
|
|
return buf.String(), nil
|
|
}
|