xcv58/Custom-JavaScript-for-Websites-2

import orderBy from 'lodash.orderby'
import uniqBy from 'lodash.uniqby'

export const setLastFocusedWindowId = (lastFocusedWindowId) => {
  chrome.storage.local.set({ lastFocusedWindowId })
}

export const getLastFocusedWindowId = async () => {
  const { lastFocusedWindowId } = await chrome.storage.local.get({
    lastFocusedWindowId: null
  })
  return lastFocusedWindowId
}

const getQueryInfo = (windowId) => {
  if (windowId) {
    return { windowId }
  }
  return { currentWindow: true }
}

export const getActiveTab = async () => {
  const windowId = await getLastFocusedWindowId()
  const queryInfo = getQueryInfo(windowId)
  const tabs = await chrome.tabs.query({ ...queryInfo, active: true })
  if (tabs.length <= 0) {
    throw new Error('No active tab! This is impossible')
  }
  return tabs[0]
}

const SOURCE_PREFIX = 'data:text/javascript'
const BASE64_PREFIX = SOURCE_PREFIX + ';base64,'
const UTF8_PREFIX = SOURCE_PREFIX + ';charset=utf-8,'

export const encodeSource = (script) => {
  // base64 may be smaller, but does not handle unicode characters
  // attempt base64 first, fall back to escaped text
  try {
    return BASE64_PREFIX + window.btoa(script)
  } catch (e) {
    return UTF8_PREFIX + encodeURIComponent(script)
  }
}

export const decodeSource = (source) => {
  if (source.startsWith(BASE64_PREFIX)) {
    return window.atob(source.replace(BASE64_PREFIX, ''))
  }
  if (source.startsWith(UTF8_PREFIX)) {
    return decodeURIComponent(source.replace(UTF8_PREFIX, ''))
  }
  if (!source) {
    return ''
  }
  return source
}

export const getHosts = async (key = '') => {
  const result = await chrome.storage.sync.get({ hosts: [] })
  if (Array.isArray(result.hosts) && result.hosts.length > 0) {
    return result.hosts
  }
  if (!key) {
    return []
  }
  const { hosts = [] } = JSON.parse(window.localStorage.getItem(key) || '{}')
  return hosts
}

const validHost = (host) => {
  if (typeof host === 'string') {
    return !!host
  }
  return !!host.pattern
}

export const setHosts = async (hosts = []) => {
  chrome.storage.sync.set({
    hosts: uniqBy(hosts.filter(validHost), JSON.stringify)
  })
}

export const clearHosts = () => {
  chrome.storage.sync.remove('hosts')
}

export const findMatchedHosts = (hosts = [], url = {}, message = {}) => {
  const { isRegex, pattern } = message
  if (isRegex && pattern) {
    return hosts.filter((host) => host.isRegex && host.pattern === pattern)
  }
  const matchedHosts = hosts.filter((host) => {
    if (typeof host === 'string') {
      return host === url.origin
    } else if (typeof host === 'object') {
      return new RegExp(host.pattern).test(url.href)
    }
    return false
  })
  return orderBy(matchedHosts, ['pattern'], ['desc'])
}

export const getHostKey = (host) => {
  if (!host) {
    throw new Error(`getHostKey get falsy host: ${host}!`)
  }
  if (typeof host === 'string') {
    return host
  } else {
    return host.pattern
  }
}

export const getHostName = (host) => {
  const { isRegex, pattern } = host
  if (isRegex) {
    return `RegExp: ${pattern}`
  }
  return host
}