X-Git-Url: https://ruderich.org/simon/gitweb/?a=blobdiff_plain;f=main.go;h=6322157f2ca9564d6523606095ee0967fd2d8daa;hb=33e06a7ef39ce42d9933ca9106a453daa7eb58ac;hp=05451172164f3783ca6a2469c1b6952b2997e20b;hpb=9231f7dbf189a0d0c19c63556a3d0fedc54d83b0;p=nsscash%2Fnsscash.git diff --git a/main.go b/main.go index 0545117..6322157 100644 --- a/main.go +++ b/main.go @@ -18,8 +18,10 @@ package main import ( + "bytes" "flag" "fmt" + "io/ioutil" "log" "os" ) @@ -28,6 +30,7 @@ func main() { flag.Usage = func() { fmt.Fprintf(os.Stderr, "usage: %[1]s [options] fetch \n"+ + "usage: %[1]s [options] convert \n"+ "", os.Args[0]) flag.PrintDefaults() @@ -46,7 +49,24 @@ func main() { break } - cfg, err := LoadConfig(args[1]) + mainFetch(args[1]) + return + + case "convert": + if len(args) != 4 { + break + } + + mainConvert(args[1], args[2], args[3]) + return + } + + flag.Usage() + os.Exit(1) +} + +func mainFetch(cfgPath string) { + cfg, err := LoadConfig(cfgPath) if err != nil { log.Fatal(err) } @@ -58,13 +78,64 @@ func main() { if err != nil { log.Fatal(err) } - err = WriteStateIfChanged(cfg.StatePath, state) + // NOTE: Make sure to call WriteState() only if there were no + // errors (see WriteState() and README) + err = WriteState(cfg.StatePath, state) if err != nil { log.Fatal(err) } - return - } +} - flag.Usage() - os.Exit(1) +func mainConvert(typ, srcPath, dstPath string) { + var t FileType + err := t.UnmarshalText([]byte(typ)) + if err != nil { + log.Fatal(err) + } + + src, err := ioutil.ReadFile(srcPath) + if err != nil { + log.Fatal(err) + } + var x bytes.Buffer + if t == FileTypePlain { + x.Write(src) + } else if t == FileTypePasswd { + pws, err := ParsePasswds(bytes.NewReader(src)) + if err != nil { + log.Fatal(err) + } + err = SerializePasswds(&x, pws) + if err != nil { + log.Fatal(err) + } + } else if t == FileTypeGroup { + grs, err := ParseGroups(bytes.NewReader(src)) + if err != nil { + log.Fatal(err) + } + err = SerializeGroups(&x, grs) + if err != nil { + log.Fatal(err) + } + } else { + log.Fatalf("unsupported file type %v", t) + } + + // We must create the file first or deployFile() will abort + f, err := os.Create(dstPath) + if err != nil { + log.Fatal(err) + } + f.Close() + + err = deployFile(&File{ + Type: t, + Url: srcPath, + Path: dstPath, + body: x.Bytes(), + }) + if err != nil { + log.Fatal(err) + } }