package com.hx.mp.util;

import com.hx.util.OSSUtil;
import com.hx.util.SimpleTool;
import com.hx.util.StringUtils;
import net.sf.json.JSONException;
import net.sf.json.JSONObject;
import org.apache.commons.codec.binary.Base64;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.springframework.stereotype.Component;

import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.text.SimpleDateFormat;
import java.util.Date;

/**
 * 微信获取权限基本工具类
 * @author ChenJiaHe
 * @Date 2020-07-14
 */
@Component
public class MPWeixinBaseUtil {

	// 类型
	private static final String GRANT_TYPE = "client_credential";

	/**获取access_token链接*/
	private static final String GETACCESS_TOKENURL = "https://api.weixin.qq.com/cgi-bin/token?grant_type=";
	/**通过code获取小程序信息链接*/
	private static final String JSCODE2SESSION_URL = "https://api.weixin.qq.com/sns/jscode2session?";
	/**发送订阅消息通知链接*/
	private static final String SEND_SUBSCRIBE_MESSAGE = "https://api.weixin.qq.com/cgi-bin/message/subscribe/send?access_token=";
	//生成小程序二维码地址（方形）
	public static final String MAKE_TWOCODE_SQUARE_URL = "https://api.weixin.qq.com/cgi-bin/wxaapp/createwxaqrcode?access_token=";
	//生成小程序二维码地址（圆形）
	public static final String MAKE_TWOCODE_ROUND_URL = "https://api.weixin.qq.com/wxa/getwxacode?access_token={0}";
	//生成无限二维码
	public static final String URL_UNLIMIT_SQUARE = "https://api.weixin.qq.com/wxa/getwxacodeunlimit?access_token={0}";
	//////////////////////////////////////////////////

	/** (小程序)通过code换取网页授权access_token/如果是snsapi_base模式的授权，这里就可以拿到openId了 ***/
	public static JSONObject getJscode2session(String appid,String secret,String code) {

		return HttpURLUtil(JSCODE2SESSION_URL+"appid="+appid+"&secret="+secret
				+"&grant_type=authorization_code&js_code=" + code,null);
	}

	/**发送订阅消息通知**/
	public static JSONObject sendSubscribeMessage(String data,String accessToken){
		return HttpURLUtil(SEND_SUBSCRIBE_MESSAGE+accessToken,data);
	}

	/** 获取access_token */
	public static JSONObject getAccessToken(String appid,String secret) throws Exception {
		String access_token;
		// 通过WX接口获取access_token
		JSONObject obj = getApplication_Access_tokenForUrl(appid,secret);
		return obj;
	}

	/** 从weixin接口获取Access_token **/
	public static JSONObject getApplication_Access_tokenForUrl(String appid,String secret) {
		return HttpURLUtil(GETACCESS_TOKENURL+GRANT_TYPE +"&appid="+appid +"&secret="+secret,null);
	}

	/**生成无限二维码，并上传到OSS*/
	public static String createUnlimitQrCode(String at, String scene, String page, int width, boolean autoColor,
											 JSONObject lineColor, boolean isHyaline, String fileName, String keyId,
											 String keySecret, String endPoint, String bucket)
	{
		String imgUrl = null;
		InputStream in = null;
		HttpURLConnection conn = null;

		try {
			//生成发送数据
			JSONObject obj = new JSONObject();
			obj.put("scene", scene);
			obj.put("width", width);
			obj.put("page", page);
			obj.put("auto_color", autoColor);
			obj.put("line_color", lineColor);
			obj.put("is_hyaline", isHyaline);

			// 创建url资源
			URL url = new URL(StringUtils.format(URL_UNLIMIT_SQUARE, at));
			// 建立http连接
			conn = (HttpURLConnection) url.openConnection();
			// 设置允许输出
			conn.setDoOutput(true);
			conn.setDoInput(true);
			// 设置不用缓存
			conn.setUseCaches(false);
			// 设置传递方式
			conn.setRequestMethod("POST");
			// 设置维持长连接
			conn.setRequestProperty("Connection", "Keep-Alive");
			// 设置文件字符集:
			conn.setRequestProperty("Charset", "UTF-8");
			// 设置文件类型:
			conn.setRequestProperty("contentType", "application/json");
			// 开始连接请求
			conn.connect();
			OutputStream out = conn.getOutputStream();
			// 写入请求的字符串
			out.write((obj.toString()).getBytes());
			out.flush();
			out.close();

			// 请求返回的状态
			if (conn.getResponseCode() == 200) {
				// 请求返回的数据
				in = conn.getInputStream();
				imgUrl = OSSUtil.uploadImg(fileName, in, keyId, keySecret, endPoint, bucket);
				conn.disconnect();
				conn = null;
			}

			if (in != null) {
				in.close();
				in = null;
			}

			if (conn != null) {
				conn.disconnect();
				conn = null;
			}
		}catch (Exception e)
		{
			e.printStackTrace();

			if (in != null) {
				try {
					in.close();
				}catch (Exception ep)
				{
					ep.printStackTrace();
				}
				in = null;
			}

			if (conn != null) {
				conn.disconnect();
				conn = null;
			}
		}

		return imgUrl;
	}

