"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : new P(function (resolve) { resolve(result.value); }).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
const debugModule = require("debug");
const snyk = require("../../../lib/");
const protect = require("../../../lib/protect");
const analytics = require("../../../lib/analytics");
const detect = require("../../../lib/detect");
const pm = require("../../../lib/package-managers");
const errors_1 = require("../../../lib/errors");
const errors = require("../../../lib/errors");
const debug = debugModule('snyk');
function protectFunc(options) {
    return __awaiter(this, void 0, void 0, function* () {
        const protectOptions = Object.assign({}, options);
        protectOptions.loose = true; // replace missing policies with empty ones
        protectOptions.vulnEndpoint = '/vuln/npm/patches';
        // TODO: fix this by providing better patch support for yarn
        // yarn hoists packages up a tree so we can't assume their location
        // on disk without traversing node_modules
        // currently the npm@2 nd npm@3 plugin resolve-deps can do this
        // but not the latest node-lockfile-parser
        // HACK: if yarn set traverseNodeModules option to
        // bypass lockfile test for wizard
        protectOptions.traverseNodeModules = true;
        try {
            const packageManager = detect.detectPackageManager(process.cwd(), protectOptions);
            const supportsProtect = pm.PROTECT_SUPPORTED_PACKAGE_MANAGERS.includes(packageManager);
            if (!supportsProtect) {
                throw new Error('Snyk protect for ' +
                    pm.SUPPORTED_PACKAGE_MANAGER_NAME[packageManager] +
                    ' projects is not currently supported');
            }
        }
        catch (error) {
            return Promise.reject(error);
        }
        if (protectOptions.interactive) {
            // silently fail
            return Promise.reject(new Error('Snyk protect interactive mode ' +
                'has moved. Please run `snyk wizard`'));
        }
        if (protectOptions['dry-run']) {
            debug('*** dry run ****');
        }
        else {
            debug('~~~~ LIVE RUN ~~~~');
        }
        try {
            const policy = yield snyk.policy.load(protectOptions['policy-path']);
            if (policy.patch) {
                return patch(protectOptions);
            }
            return 'Nothing to do';
        }
        catch (e) {
            let error;
            if (e.code === 'ENOENT') {
                error = new errors.PolicyNotFoundError();
            }
            else {
                error = new errors.FailedToLoadPolicyError();
                error.innerError = e;
            }
            throw error;
        }
    });
}
function patch(options) {
    return __awaiter(this, void 0, void 0, function* () {
        try {
            const response = (yield snyk.test(process.cwd(), options));
            // TODO: need to add support for multiple test results being returned
            // from test (for example gradle modules)
            if (!response.vulnerabilities) {
                const e = new errors_1.CustomError('Code is already patched');
                e.strCode = 'ALREADY_PATCHED';
                e.code = 204;
                throw e;
            }
            yield protect.patch(response.vulnerabilities, !options['dry-run']);
            analytics.add('success', true);
            return 'Successfully applied Snyk patches';
        }
        catch (e) {
            if (e.code === 'ALREADY_PATCHED') {
                analytics.add('success', true);
                return e.message + ', nothing to do';
            }
            analytics.add('success', false);
            throw e;
        }
    });
}
module.exports = protectFunc;
//# sourceMappingURL=index.js.map