package com.vincent.rsf.server.system.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.vincent.rsf.framework.common.R;
import com.vincent.rsf.server.common.domain.BaseParam;
import com.vincent.rsf.server.common.domain.PageParam;
import com.vincent.rsf.server.common.utils.ExcelUtil;
import com.vincent.rsf.server.system.entity.AiDiagnosisRecord;
import com.vincent.rsf.server.system.service.AiDiagnosisRecordService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.Map;

@RestController
public class AiDiagnosisController extends BaseController {

    @Autowired
    private AiDiagnosisRecordService aiDiagnosisRecordService;

    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @PostMapping("/aiDiagnosis/page")
    public R page(@RequestBody Map<String, Object> map) {
        BaseParam baseParam = buildParam(map, BaseParam.class);
        PageParam<AiDiagnosisRecord, BaseParam> pageParam = new PageParam<>(baseParam, AiDiagnosisRecord.class);
        com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<AiDiagnosisRecord> wrapper = pageParam.buildWrapper(true);
        wrapper.eq("tenant_id", getTenantId());
        return R.ok().add(aiDiagnosisRecordService.page(pageParam, wrapper));
    }

    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @GetMapping("/aiDiagnosis/{id}")
    public R get(@PathVariable("id") Long id) {
        AiDiagnosisRecord record = getTenantRecord(id);
        if (record == null) {
            return R.error("record not found");
        }
        return R.ok().add(record);
    }

    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @PostMapping("/aiDiagnosis/export")
    public void export(@RequestBody Map<String, Object> map, HttpServletResponse response) throws Exception {
        ExcelUtil.build(ExcelUtil.create(aiDiagnosisRecordService.list(new LambdaQueryWrapper<AiDiagnosisRecord>()
                .eq(AiDiagnosisRecord::getTenantId, getTenantId())), AiDiagnosisRecord.class), response);
    }

    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @GetMapping("/ai/diagnosis/list")
    public R diagnosisList() {
        return R.ok().add(aiDiagnosisRecordService.list(new LambdaQueryWrapper<AiDiagnosisRecord>()
                .eq(AiDiagnosisRecord::getTenantId, getTenantId())
                .orderByDesc(AiDiagnosisRecord::getCreateTime, AiDiagnosisRecord::getId)));
    }

    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @GetMapping("/ai/diagnosis/detail/{id}")
    public R detail(@PathVariable("id") Long id) {
        AiDiagnosisRecord record = getTenantRecord(id);
        if (record == null) {
            return R.error("record not found");
        }
        return R.ok().add(record);
    }

    @PreAuthorize("hasAuthority('system:aiDiagnosis:list')")
    @GetMapping("/ai/diagnosis/report/export/{id}")
    public void exportReport(@PathVariable("id") Long id, HttpServletResponse response) throws Exception {
        AiDiagnosisRecord record = getTenantRecord(id);
        if (record == null) {
            response.setStatus(404);
            return;
        }
        String fileName = URLEncoder.encode((record.getDiagnosisNo() == null ? "ai-diagnosis-report" : record.getDiagnosisNo()) + ".md", StandardCharsets.UTF_8.name());
        response.setCharacterEncoding(StandardCharsets.UTF_8.name());
        response.setContentType("text/markdown; charset=utf-8");
        response.setHeader("Content-Disposition", "attachment; filename=" + fileName);
        response.getWriter().write(record.getReportMarkdown() == null ? "" : record.getReportMarkdown());
        response.getWriter().flush();
    }

    private AiDiagnosisRecord getTenantRecord(Long id) {
        if (id == null) {
            return null;
        }
        return aiDiagnosisRecordService.getOne(new LambdaQueryWrapper<AiDiagnosisRecord>()
                .eq(AiDiagnosisRecord::getTenantId, getTenantId())
                .eq(AiDiagnosisRecord::getId, id)
                .last("limit 1"));
    }

}