HackBrowserData/internal/item/item.go

package item

import (
	data2 "hack-browser-data/internal/browingdata"
)

var DefaultFirefox = []Item{
	FirefoxKey4,
	FirefoxPassword,
	FirefoxCookie,
	FirefoxBookmark,
	FirefoxHistory,
	FirefoxDownload,
	FirefoxCreditCard,
	FirefoxLocalStorage,
	FirefoxExtension,
}

var DefaultYandex = []Item{
	ChromiumKey,
	ChromiumCookie,
	ChromiumBookmark,
	ChromiumHistory,
	ChromiumDownload,
	ChromiumLocalStorage,
	ChromiumExtension,
	YandexPassword,
	YandexCreditCard,
}

var DefaultChromium = []Item{
	ChromiumKey,
	ChromiumPassword,
	ChromiumCookie,
	ChromiumBookmark,
	ChromiumHistory,
	ChromiumDownload,
	ChromiumCreditCard,
	ChromiumLocalStorage,
	ChromiumExtension,
}

type Item int

const (
	ChromiumKey Item = iota
	ChromiumPassword
	ChromiumCookie
	ChromiumBookmark
	ChromiumHistory
	ChromiumDownload
	ChromiumCreditCard
	ChromiumLocalStorage
	ChromiumExtension

	YandexPassword
	YandexCreditCard

	FirefoxKey4
	FirefoxPassword
	FirefoxCookie
	FirefoxBookmark
	FirefoxHistory
	FirefoxDownload
	FirefoxCreditCard
	FirefoxLocalStorage
	FirefoxExtension
)

func (i Item) FileName() string {
	switch i {
	case ChromiumKey:
		return fileChromiumKey
	case ChromiumPassword:
		return fileChromiumPassword
	case ChromiumCookie:
		return fileChromiumCookie
	case ChromiumBookmark:
		return fileChromiumBookmark
	case ChromiumDownload:
		return fileChromiumDownload
	case ChromiumLocalStorage:
		return fileChromiumLocalStorage
	case ChromiumCreditCard:
		return fileChromiumCredit
	case ChromiumExtension:
		return UnknownItem
	case ChromiumHistory:
		return fileChromiumHistory
	case YandexPassword:
		return fileYandexPassword
	case YandexCreditCard:
		return fileYandexCredit
	case FirefoxKey4:
		return fileFirefoxKey4
	case FirefoxPassword:
		return fileFirefoxPassword
	case FirefoxCookie:
		return fileFirefoxCookie
	case FirefoxBookmark:
		return fileFirefoxData
	case FirefoxDownload:
		return fileFirefoxData
	case FirefoxLocalStorage:
		return UnsupportedItem
	case FirefoxCreditCard:
		return UnsupportedItem
	case FirefoxHistory:
		return fileFirefoxData
	case FirefoxExtension:
		return UnsupportedItem
	default:
		return UnknownItem
	}
}

func (i Item) String() string {
	switch i {
	case ChromiumKey:
		return "chromiumKey"
	case ChromiumPassword:
		return "password"
	case ChromiumCookie:
		return "cookie"
	case ChromiumBookmark:
		return "bookmark"
	case ChromiumDownload:
		return "download"
	case ChromiumLocalStorage:
		return "localStorage"
	case ChromiumCreditCard:
		return "creditCard"
	case ChromiumExtension:
		return UnsupportedItem
	case ChromiumHistory:
		return "history"
	case YandexPassword:
		return "yandexPassword"
	case YandexCreditCard:
		return "yandexCreditCard"
	case FirefoxKey4:
		return "firefoxKey4"
	case FirefoxPassword:
		return "firefoxPassword"
	case FirefoxCookie:
		return "firefoxCookie"
	case FirefoxBookmark:
		return "firefoxBookmark"
	case FirefoxDownload:
		return "firefoxDownload"
	case FirefoxHistory:
		return "firefoxHistory"
	case FirefoxLocalStorage:
		return UnsupportedItem
	case FirefoxCreditCard:
		return UnsupportedItem
	case FirefoxExtension:
		return UnsupportedItem
	default:
		return UnknownItem
	}
}

// NewBrowsingData returns a new Source instance.
// TODO: remove this function
func (i Item) NewBrowsingData() data2.Source {
	switch i {
	case ChromiumKey:
		return nil
	case ChromiumPassword:
		return &data2.ChromiumPassword{}
	case ChromiumCookie:
		return &data2.ChromiumCookie{}
	case ChromiumBookmark:
		return &data2.ChromiumBookmark{}
	case ChromiumDownload:
		return &data2.ChromiumDownload{}
	case ChromiumLocalStorage:
		return nil
	case ChromiumCreditCard:
		return &data2.ChromiumCreditCard{}
	case ChromiumExtension:
		return nil
	case ChromiumHistory:
		return &data2.ChromiumHistory{}
	case YandexPassword:
		return &data2.ChromiumPassword{}
	case YandexCreditCard:
		return &data2.ChromiumCreditCard{}
	case FirefoxPassword:
		return &data2.FirefoxPassword{}
	case FirefoxCookie:
		return &data2.FirefoxCookie{}
	case FirefoxBookmark:
		return &data2.FirefoxBookmark{}
	case FirefoxDownload:
		return &data2.FirefoxDownload{}
	case FirefoxHistory:
		return &data2.FirefoxHistory{}
	default:
		return nil
	}
}