// see API - https://github.com/yujiosaka/headless-chrome-crawler/blob/master/docs/API.md#event-requeststarted
import fs from 'fs';
import lighthouse from 'lighthouse';
import * as chromeLauncher from 'chrome-launcher';

import path from 'path';
import {saveAsJson, copyJsonToReports, uploadJson, startViewer} from './actions/index.js';
import {getUserDir} from './utils.js';
import axios from 'axios';
import HCCrawler from '@popstas/headless-chrome-crawler';
import CSVExporter from '@popstas/headless-chrome-crawler/exporter/csv.js';
// console.log("pkg:", pkg);
// const {CSVExporter} = pkg;
import url from 'url';
import {validateResults, getValidationSum} from './validate.js';
import {exec} from 'child_process';
import sanitize from "sanitize-filename";
// поля описаны в API по ссылке выше
import fieldsPresets from './presets/scraperFields.js';
import color from './color.js';
import registry from './registry.js';
import config from './config.js';

const DEBUG = true; // выключить, если не нужны console.log на каждый запрос (не будет видно прогресс)

// запреты браузеру на подгрузку статики, ускоряет
let SKIP_IMAGES = true;
let SKIP_CSS = true;
let SKIP_JS = true;

// кол-во попыток выполнить actions
const finishTries = 5;
const saveAfterEvery = 100; // saveProgress after each N requests

let isMaxRequested = false;

let disconnectedLog = [];

// resend messages while disconnected
function sendDisconnected(socket) {
  if (disconnectedLog.length === 0) return;
  const log = [...disconnectedLog];
  disconnectedLog = [];
  for (let item of log) {
    socketSend(socket, item.event, item.msg);
  }
}

function socketSend(socket, event, msg) {
  if (socket) {
    const channel = event + (socket.uid || '')
    // console.log(channel + ': ', msg);

    if (socket.connected) {
      sendDisconnected(socket);
      socket.emit(channel, msg);
    } else {
      disconnectedLog.push({ event, msg });
    }
  }
}

