mirror of
https://github.com/truewhile/MeBox.git
synced 2026-09-28 11:16:37 +08:00
truewhile
42 lines
922 B
Go
42 lines
922 B
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
// SaveDatabaseConfig updates or creates config.yaml with the specified database configuration.
|
|
func SaveDatabaseConfig(dbType, dsn string) error {
|
|
configPath := "config.yaml"
|
|
data := make(map[string]any)
|
|
|
|
content, err := os.ReadFile(configPath)
|
|
if err == nil {
|
|
if err := yaml.Unmarshal(content, &data); err != nil {
|
|
data = make(map[string]any)
|
|
}
|
|
} else if !os.IsNotExist(err) {
|
|
return fmt.Errorf("read config.yaml: %w", err)
|
|
}
|
|
|
|
dbSection, ok := data["database"].(map[string]any)
|
|
if !ok {
|
|
dbSection = make(map[string]any)
|
|
}
|
|
dbSection["type"] = dbType
|
|
dbSection["dsn"] = dsn
|
|
data["database"] = dbSection
|
|
|
|
out, err := yaml.Marshal(data)
|
|
if err != nil {
|
|
return fmt.Errorf("marshal config.yaml: %w", err)
|
|
}
|
|
|
|
if err := os.WriteFile(configPath, out, 0644); err != nil {
|
|
return fmt.Errorf("write config.yaml: %w", err)
|
|
}
|
|
return nil
|
|
}
|