import chalk from 'chalk';
import fs from 'fs-extra';
import _ from 'lodash';
import path from 'path';

import TestResult from './test-result/test-result';

import { ERROR, FAIL, SUCCESS, UPDATED } from './constants/test-statuses';

export const getReferencePath = (testResult: TestResult, stateName: string) =>
  createPath('ref', testResult, stateName);

export const getCurrentPath = (testResult: TestResult, stateName: string) =>
  createPath('current', testResult, stateName);

export const getDiffPath = (testResult: TestResult, stateName: string) =>
  createPath('diff', testResult, stateName);

export const getReferenceAbsolutePath = (
  testResult: TestResult,
  reportDir: string,
  stateName?: string,
) => {
  const referenceImagePath = getReferencePath(testResult, stateName);

  return path.resolve(reportDir, referenceImagePath);
};

export const getCurrentAbsolutePath = (
  testResult: TestResult,
  reportDir: string,
  stateName?: string,
) => {
  const currentImagePath = getCurrentPath(testResult, stateName);

  return path.resolve(reportDir, currentImagePath);
};

export const getDiffAbsolutePath = (
  testResult: TestResult,
  reportDir: string,
  stateName?: string,
) => {
  const diffImagePath = getDiffPath(testResult, stateName);

  return path.resolve(reportDir, diffImagePath);
};

/**
 * @param kind - одно из значений 'ref', 'current', 'diff'
 * @param {StateResult} result
 * @param stateName - имя стэйта для теста
 */

const createPath = (kind: string, result: TestResult, stateName: string) => {
  const attempt = result.attempt || 0;
  const imageDir = _.compact(['images', result.imageDir, stateName]);
  const components = imageDir.concat(`${result.browserId}~${kind}_${attempt}.png`);

  return path.join.apply(null, components);
};

// создает директорию и копирует скрин
export const copyImageAsync = (srcPath: string, destPath: string) => {
  return makeDirFor(destPath)
    .then(() => fs.copy(srcPath, destPath));
};

/**
 * @param {TestStateResult} result
 * @param {String} destPath
 * @returns {Promise}
 */
export const saveDiff = (result: any, destPath: string) => {
  return makeDirFor(destPath)
    .then(() => result.saveDiffTo(destPath));
};

/**
 * @param {String} destPath
 */
export const makeDirFor = (destPath: string) => {
  return fs.mkdirs(path.dirname(destPath));
};

export const logger = _.pick(console, ['log', 'warn', 'error']);

export const logPathToHtmlReport = (reportBuilder: any) => {
  const reportPath = `file://${reportBuilder.reportPath}`;

  logger.log(`Your HTML report is here: ${chalk.yellow(reportPath)}`);
};

export const logError = (e: Error) => {
  logger.error(e.stack);
};

export const getPathsFor = (status: string, formattedResult: TestResult, stateName?: string) => {
  if (status === SUCCESS || status === UPDATED) {
    return { expectedPath: getReferencePath(formattedResult, stateName) };
  }
  if (status === FAIL) {
    return {
      actualPath: getCurrentPath(formattedResult, stateName),
      diffPath: getDiffPath(formattedResult, stateName),
      expectedPath: getReferencePath(formattedResult, stateName),
    };
  }
  if (status === ERROR) {
    return {
      actualPath: formattedResult.state ? getCurrentPath(formattedResult, stateName) : '',
    };
  }

  return {};
};

export const hasImage = (formattedResult: TestResult) => {
  return !!formattedResult.getImagesInfo().length
    || !!formattedResult.currentPath
    || !!formattedResult.screenshot;
};

export const prepareCommonJSData = (data: any) => {
  return [
    `var data = ${JSON.stringify(data)};`,
    'try { module.exports = data; } catch(e) {}',
  ].join('\n');
};