chore: fix typos (#232)

2026-05-19 18:58:03 +02:00 · 2023-07-27 21:08:07 +08:00
parent c4ce4fe890
commit 038e97e2dc
20 changed files with 39 additions and 39 deletions
@@ -0,0 +1,137 @@
+package download
+
+import (
+	"database/sql"
+	"os"
+	"sort"
+	"strings"
+	"time"
+
+	// import sqlite3 driver
+	_ "github.com/mattn/go-sqlite3"
+	"github.com/tidwall/gjson"
+
+	"github.com/moond4rk/hackbrowserdata/item"
+	"github.com/moond4rk/hackbrowserdata/log"
+	"github.com/moond4rk/hackbrowserdata/utils/typeutil"
+)
+
+type ChromiumDownload []download
+
+type download struct {
+	TargetPath string
+	URL        string
+	TotalBytes int64
+	StartTime  time.Time
+	EndTime    time.Time
+	MimeType   string
+}
+
+const (
+	queryChromiumDownload = `SELECT target_path, tab_url, total_bytes, start_time, end_time, mime_type FROM downloads`
+)
+
+func (c *ChromiumDownload) Parse(_ []byte) error {
+	db, err := sql.Open("sqlite3", item.TempChromiumDownload)
+	if err != nil {
+		return err
+	}
+	defer os.Remove(item.TempChromiumDownload)
+	defer db.Close()
+	rows, err := db.Query(queryChromiumDownload)
+	if err != nil {
+		return err
+	}
+	defer rows.Close()
+	for rows.Next() {
+		var (
+			targetPath, tabURL, mimeType   string
+			totalBytes, startTime, endTime int64
+		)
+		if err := rows.Scan(&targetPath, &tabURL, &totalBytes, &startTime, &endTime, &mimeType); err != nil {
+			log.Warn(err)
+		}
+		data := download{
+			TargetPath: targetPath,
+			URL:        tabURL,
+			TotalBytes: totalBytes,
+			StartTime:  typeutil.TimeEpoch(startTime),
+			EndTime:    typeutil.TimeEpoch(endTime),
+			MimeType:   mimeType,
+		}
+		*c = append(*c, data)
+	}
+	sort.Slice(*c, func(i, j int) bool {
+		return (*c)[i].TotalBytes > (*c)[j].TotalBytes
+	})
+	return nil
+}
+
+func (c *ChromiumDownload) Name() string {
+	return "download"
+}
+
+func (c *ChromiumDownload) Len() int {
+	return len(*c)
+}
+
+type FirefoxDownload []download
+
+const (
+	queryFirefoxDownload = `SELECT place_id, GROUP_CONCAT(content), url, dateAdded FROM (SELECT * FROM moz_annos INNER JOIN moz_places ON moz_annos.place_id=moz_places.id) t GROUP BY place_id`
+	closeJournalMode     = `PRAGMA journal_mode=off`
+)
+
+func (f *FirefoxDownload) Parse(_ []byte) error {
+	db, err := sql.Open("sqlite3", item.TempFirefoxDownload)
+	if err != nil {
+		return err
+	}
+	defer os.Remove(item.TempFirefoxDownload)
+	defer db.Close()
+
+	_, err = db.Exec(closeJournalMode)
+	if err != nil {
+		log.Error(err)
+	}
+	rows, err := db.Query(queryFirefoxDownload)
+	if err != nil {
+		return err
+	}
+	defer rows.Close()
+	for rows.Next() {
+		var (
+			content, url       string
+			placeID, dateAdded int64
+		)
+		if err = rows.Scan(&placeID, &content, &url, &dateAdded); err != nil {
+			log.Warn(err)
+		}
+		contentList := strings.Split(content, ",{")
+		if len(contentList) > 1 {
+			path := contentList[0]
+			json := "{" + contentList[1]
+			endTime := gjson.Get(json, "endTime")
+			fileSize := gjson.Get(json, "fileSize")
+			*f = append(*f, download{
+				TargetPath: path,
+				URL:        url,
+				TotalBytes: fileSize.Int(),
+				StartTime:  typeutil.TimeStamp(dateAdded / 1000000),
+				EndTime:    typeutil.TimeStamp(endTime.Int() / 1000),
+			})
+		}
+	}
+	sort.Slice(*f, func(i, j int) bool {
+		return (*f)[i].TotalBytes < (*f)[j].TotalBytes
+	})
+	return nil
+}
+
+func (f *FirefoxDownload) Name() string {
+	return "download"
+}
+
+func (f *FirefoxDownload) Len() int {
+	return len(*f)
+}