package org.springblade.modules.yw.controller;

import lombok.AllArgsConstructor;
import org.springblade.common.constant.FileConstant;
import org.springblade.common.utils.ShapeFileUtil;
import org.springblade.common.utils.ZipUtil;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

@AllArgsConstructor
@RequestMapping("/yw/downZip")
@RestController
public class DownZipController {

	private final FileConstant fileConstant;

	/**
	 * 导出指定文件夹下文件(压缩成zip)
	 * @param response
	 * @throws IOException
	 */
	@GetMapping("/download-folder-as-zip")
	public void downloadFolderAsZip(HttpServletResponse response) throws IOException {
		// 要导出的文件夹路径
		String sourceFolderPath = fileConstant.getZipTempSavePath();
		// 先生成文件
		List<Map<String,Object>> propertyList = new ArrayList<>();
		for (int i = 0; i < 3; i++) {
			Map<String,Object> map = new HashMap<>();
			map.put("ID", i);
			map.put("名称", "test" + i);
			map.put("描述", "测试shp导出" + i);
			map.put("geometry", "MULTILINESTRING ((114.0888763800001 22.549298400000055, 114.0897166200001 22.54931240800005, 114.09006708000004 22.549318250000056, 114.09104754000009 22.549328150000065))");
			propertyList.add(map);
		}
		// 导出shp 文件
		ShapeFileUtil.exportShp(propertyList, "test", "MultiLineString",sourceFolderPath);
		// ZIP文件名
		String zipFileName = "exported_folder.zip";
		response.setContentType("application/zip");
		response.setHeader("Content-Disposition", "attachment; filename=\"" + zipFileName + "\"");
		try (
			BufferedOutputStream bos = new BufferedOutputStream(response.getOutputStream());
			ZipOutputStream zipOut = new ZipOutputStream(bos);
		) {
			ZipUtil.zipFolder(sourceFolderPath, zipOut);
		}
	}
}