File size: 2,686 Bytes
bee6636 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
import { codecDecode, codecEncode } from "..";
import { config } from "../../shared";
import { rewriteJs } from "./js";
export type URLMeta = {
origin: URL;
base: URL;
topFrameName?: string;
parentFrameName?: string;
};
function tryCanParseURL(url: string, origin?: string | URL): URL | null {
try {
return new URL(url, origin);
} catch {
return null;
}
}
export function rewriteBlob(url: string, meta: URLMeta) {
const blob = new URL(url.substring("blob:".length));
return "blob:" + meta.origin.origin + blob.pathname;
}
export function unrewriteBlob(url: string) {
const blob = new URL(url.substring("blob:".length));
return "blob:" + location.origin + blob.pathname;
}
export function rewriteUrl(url: string | URL, meta: URLMeta) {
if (url instanceof URL) url = url.toString();
if (url.startsWith("javascript:")) {
return (
"javascript:" +
rewriteJs(url.slice("javascript:".length), "(javascript: url)", meta)
);
} else if (url.startsWith("blob:")) {
return location.origin + config.prefix + url;
} else if (url.startsWith("data:")) {
return location.origin + config.prefix + url;
} else if (url.startsWith("mailto:") || url.startsWith("about:")) {
return url;
} else {
let base = meta.base.href;
if (base.startsWith("about:")) base = unrewriteUrl(self.location.href); // jank!!!!! weird jank!!!
const realUrl = tryCanParseURL(url, base);
if (!realUrl) return url;
const encodedHash = codecEncode(realUrl.hash.slice(1));
const realHash = encodedHash ? "#" + encodedHash : "";
realUrl.hash = "";
return (
location.origin + config.prefix + codecEncode(realUrl.href) + realHash
);
}
}
export function unrewriteUrl(url: string | URL) {
if (url instanceof URL) url = url.toString();
// remove query string
if (url.includes("?")) {
url = url.split("?")[0];
}
const prefixed = location.origin + config.prefix;
if (url.startsWith("javascript:")) {
//TODO
return url;
} else if (url.startsWith("blob:")) {
// realistically this shouldn't happen
return url;
} else if (url.startsWith(prefixed + "blob:")) {
return url.substring(prefixed.length);
} else if (url.startsWith(prefixed + "data:")) {
return url.substring(prefixed.length);
} else if (url.startsWith("mailto:") || url.startsWith("about:")) {
return url;
} else {
const realUrl = tryCanParseURL(url);
if (!realUrl) return url;
const decodedHash = codecDecode(realUrl.hash.slice(1));
const realHash = decodedHash ? "#" + decodedHash : "";
realUrl.hash = "";
return codecDecode(realUrl.href.slice(prefixed.length) + realHash);
}
}
|