/**
 * BladeX Commercial License Agreement
 * Copyright (c) 2018-2099, https://bladex.cn. All rights reserved.
 * <p>
 * Use of this software is governed by the Commercial License Agreement
 * obtained after purchasing a license from BladeX.
 * <p>
 * 1. This software is for development use only under a valid license
 * from BladeX.
 * <p>
 * 2. Redistribution of this software's source code to any third party
 * without a commercial license is strictly prohibited.
 * <p>
 * 3. Licensees may copyright their own code but cannot use segments
 * from this software for such purposes. Copyright of this software
 * remains with BladeX.
 * <p>
 * Using this software signifies agreement to this License, and the software
 * must not be used for illegal purposes.
 * <p>
 * THIS SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY. The author is
 * not liable for any claims arising from secondary or illegal development.
 * <p>
 * Author: Chill Zhuang (bladejava@qq.com)
 */
package org.springblade.system.controller;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.servlet.http.HttpServletResponse;
import jakarta.validation.Valid;
import lombok.AllArgsConstructor;
import org.springblade.core.boot.ctrl.BladeController;
import org.springblade.core.excel.util.ExcelUtil;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.mp.support.Query;
import org.springblade.core.secure.annotation.PreAuth;
import org.springblade.core.tenant.annotation.NonDS;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.DateUtil;
import org.springblade.core.tool.utils.Func;
import org.springblade.system.excel.RailwayStationExcel;
import org.springblade.system.excel.RailwayStationImporter;
import org.springblade.system.pojo.entity.RailwayStation;
import org.springblade.system.pojo.vo.RailwayStationVO;
import org.springblade.system.service.IRailwayStationService;
import org.springblade.system.wrapper.RailwayStationWrapper;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * 铁路车站主数据 控制器
 *
 * @author Chill
 */
@NonDS
@RestController
@AllArgsConstructor
@PreAuth(menu = "railway_station")
@RequestMapping("/railway-station")
@Tag(name = "铁路车站主数据", description = "铁路车站主数据")
public class RailwayStationController extends BladeController {

	private static final int DEFAULT_CURRENT = 1;
	private static final int DEFAULT_SIZE = 10;
	private static final int MAX_SIZE = 100;
	private static final String SOURCE_INITIAL = "初始化导入";
	private static final String SOURCE_BATCH = "批量";
	private static final String SOURCE_MANUAL = "手动";

	private final IRailwayStationService railwayStationService;

	/**
	 * 详情
	 */
	@GetMapping("/detail")
	@ApiOperationSupport(order = 1)
	@Operation(summary = "详情", description = "传入railwayStation")
	public R<RailwayStationVO> detail(RailwayStation railwayStation) {
		if (Func.isEmpty(railwayStation.getId())) {
			return R.fail("主键不能为空");
		}
		RailwayStation detail = railwayStationService.getById(railwayStation.getId());
		if (Func.isEmpty(detail)) {
			return R.fail("铁路车站不存在");
		}
		return R.data(RailwayStationWrapper.build().entityVO(detail));
	}

	/**
	 * 分页
	 */
	@GetMapping("/list")
	@ApiOperationSupport(order = 2)
	@Operation(summary = "分页", description = "传入railwayStation")
	public R<IPage<RailwayStationVO>> list(RailwayStationVO railwayStation, Query query) {
		IPage<RailwayStationVO> pages = railwayStationService.selectRailwayStationPage(Condition.getPage(normalizeQuery(query)), railwayStation);
		return R.data(pages);
	}

	/**
	 * 新增或修改
	 */
	@PostMapping("/submit")
	@ApiOperationSupport(order = 3)
	@Operation(summary = "新增或修改", description = "传入railwayStation")
	public R submit(@Valid @RequestBody RailwayStation railwayStation) {
		return R.status(railwayStationService.submit(railwayStation));
	}

	/**
	 * 删除
	 */
	@PostMapping("/remove")
	@ApiOperationSupport(order = 4)
	@Operation(summary = "逻辑删除", description = "传入ids")
	public R remove(@Parameter(description = "主键集合", required = true) @RequestParam String ids) {
		if (Func.isEmpty(ids)) {
			return R.fail("主键不能为空");
		}
		return R.status(railwayStationService.deleteLogic(Func.toLongList(ids)));
	}

	/**
	 * 启用或停用
	 */
	@PostMapping("/status")
	@ApiOperationSupport(order = 5)
	@Operation(summary = "启用或停用", description = "传入id和status")
	public R status(@Parameter(description = "主键", required = true) @RequestParam Long id,
					@Parameter(description = "状态", required = true) @RequestParam Integer status) {
		return R.status(railwayStationService.changeStatus(id, status));
	}

