evcc-io/packaging/toml.go
2023-12-06 18:29:32 +01:00

98 lines
1.8 KiB
Go

package main
import (
"bufio"
"bytes"
"fmt"
"io/fs"
"os"
"path/filepath"
"strings"
"github.com/BurntSushi/toml"
)
func process(filepath string) error {
in, err := os.ReadFile(filepath)
if err != nil {
return err
}
pre := new(strings.Builder)
{
sc := bufio.NewScanner(bytes.NewReader(in))
for sc.Scan() {
col := strings.SplitN(sc.Text(), "=", 2)
if len(col) < 2 {
fmt.Fprintln(pre, sc.Text())
continue
}
key := strings.TrimSpace(col[0])
val := strings.TrimSpace(col[1])
if strings.HasPrefix(val, "\"") {
fmt.Fprintln(pre, sc.Text())
continue
}
quote := `"`
if strings.Contains(val, quote) {
quote = `'`
}
fmt.Fprintf(pre, "%s = %s%s%s\n", key, quote, val, quote)
}
}
var config map[string]any
if _, err := toml.Decode(pre.String(), &config); err != nil {
return fmt.Errorf("%s: %v", filepath, err)
}
s := new(strings.Builder)
enc := toml.NewEncoder(s)
enc.Indent = ""
if err := enc.Encode(config); err != nil {
return fmt.Errorf("%s: %v", filepath, err)
}
out := new(strings.Builder)
sc := bufio.NewScanner(strings.NewReader(s.String()))
for sc.Scan() {
if strings.HasPrefix(sc.Text(), "[") && strings.Contains(sc.Text(), ".") && out.Len() > 0 {
fmt.Fprintln(out, "")
}
fmt.Fprintln(out, sc.Text())
}
if err := os.WriteFile(filepath, []byte(out.String()), 0o644); err != nil {
return fmt.Errorf("%s: %v", filepath, err)
}
return nil
}
func main() {
var err error
if len(os.Args) > 1 {
err = process(os.Args[1])
} else {
err = filepath.WalkDir("./i18n", func(filepath string, d fs.DirEntry, err error) error {
if err != nil {
return fmt.Errorf("%s: %v", filepath, err)
}
if d.IsDir() {
return nil
}
return process(filepath)
})
}
if err != nil {
panic(err)
}
}