import type { Context } from "hono";

export type Prettify<T> = {
  [K in keyof T]: T[K];
};

export function getRelativeUrl(ctx: Context, path: string) {
  const result = new URL(path, ctx.req.url);
  result.host = ctx.req.header("x-forwarded-host") || result.host;
  result.protocol = ctx.req.header("x-forwarded-proto") || result.protocol;
  result.port = ctx.req.header("x-forwarded-port") || result.port;
  return result.toString();
}

const twoPartTlds = [
  "co.uk",
  "co.jp",
  "co.kr",
  "co.nz",
  "co.za",
  "co.in",
  "com.au",
  "com.br",
  "com.cn",
  "com.mx",
  "com.tw",
  "net.au",
  "org.uk",
  "ne.jp",
  "ac.uk",
  "gov.uk",
  "edu.au",
  "gov.au",
];

export function isDomainMatch(a: string, b: string): boolean {
  if (a === b) return true;
  const partsA = a.split(".");
  const partsB = b.split(".");
  const hasTwoPartTld = twoPartTlds.some((tld) => a.endsWith(`.${tld}`) || b.endsWith(`.${tld}`));
  const numParts = hasTwoPartTld ? -3 : -2;
  const min = Math.min(partsA.length, partsB.length, numParts);
  const tailA = partsA.slice(min).join(".");
  const tailB = partsB.slice(min).join(".");
  return tailA === tailB;
}

export function lazy<T>(fn: () => T): () => T {
  let value: T | undefined;
  return () => {
    if (value === undefined) {
      value = fn();
    }
    return value;
  };
}