From 1d3b7bd214d0d2e9c60568282fbf64f860d4378c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E1=B4=8D=E1=B4=8F=E1=B4=8F=C9=B4D4=CA=80=E1=B4=8B?=
 <moond4rk13@gmail.com>
Date: Tue, 12 Apr 2022 13:48:03 +0800
Subject: [PATCH] feat: add command line and log

---
 cmd/hack-browser-data/main.go | 144 ++++++++++++++++++++++++++++++++++
 internal/log/log.go           |   8 ++
 2 files changed, 152 insertions(+)
 create mode 100644 cmd/hack-browser-data/main.go

diff --git a/cmd/hack-browser-data/main.go b/cmd/hack-browser-data/main.go
new file mode 100644
index 0000000..0024368
--- /dev/null
+++ b/cmd/hack-browser-data/main.go
@@ -0,0 +1,144 @@
+package main
+
+import (
+	"fmt"
+	"os"
+
+	"hack-browser-data/internal/browser"
+	"hack-browser-data/internal/log"
+	"hack-browser-data/internal/outputter"
+)
+
+var (
+	browserName       string
+	exportDir         string
+	outputFormat      string
+	verbose           bool
+	compress          bool
+	customProfilePath string
+)
+
+func main() {
+	Execute()
+}
+
+func Execute() {
+	browsers := browser.PickBrowser("firefox")
+	log.InitLog("debug")
+	filetype := "json"
+	dir := "results"
+	// dir := "result"
+	output := outputter.New(filetype)
+	var file *os.File
+	fmt.Println("browser number:", len(browsers))
+	for _, b := range browsers {
+		browsingData, err := b.GetBrowsingData()
+		if err != nil {
+			panic(err)
+		}
+		for _, source := range browsingData.Sources {
+			filename := fmt.Sprintf("%s_%s.%s", b.Name(), source.Name(), filetype)
+			file, err = output.CreateFile(dir, filename)
+			err = output.Write(source, file)
+			if err != nil {
+				log.Error(err)
+			}
+		}
+	}
+}
+
+// func Execute() {
+// 	app := &cli.App{
+// 		Name:  "hack-browser-data",
+// 		Usage: "Export passwords/cookies/history/bookmarks from browser",
+// 		UsageText: "[hack-browser-data -b chrome -f json -dir results -cc]\n 	Get all browingdata(password/cookie/history/bookmark) from chrome",
+// 		Version: "0.3.7",
+// 		Flags: []cli.Flag{
+// 			&cli.BoolFlag{Name: "verbose", Aliases: []string{"vv"}, Destination: &verbose, Value: false, Usage: "verbose"},
+// 			&cli.BoolFlag{Name: "compress", Aliases: []string{"cc"}, Destination: &compress, Value: false, Usage: "compress result to zip"},
+// 			&cli.StringFlag{Name: "browser", Aliases: []string{"b"}, Destination: &browserName, Value: "all", Usage: "available browsers: all|" + strings.Join(core.ListBrowser(), "|")},
+// 			&cli.StringFlag{Name: "results-dir", Aliases: []string{"dir"}, Destination: &exportDir, Value: "results", Usage: "export dir"},
+// 			&cli.StringFlag{Name: "format", Aliases: []string{"f"}, Destination: &outputFormat, Value: "csv", Usage: "format, csv|json|console"},
+// 			&cli.StringFlag{Name: "profile-dir-path", Aliases: []string{"p"}, Destination: &customProfilePath, Value: "", Usage: "custom profile dir path, get with chrome://version"},
+// 		},
+// 		HideHelpCommand: true,
+// 		Action: func(c *cli.Context) error {
+// 			var (
+// 				browsers []core.Browser
+// 				err      error
+// 			)
+// 			if verbose {
+// 				log.InitLog("debug")
+// 			} else {
+// 				log.InitLog("error")
+// 			}
+// 			if customProfilePath != "" {
+// 				browsers, err = core.PickCustomBrowser(browserName, customProfilePath, customKeyPath)
+// 				if err != nil {
+// 					log.Error(err)
+// 				}
+// 			} else {
+// 				// default select all browsers
+// 				browsers, err = core.PickBrowser(browserName)
+// 				if err != nil {
+// 					log.Error(err)
+// 				}
+// 			}
+// 			err = utils.MakeDir(exportDir)
+// 			if err != nil {
+// 				log.Error(err)
+// 			}
+// 			for _, browser := range browsers {
+// 				err := browser.InitSecretKey()
+// 				if err != nil {
+// 					log.Error(err)
+// 				}
+// 				// default select all items
+// 				// you can get single item with browser.GetItem(itemName)
+// 				items, err := browser.GetAllItems()
+// 				if err != nil {
+// 					log.Error(err)
+// 				}
+// 				name := browser.Name()
+// 				key := browser.GetSecretKey()
+// 				for _, item := range items {
+// 					err := item.CopyDB()
+// 					if err != nil {
+// 						log.Error(err)
+// 					}
+// 					switch browser.(type) {
+// 					case *core.Chromium:
+// 						err := item.ChromeParse(key)
+// 						if err != nil {
+// 							log.Error(err)
+// 						}
+// 					case *core.Firefox:
+// 						err := item.FirefoxParse()
+// 						if err != nil {
+// 							log.Error(err)
+// 						}
+// 					}
+// 					err = item.Release()
+// 					if err != nil {
+// 						log.Error(err)
+// 					}
+// 					err = item.OutPut(outputFormat, name, exportDir)
+// 					if err != nil {
+// 						log.Error(err)
+// 					}
+// 				}
+// 			}
+// 			if compress {
+// 				err = utils.Compress(exportDir)
+// 				if err != nil {
+// 					log.Error(err)
+// 				}
+// 			}
+// 			return nil
+// 		},
+// 	}
+// 	err := app.Run(os.Args)
+// 	if err != nil {
+// 		log.Error(err)
+// 	}
+// }
diff --git a/internal/log/log.go b/internal/log/log.go
index 6b6c916..c03e23a 100644
--- a/internal/log/log.go
+++ b/internal/log/log.go
@@ -97,3 +97,11 @@ func Warnf(format string, v ...interface{}) {
 func Debugf(format string, v ...interface{}) {
 	formatLogger.doLogf(LevelDebug, format, v...)
 }
+
+// NewSugaredLogger(os.Stdout, DebugLevel).Configure(func(sl *SugaredLogger) {
+// 	sl.SetName("stdLogger")
+// 	sl.ReportCaller = true
+// 	// auto enable console color
+// 	sl.Formatter.(*TextFormatter).EnableColor = color.SupportColor()
+//  sl.Formatter.SetCallerSkip(1)
+// })