@alicloud/fun/lib/vswitch.js

'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 _ = require('lodash');
const debug = require('debug')('fun:nas');
const { red } = require('colors');
const { getProfile } = require('./profile');
const { getFcClient, getNasPopClient } = require('./client');
const { promptForConfirmContinue } = require('./init/prompt');
var requestOption = {
    method: 'POST'
};
function createDefaultVSwitch(vpcClient, region, vpcId, vswitchName) {
    return __awaiter(this, void 0, void 0, function* () {
        const vswitchZoneId = yield selectAllowedVSwitchZone(vpcClient, region);
        var vswitchId;
        try {
            // 创建 vswitch
            vswitchId = yield createVSwitch(vpcClient, {
                region,
                vpcId,
                zoneId: vswitchZoneId,
                vswitchName: vswitchName
            });
        }
        catch (ex) {
            throw ex;
        }
        return vswitchId;
    });
}
function describeVSwitchAttributes(vpcClient, region, vswitchId) {
    return __awaiter(this, void 0, void 0, function* () {
        const params = {
            'RegionId': region,
            'VSwitchId': vswitchId
        };
        return yield vpcClient.request('DescribeVSwitchAttributes', params, requestOption);
    });
}
function getVSwitchZoneId(vpcClient, region, vswitchId) {
    return __awaiter(this, void 0, void 0, function* () {
        const describeRs = yield describeVSwitchAttributes(vpcClient, region, vswitchId);
        return (describeRs || {}).ZoneId;
    });
}
function getVSwitchName(vpcClient, region, vswitchId) {
    return __awaiter(this, void 0, void 0, function* () {
        const describeRs = yield describeVSwitchAttributes(vpcClient, region, vswitchId);
        return (describeRs || {}).VSwitchName;
    });
}
function findVswitchExistByName(vpcClient, region, vswitchIds, searchVSwtichName) {
    return __awaiter(this, void 0, void 0, function* () {
        if (!_.isEmpty(vswitchIds)) {
            for (let vswitchId of vswitchIds) {
                const vswitchName = yield getVSwitchName(vpcClient, region, vswitchId);
                if (_.isEqual(searchVSwtichName, vswitchName)) {
                    debug('found default vswitchId: ' + vswitchId);
                    return vswitchId;
                }
            }
        }
        debug('could not find %s from %j for region %s', searchVSwtichName, vswitchIds, region);
        return null;
    });
}
function createVSwitch(vpcClient, { region, vpcId, zoneId, vswitchName }) {
    return __awaiter(this, void 0, void 0, function* () {
        var params = {
            'RegionId': region,
            'VpcId': vpcId,
            'ZoneId': zoneId,
            'CidrBlock': '10.20.0.0/16',
            'VSwitchName': vswitchName,
            'Description': 'default vswitch created by fc fun'
        };
        debug('createVSwitch params is %j', params);
        const createRs = yield vpcClient.request('CreateVSwitch', params, requestOption);
        return createRs.VSwitchId;
    });
}
function selectVSwitchZoneId(fcAllowedZones, vpcZones, nasZones) {
    return __awaiter(this, void 0, void 0, function* () {
        const allowedZones = _.filter(vpcZones, z => {
            return _.includes(fcAllowedZones, z.ZoneId) && _.includes(nasZones.map(zone => { return zone.ZoneId; }), z.ZoneId);
        });
        const sortedZones = _.sortBy(allowedZones, ['ZoneId']);
        return (_.head(sortedZones) || {}).ZoneId;
    });
}
function getFcAllowedZones() {
    return __awaiter(this, void 0, void 0, function* () {
        const fc = yield getFcClient();
        const fcRs = yield fc.getAccountSettings();
        const fcAllowedZones = fcRs.data.availableAZs;
        debug('fc allowed zones: %j', fcAllowedZones);
        if (_.isEqual(fcAllowedZones, [''])) {
            const profile = yield getProfile();
            throw new Error(red(`No fc vswitch zones allowed, you may need login to fc console to apply for VPC feature: https://fc.console.aliyun.com/overview/${profile.defaultRegion}`));
        }
        return fcAllowedZones;
    });
}
function selectAllowedVSwitchZone(vpcClient, region) {
    return __awaiter(this, void 0, void 0, function* () {
        const nasClient = yield getNasPopClient();
        const fcAllowedZones = yield getFcAllowedZones();
        const vpcZones = yield describeVpcZones(vpcClient, region);
        const nasZones = yield require('./nas').describeNasZones(nasClient, region);
        const usedZoneId = yield selectVSwitchZoneId(fcAllowedZones, vpcZones, nasZones);
        if (!usedZoneId) {
            throw new Error('no availiable zone for vswitch');
        }
        debug('select allowed switch zone: ', usedZoneId);
        return usedZoneId;
    });
}
function describeVpcZones(vpcClient, region) {
    return __awaiter(this, void 0, void 0, function* () {
        const params = {
            'RegionId': region
        };
        const zones = yield vpcClient.request('DescribeZones', params, requestOption);
        return zones.Zones.Zone;
    });
}
function convertToFcAllowedZoneMap(vpcClient, region, vswitchIds) {
    return __awaiter(this, void 0, void 0, function* () {
        const fcAllowedZones = yield getFcAllowedZones();
        const zoneMap = new Map();
        for (const vswitchId of vswitchIds) {
            const zoneId = yield getVSwitchZoneId(vpcClient, region, vswitchId);
            if (_.includes(fcAllowedZones, zoneId)) {
                zoneMap.set(zoneId, vswitchId);
            }
        }
        if (_.isEmpty(zoneMap)) {
            throw new Error(`
Only zoneId ${fcAllowedZones} of vswitch is allowed by VpcConfig.
Check your vswitch zoneId please.`);
        }
        return zoneMap;
    });
}
function convertZones(zones, zoneMap, storageType = 'Performance') {
    const zoneId = zones.ZoneId;
    return {
        zoneId,
        vswitchId: zoneMap.get(zoneId),
        storageType
    };
}
function processDifferentZones(nasZones, FcAllowVswitchId) {
    const performance = _.find(nasZones, nasZone => !_.isEmpty(nasZone.Performance.Protocol));
    if (!_.isEmpty(performance)) {
        return {
            zoneId: performance.ZoneId,
            vswitchId: FcAllowVswitchId,
            storageType: 'Performance'
        };
    }
    const capacity = _.find(nasZones, nasZone => !_.isEmpty(nasZone.Capacity.Protocol));
    if (!_.isEmpty(capacity)) {
        return {
            zoneId: capacity.ZoneId,
            vswitchId: FcAllowVswitchId,
            storageType: 'Capacity'
        };
    }
    return null;
}
function getAvailableVSwitchId(vpcClient, region, vswitchIds, nasZones) {
    return __awaiter(this, void 0, void 0, function* () {
        const zoneMap = yield convertToFcAllowedZoneMap(vpcClient, region, vswitchIds);
        for (const zoneId of zoneMap.keys()) {
            if (!_.includes(nasZones.map(m => { return m.ZoneId; }), zoneId)) {
                zoneMap.delete(zoneId);
            }
        }
        const performances = [];
        const capacities = [];
        _.forEach(nasZones, nasZone => {
            if (_.includes([...zoneMap.keys()], nasZone.ZoneId)) {
                if (!_.isEmpty(nasZone.Performance.Protocol)) {
                    performances.push(nasZone);
                }
                if (!_.isEmpty(nasZone.Capacity.Protocol)) {
                    capacities.push(nasZone);
                }
            }
        });
        if (!_.isEmpty(performances)) {
            return convertZones(_.head(performances), zoneMap);
        }
        if (!_.isEmpty(capacities)) {
            const msg = `Region ${region} only supports capacity NAS. Do you want to create it automatically?`;
            const yes = yield promptForConfirmContinue(msg);
            if (yes) {
                return convertZones(_.head(capacities), zoneMap, 'Capacity');
            }
            throw new Error(`No NAS service available under region ${region}.`);
        }
        return processDifferentZones(nasZones, _.head([...zoneMap.values()]));
    });
}
module.exports = {
    findVswitchExistByName,
    selectVSwitchZoneId,
    createVSwitch,
    createDefaultVSwitch,
    getAvailableVSwitchId
};