feat: export and import sites and categories (#895)

Author: sheepzh

src/api/sw/merge.ts

@@ -4,4 +4,4 @@ export const listAllMergeRules = () => sendMsg2Runtime('merge.all')
 
 export const deleteMergeRule = (origin: string) => sendMsg2Runtime('merge.delete', origin)
 
-export const addMergeRule = (rule: tt4b.merge.Rule) => sendMsg2Runtime('merge.add', rule)
+export const addMergeRule = (rule: tt4b.site.MergeRule) => sendMsg2Runtime('merge.add', rule)

src/background/content-script-handler.ts

@@ -6,7 +6,7 @@
  */
 
 import { getTab } from '@api/chrome/tab'
-import { saveSite } from '@service/site-service'
+import { getSite, saveSite } from '@service/site-service'
 import { IS_ANDROID, IS_CHROME, IS_FIREFOX, IS_SAFARI, isNotTrackable } from "@util/constant/environment"
 import { extractHostname, isHomepage } from "@util/pattern"
 import { extractSiteName } from "@util/site"
@@ -30,7 +30,11 @@ async function processTabInfo(tab: ChromeTab): Promise<void> {
     IS_CHROME && /^localhost(:.+)?/.test(host) && (iconUrl = undefined)
     // Only collect site name for homepage
     const alias = isHomepage(url) ? extractSiteName(title) : undefined
-    await saveSite({ host, type: 'normal', alias, iconUrl }, false)
+    if (!iconUrl && !alias) return
+    const exist = await getSite({ host, type: 'normal' })
+    exist.iconUrl ??= iconUrl
+    exist.alias ??= alias
+    await saveSite(exist)
 }
 
 /**

src/background/database/cate-database.ts

@@ -54,7 +54,7 @@ class CateDatabase extends BaseDatabase {
 
         const ids = Object.keys(items).map(Number).filter(Number.isFinite)
         const id = (ids.length ? Math.max(...ids) : 0) + 1
-        items[id] = {n: name}
+        items[id] = { n: name }
 
         await this.saveItems(items)
         return { name, id }

src/background/database/common/constant.ts

@@ -12,12 +12,6 @@
  */
 export const REMAIN_WORD_PREFIX = '__timer__'
 
-/**
- * Key of white list
- * @since 0.0.5
- */
-export const WHITELIST_KEY = REMAIN_WORD_PREFIX + 'WHITELIST'
-
 /**
  * Key of metadata
  * @since 0.6.0

src/background/database/merge-rule-database.ts

@@ -20,7 +20,7 @@ const isMergeValue = createUnionGuard(isString, isInt)
 
 const isMergeRuleSet = createRecordGuard(isMergeValue)
 
-const isMergeRule = createObjectGuard<tt4b.merge.Rule>({
+const isMergeRule = createObjectGuard<tt4b.site.MergeRule>({
     origin: isString,
     merged: isMergeValue,
 })
@@ -42,10 +42,9 @@ class MergeRuleDatabase extends BaseDatabase implements BrowserMigratable<'__mer
         return this.setByKey(DB_KEY, data)
     }
 
-    async selectAll(): Promise<tt4b.merge.Rule[]> {
+    async selectAll(): Promise<tt4b.site.MergeRule[]> {
         const set = await this.refresh()
-        return Object.entries(set)
-            .map(([origin, merged]) => ({ origin, merged } satisfies tt4b.merge.Rule))
+        return Object.entries(set).map(([origin, merged]) => ({ origin, merged }))
     }
 
     async remove(origin: string): Promise<void> {
@@ -57,7 +56,7 @@ class MergeRuleDatabase extends BaseDatabase implements BrowserMigratable<'__mer
     /**
      * Add to the db
      */
-    async add(...toAdd: tt4b.merge.Rule[]): Promise<void> {
+    async add(...toAdd: tt4b.site.MergeRule[]): Promise<void> {
         const set = await this.refresh()
         // Not rewrite
         toAdd.forEach(({ origin, merged }) => set[origin] = set[origin] ?? merged)
@@ -87,7 +86,7 @@ class MergeRuleDatabase extends BaseDatabase implements BrowserMigratable<'__mer
         await this.update(exist)
     }
 
-    exportData(): Promise<tt4b.merge.Rule[]> {
+    exportData(): Promise<tt4b.site.MergeRule[]> {
         return this.selectAll()
     }
 }

…ckground/database/site-database/index.ts src/background/database/site-database.tssrc/background/database/site-database/index.ts renamed to src/background/database/site-database.ts src/background/database/site-database/index.ts renamed to src/background/database/site-database.ts

@@ -6,7 +6,7 @@
  */
 
 import { CATE_NOT_SET_ID } from '@util/site'
-import { BaseIDBStorage, iterateCursor, type Key, req2Promise, type Table } from '../common/indexed-storage'
+import { BaseIDBStorage, iterateCursor, type Key, req2Promise, type Table } from './common/indexed-storage'
 
 const INDEXES: (Key<tt4b.site.SiteInfo> | Key<tt4b.site.SiteInfo>[])[] = [
     'host',

src/background/database/site-database/classic.ts

@@ -9,7 +9,6 @@ import { getVersion } from "@api/chrome/runtime"
 import CateInitializer from "./cate-initializer"
 import HostMergeInitializer from "./host-merge-initializer"
 import LocalFileInitializer from "./local-file-initializer"
-import SiteMigrator from './site-migrator'
 import type { Migrator } from "./types"
 import WhitelistInitializer from "./whitelist-initializer"
 
@@ -27,7 +26,6 @@ class VersionManager {
             new LocalFileInitializer(),
             new WhitelistInitializer(),
             new CateInitializer(),
-            new SiteMigrator(),
         )
     }
 

src/background/database/whitelist-database.ts

@@ -34,7 +34,7 @@ export default class LocalFileInitializer implements Migrator {
             [TXT_HOST]: t(msg => msg.initial.localFile.txt),
         }
         for (const [host, alias] of Object.entries(hostAlias)) {
-            void saveSite({ host, type: 'normal', alias, iconUrl: undefined }, true)
+            void saveSite({ host, type: 'normal', alias, iconUrl: undefined })
         }
     }
 }