This commit is contained in:
Baobhan Sith
2025-04-20 17:05:30 +08:00
parent 80a461d337
commit e2fbdb3bd9
10 changed files with 1055 additions and 7 deletions
@@ -0,0 +1,62 @@
import { Request, Response, NextFunction } from 'express';
import { DockerService, DockerCommand } from '../services/docker.service'; // 导入服务和命令类型
// 由于没有 typedi,我们将手动实例化服务或通过其他方式获取实例
// 简单起见,这里直接 new 一个实例。在实际项目中,可能需要更复杂的实例管理。
const dockerService = new DockerService();
export class DockerController {
/**
* 处理获取 Docker 容器状态的请求 (GET /docker/status)
*/
async getStatus(req: Request, res: Response, next: NextFunction): Promise<void> {
try {
const status = await dockerService.getContainerStatus();
res.json(status); // 直接返回 { available: boolean, containers: DockerContainer[] }
} catch (error) {
// 将错误传递给 Express 的错误处理中间件
next(error);
}
}
/**
* 处理执行 Docker 命令的请求 (POST /docker/command)
*/
async executeCommand(req: Request, res: Response, next: NextFunction): Promise<void> {
// 从请求体中获取参数
const { containerId, command } = req.body;
// 基本的输入验证
if (!containerId || typeof containerId !== 'string') {
res.status(400).json({ message: 'Missing or invalid containerId in request body.' });
return;
}
// 验证 command 是否是允许的类型
const allowedCommands: DockerCommand[] = ['start', 'stop', 'restart', 'remove'];
if (!command || !allowedCommands.includes(command)) {
res.status(400).json({ message: `Invalid command. Must be one of: ${allowedCommands.join(', ')}.` });
return;
}
try {
// 调用服务执行命令
await dockerService.executeContainerCommand(containerId, command as DockerCommand);
// 成功执行,返回 200 OK 或 204 No Content
res.status(200).json({ message: `Command '${command}' executed successfully for container ${containerId}.` });
// 或者 res.status(204).send();
} catch (error: any) {
// 根据错误类型返回不同的状态码
if (error.message.includes('Docker is not available')) {
res.status(503).json({ message: error.message }); // Service Unavailable
} else if (error.message.includes('Invalid container ID') || error.message.includes('Unsupported Docker command')) {
res.status(400).json({ message: error.message }); // Bad Request
} else {
// 其他执行错误,可能是 Docker 守护进程错误等
res.status(500).json({ message: error.message || 'Failed to execute Docker command.' }); // Internal Server Error
}
// 注意:这里没有调用 next(error),因为我们已经处理了响应。
// 如果希望使用统一的错误处理中间件,则应该调用 next(error)。
}
}
}
@@ -0,0 +1,18 @@
import { Router } from 'express';
import { DockerController } from './docker.controller';
// 修改导入路径
import { isAuthenticated } from '../auth/auth.middleware';
const router = Router();
const dockerController = new DockerController(); // 同样,手动实例化
// 应用认证中间件,确保只有登录用户才能访问 Docker 相关接口
router.use(isAuthenticated);
// GET /api/docker/status - 获取 Docker 容器状态
router.get('/status', (req, res, next) => dockerController.getStatus(req, res, next));
// POST /api/docker/command - 执行 Docker 命令
router.post('/command', (req, res, next) => dockerController.executeCommand(req, res, next));
export default router;