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

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.service.AsyncListExportTaskService;
import com.vincent.rsf.server.common.utils.FileServerUtil;
import com.vincent.rsf.server.manager.utils.buildPageRowsUtils;
import com.vincent.rsf.server.system.entity.ExportTask;
import com.vincent.rsf.server.system.service.ExportTaskService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;

import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import java.io.File;
import java.util.Map;

@RestController
public class ExportTaskController extends BaseController {

    @Autowired
    private ExportTaskService exportTaskService;

    @Autowired
    private AsyncListExportTaskService asyncListExportTaskService;

    @PreAuthorize("hasAuthority('system:exportTask:list')")
    @PostMapping("/exportTask/page")
    public R page(@RequestBody Map<String, Object> map) {
        BaseParam baseParam = buildParam(map, BaseParam.class);
        if (baseParam.getOrderBy() == null || baseParam.getOrderBy().trim().isEmpty()) {
            baseParam.setOrderBy("create_time desc");
        }

        PageParam<ExportTask, BaseParam> pageParam = new PageParam<>(baseParam, ExportTask.class);

        return R.ok().add(
                buildPageRowsUtils.rowsMap(
                        exportTaskService.page(
                                pageParam,
                                pageParam.buildWrapper(true)
                                        .eq("deleted", 0)
                                        .eq("tenant_id", getTenantId())
                                        .eq("create_by", getLoginUserId())
                        )
                )
        );
    }

    @PreAuthorize("hasAuthority('system:exportTask:list')")
    @GetMapping("/exportTask/{taskId}")
    public R get(@PathVariable("taskId") Long taskId) {
        ExportTask task = asyncListExportTaskService.getTask(taskId, getTenantId(), getLoginUserId());
        if (task == null) {
            return R.error("导出任务不存在");
        }
        return R.ok().add(buildPageRowsUtils.rowsMap(task));
    }

    @PreAuthorize("hasAuthority('system:exportTask:list')")
    @GetMapping("/exportTask/{taskId}/download")
    public void download(
            @PathVariable("taskId") Long taskId,
            HttpServletResponse response,
            HttpServletRequest request
    ) {
        File file = asyncListExportTaskService.getDownloadFile(taskId, getTenantId(), getLoginUserId());
        FileServerUtil.preview(file, true, file.getName(), null, null, response, request);
    }
}