package com.ruoyi.tms.controller;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.tms.domain.vo.AddressVo;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import com.ruoyi.common.utils.file.DownloadExportUtil;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.tms.domain.TmsRegion;
import com.ruoyi.tms.service.ITmsRegionService;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.common.core.page.TableDataInfo;

/**
 * 行政区域Controller
 *
 * @author ruoyi
 * @date 2025-11-03
 */
@RestController
@RequestMapping("/tms/tmsRegion")
public class TmsRegionController extends BaseController
{
    @Autowired
    private ITmsRegionService tmsRegionService;



    /**
     * 查询行政区域列表
     */
    @PreAuthorize("@ss.hasPermi('tms:tmsRegion:list')")
    @GetMapping("/list")
    public TableDataInfo list(TmsRegion tmsRegion)
    {
        startPage();
        List<TmsRegion> list = tmsRegionService.selectTmsRegionList(tmsRegion);
        return getDataTable(list);
    }

    /**
     * 导出行政区域列表
     * @param tmsRegion 查询条件对象
     */
    @PreAuthorize("@ss.hasPermi('tms:tmsRegion:export')")
    @Log(title = "行政区域", businessType = BusinessType.EXPORT)
    @GetMapping("/export")
    public AjaxResult export(TmsRegion tmsRegion,String exportKey)
    {
        tmsRegionService.export(tmsRegion,exportKey);
        return AjaxResult.success("导出请求成功，请稍后点击下载...!");
    }



    /**
     * 获取行政区域详细信息
     */
    @PreAuthorize("@ss.hasPermi('tms:tmsRegion:query')")
    @GetMapping(value = "/{id}")
    public AjaxResult getInfo(@PathVariable("id") Integer id)
    {
        return AjaxResult.success(tmsRegionService.selectTmsRegionById(id));
    }

    @GetMapping(value = "/getRegionChildren/{parentRegionCode}")
    public AjaxResult getRegionChildren(@PathVariable("parentRegionCode") String parentRegionCode)
    {
        List<TmsRegion> tmsRegions = tmsRegionService.selectTmsRegionList(new TmsRegion() {{
            setParentRegionCode(parentRegionCode);
        }});
        return AjaxResult.success(tmsRegions);
    }

    @GetMapping(value = "/getProvince")
    public AjaxResult getProvince()
    {
        return AjaxResult.success(tmsRegionService.selectRegionVoList("0"));
    }
    @GetMapping(value = "/getCity/{code}")
    public AjaxResult getCity(@PathVariable("code") String code)
    {
        return AjaxResult.success(tmsRegionService.selectRegionVoList(code));
    }
    @GetMapping(value = "/getArea/{code}")
    public AjaxResult getArea(@PathVariable("code") String code)
    {
        return AjaxResult.success(tmsRegionService.selectRegionVoList(code));
    }
    @GetMapping(value = "/getStreet/{code}")
    public AjaxResult getStreet(@PathVariable("code") String code)
    {
        return AjaxResult.success(tmsRegionService.selectRegionVoList(code));
    }
    @GetMapping(value = "/getAddressCode")
    public AjaxResult getAddressCode(AddressVo addressVo)
    {
        Map<String,String> hasMap = new HashMap<>();

        if (StringUtils.isNotEmpty(addressVo.getProvince())){
            TmsRegion tmsRegion = tmsRegionService.getBaseMapper().selectOne(new LambdaUpdateWrapper<>(TmsRegion.class)
                    .eq(TmsRegion::getRegionName, addressVo.getProvince())
                    .last("limit 1")
            );
            if (tmsRegion != null){
                hasMap.put("provinceCode",tmsRegion.getRegionCode());
                if (StringUtils.isNotEmpty(addressVo.getCity())){
                    TmsRegion tmsRegion2 = tmsRegionService.getBaseMapper().selectOne(new LambdaUpdateWrapper<>(TmsRegion.class)
                            .eq(TmsRegion::getRegionName, addressVo.getCity())
                            .eq(TmsRegion::getParentRegionCode, tmsRegion.getRegionCode())
                            .last("limit 1")
                    );
                    if (tmsRegion2 != null){
                        hasMap.put("cityCode",tmsRegion2.getRegionCode());

                        if (StringUtils.isNotEmpty(addressVo.getDistrict())){
                            TmsRegion tmsRegion3= tmsRegionService.getBaseMapper().selectOne(new LambdaUpdateWrapper<>(TmsRegion.class)
                                    .eq(TmsRegion::getRegionName, addressVo.getDistrict())
                                    .eq(TmsRegion::getParentRegionCode, tmsRegion2.getRegionCode())
                                    .last("limit 1")
                            );
                            if (tmsRegion3 != null){
                                hasMap.put("districtCode",tmsRegion3.getRegionCode());

                                if (StringUtils.isNotEmpty(addressVo.getTownship())){
                                    TmsRegion tmsRegion4 = tmsRegionService.getBaseMapper().selectOne(new LambdaUpdateWrapper<>(TmsRegion.class)
                                            .eq(TmsRegion::getRegionName, addressVo.getTownship())
                                            .eq(TmsRegion::getParentRegionCode, tmsRegion3.getRegionCode())
                                            .last("limit 1")
                                    );
                                    if (tmsRegion4 != null){
                                        hasMap.put("townshipCode",tmsRegion4.getRegionCode());
                                    }
                                }

                            }
                        }


                    }
                }
            }
        }
        return AjaxResult.success(hasMap);
    }

    /**
     * 新增行政区域
     */
    @PreAuthorize("@ss.hasPermi('tms:tmsRegion:add')")
    @Log(title = "行政区域", businessType = BusinessType.INSERT)
    @PostMapping
    public AjaxResult add(@RequestBody TmsRegion tmsRegion)
    {
        return toAjax(tmsRegionService.insertTmsRegion(tmsRegion));
    }

    /**
     * 修改行政区域
     */
    @PreAuthorize("@ss.hasPermi('tms:tmsRegion:edit')")
    @Log(title = "行政区域", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult edit(@RequestBody TmsRegion tmsRegion)
    {
        return toAjax(tmsRegionService.updateTmsRegion(tmsRegion));
    }

    /**
     * 删除行政区域
     */
    @PreAuthorize("@ss.hasPermi('tms:tmsRegion:remove')")
    @Log(title = "行政区域", businessType = BusinessType.DELETE)
	@DeleteMapping("/{ids}")
    public AjaxResult remove(@PathVariable Integer[] ids)
    {
        return toAjax(tmsRegionService.deleteTmsRegionByIds(ids));
    }
}