import { describe, expect, it } from "vitest";
import { isAllowedOAuthRedirectUri, isPrivateOrLoopbackHost, parseUrl } from "./url-security.node";

describe("isPrivateOrLoopbackHost", () => {
	it.each([
		"0.0.0.0",
		"10.0.0.1",
		"100.64.0.1",
		"100.127.255.255",
		"127.0.0.1",
		"169.254.1.1",
		"172.16.0.1",
		"172.31.255.255",
		"192.0.0.1",
		"192.0.2.1",
		"192.88.99.1",
		"192.168.0.1",
		"198.18.0.1",
		"198.19.255.255",
		"198.51.100.1",
		"203.0.113.1",
		"224.0.0.1",
		"240.0.0.1",
		"255.255.255.255",
	])("matches non-public/special-use IPv4 address %s", (address) => {
		expect(isPrivateOrLoopbackHost(address)).toBe(true);
	});

	it.each([
		"::",
		"::1",
		"::ffff:8.8.8.8",
		"::ffff:0808:0808",
		"64:ff9b::1",
		"64:ff9b:1::1",
		"100::1",
		"100:0:0:1::1",
		"2001::1",
		"2001:2::1",
		"2001:10::1",
		"2001:db8::1",
		"2002::1",
		"3fff::1",
		"5f00::1",
		"fc00::1",
		"fd12::1",
		"fe80::1",
		"fe81::1",
		"febf::1",
		"ff00::1",
		"ff02::1",
	])("matches non-public/special-use IPv6 address %s", (address) => {
		expect(isPrivateOrLoopbackHost(address)).toBe(true);
	});

	describe("loopback hostnames", () => {
		it("matches localhost", () => {
			expect(isPrivateOrLoopbackHost("localhost")).toBe(true);
		});

		it("matches LOCALHOST (case-insensitive)", () => {
			expect(isPrivateOrLoopbackHost("LOCALHOST")).toBe(true);
		});

		it("matches subdomains of localhost", () => {
			expect(isPrivateOrLoopbackHost("api.localhost")).toBe(true);
		});

		it("matches IPv6 loopback ::1", () => {
			expect(isPrivateOrLoopbackHost("::1")).toBe(true);
		});

		it("matches bracketed IPv6 loopback [::1]", () => {
			expect(isPrivateOrLoopbackHost("[::1]")).toBe(true);
		});
	});

	describe("private IPv4 ranges", () => {
		it("matches 10.0.0.0/8", () => {
			expect(isPrivateOrLoopbackHost("10.0.0.1")).toBe(true);
			expect(isPrivateOrLoopbackHost("10.255.255.255")).toBe(true);
		});

		it("matches 127.0.0.0/8 (loopback)", () => {
			expect(isPrivateOrLoopbackHost("127.0.0.1")).toBe(true);
		});

		it("matches 169.254.0.0/16 (link-local)", () => {
			expect(isPrivateOrLoopbackHost("169.254.1.1")).toBe(true);
		});

		it("matches 172.16.0.0/12", () => {
			expect(isPrivateOrLoopbackHost("172.16.0.1")).toBe(true);
			expect(isPrivateOrLoopbackHost("172.31.255.255")).toBe(true);
		});

		it("does NOT match outside 172.16-31", () => {
			expect(isPrivateOrLoopbackHost("172.15.0.1")).toBe(false);
			expect(isPrivateOrLoopbackHost("172.32.0.1")).toBe(false);
		});

		it("matches 192.168.0.0/16", () => {
			expect(isPrivateOrLoopbackHost("192.168.0.1")).toBe(true);
			expect(isPrivateOrLoopbackHost("192.168.255.255")).toBe(true);
		});

		it("matches 0.0.0.0/8", () => {
			expect(isPrivateOrLoopbackHost("0.0.0.0")).toBe(true);
		});

		it("matches 192.0.0.0/24", () => {
			expect(isPrivateOrLoopbackHost("192.0.0.1")).toBe(true);
		});

		it("matches documentation IPv4 ranges", () => {
			expect(isPrivateOrLoopbackHost("192.0.2.1")).toBe(true);
			expect(isPrivateOrLoopbackHost("198.51.100.1")).toBe(true);
			expect(isPrivateOrLoopbackHost("203.0.113.1")).toBe(true);
		});

		it("matches deprecated 6to4 relay anycast", () => {
			expect(isPrivateOrLoopbackHost("192.88.99.1")).toBe(true);
		});

		it("matches 100.64.0.0/10 (CGNAT)", () => {
			expect(isPrivateOrLoopbackHost("100.64.0.1")).toBe(true);
			expect(isPrivateOrLoopbackHost("100.127.255.255")).toBe(true);
		});

		it("matches 198.18.0.0/15 (benchmarking)", () => {
			expect(isPrivateOrLoopbackHost("198.18.0.1")).toBe(true);
			expect(isPrivateOrLoopbackHost("198.19.255.255")).toBe(true);
		});

		it("matches multicast, reserved, and broadcast IPv4 ranges", () => {
			expect(isPrivateOrLoopbackHost("224.0.0.1")).toBe(true);
			expect(isPrivateOrLoopbackHost("240.0.0.1")).toBe(true);
			expect(isPrivateOrLoopbackHost("255.255.255.255")).toBe(true);
		});

		it("does NOT match public IPs", () => {
			expect(isPrivateOrLoopbackHost("8.8.8.8")).toBe(false);
			expect(isPrivateOrLoopbackHost("1.1.1.1")).toBe(false);
			expect(isPrivateOrLoopbackHost("93.184.216.34")).toBe(false);
			expect(isPrivateOrLoopbackHost("192.31.196.1")).toBe(false);
			expect(isPrivateOrLoopbackHost("192.52.193.1")).toBe(false);
			expect(isPrivateOrLoopbackHost("192.175.48.1")).toBe(false);
		});
	});

	describe("private IPv6 ranges", () => {
		it("matches unique-local fc00::/7", () => {
			expect(isPrivateOrLoopbackHost("fc00::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("fd12::1")).toBe(true);
		});

		it("matches link-local fe80::/10", () => {
			expect(isPrivateOrLoopbackHost("fe80::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("fe81::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("febf::1")).toBe(true);
		});

		it("matches unspecified and multicast IPv6 ranges", () => {
			expect(isPrivateOrLoopbackHost("::")).toBe(true);
			expect(isPrivateOrLoopbackHost("ff00::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("ff02::1")).toBe(true);
		});

		it("matches NAT64 discard-only, benchmarking, and 6to4 IPv6 ranges", () => {
			expect(isPrivateOrLoopbackHost("64:ff9b::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("64:ff9b:1::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("100::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("100:0:0:1::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("2001::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("2001:2::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("2001:10::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("2002::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("3fff::1")).toBe(true);
			expect(isPrivateOrLoopbackHost("5f00::1")).toBe(true);
		});

		it("matches documentation IPv6 2001:db8::/32", () => {
			expect(isPrivateOrLoopbackHost("2001:db8::1")).toBe(true);
		});

		it("does NOT match IPv6 addresses outside fe80::/10", () => {
			expect(isPrivateOrLoopbackHost("fec0::1")).toBe(false);
		});

		it("does NOT match global IPv6", () => {
			expect(isPrivateOrLoopbackHost("2606:4700:4700::1111")).toBe(false);
		});

		it("matches IPv4-mapped IPv6 private and loopback addresses", () => {
			expect(isPrivateOrLoopbackHost("::ffff:10.0.0.1")).toBe(true);
			expect(isPrivateOrLoopbackHost("::ffff:127.0.0.1")).toBe(true);
			expect(isPrivateOrLoopbackHost("::ffff:169.254.169.254")).toBe(true);
			expect(isPrivateOrLoopbackHost("[::ffff:192.168.1.1]")).toBe(true);
			expect(isPrivateOrLoopbackHost("::ffff:7f00:1")).toBe(true);
			expect(isPrivateOrLoopbackHost("::ffff:0a00:1")).toBe(true);
			expect(isPrivateOrLoopbackHost("[::ffff:7f00:1]")).toBe(true);
		});

		it("matches IPv4-mapped IPv6 public addresses because the mapped range is special-purpose", () => {
			expect(isPrivateOrLoopbackHost("::ffff:8.8.8.8")).toBe(true);
			expect(isPrivateOrLoopbackHost("::ffff:0808:0808")).toBe(true);
		});

		it("matches the broad 2001::/23 IETF protocol assignments range", () => {
			expect(isPrivateOrLoopbackHost("2001:100::1")).toBe(true);
		});
	});

	describe("non-IP, non-loopback hostnames", () => {
		it("returns false for public domain", () => {
			expect(isPrivateOrLoopbackHost("example.com")).toBe(false);
		});

		it("returns false for arbitrary string", () => {
			expect(isPrivateOrLoopbackHost("not-a-real-host")).toBe(false);
		});
	});
});

describe("parseUrl", () => {
	it("returns URL object for valid URL", () => {
		const url = parseUrl("https://example.com/path");
		expect(url).not.toBeNull();
		expect(url?.hostname).toBe("example.com");
	});

	it("returns null for invalid URL", () => {
		expect(parseUrl("not a url")).toBeNull();
	});

	it("returns null for empty string", () => {
		expect(parseUrl("")).toBeNull();
	});

	it("returns null for relative URL", () => {
		expect(parseUrl("/path/only")).toBeNull();
	});
});

describe("isAllowedOAuthRedirectUri", () => {
	const trustedOrigins = ["https://app.example.com"];

	it("returns false for malformed URI", () => {
		expect(isAllowedOAuthRedirectUri("nope", trustedOrigins)).toBe(false);
	});

	it("returns false when credentials present", () => {
		expect(isAllowedOAuthRedirectUri("https://u:p@app.example.com", trustedOrigins)).toBe(false);
	});

	it("returns false when fragment present", () => {
		expect(isAllowedOAuthRedirectUri("https://app.example.com/cb#x", trustedOrigins)).toBe(false);
	});

	it("allows http for loopback (localhost)", () => {
		expect(isAllowedOAuthRedirectUri("http://localhost:3000/cb", trustedOrigins)).toBe(true);
	});

	it("allows http for 127.0.0.1", () => {
		expect(isAllowedOAuthRedirectUri("http://127.0.0.1/cb", trustedOrigins)).toBe(true);
	});

	it("allows http for IPv6 loopback", () => {
		expect(isAllowedOAuthRedirectUri("http://[::1]/cb", trustedOrigins)).toBe(true);
	});

	it("rejects http for non-loopback hosts", () => {
		expect(isAllowedOAuthRedirectUri("http://example.com/cb", trustedOrigins)).toBe(false);
	});

	it("rejects non-https/non-http protocols", () => {
		expect(isAllowedOAuthRedirectUri("ftp://example.com/cb", trustedOrigins)).toBe(false);
	});

	it("rejects https with private/loopback host", () => {
		expect(isAllowedOAuthRedirectUri("https://192.168.1.1/cb", trustedOrigins)).toBe(false);
	});

	it("matches trusted origins", () => {
		expect(isAllowedOAuthRedirectUri("https://app.example.com/cb", trustedOrigins)).toBe(true);
	});

	it("rejects public https hosts outside trusted origins", () => {
		expect(isAllowedOAuthRedirectUri("https://api.example.com/cb", trustedOrigins)).toBe(false);
	});

	it("allows any parseable URI when unsafe mode is enabled", () => {
		const options = { allowUnsafe: true };

		expect(isAllowedOAuthRedirectUri("myapp://callback", trustedOrigins, options)).toBe(true);
		expect(isAllowedOAuthRedirectUri("http://example.com/cb", trustedOrigins, options)).toBe(true);
		expect(isAllowedOAuthRedirectUri("https://192.168.1.1/cb", trustedOrigins, options)).toBe(true);
		expect(isAllowedOAuthRedirectUri("https://u:p@app.example.com/cb#x", trustedOrigins, options)).toBe(true);
		expect(isAllowedOAuthRedirectUri("not a url", trustedOrigins, options)).toBe(false);
	});
});