Innei
@@ -1,17 +0,0 @@
|
||||
import { Get } from '@nestjs/common'
|
||||
|
||||
import { ApiController } from '~/common/decorators/api-controller.decorator'
|
||||
import { Auth } from '~/common/decorators/auth.decorator'
|
||||
|
||||
import { PTYService } from './pty.service'
|
||||
|
||||
@Auth()
|
||||
@ApiController({ path: 'pty' })
|
||||
export class PTYController {
|
||||
constructor(private readonly service: PTYService) {}
|
||||
|
||||
@Get('/record')
|
||||
async getPtyLoginRecord() {
|
||||
return this.service.getLoginRecord()
|
||||
}
|
||||
}
|
||||
@@ -1,189 +0,0 @@
|
||||
// THIS MODULE IS DISABLED
|
||||
// @ts-nocheck
|
||||
import { isNil, pick } from 'lodash'
|
||||
import { spawn } from 'node-pty'
|
||||
import { Socket } from 'socket.io'
|
||||
|
||||
import { nanoid as N } from '@mx-space/external'
|
||||
import { SubscribeMessage, WebSocketGateway } from '@nestjs/websockets'
|
||||
|
||||
import { DEMO_MODE } from '~/app.config'
|
||||
import { BusinessEvents } from '~/constants/business-event.constant'
|
||||
import { RedisKeys } from '~/constants/cache.constant'
|
||||
import { DATA_DIR } from '~/constants/path.constant'
|
||||
import { AuthService } from '~/modules/auth/auth.service'
|
||||
import { ConfigsService } from '~/modules/configs/configs.service'
|
||||
import { createAuthGateway } from '~/processors/gateway/shared/auth.gateway'
|
||||
import { JWTService } from '~/processors/helper/helper.jwt.service'
|
||||
import { CacheService } from '~/processors/redis/cache.service'
|
||||
import { getIp, getRedisKey } from '~/utils'
|
||||
import type { IPty } from 'node-pty'
|
||||
import type {
|
||||
GatewayMetadata,
|
||||
OnGatewayConnection,
|
||||
OnGatewayDisconnect,
|
||||
} from '@nestjs/websockets'
|
||||
|
||||
const { nanoid } = N
|
||||
|
||||
const AuthGateway = createAuthGateway({ namespace: 'pty', authway: 'jwt' })
|
||||
@WebSocketGateway<GatewayMetadata>({ namespace: 'pty' })
|
||||
export class PTYGateway
|
||||
extends AuthGateway
|
||||
implements OnGatewayConnection, OnGatewayDisconnect
|
||||
{
|
||||
constructor(
|
||||
protected readonly jwtService: JWTService,
|
||||
protected readonly authService: AuthService,
|
||||
protected readonly cacheService: CacheService,
|
||||
protected readonly configService: ConfigsService,
|
||||
) {
|
||||
super(jwtService, authService)
|
||||
}
|
||||
socket2ptyMap = new WeakMap<Socket, IPty>()
|
||||
|
||||
@SubscribeMessage('pty')
|
||||
async pty(
|
||||
client: Socket,
|
||||
data?: { password?: string; cols: number; rows: number },
|
||||
) {
|
||||
if (DEMO_MODE) {
|
||||
client.send(
|
||||
this.gatewayMessageFormat(
|
||||
BusinessEvents.PTY_MESSAGE,
|
||||
'PTY 在演示模式下不可用',
|
||||
),
|
||||
)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
const password = data?.password
|
||||
const terminalOptions = await this.configService.get('terminalOptions')
|
||||
if (!terminalOptions.enable) {
|
||||
client.send(
|
||||
this.gatewayMessageFormat(BusinessEvents.PTY_MESSAGE, 'PTY 已禁用'),
|
||||
)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
const isValidPassword = isNil(terminalOptions.password)
|
||||
? true
|
||||
: password === terminalOptions.password
|
||||
|
||||
if (!isValidPassword) {
|
||||
if (typeof password === 'undefined' || password === '') {
|
||||
client.send(
|
||||
this.gatewayMessageFormat(
|
||||
BusinessEvents.PTY_MESSAGE,
|
||||
'PTY 验证未通过：需要密码验证',
|
||||
10000,
|
||||
),
|
||||
)
|
||||
} else {
|
||||
client.send(
|
||||
this.gatewayMessageFormat(
|
||||
BusinessEvents.PTY_MESSAGE,
|
||||
'PTY 验证未通过：密码错误',
|
||||
10001,
|
||||
),
|
||||
)
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
const zsh = await $`zsh --version`.quiet().nothrow()
|
||||
const fish = await $`fish --version`.quiet().nothrow()
|
||||
|
||||
const pty = spawn(
|
||||
os.platform() === 'win32'
|
||||
? 'powershell.exe'
|
||||
: zsh.exitCode == 0
|
||||
? 'zsh'
|
||||
: fish.exitCode == 0
|
||||
? 'fish'
|
||||
: 'bash',
|
||||
[],
|
||||
{
|
||||
cwd: DATA_DIR,
|
||||
cols: data?.cols || 30,
|
||||
rows: data?.rows || 80,
|
||||
env: pick(process.env, [
|
||||
'PATH',
|
||||
'EDITOR',
|
||||
'SHELL',
|
||||
'USER',
|
||||
'VISUAL',
|
||||
'LANG',
|
||||
'TERM',
|
||||
'LANGUAGE',
|
||||
|
||||
// other
|
||||
'N_PREFIX',
|
||||
'N_PRESERVE_NPM',
|
||||
]) as any,
|
||||
},
|
||||
)
|
||||
|
||||
const nid = nanoid()
|
||||
const ip =
|
||||
client.handshake.headers['x-forwarded-for'] ||
|
||||
client.handshake.address ||
|
||||
getIp(client.request) ||
|
||||
client.conn.remoteAddress
|
||||
|
||||
this.cacheService.getClient().hset(
|
||||
getRedisKey(RedisKeys.PTYSession),
|
||||
nid,
|
||||
|
||||
`${new Date().toISOString()},${ip}`,
|
||||
)
|
||||
pty.onExit(async () => {
|
||||
const hvalue = await this.cacheService
|
||||
.getClient()
|
||||
.hget(getRedisKey(RedisKeys.PTYSession), nid)
|
||||
if (hvalue) {
|
||||
this.cacheService
|
||||
.getClient()
|
||||
.hset(
|
||||
getRedisKey(RedisKeys.PTYSession),
|
||||
nid,
|
||||
`${hvalue},${new Date().toISOString()}`,
|
||||
)
|
||||
}
|
||||
})
|
||||
|
||||
if (terminalOptions.script) {
|
||||
pty.write(terminalOptions.script)
|
||||
pty.write('\n')
|
||||
}
|
||||
pty.onData((data) => {
|
||||
client.send(this.gatewayMessageFormat(BusinessEvents.PTY, data))
|
||||
})
|
||||
|
||||
this.socket2ptyMap.set(client, pty)
|
||||
}
|
||||
|
||||
@SubscribeMessage('pty-input')
|
||||
async ptyInput(client: Socket, data: string) {
|
||||
const pty = this.socket2ptyMap.get(client)
|
||||
if (pty) {
|
||||
pty.write(data)
|
||||
}
|
||||
}
|
||||
|
||||
@SubscribeMessage('pty-exit')
|
||||
async ptyExit(client: Socket) {
|
||||
const pty = this.socket2ptyMap.get(client)
|
||||
if (pty) {
|
||||
pty.kill()
|
||||
}
|
||||
this.socket2ptyMap.delete(client)
|
||||
}
|
||||
|
||||
handleDisconnect(client: Socket) {
|
||||
this.ptyExit(client)
|
||||
super.handleDisconnect(client)
|
||||
}
|
||||
}
|
||||
@@ -1,13 +0,0 @@
|
||||
import { Module } from '@nestjs/common'
|
||||
|
||||
import { AuthModule } from '../auth/auth.module'
|
||||
import { PTYController } from './pty.controller'
|
||||
import { PTYGateway } from './pty.gateway'
|
||||
import { PTYService } from './pty.service'
|
||||
|
||||
@Module({
|
||||
imports: [AuthModule],
|
||||
controllers: [PTYController],
|
||||
providers: [PTYService, PTYGateway],
|
||||
})
|
||||
export class PTYModule {}
|
||||
@@ -1,38 +0,0 @@
|
||||
import { Injectable } from '@nestjs/common'
|
||||
|
||||
import { RedisKeys } from '~/constants/cache.constant'
|
||||
import { CacheService } from '~/processors/redis/cache.service'
|
||||
import { getRedisKey } from '~/utils'
|
||||
|
||||
@Injectable()
|
||||
export class PTYService {
|
||||
constructor(private readonly cacheService: CacheService) {}
|
||||
|
||||
async getLoginRecord() {
|
||||
const redis = this.cacheService.getClient()
|
||||
const keys = await redis.hkeys(getRedisKey(RedisKeys.PTYSession))
|
||||
|
||||
const values = await Promise.all(
|
||||
keys.map(async (key) => {
|
||||
return redis.hget(getRedisKey(RedisKeys.PTYSession), key)
|
||||
}),
|
||||
)
|
||||
|
||||
return values
|
||||
.filter(Boolean)
|
||||
.map((value: string) => {
|
||||
const [startTime, ip, endTime] = value.split(',') as [
|
||||
string,
|
||||
string,
|
||||
string | undefined,
|
||||
]
|
||||
|
||||
return {
|
||||
startTime: new Date(startTime),
|
||||
ip,
|
||||
endTime: endTime === '' ? null : endTime,
|
||||
}
|
||||
})
|
||||
.sort((a, b) => b.startTime.getTime() - a.startTime.getTime())
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user