From 8c65dc3404973d1bfee9ca85318cc16cda7c5a4b Mon Sep 17 00:00:00 2001 From: moonrailgun Date: Sun, 22 Jan 2023 13:43:06 +0800 Subject: [PATCH] =?UTF-8?q?style:=20=E5=8D=95=E7=8B=AC=E6=8A=BD=E5=87=BAin?= =?UTF-8?q?it=E5=88=B0=E7=8B=AC=E7=AB=8B=E6=96=87=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/cli/src/commands/docker/doctor.ts | 32 ++++ apps/cli/src/commands/docker/index.ts | 194 +------------------------ apps/cli/src/commands/docker/init.ts | 62 ++++++++ apps/cli/src/commands/docker/update.ts | 104 +++++++++++++ 4 files changed, 204 insertions(+), 188 deletions(-) create mode 100644 apps/cli/src/commands/docker/doctor.ts create mode 100644 apps/cli/src/commands/docker/init.ts create mode 100644 apps/cli/src/commands/docker/update.ts diff --git a/apps/cli/src/commands/docker/doctor.ts b/apps/cli/src/commands/docker/doctor.ts new file mode 100644 index 00000000..9e5936ff --- /dev/null +++ b/apps/cli/src/commands/docker/doctor.ts @@ -0,0 +1,32 @@ +import { CommandModule } from 'yargs'; +import Docker from 'dockerode'; +import asTable from 'as-table'; +import filesize from 'filesize'; + +export const dockerDoctorCommand: CommandModule = { + command: 'doctor', + describe: 'docker 环境检查', + builder: undefined, + async handler(args) { + const docker = new Docker(); + + const images = await docker.listImages(); + const tailchatImages = images.filter((image) => + (image.RepoTags ?? []).some((tag) => tag.includes('tailchat')) + ); + + console.log('Tailchat 镜像列表:'); + console.log( + asTable.configure({ delimiter: ' | ' })( + tailchatImages.map((image) => ({ + tag: (image.RepoTags ?? []).join(','), + id: image.Id.substring(0, 12), + size: filesize(image.Size), + })) + ) + ); + + const info = await docker.info(); + console.log('info', info); + }, +}; diff --git a/apps/cli/src/commands/docker/index.ts b/apps/cli/src/commands/docker/index.ts index 55452621..65fddd56 100644 --- a/apps/cli/src/commands/docker/index.ts +++ b/apps/cli/src/commands/docker/index.ts @@ -1,200 +1,18 @@ import { CommandModule } from 'yargs'; -import Docker from 'dockerode'; -import asTable from 'as-table'; -import filesize from 'filesize'; -import ora from 'ora'; -import Spinnies from 'spinnies'; -import fs from 'fs-extra'; -import got from 'got'; -import path from 'path'; -import chalk from 'chalk'; -import inquirer from 'inquirer'; +import { dockerInitCommand } from './init'; +import { dockerUpdateCommand } from './update'; +import { dockerDoctorCommand } from './doctor'; // https://docs.docker.com/engine/api/v1.41/ -const initWelcome = chalk.green(`================ -恭喜你已经成功安装了, 你的配置文件已经准备就绪，距离成功就差一步了! - -你的tailchat配置文件都被存储在: ${chalk.underline( - path.join(process.cwd(), './tailchat') -)} - -运行以下命令以完成镜像下载与启动: -- ${chalk.bold('cd tailchat')} ${chalk.gray('# 移动到安装目录')} -- ${chalk.bold('tailchat docker update')} ${chalk.gray('# 下载/更新官方镜像')} -- ${chalk.bold('docker compose up')} ${chalk.gray('# 启动服务')} -================`); - -const envUrl = - 'https://ghproxy.com/https://raw.githubusercontent.com/msgbyte/tailchat/master/docker-compose.env'; -const configUrl = - 'https://ghproxy.com/https://raw.githubusercontent.com/msgbyte/tailchat/master/docker-compose.yml'; - -const remoteImageName = 'moonrailgun/tailchat:latest'; -const targetImage = { - repo: 'tailchat', - tag: 'latest', -}; -const targetImageName = targetImage.repo + targetImage.tag; - export const dockerCommand: CommandModule = { command: 'docker', describe: 'Tailchat 镜像管理', builder: (yargs) => yargs - .command( - 'init', - '初始化Tailchat with docker配置', - (yargs) => {}, - async (args) => { - const spinner = ora(); - try { - spinner.start('开始下载最新的配置文件'); - const [rawEnv, rawConfig] = await Promise.all([ - got(envUrl).then((res) => res.body), - got(configUrl).then((res) => res.body), - ]); - spinner.info('配置文件下载完毕'); - - // TODO: 需要实现交互式初始化，引导用户配置SECRET，API_URL, SMTP 服务 - - spinner.info('正在创建目录 tailchat ...'); - await fs.mkdir('./tailchat'); - - spinner.info('正在写入配置文件 ...'); - - await Promise.all([ - fs.writeFile('./tailchat/docker-compose.env', rawEnv), - fs.writeFile('./tailchat/docker-compose.yml', rawConfig), - ]); - - spinner.succeed('配置初始化完毕'); - console.log(initWelcome); - } catch (err) { - spinner.fail('Tailchat with docker 初始化出现意外'); - console.error(err); - } - } - ) - .command( - 'doctor', - 'docker环境检查', - (yargs) => {}, - async (args) => { - const docker = new Docker(); - - const images = await docker.listImages(); - const tailchatImages = images.filter((image) => - (image.RepoTags ?? []).some((tag) => tag.includes('tailchat')) - ); - - console.log('Tailchat 镜像列表'); - console.log( - asTable.configure({ delimiter: ' | ' })( - tailchatImages.map((image) => ({ - tag: (image.RepoTags ?? []).join(','), - id: image.Id.substring(0, 12), - size: filesize(image.Size), - })) - ) - ); - - const info = await docker.info(); - console.log('info', info); - } - ) - .command( - 'update', - '更新Tailchat镜像版本', - (yargs) => {}, - async (args) => { - const docker = new Docker(); - - const spinner = ora().start('开始更新镜像'); - - try { - const pullSpinnies = new Spinnies(); - - await new Promise((resolve, reject) => { - const taskMap = new Map(); - - // 这里有个类型问题，不会返回任何值 - docker.pull(remoteImageName, {}, (err, stream) => { - if (err) { - reject(err); - return; - } - - spinner.info('已找到远程镜像, 开始下载'); - - docker.modem.followProgress( - stream, - (err, output) => { - // onFinish - // console.log('finish', err, output); - - if (err) { - pullSpinnies.stopAll('stopped'); - reject(err); - } else { - pullSpinnies.stopAll('succeed'); - - spinner.succeed(output[1]?.status); - spinner.succeed(output[2]?.status); - resolve(); - } - }, - (event) => { - if (!event.id) { - console.log(event.status); // 可能是完成后的信息打印，直接输出 - return; - } - const text = `[${event.id}] ${event.status}${ - event.progress ? ':' : '' - } ${event.progress ?? ''}`; - - // onProcess - if (taskMap.has(event.id)) { - pullSpinnies.update(event.id, { - text, - }); - - if (event.status === 'Pull complete') { - pullSpinnies.succeed(event.id); - } - } else { - taskMap.set( - event.id, - pullSpinnies.add(event.id, { - text, - }) - ); - } - } - ); - }); - }); - - spinner.succeed('下载镜像完毕'); - - const image = docker.getImage(remoteImageName); - - if (!image) { - spinner.fail('出现异常，没有找到下载的镜像'); - return; - } - - spinner.info('正在更新镜像标签'); - - await image.tag(targetImage); - - spinner.succeed('镜像标签更新完毕'); - } catch (err) { - spinner.fail('更新出现错误, 请检查网络配置'); - console.error(err); - } - } - ) + .command(dockerInitCommand) + .command(dockerDoctorCommand) + .command(dockerUpdateCommand) .demandCommand(), handler(args) {}, }; diff --git a/apps/cli/src/commands/docker/init.ts b/apps/cli/src/commands/docker/init.ts new file mode 100644 index 00000000..a7d6e34b --- /dev/null +++ b/apps/cli/src/commands/docker/init.ts @@ -0,0 +1,62 @@ +import { CommandModule } from 'yargs'; +import ora from 'ora'; +import fs from 'fs-extra'; +import got from 'got'; +import path from 'path'; +import chalk from 'chalk'; +import inquirer from 'inquirer'; + +// https://docs.docker.com/engine/api/v1.41/ + +const initWelcome = chalk.green(`================ +恭喜你已经成功安装了, 你的配置文件已经准备就绪，距离成功就差一步了! + +你的tailchat配置文件都被存储在: ${chalk.underline( + path.join(process.cwd(), './tailchat') +)} + +运行以下命令以完成镜像下载与启动: +- ${chalk.bold('cd tailchat')} ${chalk.gray('# 移动到安装目录')} +- ${chalk.bold('tailchat docker update')} ${chalk.gray('# 下载/更新官方镜像')} +- ${chalk.bold('docker compose up')} ${chalk.gray('# 启动服务')} +================`); + +const envUrl = + 'https://ghproxy.com/https://raw.githubusercontent.com/msgbyte/tailchat/master/docker-compose.env'; +const configUrl = + 'https://ghproxy.com/https://raw.githubusercontent.com/msgbyte/tailchat/master/docker-compose.yml'; + +export const dockerInitCommand: CommandModule = { + command: 'init', + describe: '初始化Tailchat with docker配置', + builder: undefined, + async handler(args) { + const spinner = ora(); + try { + spinner.start('开始下载最新的配置文件'); + const [rawEnv, rawConfig] = await Promise.all([ + got(envUrl).then((res) => res.body), + got(configUrl).then((res) => res.body), + ]); + spinner.info('配置文件下载完毕'); + + // TODO: 需要实现交互式初始化，引导用户配置SECRET，API_URL, SMTP 服务 + + spinner.info('正在创建目录 tailchat ...'); + await fs.mkdir('./tailchat'); + + spinner.info('正在写入配置文件 ...'); + + await Promise.all([ + fs.writeFile('./tailchat/docker-compose.env', rawEnv), + fs.writeFile('./tailchat/docker-compose.yml', rawConfig), + ]); + + spinner.succeed('配置初始化完毕'); + console.log(initWelcome); + } catch (err) { + spinner.fail('Tailchat with docker 初始化出现意外'); + console.error(err); + } + }, +}; diff --git a/apps/cli/src/commands/docker/update.ts b/apps/cli/src/commands/docker/update.ts new file mode 100644 index 00000000..3f308c99 --- /dev/null +++ b/apps/cli/src/commands/docker/update.ts @@ -0,0 +1,104 @@ +import { CommandModule } from 'yargs'; +import ora from 'ora'; +import Docker from 'dockerode'; +import Spinnies from 'spinnies'; + +const remoteImageName = 'moonrailgun/tailchat:latest'; +const targetImage = { + repo: 'tailchat', + tag: 'latest', +}; +const targetImageName = targetImage.repo + targetImage.tag; + +export const dockerUpdateCommand: CommandModule = { + command: 'update', + describe: '更新Tailchat镜像版本', + builder: undefined, + async handler(args) { + const docker = new Docker(); + + const spinner = ora().start('开始更新镜像'); + + try { + const pullSpinnies = new Spinnies(); + + await new Promise((resolve, reject) => { + const taskMap = new Map(); + + // 这里有个类型问题，不会返回任何值 + docker.pull(remoteImageName, {}, (err, stream) => { + if (err) { + reject(err); + return; + } + + spinner.info('已找到远程镜像, 开始下载'); + + docker.modem.followProgress( + stream, + (err, output) => { + // onFinish + // console.log('finish', err, output); + + if (err) { + pullSpinnies.stopAll('stopped'); + reject(err); + } else { + pullSpinnies.stopAll('succeed'); + + spinner.succeed(output[1]?.status); + spinner.succeed(output[2]?.status); + resolve(); + } + }, + (event) => { + if (!event.id) { + console.log(event.status); // 可能是完成后的信息打印，直接输出 + return; + } + const text = `[${event.id}] ${event.status}${ + event.progress ? ':' : '' + } ${event.progress ?? ''}`; + + // onProcess + if (taskMap.has(event.id)) { + pullSpinnies.update(event.id, { + text, + }); + + if (event.status === 'Pull complete') { + pullSpinnies.succeed(event.id); + } + } else { + taskMap.set( + event.id, + pullSpinnies.add(event.id, { + text, + }) + ); + } + } + ); + }); + }); + + spinner.succeed('下载镜像完毕'); + + const image = docker.getImage(remoteImageName); + + if (!image) { + spinner.fail('出现异常，没有找到下载的镜像'); + return; + } + + spinner.info('正在更新镜像标签'); + + await image.tag(targetImage); + + spinner.succeed('镜像标签更新完毕'); + } catch (err) { + spinner.fail('更新出现错误, 请检查网络配置'); + console.error(err); + } + }, +};