package com.qxueyou.scc.base.util;

import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;

import javax.imageio.ImageIO;

import org.apache.commons.lang3.StringUtils;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

//import com.qxueyou.scc.base.service.IConfigService;
import com.qxueyou.scc.shorturl.util.ShortUrlUtils;
import com.google.zxing.BarcodeFormat;
import com.google.zxing.EncodeHintType;
import com.google.zxing.MultiFormatWriter;
import com.google.zxing.WriterException;
import com.google.zxing.client.j2se.MatrixToImageConfig;
import com.google.zxing.client.j2se.MatrixToImageWriter;
import com.google.zxing.common.BitMatrix;
import com.google.zxing.qrcode.decoder.ErrorCorrectionLevel;

/**
 * ¶þάÂëÉú³É¹¤¾ßÀà
 * 
 * @author µÂ»¢
 * 
 */
public final class QrCodeUtils {
	
	private static Logger log = LogManager.getLogger(QrCodeUtils.class);

	/**
	 * Éú³É¶þάÂëͼƬ²¢Êä³öµ½Á÷
	 * @param msg ÐèÒªÉú³ÉµÄ¶þάÂëÐÅÏ¢
	 * @param os ¶þάÂëͼƬÊä³öÁ÷£¬Êä³öºó»á×Ô¶¯¹Ø±Õ
	 * @param logoFlag ÊÇ·ñÌí¼Ólogo
	 * 
	 */
	public static void createQRCodeImgAndSend(String msg, OutputStream os, boolean logoFlag) {
		try {
			//Èç¹ûÊÇÍøÖ·£¬Ôòת»»Îª¶ÌÍøÖ·
			/*if(ShortUrlUtils.IsUrl(msg)){
				IShortUrlService shortUrlService = SpringUtil.getBean(IShortUrlService.class);
				ShortUrl oldShortUrl = shortUrlService.queryShortUrl(msg);
				if(oldShortUrl != null){//ÅжÏÊÇ·ñÒѾ­´æÔÚÊý¾Ý
					msg = oldShortUrl.getShortUrl();
				}else{
					//»ñÈ¡Óò
					String domain = SpringUtil.getBean(IConfigService.class).getConfigByEnv("domain-name");
					//»ñÈ¡¶ÌÁ´½Ó±àÂë
					String strShortUrlCode = ShortUrlUtils.shortUrl(msg);
					//Æ´½Ó³É¶ÌÁ´½Ó
					String shortUrl = domain+(domain.indexOf("/qxueyou", 10)>=0?"/s/":"/qxueyou/s/")+strShortUrlCode;
					ShortUrl objShortUrl = new ShortUrl();
					//±£´æ¶ÌÁ´½ÓµÄÖµ
					objShortUrl.setShortUrl(shortUrl);
					objShortUrl.setLongUrl(msg);
					objShortUrl.setShortUrlCode(strShortUrlCode);
					objShortUrl.setDeleteFlag(false);
					TraceUtils.setCreateTrace(objShortUrl);
					msg = shortUrl;
					shortUrlService.saveShortUrl(objShortUrl);
				}
			}*/
			MultiFormatWriter multiFormatWriter = new MultiFormatWriter();
			
			Map<EncodeHintType, Object> hints = new HashMap<EncodeHintType, Object>(
					2);
			hints.put(EncodeHintType.CHARACTER_SET, "utf-8");
			hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.M);

			BitMatrix bitMatrix = multiFormatWriter.encode(
					msg,
					BarcodeFormat.QR_CODE, 300, 300, hints);
			MatrixToImageConfig config = new MatrixToImageConfig(0xFF000000, -1);
			//Ìí¼Ó»ú¹¹Í¼±ê
			if(StringUtils.isNoneBlank(ClientUtils.getOrgLogoPath()) && ShortUrlUtils.IsUrl(msg) && logoFlag){
			    int width = bitMatrix.getWidth();  
		        int height = bitMatrix.getHeight();  
		        BufferedImage image = new BufferedImage(width, height,  
		                BufferedImage.TYPE_INT_RGB);  
		        for (int x = 0; x < width; x++) {  
		            for (int y = 0; y < height; y++) {  
		                image.setRGB(x, y, bitMatrix.get(x, y) ? 0xFF000000  
		                        : 0xFFFFFFFF);  
		            }  
		        } 
				addLogo_QRCode(image, os);
			}else{
				MatrixToImageWriter.writeToStream(bitMatrix, "PNG",
						os, config);
			}

		} catch (WriterException e) {
			log.error(e, e);
		} catch (IOException e) {
			log.error(e, e);
		} catch (Exception e) {
			log.error(e, e);
		}
	}
	
	/**
	 * Ìí¼Ó»ú¹¹logo
	 * 
	 * @param buf Éú³ÉµÄ¶þάÂë
	 * @return
	 * @throws IOException 
	 */
	public static void addLogo_QRCode(BufferedImage buf, OutputStream os) throws IOException{
		CloseableHttpClient httpClient = null;
		HttpGet httpget = null;
		try{
//			String domain = SpringUtil.getBean(IConfigService.class).getConfigByEnv("oss-domain");
			String domain =null;
			httpClient = HttpClients.createDefault();
			//»ñÈ¡»ú¹¹logo
			httpget = new HttpGet(domain+ClientUtils.getOrgLogoPath());  
			HttpResponse response = httpClient.execute(httpget);
			HttpEntity entity = response.getEntity();
			/**
			* ¶ÁÈ¡LogoͼƬ
			*/
			BufferedImage logo = ImageIO.read(entity.getContent());
			/**
			* ¶ÁÈ¡¶þάÂëͼƬ£¬²¢¹¹½¨»æͼ¶ÔÏó
			*/
			Graphics2D g = buf.createGraphics();
			int widthLogo = 65, heightLogo = 65;
		
			// ¼ÆËãͼƬ·ÅÖÃλÖÃ
			int x = (buf.getWidth() - widthLogo) / 2;
			int y = (buf.getHeight() - heightLogo) / 2;
		
			//¿ªÊ¼»æÖÆͼƬ
			g.setColor(Color.WHITE);
			g.drawImage(logo, x, y, widthLogo, heightLogo, null);
	        g.setStroke(new BasicStroke(1));  
			g.dispose();
			ImageIO.write(buf, "PNG", os);
		}catch (Exception e) {
			ImageIO.write(buf, "PNG", os);
			log.error("¶þάÂëÌí¼Ólogoʧ°Ü"+e);
		}finally {
			httpClient.close();
		}
	}
}