package com.terra.system.controller.data.upload;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.metadata.OrderItem;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.terra.system.annotation.SysLog;
import com.terra.common.entity.all.ResponseMsg;
import com.terra.common.entity.all.StaticData;
import com.terra.system.entity.ctrl.TabEntity;
import com.terra.system.entity.data.*;
import com.terra.system.entity.sys.DepEntity;
import com.terra.system.entity.sys.UserEntity;
import com.terra.system.helper.ClassHelper;
import com.terra.system.helper.StringHelper;
import com.terra.system.mapper.all.BasicMapper;
import com.terra.system.service.all.BaseQueryService;
import com.terra.system.service.data.*;
import com.terra.system.service.sys.DepService;
import com.terra.system.service.sys.TokenService;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.Parameters;
import javax.annotation.Resource;

import io.swagger.v3.oas.annotations.enums.ParameterIn;
import org.springframework.web.bind.annotation.GetMapping;

import javax.servlet.http.HttpServletRequest;
import java.util.List;

/**
 * 查询控制器
 * @author WWW
 */
public class QueryController extends CheckController {
    @Resource
    protected DepService depService;

    @Resource
    protected DirService dirService;

    @Resource
    protected VerService verService;

    @Resource
    protected MetaService metaService;

    @Resource
    protected TokenService tokenService;

    @Resource
    protected UploadService uploadService;

    @Resource
    protected BaseQueryService baseQueryService;

    @Resource
    DictService dictService;