	/**生成有限二维码,返回临时文
	 * @param at 微信token
	 * @param page 跳转链接
	 * @param width 宽度
	 * @param autoColor 默认false
	 * @param lineColor 默认null
	 * @param isHyaline 默认false
	 * @return
	 */
	public static File createLimitedQrCode(String at, String page, int width, boolean autoColor,
										   JSONObject lineColor, boolean isHyaline) throws IOException {
		String imgUrl = null;
		InputStream in = null;
		HttpURLConnection conn = null;

		//创建临时文件
		File file = File.createTempFile("temp", ".jpg");
		try {
			//生成发送数据
			JSONObject obj = new JSONObject();
			obj.put("width", width);
			obj.put("path", page);
			obj.put("auto_color", autoColor);
			obj.put("line_color", lineColor);
			obj.put("is_hyaline", isHyaline);

			// 创建url资源
			URL url = new URL(StringUtils.format(MAKE_TWOCODE_ROUND_URL, at));
			// 建立http连接
			conn = (HttpURLConnection) url.openConnection();
			// 设置允许输出
			conn.setDoOutput(true);
			conn.setDoInput(true);
			// 设置不用缓存
			conn.setUseCaches(false);
			// 设置传递方式
			conn.setRequestMethod("POST");
			// 设置维持长连接
			conn.setRequestProperty("Connection", "Keep-Alive");
			// 设置文件字符集:
			conn.setRequestProperty("Charset", "UTF-8");
			// 设置文件类型:
			conn.setRequestProperty("contentType", "application/json");
			// 开始连接请求
			conn.connect();
			OutputStream out = conn.getOutputStream();
			// 写入请求的字符串
			out.write((obj.toString()).getBytes());
			out.flush();
			out.close();

			// 请求返回的状态
			if (conn.getResponseCode() == 200) {
				// 请求返回的数据
				in = conn.getInputStream();
				//输入到临时文件
				/*OutputStream os = new FileOutputStream(file);
				int bytesRead = 0;
				byte[] buffer = new byte[8192];
				while ((bytesRead = in.read(buffer, 0, 8192)) != -1) {
					os.write(buffer, 0, bytesRead);
				}*/
				FileUtils.copyInputStreamToFile(in, file);

				conn.disconnect();
				conn = null;
			}

			if (in != null) {
				in.close();
				in = null;
			}

			if (conn != null) {
				conn.disconnect();
				conn = null;
			}
		}catch (Exception e) {
			e.printStackTrace();
			if (in != null) {
				try {
					in.close();
				}catch (Exception ep) {
					ep.printStackTrace();
				}
				in = null;
			}

			if (conn != null) {
				conn.disconnect();
				conn = null;
			}
		}finally {
			file.deleteOnExit();
		}
		return file;
	}

