browser/browser.go

@@ -87,13 +87,11 @@ func pickFirefox(name, profile string) []Browser {
 			} else {
 				profile = fileutil.ParentDir(profile)
 			}
-
 			if !fileutil.IsDirExists(filepath.Clean(profile)) {
 				log.Noticef("find browser firefox %s failed, profile folder does not exist", v.name)
 				continue
 			}
-
+			if multiFirefox, err := firefox.New(v.name, v.storage, profile, v.items); err == nil {
-			if multiFirefox, err := firefox.New(profile, v.items); err == nil {
 				for _, b := range multiFirefox {
 					log.Noticef("find browser firefox %s success", b.Name())
 					browsers = append(browsers, b)
@@ -102,10 +100,8 @@ func pickFirefox(name, profile string) []Browser {
 				log.Error(err)
 			}
 		}
-
 		return browsers
 	}
-
 	return nil
 }
 

browser/firefox/firefox.go

@@ -23,11 +23,18 @@ type Firefox struct {
 
 var ErrProfilePathNotFound = errors.New("profile path not found")
 
-// New returns new Firefox instances.
+// New returns a new Firefox instance.
-func New(profilePath string, items []item.Item) ([]*Firefox, error) {
+func New(name, storage, profilePath string, items []item.Item) ([]*Firefox, error) {
-	multiItemPaths := make(map[string]map[item.Item]string)
+	f := &Firefox{
-	// ignore walk dir error since it can be produced by a single entry
+		name:        name,
-	_ = filepath.WalkDir(profilePath, firefoxWalkFunc(items, multiItemPaths))
+		storage:     storage,
+		profilePath: profilePath,
+		items:       items,
+	}
+	multiItemPaths, err := f.getMultiItemPath(f.profilePath, f.items)
+	if err != nil {
+		return nil, err
+	}
 
 	firefoxList := make([]*Firefox, 0, len(multiItemPaths))
 	for name, itemPaths := range multiItemPaths {
@@ -37,10 +44,15 @@ func New(profilePath string, items []item.Item) ([]*Firefox, error) {
 			itemPaths: itemPaths,
 		})
 	}
-
 	return firefoxList, nil
 }
 
+func (f *Firefox) getMultiItemPath(profilePath string, items []item.Item) (map[string]map[item.Item]string, error) {
+	multiItemPaths := make(map[string]map[item.Item]string)
+	err := filepath.Walk(profilePath, firefoxWalkFunc(items, multiItemPaths))
+	return multiItemPaths, err
+}
+
 func (f *Firefox) copyItemToLocal() error {
 	for i, path := range f.itemPaths {
 		filename := i.String()
@@ -51,8 +63,8 @@ func (f *Firefox) copyItemToLocal() error {
 	return nil
 }
 
-func firefoxWalkFunc(items []item.Item, multiItemPaths map[string]map[item.Item]string) fs.WalkDirFunc {
+func firefoxWalkFunc(items []item.Item, multiItemPaths map[string]map[item.Item]string) filepath.WalkFunc {
-	return func(path string, info fs.DirEntry, err error) error {
+	return func(path string, info fs.FileInfo, err error) error {
 		for _, v := range items {
 			if info.Name() == v.FileName() {
 				parentBaseDir := fileutil.ParentBaseDir(path)
@@ -63,7 +75,6 @@ func firefoxWalkFunc(items []item.Item, multiItemPaths map[string]map[item.Item]
 				}
 			}
 		}
-
 		return err
 	}
 }

cmd/hack-browser-data/main.go

@@ -53,7 +53,6 @@ func Execute() {
 				data, err := b.BrowsingData(isFullExport)
 				if err != nil {
 					log.Error(err)
-					continue
 				}
 				data.Output(outputDir, b.Name(), outputFormat)
 			}