/*---------------------------------------------------------
 * Copyright (C) Microsoft Corporation. All rights reserved.
 *--------------------------------------------------------*/

 import shelljs = require("async-shelljs");
import path = require('path');
const fs = require('async-file');

declare var __coverage__: any;

export default class CoverageWritingTestRunner {
    constructor(private baseRunner: any) {
    }

    public configure(mochaOpts: any) {
        this.baseRunner.configure(mochaOpts);
    }

    public run(testRoot: string, clb: any) {
        let promiseResolve: any;
        let clbArgsLocal: { error: any, failures?: number };

        new Promise<{ error: any, failures?: number }>(function (resolve, reject) {
            promiseResolve = (error: any, failures?: number) => resolve({ error, failures });
        })
            .then(async clbArgs => {
                clbArgsLocal = clbArgs;
                return this.writeCoverage();
            })
            .then(() => clb(clbArgsLocal.error, clbArgsLocal.failures),
                () => clb(clbArgsLocal.error, clbArgsLocal.failures));

        this.baseRunner.run(testRoot, promiseResolve);
    }

    private async writeCoverage(): Promise<void> {
        if (typeof __coverage__ !== 'undefined') {
            let nycFolderPath = path.join(process.env.CODE_WORKSPACE_ROOT, ".nyc_output", "integration");

            let rawCoverageJsonPath: string;
            let remappedCoverageJsonPath: string;
            let outFolderPath: string;
            let remapIstanbulPath: string;
            let nodePath: string;

            try {
                if (!(await fs.exists(nycFolderPath))) {
                    await fs.mkdirp(nycFolderPath);

                    rawCoverageJsonPath = path.join(nycFolderPath, `${process.env.OSVC_SUITE}.json.raw`);
                    remappedCoverageJsonPath = path.join(nycFolderPath, `${process.env.OSVC_SUITE}.json`);
                    outFolderPath = path.join(process.env.CODE_WORKSPACE_ROOT, "out");
                    remapIstanbulPath = path.join(process.env.CODE_WORKSPACE_ROOT, "node_modules", "remap-istanbul", "bin", "remap-istanbul.js");
                    nodePath = "";
                    if (process.env.NVM_BIN) {
                        nodePath = `${process.env.NVM_BIN}${path.sep}`;
                    }

                    await fs.writeTextFile(rawCoverageJsonPath, JSON.stringify(__coverage__));

                    await shelljs.asyncExec(`${nodePath}node ${remapIstanbulPath} -i ${rawCoverageJsonPath} -o ${remappedCoverageJsonPath}`, {
                        cwd: outFolderPath
                    });

                    let remappedResult = JSON.parse(await fs.readTextFile(remappedCoverageJsonPath));

                    let finalResult = <{[details: string] : { path: string }}>{};

                    for (let key in remappedResult) {
                        if (remappedResult[key].path) {
                            let realPath = key.replace("../", "./");

                            finalResult[realPath] = remappedResult[key];
                            finalResult[realPath].path = realPath;
                        }
                        else {
                            finalResult[key] = remappedResult[key];
                        }
                    }

                    await fs.writeTextFile(remappedCoverageJsonPath, JSON.stringify(finalResult));

                    console.log(`done remapping ${finalResult}`);
                }
            }
            catch (e) {
                    console.log(`Coverage remapping failure: ${JSON.stringify(e)}`);
                    console.log(`* rawCoverageJsonPath: ${rawCoverageJsonPath}`);
                    console.log(`* remappedCoverageJsonPath: ${remappedCoverageJsonPath}`);
                    console.log(`* outFolderPath: ${outFolderPath}`);
                    console.log(`* remapIstanbulPath: ${remapIstanbulPath}`);
                    console.log(`* nodePath: ${nodePath}`);
                    console.log(e);
                }
            }
    }
    }