add data export to cli

cmd/user_data.go

@@ -1,10 +1,13 @@
 package cmd
 
 import (
+	"os"
+
 	"github.com/urfave/cli/v2"
 
 	clis "github.com/mayswind/lab/pkg/cli"
 	"github.com/mayswind/lab/pkg/log"
+	"github.com/mayswind/lab/pkg/utils"
 )
 
 // UserData represents the data command
@@ -24,6 +27,23 @@ var UserData = &cli.Command{
 				},
 			},
 		},
+		{
+			Name:   "export",
+			Usage:  "Export user all transactions to csv file",
+			Action: exportUserTransaction,
+			Flags: []cli.Flag{
+				&cli.StringFlag{
+					Name:    "username",
+					Aliases: []string{"n"},
+					Usage:   "Specific user name",
+				},
+				&cli.StringFlag{
+					Name:    "file",
+					Aliases: []string{"f"},
+					Usage:   "Specific exported file path (e.g. transaction.csv)",
+				},
+			},
+		},
 	},
 }
 
@@ -55,3 +75,53 @@ func checkUserTransactionAndAccount(c *cli.Context) error {
 
 	return nil
 }
+
+func exportUserTransaction(c *cli.Context) error {
+	_, err := initializeSystem(c)
+
+	if err != nil {
+		return err
+	}
+
+	userName := c.String("username")
+	uid, err := clis.UserData.GetUserIdByUsername(c, userName)
+
+	if err != nil {
+		log.BootErrorf("[user_data.exportUserTransaction] error occurs when getting user id by user name")
+		return err
+	}
+
+	filePath := c.String("file")
+
+	if filePath == "" {
+		log.BootErrorf("[user_data.exportUserTransaction] export file path is not specified")
+		return os.ErrNotExist
+	}
+
+	fileExists, err := utils.IsExists(filePath)
+
+	if fileExists {
+		log.BootErrorf("[user_data.exportUserTransaction] specified file path already exists")
+		return os.ErrExist
+	}
+
+	log.BootInfof("[user_data.exportUserTransaction] starting exporting user \"%s\" data", userName)
+
+	content, err := clis.UserData.ExportTransaction(c, uid)
+
+	if err != nil {
+		log.BootErrorf("[user_data.exportUserTransaction] error occurs when exporting user data")
+		return err
+	}
+
+	err = utils.WriteFile(filePath, content)
+
+	if err != nil {
+		log.BootErrorf("[user_data.exportUserTransaction] failed to write to %s", filePath)
+		return err
+	}
+
+	log.BootInfof("[user_data.exportUserTransaction] user transactions have been exported to %s", filePath)
+
+	return nil
+}