import { Request, Response } from 'express';
import { ExamTaskModel } from '../models/examTask';

export class ExamTaskController {
  static async getTasks(req: Request, res: Response) {
    try {
      const tasks = await ExamTaskModel.findAll();
      res.json({
        success: true,
        data: tasks
      });
    } catch (error: any) {
      res.status(500).json({
        success: false,
        message: error.message || '获取考试任务失败'
      });
    }
  }

  static async createTask(req: Request, res: Response) {
    try {
      const { name, subjectId, startAt, endAt, userIds } = req.body;

      if (!name || !subjectId || !startAt || !endAt || !Array.isArray(userIds) || userIds.length === 0) {
        return res.status(400).json({
          success: false,
          message: '参数不完整或用户列表为空'
        });
      }

      const task = await ExamTaskModel.create({
        name,
        subjectId,
        startAt,
        endAt,
        userIds
      });

      res.json({
        success: true,
        data: task
      });
    } catch (error: any) {
      res.status(400).json({
        success: false,
        message: error.message || '新增考试任务失败'
      });
    }
  }

  static async updateTask(req: Request, res: Response) {
    try {
      const { id } = req.params;
      const { name, subjectId, startAt, endAt, userIds } = req.body;

      if (!name || !subjectId || !startAt || !endAt || !Array.isArray(userIds) || userIds.length === 0) {
        return res.status(400).json({
          success: false,
          message: '参数不完整或用户列表为空'
        });
      }

      const task = await ExamTaskModel.update(id, {
        name,
        subjectId,
        startAt,
        endAt,
        userIds
      });

      res.json({
        success: true,
        data: task
      });
    } catch (error: any) {
      const message = error.message || '更新考试任务失败';
      res.status(message === '任务不存在' ? 404 : 400).json({
        success: false,
        message
      });
    }
  }

  static async deleteTask(req: Request, res: Response) {
    try {
      const { id } = req.params;
      await ExamTaskModel.delete(id);

      res.json({
        success: true,
        message: '删除成功'
      });
    } catch (error: any) {
      const message = error.message || '删除考试任务失败';
      res.status(message === '任务不存在' ? 404 : 400).json({
        success: false,
        message
      });
    }
  }

  static async getTaskReport(req: Request, res: Response) {
    try {
      const { id } = req.params;
      const report = await ExamTaskModel.getReport(id);

      res.json({
        success: true,
        data: report
      });
    } catch (error: any) {
      const message = error.message || '获取任务报表失败';
      res.status(message === '任务不存在' ? 404 : 400).json({
        success: false,
        message
      });
    }
  }

  static async getUserTasks(req: Request, res: Response) {
    try {
      const { userId } = req.params;
      
      if (!userId) {
        return res.status(400).json({
          success: false,
          message: '用户ID不能为空'
        });
      }

      const tasks = await ExamTaskModel.getUserTasks(userId);
      
      res.json({
        success: true,
        data: tasks
      });
    } catch (error: any) {
      res.status(500).json({
        success: false,
        message: error.message || '获取用户任务失败'
      });
    }
  }
}