	/**
	 * 导入铁路车站主数据
	 */
	@PostMapping("/import-railway-station")
	@ApiOperationSupport(order = 6)
	@Operation(summary = "导入铁路车站主数据", description = "传入excel")
	public R importRailwayStation(MultipartFile file, HttpServletResponse response) {
		if (file == null || file.isEmpty()) {
			return R.fail("上传文件不能为空");
		}
		String fileName = Func.toStrWithEmpty(file.getOriginalFilename(), "").toLowerCase();
		if (!fileName.endsWith(".xls") && !fileName.endsWith(".xlsx")) {
			return R.fail("请上传 .xls,.xlsx 标准格式文件");
		}
		List<RailwayStationExcel> failureList = railwayStationService.importRailwayStation(ExcelUtil.read(file, RailwayStationExcel.class));
		if (Func.isNotEmpty(failureList)) {
			org.springblade.common.excel.ImportFailureExcelUtil.export(response, "铁路车站主数据导入失败明细" + DateUtil.time(), "导入失败明细", failureList, RailwayStationExcel.class);
			return null;
		}
		return R.success("操作成功");
	}

	/**
	 * 导出铁路车站主数据
	 */
	@GetMapping("/export-railway-station")
	@ApiOperationSupport(order = 7)
	@Operation(summary = "导出铁路车站主数据")
	public void exportRailwayStation(@Parameter(hidden = true) @RequestParam Map<String, Object> railwayStation, HttpServletResponse response) {
		Object ids = railwayStation.remove("ids");
		Object dataSource = railwayStation.remove("dataSource");
		railwayStation.remove("Blade-Auth");
		railwayStation.remove("Authorization");
		railwayStation.remove("access_token");
		normalizeRegionCodeCondition(railwayStation);
		QueryWrapper<RailwayStation> queryWrapper = Condition.getQueryWrapper(railwayStation, RailwayStation.class);
		applyDataSourceCondition(queryWrapper, dataSource);
		if (Func.isNotEmpty(ids)) {
			queryWrapper.lambda().in(RailwayStation::getId, Func.toLongList(ids.toString()));
		}
		List<RailwayStationExcel> list = railwayStationService.exportRailwayStation(queryWrapper);
		ExcelUtil.export(response, "铁路车站主数据" + DateUtil.time(), "铁路车站主数据表", list, RailwayStationExcel.class);
	}

	/**
	 * 导出模板
	 */
	@GetMapping("/export-template")
	@ApiOperationSupport(order = 8)
	@Operation(summary = "导出模板")
	public void exportTemplate(HttpServletResponse response) {
		List<RailwayStationExcel> list = new ArrayList<>();
		ExcelUtil.export(response, "铁路车站主数据模板", "铁路车站主数据表", list, RailwayStationExcel.class);
	}

	private Query normalizeQuery(Query query) {
		if (query == null) {
			query = new Query();
		}
		if (query.getCurrent() == null || query.getCurrent() < DEFAULT_CURRENT) {
			query.setCurrent(DEFAULT_CURRENT);
		}
		if (query.getSize() == null || query.getSize() <= 0) {
			query.setSize(DEFAULT_SIZE);
		}
		if (query.getSize() > MAX_SIZE) {
			query.setSize(MAX_SIZE);
		}
		return query;
	}

	private void normalizeRegionCodeCondition(Map<String, Object> params) {
		Object regionCode = params.remove("regionCode");
		if (Func.isNotEmpty(regionCode) && Func.isEmpty(params.get("districtCode"))) {
			params.put("districtCode", regionCode);
		}
	}

	private void applyDataSourceCondition(QueryWrapper<RailwayStation> queryWrapper, Object dataSource) {
		String value = Func.toStrWithEmpty(dataSource, "");
		if (Func.isEmpty(value)) {
			return;
		}
		if (SOURCE_INITIAL.equals(value)) {
			queryWrapper.in("data_source", SOURCE_INITIAL, "初始导入");
		} else if (SOURCE_BATCH.equals(value)) {
			queryWrapper.in("data_source", SOURCE_BATCH, "批量导入");
		} else if (SOURCE_MANUAL.equals(value)) {
			queryWrapper.in("data_source", SOURCE_MANUAL, "手动录入", "手工导入");
		} else {
			queryWrapper.eq("data_source", value);
		}
	}

}
