utils.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
var vscode = require("vscode");
var path = require("path");
var aws4 = require("aws4");
var request = require("request");
var xml2js = require("xml2js");
var awsSession = null;
function promptForMFAIfRequired() {
    return new Promise(function (resolve, reject) {
        var extensionConfig = vscode.workspace.getConfiguration('cloud9sync');
        if (extensionConfig.get('mfaSerial') != "") {
            vscode.window.showInputBox({
                placeHolder: "",
                prompt: "Enter your MFA code.",
                value: "",
                ignoreFocusOut: false
            }).then(function (mfa_token) {
                resolve("&TokenCode=" + mfa_token);
            });
        }
        else {
            resolve("");
        }
    });
}
function GetAWSCreds() {
    return new Promise(function (resolve, reject) {
        var extensionConfig = vscode.workspace.getConfiguration('cloud9sync');
        var awsregion = extensionConfig.get('region');
        if (extensionConfig.get('accessKey') == "" ||
            extensionConfig.get('secretKey') == "" ||
            !extensionConfig.get('accessKey') ||
            !extensionConfig.get('secretKey')) {
            resolve(null);
        }
        if (awsSession) {
            resolve(awsSession['value']);
        }
        else if (extensionConfig.get('mfaSerial') != "" || extensionConfig.get('assumeRole') != "") {
            var path_1 = "/?Version=2011-06-15&DurationSeconds=" + extensionConfig.get('sessionDuration');
            if (extensionConfig.get('mfaSerial') != "") {
                path_1 += "&SerialNumber=" + extensionConfig.get('mfaSerial');
            }
            if (extensionConfig.get('assumeRole') != "") {
                path_1 += '&Action=AssumeRole&RoleSessionName=VSCodeCloud9&RoleArn=' + extensionConfig.get('assumeRole');
            }
            else {
                path_1 += '&Action=GetSessionToken';
            }
            promptForMFAIfRequired().then(function (mfa_token) {
                path_1 += mfa_token;
                var awsreq = aws4.sign({
                    service: 'sts',
                    region: 'us-east-1',
                    method: 'POST',
                    path: path_1,
                    headers: {
                        'Content-Type': 'application/x-amz-json-1.1'
                    },
                }, {
                    secretAccessKey: extensionConfig.get('secretKey'),
                    accessKeyId: extensionConfig.get('accessKey')
                });
                request.post({
                    url: "https://" + awsreq.hostname + awsreq.path,
                    headers: awsreq.headers,
                    body: awsreq.body,
                    rejectUnauthorized: false,
                    proxy: GetProxy()
                }, function (err, httpResponse, cred_response) {
                    console.log(httpResponse);
                    xml2js.parseString(cred_response, {
                        ignoreAttrs: true
                    }, function (err, result) {
                        // TODO: Handle failure here
                        var creds = null;
                        if (extensionConfig.get('assumeRole') != "") {
                            creds = result['AssumeRoleResponse']['AssumeRoleResult'][0]['Credentials'][0];
                        }
                        else {
                            creds = result['GetSessionTokenResponse']['GetSessionTokenResult'][0]['Credentials'][0];
                        }
                        awsSession = {
                            'value': {
                                secretAccessKey: creds['SecretAccessKey'][0],
                                accessKeyId: creds['AccessKeyId'][0],
                                sessionToken: creds['SessionToken'][0]
                            },
                            'expiry': creds['Expiration'][0]
                        };
                        // TODO: Handle refresh here
                        resolve(awsSession['value']);
                    });
                });
            });
        }
        else {
            resolve({
                secretAccessKey: extensionConfig.get('secretKey'),
                accessKeyId: extensionConfig.get('accessKey')
            });
        }
    });
}
exports.GetAWSCreds = GetAWSCreds;
function FileNameToUri(fileName) {
    var rootPath = "";
    vscode.workspace.workspaceFolders.forEach(function (workspaceFolder) {
        if (workspaceFolder.uri.scheme == "file") {
            rootPath = workspaceFolder.uri.fsPath;
        }
    });
    var filePath = path.join(rootPath, fileName);
    return vscode.Uri.file(filePath);
}
exports.FileNameToUri = FileNameToUri;
function ShortenFilePath(filePath) {
    var rootPath = "";
    vscode.workspace.workspaceFolders.forEach(function (workspaceFolder) {
        if (workspaceFolder.uri.scheme == "file") {
            rootPath = workspaceFolder.uri.fsPath;
        }
    });
    return EnsureLeadingSlash(filePath.replace(rootPath, "").replace("\\", "/"));
}
exports.ShortenFilePath = ShortenFilePath;
function GetShortFilePath(document) {
    var filePath = document.fileName;
    var rootPath = "";
    if (document.uri.scheme == "cloud9") {
        filePath = "/" + document.uri.path.split("/").slice(2).join('/');
    }
    vscode.workspace.workspaceFolders.forEach(function (workspaceFolder) {
        if (workspaceFolder.uri.scheme == "file") {
            rootPath = workspaceFolder.uri.fsPath;
        }
    });
    return filePath.replace(rootPath, "").replace("\\", "/");
}
exports.GetShortFilePath = GetShortFilePath;
function EnsureLeadingSlash(str) {
    if (str[0] == '/' || str[0] == '\\')
        return str;
    return "/" + str;
}
exports.EnsureLeadingSlash = EnsureLeadingSlash;
function GetRegion() {
    var extensionConfig = vscode.workspace.getConfiguration('cloud9sync');
    return extensionConfig.get('region');
}
exports.GetRegion = GetRegion;
function GetProxy() {
    var extensionConfig = vscode.workspace.getConfiguration('cloud9sync');
    var proxy = extensionConfig.get('proxy');
    if (proxy == "")
        return null;
    return proxy;
}
exports.GetProxy = GetProxy;
function ReducePromises(array, fn) {
    var results = [];
    return array.reduce(function (p, item) {
        return p.then(function () {
            return fn(item).then(function (data) {
                results.push(data);
                return results;
            }).catch(function (y) {
                console.error(y);
            });
        }).catch(function (x) {
            console.error(x);
        });
    }, Promise.resolve());
}
exports.ReducePromises = ReducePromises;