|
| 1 | +import * as path from 'path' |
| 2 | +import { ExtensionInstaller, Extension } from '@ali/ide-extension-installer'; |
| 3 | +import log from 'npmlog' |
| 4 | +import semver from 'semver' |
| 5 | +import * as fse from 'fs-extra'; |
| 6 | +import { parallelLimit } from './../util' |
| 7 | +import { EXTENSION_DIR, FRAMEWORK_PATH, FRAMEWORK_NAME, LOG_PREFIX, EXTENSION_METADATA_DIR } from './constant' |
| 8 | +import { ExtensionScanner } from './scanner' |
| 9 | +import { IWorkExtensionMetaData } from './type'; |
| 10 | + |
| 11 | +if (!FRAMEWORK_PATH) { |
| 12 | + log.error(LOG_PREFIX, `cli 无法单独使用，需要配合 spacex 一起使用，请直接安装 ${FRAMEWORK_NAME}`) |
| 13 | + process.exit(1) |
| 14 | +} |
| 15 | + |
| 16 | +const pkgJSON = fse.readJSONSync(path.join(__dirname, '../../package.json')) |
| 17 | + |
| 18 | +const extensionInstaller = new ExtensionInstaller({ |
| 19 | + accountId: 'nGJBcqs1D-ma32P3mBftgsfq', |
| 20 | + masterKey: '-nzxLbuqvrKh8arE0grj2f1H', |
| 21 | + frameworkVersion: pkgJSON.engines.kaitian, |
| 22 | + dist: EXTENSION_DIR, |
| 23 | + ignoreIncreaseCount: true, |
| 24 | +}) |
| 25 | + |
| 26 | +export const install = async (extensions: Extension[]) => { |
| 27 | + if (!extensions?.length) { |
| 28 | + // 使用 package.json 中的配置 |
| 29 | + extensions = await getExtensionFromPackage() |
| 30 | + } |
| 31 | + checkExtensions(extensions) |
| 32 | + await parallelLimit(extensions.map(ext => () => installExtension(ext)), 5) |
| 33 | + const metadata = await new ExtensionScanner([EXTENSION_DIR], {}).run() |
| 34 | + await writeMetadata(metadata) |
| 35 | + log.info(FRAMEWORK_NAME, 'extensions installed successfully') |
| 36 | +} |
| 37 | + |
| 38 | +async function getExtensionFromPackage() { |
| 39 | + try { |
| 40 | + const projectPkgJSON = await fse.readJSON(path.resolve('package.json')) |
| 41 | + return projectPkgJSON?.['kaitian-extensions'] ?? [] |
| 42 | + } catch (err) { |
| 43 | + return [] |
| 44 | + } |
| 45 | +} |
| 46 | + |
| 47 | +function checkExtensions(extensions: Extension[]) { |
| 48 | + extensions.forEach((ext) => { |
| 49 | + if (!ext.publisher) { |
| 50 | + log.error(FRAMEWORK_NAME, `${JSON.stringify(ext)} 缺少 publisher`) |
| 51 | + throw new Error('exit') |
| 52 | + } |
| 53 | + if (!ext.name) { |
| 54 | + log.error(FRAMEWORK_NAME, `${JSON.stringify(ext)} 缺少 name`) |
| 55 | + throw new Error('exit') |
| 56 | + } |
| 57 | + if (!ext.version) { |
| 58 | + log.error(FRAMEWORK_NAME, `${JSON.stringify(ext)} 缺少 version`) |
| 59 | + throw new Error('exit') |
| 60 | + } else if (!semver.valid(ext.version)) { |
| 61 | + log.error(FRAMEWORK_NAME, `${JSON.stringify(ext)} 中的 version 无效`) |
| 62 | + throw new Error('exit') |
| 63 | + } |
| 64 | + }) |
| 65 | +} |
| 66 | + |
| 67 | +async function installExtension(extension: Extension) { |
| 68 | + try { |
| 69 | + await extensionInstaller.install({ |
| 70 | + publisher: extension.publisher, |
| 71 | + name: extension.name, |
| 72 | + version: extension.version |
| 73 | + }) |
| 74 | + } catch(err) { |
| 75 | + log.error(FRAMEWORK_NAME, `${extension.publisher}.${extension.name}@${extension.version} 安装失败，请稍后重试`) |
| 76 | + } |
| 77 | +} |
| 78 | + |
| 79 | +async function writeMetadata(metadata: IWorkExtensionMetaData[]) { |
| 80 | + await fse.ensureDir(EXTENSION_METADATA_DIR) |
| 81 | + return Promise.all(metadata.map(async data => { |
| 82 | + try { |
| 83 | + const p = path.join(EXTENSION_METADATA_DIR, `${data.extensionId}.js`) |
| 84 | + await fse.writeFile(p, ` |
| 85 | +module.exports = ${JSON.stringify(data, null, 2)} |
| 86 | + `.trim() + '\n') |
| 87 | + } catch (err) { |
| 88 | + log.error(FRAMEWORK_NAME, `${data.extensionId} 安装失败`) |
| 89 | + } |
| 90 | + })) |
| 91 | +} |
0 commit comments