refactor: Optimize traversal browser data logic (#311)

* refactor: Refactor package names and imports for better code organization.
* refactor: Package imports and variable types for consistency
* chore: Disable unused-parameter rule in revive.
* refactor: Refactor and organize data extraction and browserdata parse.
* fix: rename wrong error message info

types/types_test.go

@@ -0,0 +1,129 @@
+package types
+
+import (
+	"fmt"
+	"os"
+	"strconv"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestDataType_FileName(t *testing.T) {
+	for _, item := range DefaultChromiumTypes {
+		assert.Equal(t, item.Filename(), item.filename())
+	}
+	for _, item := range DefaultFirefoxTypes {
+		assert.Equal(t, item.Filename(), item.filename())
+	}
+	for _, item := range DefaultYandexTypes {
+		assert.Equal(t, item.Filename(), item.filename())
+	}
+}
+
+func TestDataType_TempFilename(t *testing.T) {
+	asserts := assert.New(t)
+
+	testCases := []struct {
+		item     DataType
+		expected string
+	}{
+		{ChromiumKey, "Local State"},
+		{ChromiumPassword, "Login Data"},
+		{ChromiumLocalStorage, "Local Storage/leveldb"},
+		{FirefoxSessionStorage, "unsupported item"},
+		{FirefoxLocalStorage, "webappsstore.sqlite"},
+		{YandexPassword, "Ya Passman Data"},
+		{YandexCreditCard, "Ya Credit Cards"},
+	}
+
+	for _, tc := range testCases {
+		expectedPrefix := tc.expected + "_" + strconv.Itoa(int(tc.item)) + ".temp"
+		actualPath := tc.item.TempFilename()
+		asserts.Contains(actualPath, expectedPrefix, "TempFilename should contain the correct prefix for "+tc.expected)
+		asserts.Contains(actualPath, os.TempDir(), "TempFilename should be in the system temp directory for "+tc.expected)
+	}
+}
+
+func TestDataType_IsSensitive(t *testing.T) {
+	asserts := assert.New(t)
+	testCases := []struct {
+		item     DataType
+		expected bool
+	}{
+		{ChromiumKey, true},
+		{ChromiumPassword, true},
+		{ChromiumBookmark, false},
+	}
+	for _, tc := range testCases {
+		asserts.Equal(tc.expected, tc.item.IsSensitive(), fmt.Sprintf("IsSensitive for %v should be %v", tc.item, tc.expected))
+	}
+}
+
+func TestFilterSensitiveItems(t *testing.T) {
+	asserts := assert.New(t)
+	testCases := []struct {
+		items    []DataType
+		expected int
+	}{
+		{[]DataType{ChromiumKey, ChromiumBookmark, ChromiumPassword}, 2},
+		{[]DataType{ChromiumBookmark, ChromiumHistory}, 0},
+	}
+
+	for _, tc := range testCases {
+		filteredItems := FilterSensitiveItems(tc.items)
+		asserts.Len(filteredItems, tc.expected, "FilterSensitiveItems should return the correct number of sensitive items")
+		for _, item := range filteredItems {
+			asserts.True(item.IsSensitive(), "Filtered items should be sensitive")
+		}
+	}
+}
+
+func (i DataType) 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 ChromiumSessionStorage:
+		return fileChromiumSessionStorage
+	case ChromiumCreditCard:
+		return fileChromiumCredit
+	case ChromiumExtension:
+		return fileChromiumExtension
+	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 fileFirefoxLocalStorage
+	case FirefoxHistory:
+		return fileFirefoxData
+	case FirefoxExtension:
+		return fileFirefoxExtension
+	case FirefoxCreditCard:
+		return UnsupportedItem
+	default:
+		return UnsupportedItem
+	}
+}