	/**生成无限二维码,返回临时文
	 * @param at 微信token
	 * @param scene 参数，只能32位，最好不要中文
	 * @param page 跳转链接
	 * @param width 宽度
	 * @param autoColor 默认false
	 * @param lineColor 默认null
	 * @param isHyaline 默认false
	 * @return
	 */
	public static File createUnlimitQrCode(String at, String scene, String page, int width, boolean autoColor,
											 JSONObject lineColor, boolean isHyaline) throws IOException {
		String imgUrl = null;
		InputStream in = null;
		HttpURLConnection conn = null;

		//创建临时文件
		File file = File.createTempFile("temp", ".jpg");

		try {
			//生成发送数据
			JSONObject obj = new JSONObject();
			obj.put("scene", scene);
			obj.put("width", width);
			obj.put("page", page);
			obj.put("auto_color", autoColor);
			obj.put("line_color", lineColor);
			obj.put("is_hyaline", isHyaline);

			// 创建url资源
			URL url = new URL(StringUtils.format(URL_UNLIMIT_SQUARE, at));
			// 建立http连接
			conn = (HttpURLConnection) url.openConnection();
			// 设置允许输出
			conn.setDoOutput(true);
			conn.setDoInput(true);
			// 设置不用缓存
			conn.setUseCaches(false);
			// 设置传递方式
			conn.setRequestMethod("POST");
			// 设置维持长连接
			conn.setRequestProperty("Connection", "Keep-Alive");
			// 设置文件字符集:
			conn.setRequestProperty("Charset", "UTF-8");
			// 设置文件类型:
			conn.setRequestProperty("contentType", "application/json");
			// 开始连接请求
			conn.connect();
			OutputStream out = conn.getOutputStream();
			// 写入请求的字符串
			out.write((obj.toString()).getBytes());
			out.flush();
			out.close();

			// 请求返回的状态
			if (conn.getResponseCode() == 200) {
				// 请求返回的数据
				in = conn.getInputStream();
				//输入到临时文件
				/*OutputStream os = new FileOutputStream(file);
				int bytesRead = 0;
				byte[] buffer = new byte[8192];
				while ((bytesRead = in.read(buffer, 0, 8192)) != -1) {
					os.write(buffer, 0, bytesRead);
				}*/
				FileUtils.copyInputStreamToFile(in, file);

				conn.disconnect();
				conn = null;
			}

			if (in != null) {
				in.close();
				in = null;
			}

			if (conn != null) {
				conn.disconnect();
				conn = null;
			}
		}catch (Exception e) {
			e.printStackTrace();
			if (in != null) {
				try {
					in.close();
				}catch (Exception ep) {
					ep.printStackTrace();
				}
				in = null;
			}

			if (conn != null) {
				conn.disconnect();
				conn = null;
			}
		}finally {
			file.deleteOnExit();
		}
		return file;
	}


	/**生成小程序二维码工具（方形）
	 * path 二维码跳转链接
	 * width 二维码宽度,默认是430
	 * saveUrl 保存图片的全路径
	 * seeUrl 显示图片路径
	 * appid 小程序配置
	 * secret 小程序配置
	 * return 二维码显示图片链接
	 * @throws Exception
	 * */
	public static String twoCodeImgSquare(String at, String path,Integer width,String saveUrl,
										  String seeUrl,String appid,String secret) throws Exception {

		if(!SimpleTool.checkNotNull(width)) {
			width = 430;
		}
		//生成发送数据
		JSONObject obj = new JSONObject();
		obj.put("path", path);
		obj.put("width", width);

		//二维码图片路径
		String codeUrl = "";
		codeUrl = HttpURLUtilMakeCodeImg(MAKE_TWOCODE_SQUARE_URL+at,obj,saveUrl,seeUrl);
		if(SimpleTool.checkNotNull(codeUrl)){
			if(!codeUrl.startsWith("/")){
				codeUrl = "/"+codeUrl;
			}
		}
		return codeUrl;
	}

	public static String twoCodeImgSquare(String at, String path,Integer width, String fileName, String keyId,
										  String keySecret, String endPoint, String bucket) throws Exception {

		if(!SimpleTool.checkNotNull(width)) {
			width = 430;
		}
		//生成发送数据
		JSONObject obj = new JSONObject();
		obj.put("path", path);
		obj.put("width", width);

		String imgUrl = null;
		InputStream in = null;
		HttpURLConnection conn = null;

		try {

			// 创建url资源
			URL url = new URL(MAKE_TWOCODE_SQUARE_URL + at);
			// 建立http连接
			conn = (HttpURLConnection) url.openConnection();
			// 设置允许输出
			conn.setDoOutput(true);
			conn.setDoInput(true);
			// 设置不用缓存
			conn.setUseCaches(false);
			// 设置传递方式
			conn.setRequestMethod("POST");
			// 设置维持长连接
			conn.setRequestProperty("Connection", "Keep-Alive");
			// 设置文件字符集:
			conn.setRequestProperty("Charset", "UTF-8");
			// 设置文件类型:
			conn.setRequestProperty("contentType", "application/json");
			// 开始连接请求
			conn.connect();
			OutputStream out = conn.getOutputStream();
			// 写入请求的字符串
			out.write((obj.toString()).getBytes());
			out.flush();
			out.close();

			// 请求返回的状态
			if (conn.getResponseCode() == 200) {
				// 请求返回的数据
				in = conn.getInputStream();
				imgUrl = OSSUtil.uploadImg(fileName, in, keyId, keySecret, endPoint, bucket);
				conn.disconnect();
				conn = null;
			}

			if (in != null) {
				in.close();
				in = null;
			}

			if (conn != null) {
				conn.disconnect();
				conn = null;
			}
		}catch (Exception e)
		{
			e.printStackTrace();

			if (in != null) {
				try {
					in.close();
				}catch (Exception ep)
				{
					ep.printStackTrace();
				}
				in = null;
			}

			if (conn != null) {
				conn.disconnect();
				conn = null;
			}
		}

		return imgUrl;


	}

