86 lines
1.8 KiB
Go
86 lines
1.8 KiB
Go
// Copyright © 2022 Roberto Hidalgo <joao@un.rob.mx>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
package config
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"text/template"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type opDetails struct {
|
|
Vault string `yaml:"vault"`
|
|
Name string `yaml:"name"`
|
|
NameTemplate string `yaml:"nameTemplate"` // nolint: tagliatelle
|
|
Repo string
|
|
}
|
|
|
|
type singleModeConfig struct {
|
|
Config *opDetails `yaml:"_config,omitempty"` // nolint: tagliatelle
|
|
}
|
|
|
|
func argIsYAMLFile(path string) bool {
|
|
return strings.HasSuffix(path, ".yaml") || strings.HasSuffix(path, ".yml")
|
|
}
|
|
|
|
func VaultAndNameFrom(path string, buf []byte) (name string, vault string, err error) {
|
|
smc := &singleModeConfig{}
|
|
if buf == nil {
|
|
var err error
|
|
buf, err = os.ReadFile(path)
|
|
if err != nil {
|
|
return "", "", fmt.Errorf("could not read file %s", path)
|
|
}
|
|
}
|
|
|
|
if err = yaml.Unmarshal(buf, &smc); err == nil && smc.Config != nil {
|
|
return smc.Config.Name, smc.Config.Vault, nil
|
|
}
|
|
|
|
rmc, err := findRepoConfig(path)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
|
|
if rmc == nil {
|
|
return "", "", fmt.Errorf("could not find repo config for %s", path)
|
|
}
|
|
|
|
if rmc.NameTemplate == "" {
|
|
rmc.NameTemplate = "{{ DirName }}:{{ FileName}}"
|
|
}
|
|
|
|
logrus.Debugf("Found repo config at %s", rmc.Repo)
|
|
|
|
tpl := template.Must(template.New("help").Funcs(template.FuncMap{
|
|
"DirName": func() string {
|
|
return filepath.Base(filepath.Dir(path))
|
|
},
|
|
"FileName": func() string {
|
|
return strings.Split(filepath.Base(path), ".")[0]
|
|
},
|
|
}).Parse(rmc.NameTemplate))
|
|
|
|
var nameBuf bytes.Buffer
|
|
err = tpl.Execute(&nameBuf, nil)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
return nameBuf.String(), rmc.Vault, nil
|
|
}
|
|
|
|
func isNumeric(s string) bool {
|
|
for _, v := range s {
|
|
if v < '0' || v > '9' {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|