stylus/content/install-user-css.js

/* global semverCompare escapeHtml */

'use strict';

let pendingResource;

function install(style) {
  const request = Object.assign(style, {
    method: 'saveUsercss',
    reason: 'install',
    updateUrl: location.href
  });
  return runtimeSend(request)
    .then(result => {
      $$('.warning')
        .forEach(el => el.remove());
      $('button.install').textContent = 'Installed';
      $('button.install').disabled = true;
      window.dispatchEvent(new CustomEvent('installed', {detail: result}));
    })
    .catch(err => {
      alert(chrome.i18n.getMessage('styleInstallFailed', String(err)));
    });
}

function runtimeSend(request) {
  return new Promise((resolve, reject) => {
    chrome.runtime.sendMessage(request, result => {
      if (result.status === 'error') {
        reject(result.error);
      } else {
        resolve(result);
      }
    });
  });
}

function getAppliesTo(style) {
  function *_gen() {
    for (const section of style.sections) {
      for (const type of ['urls', 'urlPrefixes', 'domains', 'regexps']) {
        if (section[type]) {
          yield *section[type];
        }
      }
    }
  }
  const result = [..._gen()];
  if (!result.length) {
    result.push(chrome.i18n.getMessage('appliesToEverything'));
  }
  return result;
}

function initInstallPage({style, dup}, sourceLoader) {
  return pendingResource.then(() => {
    const versionTest = dup && semverCompare(style.version, dup.version);
    document.body.innerHTML = '';
    document.body.appendChild(tHTML(`
      <div class="container">
        <div class="header">
          <h1>
            ${escapeHtml(style.name)}
            <small class="meta-version">v${escapeHtml(style.version)}</small>
          </h1>
          <p>${escapeHtml(style.description)}</p>
          <h3 i18n-text="author"></h3>
          ${escapeHtml(style.author)}
          <h3 i18n-text="license"></h3>
          ${escapeHtml(style.license)}
          <h3 i18n-text="appliesLabel"></h3>
          <ul>
            ${getAppliesTo(style).map(s => `<li>${escapeHtml(s)}</li>`)}
          </ul>
          <div class="actions">
            <button class="install">${t(
              !dup ? 'installButton' : versionTest > 0 ?
                'installButtonUpdate' : 'installButtonReinstall'
            )}</button>
          </div>
          <div class="external">
            <a href="${encodeURI(style.url)}" target="_blank" i18n-text="externalHomepage" rel="noopener"></a>
            <a href="${encodeURI(style.supportURL)}" target="_blank" i18n-text="externalSupport" rel="noopener"></a>
          </div>
        </div>
        <div class="main">
          <div class="code"></div>
        </div>
      </div>
    `));
    if (versionTest < 0) {
      $('.actions').parentNode.insertBefore(tHTML(`
        <div class="warning" i18n-text="versionInvalidOlder"></div>
      `), $('.actions'));
    }
    $('.code').textContent = style.source;
    $('button.install').onclick = () => {
      if (dup) {
        if (confirm(chrome.i18n.getMessage('styleInstallOverwrite', [style.name, dup.version, style.version]))) {
          install(style);
        }
      } else if (confirm(chrome.i18n.getMessage('styleInstall', [style.name]))) {
        install(style);
      }
    };

    if (location.protocol === 'file:') {
      initLiveReload(sourceLoader);
    }
  });
}

function initLiveReload(sourceLoader) {
  let installed;
  const watcher = sourceLoader.watch(source => {
    $('.code').textContent = source;
    return runtimeSend({
      method: 'saveUsercss',
      id: installed.id,
      source: source
    }).then(() => {
      $$('.main .warning').forEach(e => e.remove());
    }).catch(err => {
      const oldWarning = $('.main .warning');
      const warning = tHTML(`
        <div class="warning" i18n-text="parseUsercssError">
          <pre>${escapeHtml(err)}</pre>
        </div>
      `);
      if (oldWarning) {
        oldWarning.replaceWith(warning);
      } else {
        $('.main').insertBefore(warning, $('.main').childNodes[0]);
      }
    });
  });
  window.addEventListener('installed', ({detail: {style}}) => {
    installed = style;
    if ($('.live-reload-checkbox').checked) {
      watcher.start();
    }
  });
  $('.actions').appendChild(tHTML(`
    <label class="live-reload">
      <input type="checkbox" class="live-reload-checkbox">
      <span i18n-text="liveReloadLabel"></span>
    </label>
  `));
  $('.live-reload-checkbox').onchange = e => {
    if (!installed) {
      return;
    }
    if (e.target.checked) {
      watcher.start();
    } else {
      watcher.stop();
    }
  };
}

function initErrorPage(err, source) {
  return pendingResource.then(() => {
    document.body.innerHTML = '';
    document.body.appendChild(tHTML(`
      <div class="warning" i18n-text="parseUsercssError">
        <pre>${escapeHtml(err)}</pre>
      </div>
      <div class="code"></div>
    `));
    $('.code').textContent = source;
  });
}

function createSourceLoader() {
  let source;

  function fetchText(url) {
    return new Promise((resolve, reject) => {
      // you can't use fetch in Chrome under 'file:' protocol
      const xhr = new XMLHttpRequest();
      xhr.open('GET', url);
      xhr.addEventListener('load', () => resolve(xhr.responseText));
      xhr.addEventListener('error', () => reject(xhr));
      xhr.send();
    });
  }

  function load() {
    return fetchText(location.href)
      .then(_source => {
        source = _source;
        return source;
      });
  }

  function watch(cb) {
    let timer;
    const DELAY = 1000;

    function start() {
      if (timer) {
        return;
      }
      timer = setTimeout(check, DELAY);
    }

    function stop() {
      clearTimeout(timer);
      timer = null;
    }

    function check() {
      fetchText(location.href)
        .then(_source => {
          if (source !== _source) {
            source = _source;
            return cb(source);
          }
        })
        .catch(console.log)
        .then(() => {
          timer = setTimeout(check, DELAY);
        });
    }

    return {start, stop};
  }

  return {load, watch, source: () => source};
}

function initUsercssInstall() {
  pendingResource = runtimeSend({
    method: 'injectContent',
    files: [
      '/js/dom.js',
      '/js/localization.js',
      '/js/usercss.js',
      '/vendor/node-semver/semver.js',
      '/content/install-user-css.css'
    ]
  });

  const sourceLoader = createSourceLoader();
  sourceLoader.load()
    .then(() =>
      runtimeSend({
        method: 'filterUsercss',
        source: sourceLoader.source(),
        checkDup: true
      })
    )
    .then(result => initInstallPage(result, sourceLoader))
    .catch(err => initErrorPage(err, sourceLoader.source()));
}

function isUsercss() {
  if (!/\.user\.(css|styl|less|scss|sass)$/i.test(location.pathname)) {
    return false;
  }
  if (!/text\/(css|plain)/.test(document.contentType)) {
    return false;
  }
  if (!/==userstyle==/i.test(document.body.textContent)) {
    return false;
  }
  return true;
}

if (isUsercss()) {
  initUsercssInstall();
}