validator/js/chromeextension/content_script.js - main - githubmirror/amphtml

validator/js/chromeextension/content_script.js
122 строки3 KB
honeybadgerdontcare
remove notices from validator/js/... (#35734)
19 авг 2021, 01:41
19 авг 2021, 01:417a3474d
const globals = {};
globals.amphtmlRegex = new RegExp('(^\\s*)amphtml(\\s*$)');
globals.ampCaches = [
  {
    'getAmpHref': function() {
      if (window.location.pathname.startsWith('/a/s') ||
          window.location.pathname.startsWith('/c/s') ||
          window.location.pathname.startsWith('/v/s')) {
        return 'https://' + window.location.pathname.slice(5);
      } else if (window.location.pathname.startsWith('/c')) {
        return 'http://' + window.location.pathname.slice(3);
      } else {
        return '';
      }
    },
    'isAmpCache': function() {
      return window.location.hostname.endsWith('cdn.ampproject.org'); // lgtm [js/incomplete-url-substring-sanitization]
    },
  },
];

/**
 * Returns the AMP page href from an AMP Cache. Otherwise returns empty string.
 *
 * @return {string}
 * @private
 */
function getAmpCacheHref() {
  for (const index in globals.ampCaches) {
    const ampCache = globals.ampCaches[index];
    if (ampCache.isAmpCache()) {
      return ampCache.getAmpHref();
    }
  }
  return '';
}

/**
 * Determine if <link rel="amphtml" href="..."> exists in the page and
 * return the href's value iff it starts with http:// or https://.
 * Otherwise returns empty string.
 *
 * @return {string}
 * @private
 */
function getAmpHtmlLinkHref() {
  let ampHtmlLinkHref = '';
  // URL must begin with http:// or https://
  const validUrlPrefixRe = /^[\s\xa0]*https?:\/\//;
  const headLinks = document.head.getElementsByTagName('link');
  if (headLinks.length > 0) {
    for (const index in headLinks) {
      const link = headLinks[index];
      if (link instanceof HTMLLinkElement &&
          link.hasAttribute('rel') &&
          globals.amphtmlRegex.test(link.getAttribute('rel')) &&
          link.hasAttribute('href') &&
          validUrlPrefixRe.test(link.getAttribute('href'))) {
        ampHtmlLinkHref = link.getAttribute('href');
        break;
      }
    }
  }
  return ampHtmlLinkHref;
}

/**
 * Determine if the page is from an AMP Cache.
 *
 * @return {boolean}
 * @private
 */
function isAmpCache() {
  for (const index in globals.ampCaches) {
    const ampCache = globals.ampCaches[index];
    if (ampCache.isAmpCache()) {
      return true;
    }
  }
  return false;
}

/**
 * Determine if the page is an AMP page.
 *
 * @return {boolean}
 * @private
 */
function isAmpDocument() {
  return (document.documentElement.hasAttribute('amp') ||
      document.documentElement.hasAttribute('⚡'));
}

/**
 * Listener for requests from the extension.
 *
 * Requests for getAmpDetails. Return to the extension:
 * - isAmp: Is the page marked as an AMP page.
 * - fromAmpCache: Is the page from an AMP Cache.
 * - ampHref: the href to an AMP page if the page is not an AMP page but there
 *   is an <link rel="amphtml"> or if the page is from an AMP Cache.
 * - userAgent: Tab's current userAgent, which may have been modified by
 *   device emulation.
 *
 * Requests for loadAmp and has ampHref, then redirects the browser to ampHref.
 */
chrome.runtime.onMessage.addListener(function(request, sender, sendResponse) {
  if (request.getAmpDetails) {
    const isAmp = isAmpDocument();
    const fromAmpCache = isAmpCache();
    let ampHref = '';
    if (!isAmp) {ampHref = getAmpHtmlLinkHref();}
    if (fromAmpCache) {ampHref = getAmpCacheHref();}
    sendResponse({
      'isAmp': isAmp, 'fromAmpCache': fromAmpCache, 'ampHref': ampHref,
      'userAgent': navigator.userAgent,
    });
  }
  if (request.loadAmp && request.ampHref) {
    window.location = request.ampHref;
  }
});