    @SysLog()
    @Operation(summary = "根据父ID分页查询并返回记录数")
    @Parameters({
            @Parameter(name = "metaid", description = "父ID", in = ParameterIn.QUERY, example = "0"),
            @Parameter(name = "name", description = "名称", in = ParameterIn.QUERY, example = ""),
            @Parameter(name = "pageSize", description = "每页条数", in = ParameterIn.QUERY, example = "10"),
            @Parameter(name = "pageIndex", description = "分页数（从1开始）", in = ParameterIn.QUERY, example = "1")
    })
    @GetMapping(value = "/selectPageAndCountByPid")
    public ResponseMsg<List<MetaEntity>> selectPageAndCountByPid(Integer metaid, String name, Integer pageSize, Integer pageIndex) {
        try {
            if (pageSize < 1 || pageIndex < 1) {
                return fail("每页页数或分页数小于1", null);
            }
            if (null == metaid || metaid < 1) {
                return fail("父ID不能为空且大于1", null);
            }

            int count = metaService.selectCountByPid(metaid, name);
            if (count == 0) {
                return success(0, null);
            }

            List<MetaEntity> rs = metaService.selectPageByPid(metaid, name, pageSize, pageSize * (pageIndex - 1));

            return success(count, rs);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询所有单位")
    @GetMapping(value = "/selectDepAll")
    public ResponseMsg<List<DepEntity>> selectDepAll() {
        try {
            List<DepEntity> list = depService.selectDepAll();

            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询所有目录")
    @GetMapping(value = "/selectDirAll")
    public ResponseMsg<List<DirEntity>> selectDirAll() {
        try {
            List<DirEntity> list = dirService.selectDirAll();

            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "根据目录ID查询版本列表")
    @Parameters({
            @Parameter(name = "dirid", description = "目录ID", in = ParameterIn.QUERY)
    })
    @GetMapping(value = "/selectVerByDirid")
    public ResponseMsg<List<VerEntity>> selectVerByDirid(Integer dirid) {
        try {
            if (null == dirid || dirid < 0) {
                dirid = 0;
            }

            List<VerEntity> list = verService.selectByDirid(dirid);
            if (null == list || list.isEmpty()) {
                list = verService.selectByDirid(0);
            }

            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询坐标系")
    @Parameters({
            @Parameter(name = "zoning", description = "带号", in = ParameterIn.QUERY, example = "6度有带号")
    })
    @GetMapping(value = "/selectCoords")
    public ResponseMsg<Object> selectCoords(String zoning) {
        try {
            List<CoordEntity> list = uploadService.selectCoords(zoning);

            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询项目名称")
    @GetMapping(value = "/selectProject")
    public ResponseMsg<Object> selectProject() {
        try {
            List<DirEntity> list = uploadService.selectProject();

            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询所有表")
    @Parameters({
            @Parameter(name = "name", description = "名称", in = ParameterIn.QUERY, example = "点"),
            @Parameter(name = "hasGeom", description = "含有Geom字段", in = ParameterIn.QUERY, example = "false")
    })
    @GetMapping(value = "/selectTabs")
    public ResponseMsg<List<TabEntity>> selectTabs(String name, Boolean hasGeom) {
        try {
            List<TabEntity> list = dictService.selectDictTab(name, null == hasGeom || !hasGeom ? "gid" : "geom");

            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询字段信息")
    @Parameters({
            @Parameter(name = "ns", description = "名称空间", in = ParameterIn.QUERY, example = "bd"),
            @Parameter(name = "tab", description = "表名", in = ParameterIn.QUERY, example = "dlg25wAanp")
    })
    @GetMapping(value = "/selectFields")
    public ResponseMsg<List<DictEntity>> selectFields(String ns, String tab) {
        try {
            if (StringHelper.isEmpty(ns) || StringHelper.isEmpty(tab)) {
                return fail("名称空间和表名不能为空", null);
            }

            List<DictEntity> list = baseQueryService.selectFields(ns, tab);

            return success(list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "查询表中数据")
    @Parameters({
            @Parameter(name = "id", description = "元数据ID", in = ParameterIn.QUERY, example = "115"),
            @Parameter(name = "pageIndex", description = "分页数（从1开始）", in = ParameterIn.QUERY, example = "1"),
            @Parameter(name = "pageSize", description = "每页条数", in = ParameterIn.QUERY, example = "10")
    })
    @GetMapping(value = "/selectDbData")
    public ResponseMsg<Object> selectDbData(Integer id, Integer pageIndex, Integer pageSize) {
        // noinspection AlibabaRemoveCommentedCode
        try {
            if (null == id || id < 0) {
                return fail("元数据ID不能为空或小于0", null);
            }

            MetaEntity meta = metaService.selectById(id);
            if (null == meta || null == meta.getTab() || !meta.getTab().contains(StaticData.POINT)) {
                return fail("找不到元数据信息", null);
            }

            String entity = meta.getTab().substring(meta.getTab().indexOf(".") + 1).replace("_", "").toLowerCase();
            BasicMapper baseMapper = ClassHelper.getBasicMapper(entity);
            if (null == baseMapper) {
                return null;
            }

            QueryWrapper wrapper = new QueryWrapper();
            wrapper.eq("parentid", meta.getEventid());

            Page<Object> page = new Page<>(pageIndex, pageSize);
            page.addOrder(OrderItem.desc("gid"));
            IPage<Object> paged = baseMapper.selectPage(page, wrapper);

            return success(paged.getTotal(), paged.getRecords());
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "分页查询上传数据并返回记录数")
    @Parameters({
            @Parameter(name = "name", description = "名称", in = ParameterIn.QUERY, example = ""),
            @Parameter(name = "pageSize", description = "每页条数", in = ParameterIn.QUERY, example = "10"),
            @Parameter(name = "pageIndex", description = "分页数（从1开始）", in = ParameterIn.QUERY, example = "1")
    })
    @GetMapping(value = "/selectByPageForUpload")
    public ResponseMsg<Object> selectByPageForUpload(String name, Integer pageSize, Integer pageIndex, HttpServletRequest req) {
        try {
            if (pageSize < 1 || pageIndex < 1) {
                return fail("每页页数或分页数小于1", null);
            }

            UserEntity ue = tokenService.getCurrentUser(req);

            int count = metaService.selectCountForUpload(name, ue.getId(), null);
            if (count == 0) {
                return success(0, null);
            }

            List<MetaEntity> list = metaService.selectByPageForUpload(name, ue.getId(), null, pageSize, pageSize * (pageIndex - 1));

            return success(count, list);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }

    @SysLog()
    @Operation(summary = "根据元数据ID查询")
    @Parameters({
            @Parameter(name = "id", description = "ID", in = ParameterIn.QUERY, example = "1")
    })
    @GetMapping(value = "/selectMetaById")
    public ResponseMsg<MetaEntity> selectMetaById(int id) {
        try {
            MetaEntity entity = metaService.selectById(id);

            return success(entity);
        } catch (Exception ex) {
            return fail(ex, null);
        }
    }
}