package org.sxkj.common.utils;

import com.alibaba.cloud.commons.lang.StringUtils;
import org.hsqldb.lib.StringUtil;
import org.springframework.http.HttpHeaders;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import org.sxkj.common.constant.CommonConstant;

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

/**
 * HTTP 请求头工具类
 */
public class HeaderUtils {


	/**
	 * 从当前请求中获取指定请求头（适用于Spring MVC环境）
	 *
	 * @param headerName 请求头名称（不区分大小写）
	 * @return 请求头值，不存在返回Optional.empty()
	 */
	public static Optional<String> getHeader(String headerName) {
		return Optional.ofNullable(getServletRequest())
			.flatMap(request -> getHeader(request, headerName));
	}

	/**
	 * 获取头信息中的城市code
	 *
	 * @return
	 */
	public static String getAreaCode() {
		return getAreaCode(null);

	}

	/**
	 * * 获取城市code,没有传参按头信息中的
	 *
	 * @param areaCodeParam 待截取的行政区划
	 * @return 截取之后的行政区划code
	 */
	public static String getAreaCode(String areaCodeParam) {
		return getAreaCodeHeaderAreaCode(areaCodeParam);
	}

	/**
	 * 格式化区域数据  360000 将后面的0000去掉
	 *
	 * @param areaCode 区域
	 * @return 后面无0 的数据
	 */
	public static String formatAreaCode(String areaCode) {
		if (StringUtils.isEmpty(areaCode)) {
			return areaCode;
		}
		areaCode = areaCode.replaceAll("0+$", "");
		if (areaCode.length() % 2 == 1) {
			return areaCode + "0";
		}
		return areaCode;
	}


	/**
	 * 获取行政区划，不删除长度
	 *
	 * @param areaCodeParam 行政区划 优先返回
	 * @return 行政区划
	 */
	public static String getAreaCodeHeaderAreaCode(String areaCodeParam) {
		String areaCode = getHeader(CommonConstant.AREA_CODE).orElse(null);
		String code = StringUtils.isEmpty(areaCodeParam) ? areaCode : areaCodeParam;
		return formatAreaCode(code);
	}

	/**
	 * 获取行政区划，不处理长度
	 * @param areaCodeParam 行政区划 优先返回
	 * @return 行政区划code
	 */
	public static String getAreaCodeHeaderAreaCodeNot(String areaCodeParam) {
		String areaCode = getHeader(CommonConstant.AREA_CODE).orElse(null);
		return StringUtils.isEmpty(areaCodeParam) ? areaCode : areaCodeParam;
	}


	/**
	 * 从HttpServletRequest中获取指定请求头
	 *
	 * @param request    HttpServletRequest对象
	 * @param headerName 请求头名称（不区分大小写）
	 * @return 请求头值，不存在返回Optional.empty()
	 */
	public static Optional<String> getHeader(HttpServletRequest request, String headerName) {
		if (request == null || headerName == null || headerName.isEmpty()) {
			return Optional.empty();
		}

		// 先尝试直接获取（性能更优）
		String value = request.getHeader(headerName);
		if (value != null) {
			return Optional.of(value);
		}

		// 如果直接获取不到，再遍历所有header进行不区分大小写的匹配
		Enumeration<String> headerNames = request.getHeaderNames();
		while (headerNames != null && headerNames.hasMoreElements()) {
			String name = headerNames.nextElement();
			if (headerName.equalsIgnoreCase(name)) {
				return Optional.ofNullable(request.getHeader(name));
			}
		}

		return Optional.empty();
	}

	/**
	 * 从ServerHttpRequest中获取指定请求头（适用于WebFlux环境）
	 *
	 * @param request    ServerHttpRequest对象
	 * @param headerName 请求头名称（不区分大小写）
	 * @return 请求头值，不存在返回Optional.empty()
	 */
	public static Optional<String> getHeader(ServerHttpRequest request, String headerName) {
		if (request == null || headerName == null || headerName.isEmpty()) {
			return Optional.empty();
		}

		HttpHeaders headers = request.getHeaders();
		// 先尝试直接获取（性能更优）
		if (headers.containsKey(headerName)) {
			return Optional.ofNullable(headers.getFirst(headerName));
		}

		// 如果直接获取不到，再遍历所有header进行不区分大小写的匹配
		for (Map.Entry<String, List<String>> entry : headers.entrySet()) {
			if (headerName.equalsIgnoreCase(entry.getKey())) {
				return Optional.ofNullable(entry.getValue().get(0));
			}
		}

		return Optional.empty();
	}

	/**
	 * 获取当前请求的所有请求头（适用于Spring MVC环境）
	 *
	 * @return 请求头Map，键已转为小写
	 */
	public static Map<String, String> getAllHeaders() {
		return Optional.ofNullable(getServletRequest())
			.map(HeaderUtils::getAllHeaders)
			.orElse(Collections.emptyMap());
	}

	/**
	 * 从HttpServletRequest获取所有请求头
	 *
	 * @param request HttpServletRequest对象
	 * @return 请求头Map，键已转为小写
	 */
	public static Map<String, String> getAllHeaders(HttpServletRequest request) {
		Map<String, String> headers = new HashMap<>();
		if (request != null) {
			Enumeration<String> headerNames = request.getHeaderNames();
			while (headerNames != null && headerNames.hasMoreElements()) {
				String name = headerNames.nextElement();
				headers.put(name.toLowerCase(), request.getHeader(name));
			}
		}
		return headers;
	}

