import type { IdResolver } from '@atproto/identity';

import axios from 'axios';

import type { TestPdsServer } from './pds.js';

export const mockNetworkUtilities = (pds: TestPdsServer) => {
	mockResolvers(pds.ctx.idResolver, pds);
};

export const mockResolvers = (idResolver: IdResolver, pds: TestPdsServer) => {
	// Map pds public url to its local url when resolving from plc
	const origResolveDid = idResolver.did.resolveNoCache;
	idResolver.did.resolveNoCache = async (did: string) => {
		const result = await (origResolveDid.call(idResolver.did, did) as ReturnType<typeof origResolveDid>);
		const service = result?.service?.find((svc) => svc.id === '#atproto_pds');

		if (typeof service?.serviceEndpoint === 'string') {
			service.serviceEndpoint = service.serviceEndpoint.replace(
				pds.ctx.cfg.service.publicUrl,
				`http://localhost:${pds.port}`,
			);
		}

		return result;
	};

	const origResolveHandleDns = idResolver.handle.resolveDns;
	idResolver.handle.resolve = async (handle: string) => {
		const isPdsHandle = pds.ctx.cfg.identity.serviceHandleDomains.some((domain) => handle.endsWith(domain));

		if (!isPdsHandle) {
			return origResolveHandleDns.call(idResolver.handle, handle);
		}

		const url = `${pds.url}/.well-known/atproto-did`;
		try {
			const res = await axios.get(url, { headers: { host: handle } });
			return res.data;
		} catch (err) {
			return undefined;
		}
	};
};