extension/src/utils/storage.ts

import { Storage } from '@plasmohq/storage';
import { useStorage } from '@plasmohq/storage/hook';

import { makeUrlIntoDomain } from '~utils/domains';

export const DEFAULT_DOMAIN_WHITELIST = ['movie-web.app', 'localhost:5173'];

export const storage = new Storage();

const domainIsInWhitelist = async (domain: string) => {
  const whitelist = await storage.get<string[]>('domainWhitelist');
  return whitelist?.some((d) => d.includes(domain)) ?? false;
};

export function useDomainStorage() {
  return useStorage<string[]>('domainWhitelist', (v) => v ?? DEFAULT_DOMAIN_WHITELIST);
}

export const isDomainWhitelisted = async (url: string | undefined) => {
  if (!url) return false;
  const domain = makeUrlIntoDomain(url);
  if (!domain) return false;
  return domainIsInWhitelist(domain);
};

export const assertDomainWhitelist = async (url: string) => {
  const isWhiteListed = await isDomainWhitelisted(url);
  if (!isWhiteListed) throw new Error('Domain is not whitelisted');
};