HackBrowserData/cmd/hack-browser-data/extract.go

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
}