ttv-lol-pro/src/common/ts/proxyInfo.ts

import { Address6 } from "ip-address";
import type { ProxyInfo, ProxyType } from "../../types";

const DEFAULT_PORTS: Readonly<Record<ProxyType, number>> = Object.freeze({
  direct: 0,
  http: 80,
  https: 443,
  socks: 1080,
  socks4: 1080,
});

export function getProxyInfoFromUrl(
  url: string
): ProxyInfo & { host: string; port: number } {
  let type: ProxyType | undefined = undefined;
  if (url.includes("://")) {
    const [protocol, urlWithoutProtocol] = url.split("://", 2);
    type = protocol as ProxyType;
    url = urlWithoutProtocol;
  }

  const lastIndexOfAt = url.lastIndexOf("@");
  const hostname = url.substring(lastIndexOfAt + 1, url.length);
  const lastIndexOfColon = getLastIndexOfColon(hostname);

  let host: string | undefined = undefined;
  let port: number | undefined = undefined;
  if (lastIndexOfColon === -1) {
    host = hostname;
    port = !type ? 3128 : DEFAULT_PORTS[type]; // Default port
  } else {
    host = hostname.substring(0, lastIndexOfColon);
    port = Number(hostname.substring(lastIndexOfColon + 1, hostname.length));
  }
  if (host.startsWith("[") && host.endsWith("]")) {
    host = host.substring(1, host.length - 1);
  }

  let username: string | undefined = undefined;
  let password: string | undefined = undefined;
  if (lastIndexOfAt !== -1) {
    const credentials = url.substring(0, lastIndexOfAt);
    const indexOfColon = credentials.indexOf(":");
    username = credentials.substring(0, indexOfColon);
    password = credentials.substring(indexOfColon + 1, credentials.length);
  }

  return {
    type: type ?? "http",
    host,
    port,
    username,
    password,
  };
}

/**
 * Returns the last index of a colon in a hostname, ignoring colons inside brackets.
 * Supports IPv6 addresses.
 * @param hostname
 * @returns Returns -1 if no colon is found.
 */
function getLastIndexOfColon(hostname: string): number {
  let lastIndexOfColon = -1;
  let bracketDepth = 0;
  for (let i = hostname.length - 1; i >= 0; i--) {
    const char = hostname[i];
    if (char === "]") {
      bracketDepth++;
    } else if (char === "[") {
      bracketDepth--;
    } else if (char === ":" && bracketDepth === 0) {
      lastIndexOfColon = i;
      break;
    }
  }
  return lastIndexOfColon;
}

export function getUrlFromProxyInfo(proxyInfo: ProxyInfo): string {
  const { host, port, username, password } = proxyInfo;
  if (!host) return "";
  let url = "";
  if (username && password) {
    url = `${username}:${password}@`;
  } else if (username) {
    url = `${username}@`;
  }
  const isIPv6 = Address6.isValid(host);
  if (isIPv6) {
    url += `[${host}]`;
  } else {
    url += host;
  }
  if (port) {
    url += `:${port}`;
  }
  return url;
}