create-chuntianxiaozhu/template-plugin-chrome/src/utils/utils.ts

春天小猪模板工具

import { CheerioAPI } from "cheerio";

export function generateWangLink($: CheerioAPI) {
  const host = "https://amos.alicdn.com/";
  const config = "groupid=0&v=2&site=cntaobao&s=1&charset=utf-8&uid=";

  const userNick = $(".wangwang").first().attr("data-nick");
  return `${host}getcid.aw?${config}${userNick}&fromid=cntaobao`;
}

export function queryCookies(details: chrome.cookies.GetAllDetails) {
  return chrome.cookies.getAll(details).then((cookies) => {
    const cookieKeyValue: string[] = [];
    const cookieMap: any = {};
    cookies.map((cookie) => {
      cookieKeyValue.push(`${cookie.name}=${cookie.value}`);
      cookieMap[cookie.name] = cookie.value;
    });
    return {
      cookieMap,
      cookieStr: cookieKeyValue.join("; "),
    };
  });
}

export interface IUri {
  scheme: string;
  authority?: string;
  host: string;
  port: number;
  tld: string;
  path: string;
  query?: string;
  fragment?: string;
  request: string;
  valid: boolean;
}

export const REGEX_URI =
  // eslint-disable-next-line no-useless-escape
  /(https?|s?ftp):\/?\/?([^\/ ]+)\.([a-zA-Z]+):?([0-9]*)?\/?([^\?]*)\??([^#]*)#?(.*)/gi;
export const REGEX_AUTH_URI =
  // eslint-disable-next-line no-useless-escape
  /(https?|s?ftp):\/?\/?([a-zA-Z:]*)?@?([^\/ ]+)\.([a-zA-Z]+):?([0-9]*)?\/?([^\?]*)\??([^#]*)#?(.*)/gi;

/* No special characters */
export const REGEX_URI_STRICT =
  // eslint-disable-next-line no-useless-escape
  /(https?|s?ftp):\/?\/?([^\/ ]+)\.([a-zA-Z]+):?([0-9]*)?\/?([a-zA-Z0-9/%_-~.]*)\??([a-zA-Z0-9-._~:\/\?\[\]@!$&'()%\*\+,;=]*)#?([\[\]a-zA-Z0-9-._~:\/\?#\[\]@!$&'()%\*\+,;=]*)/gi;
export const REGEX_AUTH_URI_STRICT =
  // eslint-disable-next-line no-useless-escape
  /(https?|s?ftp):\/?\/?([a-zA-Z:]*)?@?([^\/ ]+)\.([a-zA-Z]+):?([0-9]*)?\/?([a-zA-Z0-9/%_\-~.]*)\??([\[\]a-zA-Z0-9-._~:\/\?\[\]@!$&'()%\*\+,;=]*)#?([a-zA-Z0-9-._~:\/\?#\[\]@!$&'()%\*\+,;=]*)/gi;

export const parseUri = (url: string = "", strict: boolean = false): IUri => {
  let port: number = 80;
  // tslint:disable-next-line: no-bitwise
  if (~url.indexOf("@")) {
    const result = strict
      ? new RegExp(REGEX_URI_STRICT).exec(url) || []
      : new RegExp(REGEX_AUTH_URI).exec(url) || [];
    const [
      request = "",
      scheme,
      authority,
      host,
      tld,
      portNumber,
      path,
      params,
      fragment,
    ] = result;
    if (portNumber) {
      port = parseInt(portNumber, 10);
    }
    let domain: string | undefined = host;
    // tslint:disable-next-line: no-bitwise
    // eslint-disable-next-line no-useless-escape
    if (!host || ~host.indexOf("..") || host.match(/([^a-zA-Z0-9._\-])/)) {
      domain = undefined;
    }
    if (!portNumber && scheme && scheme.toLowerCase() === "https") {
      port = 443;
    }
    const valid: boolean =
      request !== "" &&
      scheme !== undefined &&
      domain !== undefined &&
      tld !== undefined &&
      request.length < 2048;
    const uri: IUri = {
      authority,
      fragment,
      host: domain ? domain + "." + tld : "",
      path,
      port,
      query: params,
      request: url,
      scheme,
      tld,
      valid,
    };
    return uri;
  } else {
    const result = strict
      ? new RegExp(REGEX_URI_STRICT).exec(url) || []
      : new RegExp(REGEX_URI).exec(url) || [];
    const [
      request = "",
      scheme,
      host,
      tld,
      portNumber,
      path,
      params,
      fragment,
    ] = result;
    let domain: string | undefined = host;
    // tslint:disable-next-line: no-bitwise
    // eslint-disable-next-line no-useless-escape
    if (!host || ~host.indexOf("..") || host.match(/([^a-zA-Z0-9._\-])/)) {
      domain = undefined;
    }
    if (portNumber) {
      port = parseInt(portNumber, 10);
    }
    if (!portNumber && scheme && scheme.toLowerCase() === "https") {
      port = 443;
    }
    const valid =
      request !== "" &&
      scheme !== undefined &&
      domain !== undefined &&
      tld !== undefined &&
      request.length < 2048;
    const uri: IUri = {
      fragment,
      host: domain ? domain + "." + tld : "",
      path,
      port,
      query: params,
      request: url,
      scheme,
      tld,
      valid,
    };
    return uri;
  }
};

/**
 * 从脚本中获取
 * @param scriptText
 * @param varName
 * @returns
 */
export function getVarFromScript(scriptText: string, varName: string) {
  let value = null;
  try {
    value = eval(
      `(function anonymous() { try { ${scriptText} } catch{} return ${varName};})()`
    );
  } catch {
    return value;
  }
  return value;
}