package cn.iocoder.yudao.module.finance.controller.admin.asset;

import cn.hutool.core.collection.CollUtil;
import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.util.collection.MapUtils;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
import cn.iocoder.yudao.module.finance.controller.admin.asset.vo.FinFixedAssetDepreciateRespVO;
import cn.iocoder.yudao.module.finance.controller.admin.asset.vo.FinFixedAssetPageReqVO;
import cn.iocoder.yudao.module.finance.controller.admin.asset.vo.FinFixedAssetRespVO;
import cn.iocoder.yudao.module.finance.controller.admin.asset.vo.FinFixedAssetSaveReqVO;
import cn.iocoder.yudao.module.finance.dal.dataobject.asset.FinFixedAssetDO;
import cn.iocoder.yudao.module.finance.service.asset.FinAssetAmountUtils;
import cn.iocoder.yudao.module.finance.service.asset.FinFixedAssetService;
import cn.iocoder.yudao.module.system.api.dept.DeptApi;
import cn.iocoder.yudao.module.system.api.dept.dto.DeptRespDTO;
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.annotation.Resource;
import jakarta.servlet.http.HttpServletResponse;
import jakarta.validation.Valid;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;

import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Set;

import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.EXPORT;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertList;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;

@Tag(name = "管理后台 - 固定资产")
@RestController
@RequestMapping("/finance/fixed-asset")
@Validated
public class FinFixedAssetController {

    @Resource
    private FinFixedAssetService fixedAssetService;
    @Resource
    private DeptApi deptApi;
    @Resource
    private AdminUserApi adminUserApi;

    @PostMapping("/create")
    @Operation(summary = "创建固定资产")
    @PreAuthorize("@ss.hasPermission('finance:fixed-asset:create')")
    public CommonResult<Long> createFixedAsset(@Valid @RequestBody FinFixedAssetSaveReqVO createReqVO) {
        return success(fixedAssetService.createFixedAsset(createReqVO));
    }

    @PutMapping("/update")
    @Operation(summary = "更新固定资产")
    @PreAuthorize("@ss.hasPermission('finance:fixed-asset:update')")
    public CommonResult<Boolean> updateFixedAsset(@Valid @RequestBody FinFixedAssetSaveReqVO updateReqVO) {
        fixedAssetService.updateFixedAsset(updateReqVO);
        return success(true);
    }

    @DeleteMapping("/delete")
    @Operation(summary = "删除固定资产")
    @Parameter(name = "id", description = "编号", required = true)
    @PreAuthorize("@ss.hasPermission('finance:fixed-asset:delete')")
    public CommonResult<Boolean> deleteFixedAsset(@RequestParam("id") Long id) {
        fixedAssetService.deleteFixedAsset(id);
        return success(true);
    }

    @DeleteMapping("/delete-list")
    @Operation(summary = "批量删除固定资产")
    @Parameter(name = "ids", description = "编号列表", required = true)
    @PreAuthorize("@ss.hasPermission('finance:fixed-asset:delete')")
    public CommonResult<Boolean> deleteFixedAssetList(@RequestParam("ids") List<Long> ids) {
        fixedAssetService.deleteFixedAssetList(ids);
        return success(true);
    }

    @PostMapping("/depreciate")
    @Operation(summary = "计提折旧", description = "按资产逐个计提一个月的折旧；ids 为空时对所有未报废资产计提")
    @Parameter(name = "ids", description = "资产编号列表，为空表示全部", required = false)
    @PreAuthorize("@ss.hasPermission('finance:fixed-asset:depreciate')")
    public CommonResult<FinFixedAssetDepreciateRespVO> depreciateFixedAssetList(
            @RequestParam(value = "ids", required = false) List<Long> ids) {
        return success(fixedAssetService.depreciateFixedAssetList(ids));
    }

    @GetMapping("/get")
    @Operation(summary = "获得固定资产")
    @Parameter(name = "id", description = "编号", required = true, example = "1024")
    @PreAuthorize("@ss.hasPermission('finance:fixed-asset:query')")
    public CommonResult<FinFixedAssetRespVO> getFixedAsset(@RequestParam("id") Long id) {
        FinFixedAssetDO asset = fixedAssetService.getFixedAsset(id);
        if (asset == null) {
            return success(null);
        }
        FinFixedAssetRespVO respVO = toRespVO(asset);
        fillDepartmentAndKeeper(List.of(respVO));
        return success(respVO);
    }

