BewlyBewly/src/background/main.ts

import { createSharedComposable } from '@vueuse/core'
import { sendMessage, onMessage } from 'webext-bridge'
import { browserSettings, Tabs } from 'webextension-polyfill'

// only on dev mode
if (import.meta.hot) {
  // @ts-expect-error for background HMR
  import('/@vite/client')
  // load latest content script
  import('./contentScriptHMR')
}

let previousTabId = 0

// communication example: send previous tab title from background page
// see shim.d.ts for type declaration
browser.tabs.onActivated.addListener(async({ tabId }) => {
  if (!previousTabId) {
    previousTabId = tabId
    return
  }

  let tab: Tabs.Tab

  try {
    tab = await browser.tabs.get(previousTabId)
    previousTabId = tabId
  }
  catch {
    return
  }

  // eslint-disable-next-line no-console
  console.log('previous tab', tab)
  sendMessage('tab-prev', { title: tab.title }, { context: 'content-script', tabId })
})

onMessage('get-current-tab', async() => {
  try {
    const tab = await browser.tabs.get(previousTabId)
    return {
      title: tab?.title,
    }
  }
  catch {
    return {
      title: undefined,
    }
  }
})

// preinsert css
browser.tabs.onUpdated.addListener((tabId: number, changInfo: Tabs.OnUpdatedChangeInfoType, tab: Tabs.Tab) => {
  if (/https?:\/\/bilibili.com\/?$/.test(`${tab.url}`)
  || /https?:\/\/www.bilibili.com\/?$/.test(`${tab.url}`)
  || /https?:\/\/bilibili.com\/\?spm_id_from=.*/.test(`${tab.url}`)
  || /https?:\/\/www.bilibili.com\/\?spm_id_from=(.)*/.test(`${tab.url}`)) {
    if (changInfo.status === 'loading') {
      const css = `
      body {
        opacity: 0;
        transition: opacity 0.5s;
        overflow-y: hidden;
        pointer-events: none;
      }
      `

      browser.tabs.insertCSS(tabId, {
        code: css,
        runAt: 'document_start',
        matchAboutBlank: true,
      })
    }

    else if (changInfo.status === 'complete') {
      const css = `
      body {
        opacity: 1;
        overflow-y: auto;
        pointer-events: auto;
      }
      `

      browser.tabs.insertCSS(tabId, {
        code: css,
        runAt: 'document_start',
        matchAboutBlank: true,
      })
    }
  }
})

chrome.runtime.onMessage.addListener((message: any, sender: chrome.runtime.MessageSender, sendResponse: any) => {
  const APP_URL = 'https://app.bilibili.com'
  const API_URL = 'https://api.bilibili.com'

  if (message.contentScriptQuery === 'getAccessKey') {
    const url = message.confirmUri
    fetch(url)
      .then(response => sendResponse({ accessKey: `${response.url}`.match(/access_key=([0-9a-z]{32})/)![1] }))
      .catch(error => console.error(error))
    return true
  }
  if (message.contentScriptQuery === 'getRecommendVideo') {
    const url = `${APP_URL}/x/feed/index?build=1&idx=${message.idx}&appkey=27eb53fc9058f8c3&access_key=${message.accessKey}`
    fetch(url)
      .then(response => response.json())
      .then(data => sendResponse(data))
      .catch(error => console.error(error))
    return true
  }
  if (message.contentScriptQuery === 'getUserInfo') {
    // https://api.bilibili.com/x/web-interface/nav
    // const url = `${API_URL}/x/web-interface/card?mid=${message.mid}&photo=true`
    const url = `${API_URL}/x/web-interface/nav`
    fetch(url)
      .then(response => response.json())
      .then(data => sendResponse(data))
      .catch(error => console.error(error))
    return true // Will respond asynchronously.
  }
  if (message.contentScriptQuery === 'getUserStat') {
    const url = `${API_URL}/x/web-interface/nav/stat`
    fetch(url)
      .then(response => response.json())
      .then(data => sendResponse(data))
      .catch(error => console.error(error))
    return true
  }
  if (message.contentScriptQuery === 'getSearchSuggestion') {
    const url = `https://s.search.bilibili.com/main/suggest?term=${message.term}`
    fetch(url)
      .then(response => response.json())
      .then(data => sendResponse(data))
      .catch(error => console.error(error))
    return true
  }
  if (message.contentScriptQuery === 'logout') {
    const url = `https://passport.bilibili.com/login/exit/v2?biliCSRF=${message.biliCSRF}`
    fetch(url, {
      method: 'POST',
      body: JSON.stringify({
        biliCSRF: message.biliJct,
      }),
    })
      .then(response => response.json())
      .then(data => sendResponse(data))
      .catch(error => console.error(error))
    return true
  }
  if (message.contentScriptQuery === 'getUnreadMsg') {
    const url = `${API_URL}/x/msgfeed/unread?build=0&mobi_app=web`
    fetch(url)
      .then(response => response.json())
      .then(data => sendResponse(data))
      .catch(error => console.error(error))
    return true
  }
  if (message.contentScriptQuery === 'getUnreadDm') {
    const url = 'https://api.vc.bilibili.com/session_svr/v1/session_svr/single_unread?build=0&mobi_app=web&unread_type=0'
    fetch(url)
      .then(response => response.json())
      .then(data => sendResponse(data))
      .catch(error => console.error(error))
    return true
  }
  if (message.contentScriptQuery === 'getNewMomentsCount') {
    const url = `${API_URL}/x/web-interface/dynamic/entrance`
    fetch(url)
      .then(response => response.json())
      .then(data => sendResponse(data))
      .catch(error => console.error(error))
    return true
  }
  if (message.contentScriptQuery === 'submitDislike') {
    // https://github.com/indefined/UserScripts/blob/master/bilibiliHome/bilibiliHome.API.md#%E6%8F%90%E4%BA%A4%E4%B8%8D%E5%96%9C%E6%AC%A2
    const url = `${APP_URL}/x/feed/dislike?access_key=${message.accessKey}
      &goto=${message.goto}
      &id=${message.id}
      &mid=${message.mid}
      &reason_id=${message.reasonID}
      &rid=${message.rid}
      &tag_id=${message.tagID}`
    fetch(url)
      .then(response => response.json())
      .then(data => sendResponse(data))
      .catch(error => console.error(error))
    return true
  }
})