async function scrapSite ({baseUrl, options = {}}) {
  const domain = url.parse(baseUrl).hostname || baseUrl;
  const protocol = url.parse(baseUrl).protocol;

  const log = (msg) => {
    const socketId = options.socket ? `${options.socket.id} ` : '';
    const disconnected = options.socket && !options.socket.connected ? '(disconnected) ' : '';
    const hhmmss = new Date().toTimeString().split(' ')[0];
    if (DEBUG) console.log(`${hhmmss} ${socketId}${disconnected}${msg}`);
    socketSend(options.socket, 'status', msg);
  };
  options.log = log;

  // const plugins = registry.getPlugins();
  // if (plugins) console.log('loaded plugins: ', plugins.map(p => p.name).join(', '));

  const parseUrls = async (url) => {
    const urls = [];
    const regex = /(?:(?:https?|ftp|file):\/\/|www\.|ftp\.)(?:\([-A-ZА-Я0-9+&#\/%=~_|$?!:,.]*\)|[-A-ZА-Я0-9+&#\/%=~_|$?!:,.])*(?:\([-A-ZА-Я0-9+&#\/%=~_|$?!:,.]*\)|[A-ZА-Я0-9+&#\/%=~_|$])/ig

    let content;
    /* if (fs.existsSync(url)) { // TODO: url list from file
      content = fs.readFileSync(options.file, 'utf8');
    } */
    const res = await axios.get(url);
    content = res.data;
    content = content.replace(/^#.*$/gm, ''); // remove comments

    let pageUrl;
    while (pageUrl = regex.exec(content)){
      if (pageUrl[0].match(/\.(png|jpg|js|css)$/)) continue;
      urls.push(pageUrl[0]);
    }

    const onlyUnique = (value, index, self) => self.indexOf(value) === index;
    return urls.filter(onlyUnique);
  }

  let itemsPartial = [];
  const getItemsPartial = () => {
    if (!options.partialReport) return itemsPartial;
    if (itemsPartial.length > 0) return itemsPartial;

    // load partial path
    const partialPath = path.normalize(`${options.partialReport}`);
    const partialJson = fs.readFileSync(partialPath, 'utf8');
    const partialItems = JSON.parse(partialJson).items;
    itemsPartial.push(...partialItems);

    return itemsPartial;
  }

  let urls = [];
  let lastSuccessUrl = '';
  if (options.urlList) {
    if (options.urls && options.urls.length > 1) urls = options.urls;
    else urls = await parseUrls(baseUrl);

    if (options.urlsRemain) urls = options.urlsRemain;
  }
  let fullUrls = options.urlsRemain ? options.urls : urls;

  // return urls list from url index
  const getUrlsRemain = (url) => {
    const index = fullUrls.indexOf(url);
    return index === -1 ? fullUrls : fullUrls.slice(index + 1);
  }

  if (options.partialReport && !options.urlsRemain) {
    options.partial = true;
    options.partNum = 1;
    // console.log("build urlsRemain from partialReport:");
    const itemsPartial = getItemsPartial();
    console.log(`Found previous items: ${itemsPartial.length}`);
    if (itemsPartial.length > 0) {
      const itemLast = itemsPartial[itemsPartial.length - 1];
      options.urlsRemain = getUrlsRemain(itemLast.url);
      urls = options.urlsRemain;
      // console.log("itemLast.url:", itemLast.url);
      // console.log("options.urlsRemain.length:", options.urlsRemain.length);
    }
  }

  // console.log('urls: ', urls);
  const baseName = sanitize(options.outName || domain);
  const csvPath = path.normalize(`${options.outDir}/${baseName}.csv`);
  const jsonPath = path.normalize(`${options.outDir}/${baseName}.json`);
  let screenshotPath;
  const screenshotExt = 'png'; // or jpg

  if (options.screenshot) {
    screenshotPath = getUserDir(options.socket.uid) + `/${baseName}-screenshots`;
    console.log("make screenshot directory:", screenshotPath);
    if (!fs.existsSync(screenshotPath)) fs.mkdirSync(screenshotPath);
  }
  function getScreenshotPath(url) {
    return `${screenshotPath}/${sanitize(url)}.${screenshotExt}`
  }
  function getScreenshotUrl(url) {
    // console.log("process.env:", process.env);
    return `${screenshotPath}/${sanitize(url)}.${screenshotExt}`.replace('data/reports', './reports');
  }


  let webPath;

  if (!options.color) color.white = color.red = color.reset = color.yellow = '';

  if (!options.fieldsPreset || !fieldsPresets[options.fieldsPreset]) {
    options.fieldsPreset = 'default';
  }

  let fields = [...(fieldsPresets[options.fieldsPreset] || [])];

  // exclude fields
  if (options.fieldsExclude && options.fieldsExclude.length > 0) {
    fields = fields.filter(f => {
      const fName = f.replace(/.*\./g, '');
      return !options.fieldsExclude.includes(fName);
    });
  }

  // custom fields
  if (Object.keys(options.fields).length > 0) {
    // console.log('options.fields: ', options.fields);
    const newFields = Object.keys(options.fields).map(f => 'result.' + f);
    fields = [...fields, ...newFields];
  }

  // lighthouse fields
  if (options.lighthouse) {
    for (let fName of fieldsPresets['lighthouse-all']) {
      if (fields.indexOf(fName) === -1) {
        fields.push(fName);
      }
    }
  }

  // screenshot fields
  if (options.screenshot && !fields.includes('screenshot')) {
    fields.push('screenshot');
  }

  // plugins fields
  const plugins = registry.getPlugins();
  for (let plugin of plugins) {
    if (options.disablePlugins.includes(plugin.name)){
      continue;
    }
    if (plugin.fields) for(let field of plugin.fields) {
      if (typeof field === 'string') {
        fields.push(field);
      } else {
        fields.push(field.name);
      }
      // console.log(`push ${field}`);
    }
  }

  // skip static
  if (options.skipStatic !== undefined) {
    SKIP_IMAGES = SKIP_CSS = SKIP_JS = options.skipStatic;
  }

  // open second chrome for lighthouse
  let lighthouseChrome;
  // const chromeLauncher = await import('chrome-launcher');
  if (options.lighthouse) {
    const chromeFlags = ['--no-sandbox'];
    if (options.headless) chromeFlags.push('--headless');
    lighthouseChrome = await chromeLauncher.launch({chromeFlags});
  }

  const exporter = new CSVExporter({
    file: csvPath,
    fields: fields,
    separator: ';',
  });

  const failedUrls = [];

  let crawler;
  const defaultOptions = {
    allowedDomains: options.limitDomain ? [domain] : undefined,
    skipRequestedRedirect: true, // all redirects marks as visited
    depthPriority: false, // without it find not all pages
    retryCount: 1,
    retryDelay: 5000,
    args: ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage'], // puppeteer freezes without it
    exporter,

    // url ignore rules
    preRequest(options) {
      // console.log(options.url);
      if (options.url.match(/\.(jpg|jpeg|png|gif)/i)) return false; // картинки
      if (options.url.match(/\?width=\d+&height=\d+/)) return false; // визитки, сотрудники
      if (options.url.includes('?vi=y')) return false; // версия для слабовидящих
      if (options.url.includes('gallery/?page=detail')) return false; // Битрикс Галерея 2.0
      if (options.url.includes('/?lightbox=')) return false; // lightbox
      if (options.url.includes('rk.php')) return false; // bitrix rk
      if (options.url.includes('/?catalog_view=')) return false; // bitrix display
      if (options.url.includes('/?SORT=')) return false; // bitrix sort
      if (options.url.includes('/filter/clear/apply/')) return false; // bitrix filter
      // if (options.url.match(/\?(category|age|usage|madein|season|brand)=/)) return false; // bitrix filter

      // http scan while first page was https
      return !(!options.urlList && url.parse(options.url).protocol !== protocol);
    },

    // сюда можно дописывать сборщики данных со страницы
    // поля надо добавить в fields выше
    evaluatePage: async () => {
      try {
        const customFields = await window.__customFields();
        const removeSelectors = await window.__removeSelectors();
        // console.log('window.__customFields(): ', JSON.stringify(customFields));

        let domainParts = location.host.split('.');
        const domain2level = domainParts.slice(domainParts.length - 2).
          join('.');
        const canonical = $('link[rel="canonical"]');
        const canonicalHref = canonical.attr('href');
        const relUrl = window.location.href.replace(`${window.location.protocol}//${window.location.host}`, '');
        const isCanonical = canonicalHref ?
          (canonicalHref === decodeURI(window.location.href) ||
          canonicalHref === decodeURI(relUrl) ? 1 : 0) : '';
        const h1 = $('h1');
        const description = $('meta[name="description"]');

        const result = {
          request_time:
            window.performance.timing.responseEnd -
            window.performance.timing.requestStart,
          title: $('title').text(),
          page_date: '',
          h1: h1.text().trim(),
          h1_count: h1.length,
          h2_count: $('h2').length,
          h3_count: $('h3').length,
          h4_count: $('h4').length,
          google_amp: $('link[rel="amphtml"]').length,
          dom_size: document.getElementsByTagName('*').length,
          head_size: document.head.innerHTML.length,
          body_size: document.body.innerHTML.length,
          html_size_rendered: document.head.innerHTML.length +
            document.body.innerHTML.length,
          text_ratio_percent: Math.round(
            document.body.innerText.length / document.body.innerHTML.length *
            100),
          images: $('img').length,
          images_without_alt: $('img:not([alt])').length,
          images_alt_empty: $('img[alt=""]').length,
          images_outer: $(
            'img[src^="http"]:not([src^="/"]):not([src*="' + domain2level +
            '"])').length,
          links: $('a[href]:not([href^="javascript"]):not([href^="#"])').length,
          links_inner: $(
            'a[href^="/"], a[href*="' + domain2level + '"]').length,
          links_outer: $(
            'a[href^="http"]:not([href^="javascript"]):not([href^="#"]):not([href^="/"]):not([href*="' +
            domain2level + '"])').length,
          // links_absolute: $('').length,
          description:
            (description.attr('content') &&
              description.
                attr('content').
                split('\n').
                join(' ')) ||
            '',
          keywords: $('meta[name="keywords"]').attr('content'),
          canonical: canonicalHref,
          is_canonical: isCanonical,
          canonical_count: canonical.length,
          og_title: $('meta[property="og:title"]').attr('content'),
          og_image: $('meta[property="og:image"]').attr('content'),
          schema_types: $.unique($('[itemtype]').
            map((i, item) => $(item).
              attr('itemType').
              replace(/https?:\/\/schema\.org\//, ''))).toArray().join(', '),
        };

        // page date from microformat
        const pageDate = $('[itemprop="datePublished"]').first();
        if (pageDate.length === 1) {
          const dateStr = pageDate.attr('datetime') || pageDate.attr('content');
          const d = new Date(dateStr);
          if (!isNaN(d.getTime())) {
            result.page_date = dateStr.substring(0, 10);
          }
        }

        for (let name in customFields) {
          try {
            result[name] = eval(customFields[name].replace(/`/g, '\''));
          } catch(e) {
            result[name] = `field ${name}: error parse ${customFields[name]}`;
          }
          // if(name == 'section') result[name] = $('.views-field.views-field-field-section a').text();
        }

        // remove elements before screenshot
        if (removeSelectors && removeSelectors.length > 0) {
          const selector = Array.isArray(removeSelectors) ? removeSelectors.join(',') : removeSelectors;
          $(selector).remove();
        }

        return result;
      } catch (e) {
        return {
          error: JSON.stringify(e),
        };
      }
    },

    onSuccess: result => {
      if (!result.result) return;

      if (result.result.error && result.response.status) {
        const msg = `${color.red}${result.response.url}: evaluatePage: Error collect page data: ${result.result.error}${color.reset}`;
        console.error(msg);
      }
    },

    customCrawl: async (page, crawl, crawler) => {
      // You can access the page object before requests
      await page.setRequestInterception(true);
      await page.setBypassCSP(true);

      //page.on('console', msg => console.log(msg.text()));
      await page.exposeFunction('__customFields', () => {
        return options.fields;
      });

      await page.exposeFunction('__removeSelectors', () => {
        return options.removeSelectors;
      });

      let mixedContentUrl = '';

      page.on('requestfinished', async request => {
        // console.log("requestfinished:", request.url());
        if (config.featureScreenshot) {
          await page.screenshot({
            path: screenshotPath + sanitize(request.url()) + '.' + screenshotExt,
          });
        }
      });

      // это событие срабатывает, когда chrome подгружает статику на странице (и саму страницу)
      page.on('request', request => {
        //console.log('request.url(): ', request.url());

        // check for mixed content, thanks to https://github.com/busterc/mixed-content-crawler/
        if (protocol === 'https:' &&
          ['image', 'stylesheet', 'script'].includes(request.resourceType()) &&
          request.url().match(/^http:/)) {
          request.notHTTPS = true;
          mixedContentUrl = request.url();
          return request.abort();
        }

        const isDoc = options.docsExtensions.some(
          ext => request.url().includes(`.${ext}`));
        if (isDoc) {
          // досюда как-то доходит
          request.abort();
        } else if (SKIP_IMAGES && request.resourceType() === 'image') {
          request.abort();
        } else if (SKIP_CSS && request.resourceType() === 'stylesheet') {
          request.abort();
        } else if (SKIP_JS && request.resourceType() === 'script') {
          request.abort();
        } else {
          request.continue();
        }
      });

      page.on('requestfailed', request => {
        // skip adv errors
        if (request.url().includes('googlesyndication')
          || request.url().includes('googleads')
          || request.url().includes('adfox')
          || request.url().includes('an.yandex.ru')
          || request.url().includes('nativeroll.tv')
        ) {
          return;
        }

        if (request.notHTTPS) {
          // console.error(
          //   `${color.red}${crawler._options.url}: mixed content: ${request.url()}${color.reset}`);
          return;
        }

        const isStatic = ['image', 'script', 'stylesheet'].includes(request.resourceType());
        if (!isStatic) {
          // console.error(`Request failed: ${request.response()?.status()}, ${request.url()}, ${request.failure().errorText}`);

          // add to result table when first error
          if (!failedUrls.includes(crawler._options.url)) {
            failedUrls.push(crawler._options.url);
            const result = {
              options: {},
              depth: 0,
              previousUrl: '',
              response: {
                url: crawler._options.url,
                status: request.response()?.status() || 0,
              },
              redirectChain: [],
              screenshot: null,
              cookies: [],
              links: [],
            };
            exporter.writeLine(result);
            lastSuccessUrl = crawler._options.url;
          }
        }
      });

      // count html size before render scripts
      let pageSize = 0;
      page.on('response', response => {
        if (response.url() !== crawler._options.url) return;
        response.buffer().then(buffer => {
          pageSize += buffer.length
        })
      });

      /*page.on('requestfinished', async request => {
        console.log("requestfinished:", request.url().substring(0, 128));
      });*/

      /*page.on('error', function(err) {
        console.error(`${color.red}Page error:${color.reset} ` + err.toString());
      });*/

      /*page.on('close', function() {
        console.log("Page closed");
        // console.error(`${color.red}Page closed${color.reset} `);
      });*/

      // js error at requested page
      /*page.on('pageerror', function(err) {
        console.error(`${color.red}pegeerror:${color.reset} ` + err.toString());
      });*/

      // console.log('co '+ crawler._options.url);

      // костыль, который возвращает фейково обойдённый документ, если он признан документом
      // нужно, чтобы доки не сканировались (выдают ошибку), но при этом добавлялись в csv
      // т.к. в этом контексте нет текущего урла, он задаётся в глобал через событие requeststarted
      const isDoc = crawler._options.url && options.docsExtensions.some(
        ext => crawler._options.url.includes(`.${ext}`));
      if (isDoc) {
        const answer = await axios({
          url: encodeURI(crawler._options.url),
          method: "HEAD",
          validateStatus: () => true,
        });
        return {
          options: {},
          depth: 0,
          previousUrl: '',
          response: {
            url: crawler._options.url,
            status: answer.status,
          },
          redirectChain: [],
          result: {
            html_size: answer.headers['content-length'] || 0,
          },
          screenshot: null,
          cookies: [],
          links: [],
        };
      }

      // The result contains options, links, cookies etc.
      // log(`before crawl(), url: ${crawler._options.url}`);
      let result = {};
      try {
        result = await crawl();
        // console.log("after crawl(), result.response.status: ", result?.response?.status);

        if (!result.result) result.result = {};
        result.result.x_cache = 0;
        for (let header in result.response.headers) {
          if (header.match(/x-cache/) && result.response.headers[header].match(/hit/i)) {
            result.result.x_cache = 1;
            break;
          }
        }

        if (options.lighthouse) {
          const opts = {
            logLevel: 'info',
            maxWaitForLoad: 30000,
            // extends: 'lighthouse:default',
            /*onlyAudits: [
              'first-meaningful-paint',
              'speed-index',
              'first-cpu-idle',
              'interactive',
            ],*/
            // onlyCategories : [ 'performance'/*, 'pwa', 'accessibility', 'best-practices', 'seo'*/ ],
            port: lighthouseChrome.port,
            locale: options.lang,
          };
          // console.log("lighthouse:", crawler._options.url);
          // const lighthouse = await import('lighthouse');
          // console.log("lighthouse:", lighthouse);
          const res = await lighthouse(crawler._options.url, opts);
          const data = JSON.parse(res.report);

          const audits = [
            'first-contentful-paint',
            'speed-index',
            'largest-contentful-paint',
            'interactive',
            'total-blocking-time',
            'cumulative-layout-shift',
          ];
          const categories = [
            'performance',
            'accessibility',
            'best-practices',
            'seo',
            'pwa'];
          const lighthouseData = {
            scores: {},
          };

          // const fieldConfigs = []; // for code-generation of presets/fields.js

          for (let auditName of audits) {
            if (!data.audits[auditName]) continue;
            lighthouseData[auditName] = parseInt(
              data.audits[auditName].numericValue);
          }

          for (let categoryId of categories) {
            if (!data.categories[categoryId]) continue;

            // lighthouse.scores
            lighthouseData.scores[categoryId] = parseInt(data.categories[categoryId].score) * 100;

            // all audits
            for (let auditRef of data.categories[categoryId].auditRefs) {
              let value;
              const auditName = auditRef.id;
              const audit = data.audits[auditName];

              if (audit.numericValue) value = parseInt(audit.numericValue);
              else value = audit.score;

              lighthouseData[auditName] = value;

              // add to sheet fields
              if (options.fieldsPreset === 'lighthouse-all') {
                const fieldId = 'lighthouse.' + auditName;
                if (fields.indexOf(fieldId) === -1) {
                  fields.push(fieldId);
                }
              }

              // generate field config
              const fieldConfig = {
                name: 'lighthouse_' + audit.id,
                comment: audit.title,
                description: audit.description,
                groups: ['# Lighthouse: ' + data.categories[categoryId].title],
                type: 'integer',
              };
              if (auditRef.group) {
                const groupTitle = data.categories[categoryId].title + ': ' +
                  data.categoryGroups[auditRef.group].title;
                fieldConfig.groups.push(groupTitle);
                // fieldConfig.groups = [groupTitle];
              }
              // fieldConfigs.push(fieldConfig);
            }
          }

          result.lighthouse = lighthouseData;
          // console.log(JSON.stringify(fieldConfigs)); // copy to fields.js
          // console.log(lighthouseData);
        }

        if (options.screenshot) {
          result.screenshot = getScreenshotUrl(result.response.url)
        }

        if (!result.result) result.result = {}; // TODO: find out why result.result is not defined sometimes, possible js error into page

        result.result.mixed_content_url = mixedContentUrl;
        if (result.response.url) {
          // url might be malformed
          try {
            result.response.url = decodeURI(result.response.url);
          } catch (e) {
            // do nothing
          }
        }

        if (result.redirectChain.length > 0) {
          result.result.redirected_from = result.redirectChain[0]?.url;
          if (result.response.status === 200) result.response.status = 301;
        }
        result.result.redirects = result.redirectChain.length;

        lastSuccessUrl = result.result.redirected_from || result.response.url;

        // console validate output
        // was in onSuccess(), but causes exception on docs
        if (options.consoleValidate) {
          const msgs = [];
          const validate = validateResults(result, fields); // TODO: fields declared implicitly
          for (let name in validate) {
            const res = validate[name];
            const msgColor = {warning: color.yellow, error: color.red}[res.type];
            msgs.push(`${name}: ${msgColor}${res.msg}${color.reset}`);
          }
          if (msgs.length > 0) console.log(msgs.join(', '));
        }

        // You can access the page object after requests
        // TODO: skip depending of fields preset
        result.content = await page.content();
        result.result.html_size = pageSize;
      }
      catch (e) {
        const err = e.message?.substring(0, 512);
        if (isCancalling || isMaxRequested) {
          // Scan is cancelled, suppress errors
          return;
        }
        log(`Error while scan ${crawler._options.url}`);
        // console.log("result:", result);
        if (!result.result) {
          result = {
            options: {},
            depth: 0,
            previousUrl: '',
            response: {
              url: crawler._options.url,
              status: 0,
            },
            result: {},
            redirectChain: [],
            screenshot: null,
            cookies: [],
            links: [],
          }
        }
        // if (!result.response) result.response = { ok: true, url: crawler._options.url };
        // console.log("err:", err);
        const errText = `${err}`;
        let isRetry = false;
        if (errText.includes('net::ERR_CERT')) { // ERR_CERT_DATE_INVALID, net::ERR_CERT_AUTHORITY_INVALID, net::ERR_CERT_COMMON_NAME_INVALID
          result.result.error = 'ssl_err';
          result.response.status = -1;
        }
        else if (errText.includes('net::ERR_NAME_NOT_RESOLVED')) {
          result.result.error = 'dns_err';
          result.response.status = -1;
        }
        else if (errText.includes('Navigation Timeout Exceeded')) {
          result.result.error = 'timeout';
          crawler._options.timeout = 30000;
          isRetry = true;
        }
        else if (errText.includes('browser has disconnected')) {
          result.result.error = 'browser_err';
          isRetry = true;
        }
        else if (errText.includes('net::ERR_INVALID_RESPONSE')) {
          result.result.error = 'invalid response';
          result.response.status = -1;
        }
        else if (errText.includes('Execution context was destroyed')) {
          result.result.error = 'context destroyed';
        }
        else if (errText.includes('net::ERR_ABORTED') || errText.includes('net::ERR_CONNECTION_REFUSED')) {
          result.result.error = 'aborted (blocked)';
        }
        else if (errText.includes('net::ERR_EMPTY_RESPONSE')) {
          result.result.error = 'empty (blocked)';
        }
        else if (errText.includes('URI malformed')) result.result.error = 'bad_url'; // not used?
        else {
          // console.log(err);
          log("Unknown error, errText:", errText);
          console.log(e.stack);
        }
        // console.log("result.result.error:", result.result.error);
        // console.log("errText:", errText);
        if (result.result.error) log(`${result.result.error}${isRetry ? ', retry' : ''}`);
        if (isRetry) throw e; // for retry
      }


      // plugins afterRequest
      if (result.response?.status) {
        try {
          socketSend(options.socket, 'ping', 'ping'); // https://github.com/socketio/socket.io/issues/3025
          // log('ping', opts.socket, true);
          await registry.execPlugins(result, options, 'afterRequest');
        } catch (e) {
          console.log('Error while plugins afterRequest');
          console.log(e);
        }
      }

      // console.log('result.readability_reading: ', result.readability_reading);
      // console.log('result.readability_readingMinutes: ', result.readability_readingMinutes);
      // console.log('result.readability_readingTime: ', result.readability_readingTime);

      // You need to extend and return the crawled result
      return result;
    }, // /customCrawl
  };

  // for suppress jetbrains warning: function not used
  if (!defaultOptions.preRequest) return;
  if (!defaultOptions.evaluatePage) return;
  if (!defaultOptions.onSuccess) return;
  if (!defaultOptions.customCrawl) return;

  const crawlerOptions = {...defaultOptions, ...options};

  if ('screenshot' in crawlerOptions && crawlerOptions.screenshot) {
    //delete crawlerOptions.screenshot; // don't send `screenshot` arg to crawler, for custom screenshots
    crawlerOptions.screenshot = {
      path: screenshotPath + '/screenshot.' + screenshotExt, // replace bool to screennshot dir
      fullPage: true,
    };
  }

  crawlerOptions.args = defaultOptions.args; // override args for chromium


  // start
  const start = Date.now();
  if (!options.partialFirstStartTime) options.partialFirstStartTime = start;
  if (!options.partNum) options.partNum = 0;

  // console.log(`${color.yellow}Scrapping ${baseUrl}...${color.reset}`);
  let requestedCount = 0;

  let isSaving = false;

  const saveProgress = async () => {
    if (isSaving) return {};
    isSaving = true;
    const saveStart = Date.now();
    const scanTime = Math.round((Date.now() - start) / 1000);
    const startTime = options.partialFirstStartTime;
    const itemsPartial = getItemsPartial();
    const { items } = await saveAsJson({
      csvPath,
      jsonPath,
      lang: options.lang,
      preset: options.preset,
      defaultFilter: options.defaultFilter,
      url: baseUrl,
      args: options.args,
      scanTime,
      itemsPartial,
      startTime,
      partNum: options.partNum,
    })
    const { jsonName, localPath } = copyJsonToReports(jsonPath, options.socket?.uid, undefined, startTime, true);

    // send result json to socket
    socketSend(options.socket, 'result', {name: jsonName, isProgress: true, count: items.length});
    const saveTime = Date.now() - saveStart;
    console.log(`Save to ${jsonName} (${items.length} items, ${saveTime} ms)`);
    isSaving = false;

    return { jsonName, localPath };
  }

  try {
    crawler = await HCCrawler.launch(crawlerOptions);
  } catch (e) {
    console.log(e);
  }


  let tryingRestart = false;
  let isCancalling = false;

  crawler.on('requeststarted', async opts => {
    // const failedEmulated10Percent = requestedCount > 10 && Math.random() < 0.1;

    // catch error after scan
    if (crawler._browser._connection._closed /*|| failedEmulated10Percent*/) {
      // log("Browser connection closed (requeststarted)");
      // 11.03.2021 12:22 fix: suppress headless-chrome-crawler exceptions after max requests reached
      // but if return it can cause infinite loop
      // return;

      // start new crawler for remain urls
      if (urls.length > 0) {
        if (!tryingRestart) {
          console.log(`${color.yellow}Trying to restart urls scan from last known url:${color.reset}`);
          tryingRestart = true;
          const urlsRemain = getUrlsRemain(lastSuccessUrl);
          const seconds = 30;
          log (`Trying to restart scanning of ${fullUrls.length} urls, from: ${lastSuccessUrl}`);

          // save json, pass report to next crawler
          try {
            const { localPath} = await saveProgress();

            // output next url
            options.urlsRemain = options.urlsRemain || [];
            // const itemsPartial = getItemsPartial();
            let scanned = options.urlsRemain.length ? fullUrls.length - options.urlsRemain.length : 0;
            let requestedCount = crawler.requestedCount() + scanned + 1;
            let queueCount = await crawler.queueSize();
            log(`${requestedCount} ${urlsRemain[0]}, restarting in ${seconds} seconds... (${queueCount})`);

            // start new crawler
            // socketSend(options.socket, 'status', 'waiting');
            setTimeout(() => {
              const newOptions = {
                ...options, ...{
                  urls: [...fullUrls],
                  cancel: false,
                  urlsRemain: urlsRemain,
                  partialReport: [localPath],
                  partNum: options.partNum + 1,
                  removeCsv: false,
                  partialFirstStartTime: options.partialFirstStartTime || start,
                }
              };

              // console.log("newOptions:", newOptions);
              scrapSite({baseUrl: baseUrl, options: newOptions});
            }, seconds * 1000);
          } catch (e) {
            console.log(e);
          }
        }
      } else {
        log("Aborting scan...");
        socketSend(options.socket, 'status', 'cancel command');
      }

      // should be after module.exports() recursive call
      options.cancel = true;
      options.partial = true;
      // return;
    }

    // set screenshot path before crarl
    if (options.screenshot) {
      const path = getScreenshotPath(opts.url);
      console.log("save screenshot:", path);
      crawler._options.screenshot.path = path;
    }

    let queueCount = await crawler.queueSize();
    requestedCount = crawler.requestedCount() + 1;

    let requestedCountRelative = requestedCount;
    if (options.urlList && options.partial && options.urlsRemain?.length) {
      // const itemsPartial = getItemsPartial();
      const itemsScannedCount = fullUrls.length - options.urlsRemain.length;
      requestedCountRelative = requestedCount + itemsScannedCount;
      // queueCount = options.urls.length;
    }

    if (options.cancel) {
      if (!isCancalling) {
        console.log("cancel scrapping, setMaxRequest:", requestedCount);
        crawler.setMaxRequest(requestedCount); // cancel command
        void saveProgress();
        isCancalling = true;
      }
    }

    // console.log('crawler: ', crawler);
    try {
      const url = new URL(opts.url);
      // console.log('url: ', url);
      log(`${requestedCountRelative} ${url.href} (${queueCount})`);
    } catch(e) {
      // possible URIError: URI malformed
      console.error(e);
      log(`${requestedCountRelative} ${opts.url} (${queueCount})`);
    }

    if (requestedCountRelative % saveAfterEvery === 0) {
      void saveProgress();
    }
  });
  crawler.on('requestfailed', error => {
    if (crawler._browser._connection._closed) {
      // log("Browser connection closed (requestfailed)");
      return;
    }
    try {
      console.error(
        `Failed: ${decodeURI(error.options.url)}`);
      crawler._options.timeout = 30000; // TODO: to config, extend timeout on error
    } catch(e) {
      // possible URIError: URI malformed
      console.error(e);
    }
  });
  crawler.on('requestdisallowed', options => {
    log(`Disallowed in robots.txt: ${decodeURI(options.url)}`, options.socket);
    console.error(`${color.yellow}Disallowed in robots.txt: ${decodeURI(
      options.url)}${color.reset}`);
  });
  crawler.on('maxdepthreached', () => {
    if (options.maxDepth > 1) console.log(`${color.yellow}Max depth reached${color.reset}`);
  });
  crawler.on('maxrequestreached', () => {
    if (crawler._browser._connection._closed) return; // catch error after scan
    console.log(`\n${color.yellow}Max requests reached${color.reset}`);
    isMaxRequested = true;
    // console.log(`${color.yellow}Please, ignore this error:${color.reset}`);
  });

  if (options.urlList) {
    log('Queue ' + urls.length + ' urls', options.socket);
    // console.log(`${urls.join('\n')}`);
    for (let url of urls) {
      await crawler.queue(url);
    }
  } else {
    await crawler.queue(baseUrl);
  }

  await crawler.onIdle();
  await crawler.close();

  // after scan
  const scanTime = Math.round((Date.now() - start) / 1000);
  const perPage = Math.round((scanTime / requestedCount) * 100) / 100;

  // close lighthouse's chrome
  await chromeLauncher.killAll();

  const outValidationSummary = () => {
    const sum = getValidationSum();
    if (Object.entries(sum).length === 0) return;
    console.log(`\n\n${color.white}Validation summary:${color.reset}`);
    for (let colName in sum) {
      console.log(`\n${color.white}${colName}:${color.reset}`);
      for (let res of sum[colName]) {
        const msgColor = {warning: color.yellow, error: color.red}[res.type];
        console.log(`${msgColor}${res.msg}${color.reset}\t${res.url}`);
      }
    }
  };

  const finishScan = async () => {
    console.log('');

    if (options.consoleValidate) {
      outValidationSummary();
    }

    if (options.json) {
      const itemsPartial = getItemsPartial();
      const data = await saveAsJson({
        csvPath,
        jsonPath,
        lang: options.lang,
        preset: options.preset,
        defaultFilter: options.defaultFilter,
        url: baseUrl,
        args: options.args,
        scanTime,
        itemsPartial,
        startTime: options.partialFirstStartTime,
        partNum: options.partNum,
      });


      if (!options.removeJson) console.log(`Saved ${data.items.length} items to ${jsonPath}`);

      // user plugins
      await registry.execPlugins(jsonPath, options, 'afterScan');

      if (!options.webService) {
        if (options.upload) webPath = await uploadJson(jsonPath);
        await startViewer(jsonPath, webPath);
      }

      if (options.webService) {
        const { jsonName } = copyJsonToReports(jsonPath, options.socket.uid, undefined, options.partialFirstStartTime, true);

        // send result json to socket
        socketSend(options.socket, 'result', {name: jsonName, count: data.items.length});
        console.log(`Save to ${jsonName}`);

        if (!options.cancel) log(`Finish audit: ${jsonName}`);


        // TODO: error upload 8MB+
        if (options.upload) {
          webPath = await uploadJson(jsonPath);
          socketSend(options.socket, 'result', {json: webPath, count: data.items.length});
        }

        const mins = Number(scanTime / 60).toFixed(1);
        log(
          `Finish: ${mins} mins, ${data.items.length} items (${perPage} sec per page)` +
          `${options.partNum > 0 ? `, ${options.partNum} resumes` : ''}`,
          options.socket);

        // return stats
        return {
          time: scanTime,
          perPage: perPage,
          pages: requestedCount,
        }
      }

      if (options.removeJson) fs.unlinkSync(jsonPath);
    }

    if (options.removeCsv) fs.unlinkSync(csvPath);

    const mins = Number(scanTime / 60).toFixed(1);
    log(`Finish: ${mins} mins (${perPage} sec per page)`, options.socket);
  };

  const tryFinish = async (tries) => {
    try {
      return await finishScan();
    } catch (e) {
      log('error after scan: ' + e.message.substring(0, 512));
      console.error(e);
      console.error(e.stacktrace);
    }
  };

  return await tryFinish(finishTries);
}

// export default {scrapSite};
export default scrapSite;