perf: 支持windows文件上传

This commit is contained in:
xiaojunnuo
2024-06-27 16:38:43 +08:00
parent 37caef38ad
commit 7f61cab101
7 changed files with 113 additions and 30 deletions

View File

@@ -60,6 +60,32 @@ export class SshAccess implements IAccess, ConnectConfig {
},
})
passphrase!: string;
@AccessInput({
title: '是否Windows',
helper: '如果是Windows主机请勾选此项',
component: {
name: 'a-switch',
vModel: 'checked',
},
})
windows: boolean = false;
@AccessInput({
title: '命令编码',
helper: '如果是Windows主机且出现乱码了请尝试设置为GBK',
component: {
name: 'a-select',
vModel: 'value',
options:[
{value:"","label":"默认"},
{value:"GBK","label":"GBK"},
{value:"UTF8","label":"UTF-8"},
]
},
})
encoding: string;
}
new SshAccess();

View File

@@ -3,14 +3,26 @@ import ssh2, { ConnectConfig } from 'ssh2';
import path from 'path';
import _ from 'lodash';
import { ILogger } from '@certd/pipeline';
import iconv from 'iconv-lite';
import {SshAccess} from "../access";
export class AsyncSsh2Client {
conn: ssh2.Client;
logger: ILogger;
connConf: ssh2.ConnectConfig;
constructor(connConf: ssh2.ConnectConfig, logger: ILogger) {
windows:boolean = false;
encoding:string;
constructor(connConf: SshAccess, logger: ILogger) {
this.connConf = connConf;
this.logger = logger;
this.windows = connConf.windows || false;
this.encoding = connConf.encoding;
}
convert(buffer: Buffer) {
if(this.encoding){
return iconv.decode(buffer, this.encoding);
}
return buffer.toString();
}
async connect() {
@@ -59,30 +71,29 @@ export class AsyncSsh2Client {
async exec(script: string) {
return new Promise((resolve, reject) => {
this.logger.info(`执行脚本[${this.connConf.host}][exec]: ` + script);
this.logger.info(`执行命令[${this.connConf.host}][exec]: ` + script);
this.conn.exec(script, (err: Error, stream: any) => {
if (err) {
reject(err);
return;
}
let data: any = null;
let data: string = null;
stream
.on('close', (code: any, signal: any) => {
this.logger.info(`[${this.connConf.host}][close]:code:${code}`);
data = data ? data.toString() : null;
if (code === 0) {
resolve(data);
} else {
reject(new Error(data));
}
})
.on('data', (ret: any) => {
this.logger.info(`[${this.connConf.host}][info]: ` + ret);
data = ret;
.on('data', (ret: Buffer) => {
data = this.convert(ret)
this.logger.info(`[${this.connConf.host}][info]: ` + data);
})
.stderr.on('data', (err: Error) => {
this.logger.info(`[${this.connConf.host}][error]: ` + err);
data = err;
.stderr.on('data', (ret:Buffer) => {
data = this.convert(ret)
this.logger.info(`[${this.connConf.host}][error]: ` + data);
});
});
});
@@ -104,10 +115,16 @@ export class AsyncSsh2Client {
this.logger.info('Stream :: close');
resolve(output);
})
.on('data', (data: any) => {
.on('data', (ret: Buffer) => {
const data = this.convert(ret)
this.logger.info('' + data);
output.push('' + data);
});
output.push(data);
})
.stderr.on('data', (ret:Buffer) => {
const data = this.convert(ret)
output.push(data);
this.logger.info(`[${this.connConf.host}][error]: ` + data);
});
stream.end(script + '\nexit\n');
});
});
@@ -134,7 +151,7 @@ export class SshClient {
}
* @param options
*/
async uploadFiles(options: { connectConf: ConnectConfig; transports: any }) {
async uploadFiles(options: { connectConf: SshAccess; transports: any }) {
const { connectConf, transports } = options;
await this._call({
connectConf,
@@ -142,7 +159,17 @@ export class SshClient {
const sftp = await conn.getSftp();
this.logger.info('开始上传');
for (const transport of transports) {
await conn.exec(`mkdir -p ${path.dirname(transport.remotePath)} `);
let filePath = path.dirname(transport.remotePath);
let mkdirCmd = `mkdir -p ${filePath} `;
if(conn.windows){
if(filePath.indexOf("/") > -1){
this.logger.info("--------------------------")
this.logger.info("请注意windows下文件目录分隔应该写成\\而不是/")
this.logger.info("--------------------------")
}
mkdirCmd = `if not exist "${filePath}" mkdir ${filePath} `
}
await conn.exec(mkdirCmd);
await conn.fastPut({ sftp, ...transport });
}
this.logger.info('文件全部上传成功');
@@ -151,7 +178,7 @@ export class SshClient {
}
async exec(options: {
connectConf: ConnectConfig;
connectConf: SshAccess;
script: string | Array<string>;
}) {
let { script } = options;
@@ -170,7 +197,7 @@ export class SshClient {
}
async shell(options: {
connectConf: ConnectConfig;
connectConf: SshAccess;
script: string;
}): Promise<string[]> {
const { connectConf, script } = options;
@@ -183,7 +210,7 @@ export class SshClient {
}
async _call(options: {
connectConf: ConnectConfig;
connectConf: SshAccess;
callable: any;
}): Promise<string[]> {
const { connectConf, callable } = options;

View File

@@ -10,6 +10,7 @@ import {
import { SshClient } from '../../lib/ssh';
import { CertInfo, CertReader } from '@certd/plugin-cert';
import * as fs from 'fs';
import {SshAccess} from "../../access";
@IsTaskPlugin({
name: 'uploadCertToHost',
@@ -112,7 +113,7 @@ export class UploadCertToHostPlugin extends AbstractTaskPlugin {
throw new Error('主机登录授权配置不能为空');
}
this.logger.info('准备上传到服务器');
const connectConf = await this.accessService.getById(accessId);
const connectConf:SshAccess = await this.accessService.getById(accessId);
const sshClient = new SshClient(this.logger);
await sshClient.uploadFiles({
connectConf,