refactor: format project layout

internal/browser/data/history.go

@@ -0,0 +1,101 @@
+package data
+
+import (
+	"database/sql"
+	"fmt"
+	"sort"
+
+	"hack-browser-data/internal/browser/consts"
+	"hack-browser-data/internal/utils"
+
+	_ "github.com/mattn/go-sqlite3"
+)
+
+type ChromiumHistory []history
+
+func (c *ChromiumHistory) Parse(masterKey []byte) error {
+	historyDB, err := sql.Open("sqlite3", consts.ChromiumHistoryFilename)
+	if err != nil {
+		return err
+	}
+	defer historyDB.Close()
+	rows, err := historyDB.Query(queryChromiumHistory)
+	if err != nil {
+		return err
+	}
+	defer rows.Close()
+	for rows.Next() {
+		var (
+			url, title    string
+			visitCount    int
+			lastVisitTime int64
+		)
+		// TODO: handle rows error
+		if err := rows.Scan(&url, &title, &visitCount, &lastVisitTime); err != nil {
+			fmt.Println(err)
+		}
+		data := history{
+			Url:           url,
+			Title:         title,
+			VisitCount:    visitCount,
+			LastVisitTime: utils.TimeEpochFormat(lastVisitTime),
+		}
+		*c = append(*c, data)
+	}
+	sort.Slice(*c, func(i, j int) bool {
+		return (*c)[i].VisitCount > (*c)[j].VisitCount
+	})
+	return nil
+}
+
+func (c *ChromiumHistory) Name() string {
+	return "history"
+}
+
+type FirefoxHistory []history
+
+func (f *FirefoxHistory) Parse(masterKey []byte) error {
+	var (
+		err         error
+		keyDB       *sql.DB
+		historyRows *sql.Rows
+	)
+	keyDB, err = sql.Open("sqlite3", consts.FirefoxHistoryFilename)
+	if err != nil {
+		return err
+	}
+	_, err = keyDB.Exec(closeJournalMode)
+	if err != nil {
+		return err
+	}
+	defer keyDB.Close()
+	historyRows, err = keyDB.Query(queryFirefoxHistory)
+	if err != nil {
+		return err
+	}
+	defer historyRows.Close()
+	for historyRows.Next() {
+		var (
+			id, visitDate int64
+			url, title    string
+			visitCount    int
+		)
+		if err = historyRows.Scan(&id, &url, &visitDate, &title, &visitCount); err != nil {
+			fmt.Println(err)
+		}
+		*f = append(*f, history{
+			Title:         title,
+			Url:           url,
+			VisitCount:    visitCount,
+			LastVisitTime: utils.TimeStampFormat(visitDate / 1000000),
+		})
+	}
+	sort.Slice(*f, func(i, j int) bool {
+		return (*f)[i].VisitCount < (*f)[j].VisitCount
+	})
+	return nil
+}
+
+func (f *FirefoxHistory) Name() string {
+	return "history"
+}