You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
140 lines
3.4 KiB
140 lines
3.4 KiB
package browser
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
|
|
"hack-browser-data/internal/browingdata"
|
|
"hack-browser-data/internal/browser/chromium"
|
|
"hack-browser-data/internal/browser/firefox"
|
|
"hack-browser-data/internal/log"
|
|
"hack-browser-data/internal/utils/fileutil"
|
|
)
|
|
|
|
type Browser interface {
|
|
Name() string
|
|
|
|
GetMasterKey() ([]byte, error)
|
|
// GetBrowsingData returns the browsing data for the browser.
|
|
GetBrowsingData() (*browingdata.Data, error)
|
|
}
|
|
|
|
func PickBrowser(name, profile string) ([]Browser, error) {
|
|
var browsers []Browser
|
|
clist := pickChromium(name, profile)
|
|
for _, b := range clist {
|
|
if b != nil {
|
|
browsers = append(browsers, b)
|
|
}
|
|
}
|
|
flist := pickFirefox(name, profile)
|
|
for _, b := range flist {
|
|
if b != nil {
|
|
browsers = append(browsers, b)
|
|
}
|
|
}
|
|
return browsers, nil
|
|
}
|
|
|
|
func pickChromium(name, profile string) []Browser {
|
|
var browsers []Browser
|
|
name = strings.ToLower(name)
|
|
// TODO: add support for 「all」 flag and set profilePath
|
|
if name == "all" {
|
|
for _, v := range chromiumList {
|
|
if b, err := chromium.New(v.name, v.storage, v.profilePath, v.items); err == nil {
|
|
log.Infof("find browser %s success", b.Name())
|
|
browsers = append(browsers, b)
|
|
} else {
|
|
// TODO: show which browser find failed
|
|
if strings.Contains(err.Error(), "profile path is not exist") {
|
|
log.Infof("find browser %s failed, profile path is not exist", v.name)
|
|
continue
|
|
} else {
|
|
log.Error("new chromium error:", err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if c, ok := chromiumList[name]; ok {
|
|
if profile == "" {
|
|
profile = c.profilePath
|
|
}
|
|
b, err := chromium.New(c.name, c.storage, profile, c.items)
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), "profile path is not exist") {
|
|
log.Infof("find browser %s failed, profile path is not exist", c.name)
|
|
} else {
|
|
log.Error("new chromium error:", err)
|
|
}
|
|
}
|
|
browsers = append(browsers, b)
|
|
}
|
|
return browsers
|
|
}
|
|
|
|
func pickFirefox(name, profile string) []Browser {
|
|
var browsers []Browser
|
|
name = strings.ToLower(name)
|
|
if name == "all" || name == "firefox" {
|
|
for _, v := range firefoxList {
|
|
if profile == "" {
|
|
profile = v.profilePath
|
|
} else {
|
|
profile = fileutil.ParentDir(profile)
|
|
}
|
|
if multiFirefox, err := firefox.New(v.name, v.storage, profile, v.items); err == nil {
|
|
for _, b := range multiFirefox {
|
|
log.Infof("find browser: %s success", b.Name())
|
|
browsers = append(browsers, b)
|
|
}
|
|
} else {
|
|
if strings.Contains(err.Error(), "profile path is not exist") {
|
|
log.Infof("find browser: %s failed, profile path is not exist", v.name)
|
|
} else {
|
|
log.Error(err)
|
|
}
|
|
}
|
|
|
|
}
|
|
return browsers
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ListBrowser() []string {
|
|
var l []string
|
|
for c := range chromiumList {
|
|
l = append(l, c)
|
|
}
|
|
for f := range firefoxList {
|
|
l = append(l, f)
|
|
}
|
|
return l
|
|
}
|
|
|
|
var (
|
|
// home dir path for all platforms
|
|
homeDir, _ = os.UserHomeDir()
|
|
)
|
|
|
|
const (
|
|
chromeName = "Chrome"
|
|
chromeBetaName = "Chrome Beta"
|
|
chromiumName = "Chromium"
|
|
edgeName = "Microsoft Edge"
|
|
speed360Name = "360speed"
|
|
qqBrowserName = "QQ"
|
|
braveName = "Brave"
|
|
operaName = "Opera"
|
|
operaGXName = "OperaGX"
|
|
vivaldiName = "Vivaldi"
|
|
coccocName = "CocCoc"
|
|
yandexName = "Yandex"
|
|
|
|
firefoxName = "Firefox"
|
|
firefoxBetaName = "Firefox Beta"
|
|
firefoxDevName = "Firefox Dev"
|
|
firefoxNightlyName = "Firefox Nightly"
|
|
firefoxESRName = "Firefox ESR"
|
|
)
|
|
|