nsscash: split main() into separate functions

[nsscash/nsscash.git] / main.go

diff --git a/main.go b/main.go
index cef6c61d41bfe9dfef0c8995ab25cf8b4762d7c9..6322157f2ca9564d6523606095ee0967fd2d8daa 100644 (file)
--- a/main.go
+++ b/main.go
@@ -49,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)
                }
@@ -61,39 +78,52 @@ 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
-
-       case "convert":
-               if len(args) != 4 {
-                       break
-               }
+}
 
+func mainConvert(typ, srcPath, dstPath string) {
                var t FileType
-               err := t.UnmarshalText([]byte(args[1]))
+               err := t.UnmarshalText([]byte(typ))
                if err != nil {
                        log.Fatal(err)
                }
 
-               src, err := ioutil.ReadFile(args[2])
-               if err != nil {
-                       log.Fatal(err)
-               }
-               pws, err := ParsePasswds(bytes.NewReader(src))
+               src, err := ioutil.ReadFile(srcPath)
                if err != nil {
                        log.Fatal(err)
                }
                var x bytes.Buffer
-               err = SerializePasswds(&x, pws)
-               if err != nil {
-                       log.Fatal(err)
+               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(args[3])
+               f, err := os.Create(dstPath)
                if err != nil {
                        log.Fatal(err)
                }
@@ -101,16 +131,11 @@ func main() {
 
                err = deployFile(&File{
                        Type: t,
-                       Url:  args[2],
-                       Path: args[3],
+                       Url:  srcPath,
+                       Path: dstPath,
                        body: x.Bytes(),
                })
                if err != nil {
                        log.Fatal(err)
                }
-               return
-       }
-
-       flag.Usage()
-       os.Exit(1)
 }