netris-nestri/packages/auth/src/util.ts

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;
	};
}