package org.springblade.modules.yw.service.impl;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.geotools.geojson.geom.GeometryJSON;
import org.geotools.geometry.jts.WKBReader;
import org.locationtech.jts.geom.Geometry;
import org.locationtech.jts.io.ParseException;
import org.springblade.common.utils.FileUtil;
import org.springblade.common.utils.ShapeFileUtil;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.modules.yw.entity.GeomInfoEntity;
import org.springblade.modules.yw.mapper.GeomInfoMapper;
import org.springblade.modules.yw.service.IGeomInfoService;
import org.springblade.modules.yw.vo.GeomInfoVO;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Objects;

/**
 * 空间信息表 服务实现类
 *
 * @author BladeX
 * @since 2024-11-05
 */
@Service
public class GeomInfoServiceImpl extends ServiceImpl<GeomInfoMapper, GeomInfoEntity> implements IGeomInfoService {

	/**
	 * 自定义分页
	 * @param page
	 * @param geomInfo
	 * @return
	 */
	@Override
	public IPage<GeomInfoVO> selectGeomInfoPage(IPage<GeomInfoVO> page, GeomInfoVO geomInfo) {
		return page.setRecords(baseMapper.selectGeomInfoPage(page, geomInfo));
	}

	/**
	 * 导入 shp zip 包文件解析空间信息
	 * @param multipartFile
	 * @return
	 */
	@Override
	public R importShpZip(MultipartFile multipartFile) {
		// 判断文件是否为 zip 文件
		if (!FileUtil.getFileExtension(multipartFile).equals("zip")){
			return R.data(400,"文件格式不对，必须是 zip 文件","文件格式不对，必须是 zip 文件");
		}
		// 转 file
		File file = FileUtil.toFile(multipartFile);
		// 获取文件信息
		try {
			List<Map<String, Object>> list = ShapeFileUtil.shpToGeoJson(file);
			// 写入数据库,暂时考虑只有一组数据的情况
			if (list.size()>0){
				Map<String, Object> map = list.get(0);
				GeomInfoEntity geomInfoEntity = Objects.requireNonNull(BeanUtil.copy(map, GeomInfoEntity.class));
				String geometry = map.get("geometry").toString();
				geomInfoEntity.setGeom("'" + geometry +"'");
				// 保存
				baseMapper.saveGeoInfo(geomInfoEntity);
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
		// 返回
		return R.data(200,"操作成功","操作成功");
	}
}