feat(keys): add keys import subcommand (#601)

2026-06-04 19:48:01 +02:00 · 2026-05-24 20:41:30 +08:00
parent b468c5d0dc
commit d5dc81f1c0
3 changed files with 142 additions and 94 deletions
@@ -0,0 +1,37 @@
+package main
+
+import (
+	"fmt"
+	"path/filepath"
+
+	"github.com/moond4rk/hackbrowserdata/browser"
+	"github.com/moond4rk/hackbrowserdata/log"
+	"github.com/moond4rk/hackbrowserdata/output"
+	"github.com/moond4rk/hackbrowserdata/types"
+	"github.com/moond4rk/hackbrowserdata/utils/fileutil"
+)
+
+func extractAndWrite(browsers []browser.Browser, categories []types.Category, outputDir, outputFormat string, compress bool) error {
+	w, err := output.NewWriter(outputDir, outputFormat)
+	if err != nil {
+		return err
+	}
+	for _, b := range browsers {
+		log.Infof("Extracting %s/%s...", b.BrowserName(), b.ProfileName())
+		data, extractErr := b.Extract(categories)
+		if extractErr != nil {
+			log.Errorf("extract %s/%s: %v", b.BrowserName(), b.ProfileName(), extractErr)
+		}
+		w.Add(b.BrowserName(), b.ProfileName(), data)
+	}
+	if err := w.Write(); err != nil {
+		return err
+	}
+	if compress {
+		if err := fileutil.CompressDir(outputDir); err != nil {
+			return fmt.Errorf("compress: %w", err)
+		}
+		log.Infof("Compressed: %s/%s.zip", outputDir, filepath.Base(outputDir))
+	}
+	return nil
+}