	/**
	 * 从ServerHttpRequest获取所有请求头（适用于WebFlux环境）
	 *
	 * @param request ServerHttpRequest对象
	 * @return 请求头Map，键已转为小写
	 */
	public static Map<String, String> getAllHeaders(ServerHttpRequest request) {
		Map<String, String> headers = new HashMap<>();
		if (request != null) {
			request.getHeaders().forEach((key, values) ->
				headers.put(key.toLowerCase(), values.get(0)));
		}
		return headers;
	}

	/**
	 * 处理区域代码，提取最后一组并补零到12位
	 *
	 * @param areaCode 原始区域代码，可能包含多个以逗号分隔的代码
	 * @return 处理后的区域代码，补零到12位
	 */
	public static String processAreaCode(String areaCode) {
		if (areaCode != null && !areaCode.isEmpty()) {
			// 分割区域代码并只取最后一组
			String[] codes = areaCode.split(",");
			if (codes.length > 0) {
				// 取最后一组并去除空格
				String lastCode = codes[codes.length - 1].trim();
				if (!lastCode.isEmpty()) {
					// 补零到12位
					return lastCode + "000000000000".substring(lastCode.length());
				}
			}
		}
		return areaCode;
	}

	/**
	 * 补零
	 * @param areaCode
	 * @return
	 */
	public static String fillZero(String areaCode) {
		return areaCode + "000000000000".substring(areaCode.length());
	}

	/**
	 * 根据URL获取附件类型
	 * 通过文件扩展名判断附件类型：
	 * - 图片类型：jpg, jpeg, png, gif, bmp, webp, svg 等 -> 返回 1
	 * - 视频类型：mp4, avi, mov, wmv, flv, mkv, mpeg, mpg 等 -> 返回 2
	 * - 文档类型：pdf, doc, docx, xls, xlsx, ppt, pptx, txt 等 -> 返回 3
	 * - 其他类型 -> 返回 0
	 *
	 * @param url 文件URL地址
	 * @return 附件类型：1-图片，2-视频，3-文档，0-其他
	 */
	public static Integer getAttachmentType(String url) {
		if (StringUtils.isEmpty(url)) {
			return 0;
		}

		// 提取文件扩展名
		String extension = getFileExtension(url);
		if (StringUtils.isEmpty(extension)) {
			return 0;
		}

		// 转换为小写进行比较
		String extLower = extension.toLowerCase();

		// 图片类型
		if (isImageType(extLower)) {
			return 1;
		}

		// 视频类型
		if (isVideoType(extLower)) {
			return 2;
		}

		// 文档类型
		if (isDocumentType(extLower)) {
			return 3;
		}

		// 其他类型
		return 0;
	}

	/**
	 * 从URL中提取文件扩展名
	 *
	 * @param url 文件URL地址
	 * @return 文件扩展名（不包含点），如果无法提取则返回null
	 */
	private static String getFileExtension(String url) {
		if (StringUtils.isEmpty(url)) {
			return null;
		}

		// 去除URL参数部分（?后面的内容）
		int questionMarkIndex = url.indexOf('?');
		if (questionMarkIndex != -1) {
			url = url.substring(0, questionMarkIndex);
		}

		// 查找最后一个点的位置
		int lastDotIndex = url.lastIndexOf('.');
		if (lastDotIndex == -1 || lastDotIndex == url.length() - 1) {
			return null;
		}

		// 提取扩展名
		return url.substring(lastDotIndex + 1);
	}

	/**
	 * 判断是否为图片类型
	 *
	 * @param extension 文件扩展名（小写）
	 * @return 是否为图片类型
	 */
	private static boolean isImageType(String extension) {
		return extension.equals("jpg") || extension.equals("jpeg") ||
			extension.equals("png") || extension.equals("gif") ||
			extension.equals("bmp") || extension.equals("webp") ||
			extension.equals("svg") || extension.equals("tiff") ||
			extension.equals("tif") || extension.equals("ico");
	}

	/**
	 * 判断是否为视频类型
	 *
	 * @param extension 文件扩展名（小写）
	 * @return 是否为视频类型
	 */
	private static boolean isVideoType(String extension) {
		return extension.equals("mp4") || extension.equals("avi") ||
			extension.equals("mov") || extension.equals("wmv") ||
			extension.equals("flv") || extension.equals("mkv") ||
			extension.equals("mpeg") || extension.equals("mpg") ||
			extension.equals("webm") || extension.equals("m4v") ||
			extension.equals("3gp");
	}

	/**
	 * 判断是否为文档类型
	 *
	 * @param extension 文件扩展名（小写）
	 * @return 是否为文档类型
	 */
	private static boolean isDocumentType(String extension) {
		return extension.equals("pdf") || extension.equals("doc") ||
			extension.equals("docx") || extension.equals("xls") ||
			extension.equals("xlsx") || extension.equals("ppt") ||
			extension.equals("pptx") || extension.equals("txt") ||
			extension.equals("csv") || extension.equals("zip") ||
			extension.equals("rar");
	}

	/**
	 * 获取当前HttpServletRequest对象
	 *
	 * @return HttpServletRequest对象，可能为null
	 */
	private static HttpServletRequest getServletRequest() {
		try {
			ServletRequestAttributes attributes = (ServletRequestAttributes)
				RequestContextHolder.getRequestAttributes();
			return attributes != null ? attributes.getRequest() : null;
		} catch (IllegalStateException e) {
			// 非Web请求上下文
			return null;
		}
	}
}