    @GetMapping("/page")
    @Operation(summary = "获得固定资产分页")
    @PreAuthorize("@ss.hasPermission('finance:fixed-asset:query')")
    public CommonResult<PageResult<FinFixedAssetRespVO>> getFixedAssetPage(@Valid FinFixedAssetPageReqVO pageReqVO) {
        PageResult<FinFixedAssetDO> pageResult = fixedAssetService.getFixedAssetPage(pageReqVO);
        PageResult<FinFixedAssetRespVO> result = BeanUtils.toBean(pageResult, FinFixedAssetRespVO.class,
                this::fillMonthlyDepreciation);
        fillDepartmentAndKeeper(result.getList());
        return success(result);
    }

    @GetMapping("/list")
    @Operation(summary = "获得固定资产列表")
    @PreAuthorize("@ss.hasPermission('finance:fixed-asset:query')")
    public CommonResult<List<FinFixedAssetRespVO>> getFixedAssetList() {
        List<FinFixedAssetDO> list = fixedAssetService.getFixedAssetList();
        List<FinFixedAssetRespVO> voList = convertList(list, this::toRespVO);
        fillDepartmentAndKeeper(voList);
        return success(voList);
    }

    @GetMapping("/export-excel")
    @Operation(summary = "导出固定资产 Excel")
    @PreAuthorize("@ss.hasPermission('finance:fixed-asset:export')")
    @ApiAccessLog(operateType = EXPORT)
    public void exportFixedAssetExcel(@Valid FinFixedAssetPageReqVO pageReqVO,
                                      HttpServletResponse response) throws IOException {
        pageReqVO.setPageSize(PageParam.PAGE_SIZE_NONE);
        List<FinFixedAssetDO> list = fixedAssetService.getFixedAssetPage(pageReqVO).getList();
        List<FinFixedAssetRespVO> voList = BeanUtils.toBean(list, FinFixedAssetRespVO.class,
                this::fillMonthlyDepreciation);
        fillDepartmentAndKeeper(voList);
        // 导出 Excel
        ExcelUtils.write(response, "固定资产.xls", "数据", FinFixedAssetRespVO.class, voList);
    }

    private FinFixedAssetRespVO toRespVO(FinFixedAssetDO asset) {
        return fillMonthlyDepreciation(BeanUtils.toBean(asset, FinFixedAssetRespVO.class));
    }

    /**
     * 回填月折旧额（不落库，仅用于展示，口径与计提折旧一致）
     */
    private FinFixedAssetRespVO fillMonthlyDepreciation(FinFixedAssetRespVO respVO) {
        if (respVO.getUsefulLife() == null || respVO.getUsefulLife() <= 0
                || respVO.getOriginalValue() == null) {
            return respVO;
        }
        return respVO.setMonthlyDepreciation(FinAssetAmountUtils.calcMonthlyAmount(
                respVO.getOriginalValue(), respVO.getResidualRate(), respVO.getUsefulLife()));
    }

    /**
     * 批量回填「使用部门名称」与「保管人昵称」。
     *
     * DO 上存的是 departmentId / keeperId，而列表与导出要展示名称，
     * 所以按 ERP 收款单的既有口径，在这里把部门、用户各批量查一次再回填，避免逐行查库。
     * 解析不到的（部门/用户被删除、编号为空）就保持为空，不抛异常。
     */
    private void fillDepartmentAndKeeper(List<FinFixedAssetRespVO> list) {
        if (CollUtil.isEmpty(list)) {
            return;
        }
        Set<Long> deptIds = convertSet(list, FinFixedAssetRespVO::getDepartmentId,
                respVO -> respVO.getDepartmentId() != null);
        Set<Long> keeperIds = convertSet(list, FinFixedAssetRespVO::getKeeperId,
                respVO -> respVO.getKeeperId() != null);
        Map<Long, DeptRespDTO> deptMap = deptApi.getDeptMap(deptIds);
        Map<Long, AdminUserRespDTO> userMap = adminUserApi.getUserMap(keeperIds);
        list.forEach(respVO -> {
            MapUtils.findAndThen(deptMap, respVO.getDepartmentId(),
                    dept -> respVO.setDepartmentName(dept.getName()));
            MapUtils.findAndThen(userMap, respVO.getKeeperId(),
                    user -> respVO.setKeeperName(user.getNickname()));
        });
    }

}