	/** 生成二维码的图片 返回图片路径
	 * urlx 执行链接
	 * objx 数据
	 * saveUrl 保存图片路径
	 * seeUrl 显示图片路径
	 *
	 *  **/
	public static String HttpURLUtilMakeCodeImg(String urlx,JSONObject objx,String saveUrl,String seeUrl) throws IOException {
		String realpath = null;

		// 创建url资源
		URL url = new URL(urlx);
		// 建立http连接
		HttpURLConnection conn = (HttpURLConnection) url.openConnection();
		// 设置允许输出
		conn.setDoOutput(true);

		conn.setDoInput(true);

		// 设置不用缓存
		conn.setUseCaches(false);
		// 设置传递方式
		conn.setRequestMethod("POST");
		// 设置维持长连接
		conn.setRequestProperty("Connection", "Keep-Alive");
		// 设置文件字符集:
		conn.setRequestProperty("Charset", "UTF-8");
		// 转换为字节数组
		// byte[] data = (objx.toString()).getBytes();
		// 设置文件长度
		// conn.setRequestProperty("Content-Length",String.valueOf(data.length));

		// 设置文件类型:
		conn.setRequestProperty("contentType", "application/json");
		// 开始连接请求
		conn.connect();
		OutputStream out = conn.getOutputStream();
		// 写入请求的字符串
		out.write((objx.toString()).getBytes());
		out.flush();
		out.close();

		// 请求返回的状态
		if (conn.getResponseCode() == 200) {
			System.out.println("连接成功");
			// 请求返回的数据
			InputStream in = conn.getInputStream();
			ByteArrayOutputStream swapStream = new ByteArrayOutputStream();
			byte[] temp = new byte[1024];
			int rc = 0;
			while ((rc = in.read(temp, 0, 100)) > 0) {
				swapStream.write(temp, 0, rc);
			}
			byte[] buffer2 = swapStream.toByteArray();
			String base64 = Base64.encodeBase64String(buffer2);

			//生成名称
			java.util.Random r=new java.util.Random();
			StringBuilder str=new StringBuilder();//定义变长字符串
			for(int i=0;i<8;i++){
				str.append(r.nextInt(10));
			}
			Date date = new Date();
			SimpleDateFormat sf = new SimpleDateFormat("yyyyMMddHHmmss");
			String imgName = sf.format(date);
			imgName = imgName+str.toString();

			//文件保存位置
			File saveDir = new File(saveUrl);
			if(!saveDir.exists()){
				saveDir.mkdir();
			}
			File file = new File(saveDir+"/"+imgName+".png");
			FileOutputStream fos = new FileOutputStream(file);
			fos.write(buffer2);
			if(fos!=null){
				fos.close();
			}
			if(in!=null){
				in.close();
			}
			realpath = seeUrl+"/"+imgName+".png";
		} else {
			System.out.println("no++");
		}
		return realpath;
	}

	/** 请求http协议 获取信息工具 **/
	private static JSONObject HttpURLUtil(String url,String data) {
		HttpURLConnection con = null;
		URL u;
		String wxMsgXml;
		JSONObject obj;
		try {
			u = new URL(url);
			con = (HttpURLConnection) u.openConnection();
			con.setRequestMethod("POST");
			con.setDoOutput(true);
			con.setDoInput(true);
			con.setUseCaches(false);
			con.setReadTimeout(5000);
			con.setRequestProperty("Charset", "UTF-8");
			con.setRequestProperty("Content-Type",
					"application/x-www-form-urlencoded");
			if (data != null) {
				OutputStream os = con.getOutputStream();
				os.write(data.getBytes("utf-8"));
			}
			if (con.getResponseCode() != 200)
				throw new RuntimeException("请求url失败");
			// 读取返回内容
			wxMsgXml = IOUtils.toString(con.getInputStream(), StandardCharsets.UTF_8);
			// 判断返回参数是否正确/成功
			obj = JSONObject.fromObject(wxMsgXml);
			// System.out.println("HttpURLUtil:"+wxMsgXml);
		} catch (Exception e) {
			e.printStackTrace();
			obj = new JSONObject();
			try {
				obj.put("status", 1);
				obj.put("errMsg", e.getMessage());
			} catch (JSONException e1) {
				e1.printStackTrace();
			}
		} finally {
			if (con != null) {
				con.disconnect();
			}
		}
		return obj;
	}

}