package cn.ksource.core.util;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.Map.Entry;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.poi.poifs.filesystem.DirectoryEntry;
import org.apache.poi.poifs.filesystem.DocumentEntry;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;

import cn.ksource.config.SysConfigConstants;

public class HtmlToword {

	
public static String writeWordFile(HttpServletRequest request,HttpServletResponse response,String url,Map<String, String> addParams,String fileName) {

  
  try {
     String content = get(url, addParams);
     
    /* Map<String,String> map = zoomImage(request,content);
     content = map.get("result");
     String imgSrc = map.get("imgSrc");
     System.out.println(imgSrc);*/
     byte b[] = content.getBytes();
     ByteArrayInputStream bais = new ByteArrayInputStream(b);
     POIFSFileSystem poifs = new POIFSFileSystem();
     DirectoryEntry directory = poifs.getRoot();
     DocumentEntry documentEntry = directory.createDocument("WordDocument", bais);
     ByteArrayOutputStream baos = new ByteArrayOutputStream();
     poifs.writeFilesystem(baos);
     bais.close();
     toClientForReport(baos,fileName,response);
     
     //删除临时图像
    /* String path = request.getSession().getServletContext().getRealPath("/")+imgSrc;
     return path;*/
     
  } catch (IOException e) {
   e.printStackTrace();
  }
  	return "";
}

public static Map<String, String> zoomImage(HttpServletRequest request,String content) {
	Document document = Jsoup.parse(content);
	Elements elements = document.select("img");
	Map map = new HashMap();
	if(elements.size()>0) {
		Element element = elements.get(0);
		String imgSrc = element.attr("src");
		System.out.println(imgSrc);
		imgSrc = imgSrc.replace(SysConfigConstants.SERVER_IP, "");
		String pathPix = request.getSession().getServletContext().getRealPath("/");
		System.out.println(imgSrc);
		String newName = FileUtil.resizeForEqualImg(new File(pathPix+imgSrc), 196, 184, false);
		System.out.println(newName);
		imgSrc = imgSrc.replace(imgSrc.substring(imgSrc.lastIndexOf("/")+1), newName);
		map.put("imgSrc", imgSrc);
		System.out.println(SysConfigConstants.SERVER_IP+imgSrc);
		element.attr("src", SysConfigConstants.SERVER_IP+imgSrc);
	}
	String result = document.toString();
	map.put("result", result);
	return map;
}


public static String downloadWordFile(HttpServletRequest request,HttpServletResponse response,String url,String path,Map<String, String> addParams) {
	  boolean w = false;
	  
	  try {
	   if (!"".equals(path)) {
	    // 检查目录是否存在
	    File fileDir = new File(path);
	    if (!fileDir.exists()) {
	    	fileDir.mkdirs();
	    }
	     // 生成临时文件名称
	     String fileName = StringUtil.getUUID()+".doc";
	     String content = get(url, addParams);
	     
	     
	     byte b[] = content.getBytes();
	     ByteArrayInputStream bais = new ByteArrayInputStream(b);
	     POIFSFileSystem poifs = new POIFSFileSystem();
	     DirectoryEntry directory = poifs.getRoot();
	     DocumentEntry documentEntry = directory.createDocument("WordDocument", bais);
	     FileOutputStream ostream = new FileOutputStream(path+ fileName);
	     ByteArrayOutputStream baos = new ByteArrayOutputStream();
	     poifs.writeFilesystem(ostream);
	     bais.close();
	     ostream.close();
	     
	     return fileName;
	   }

	  } catch (IOException e) {
	   e.printStackTrace();
	  }
	  return "";
}


public static void downloadhtml(HttpServletResponse response,String url,Map<String, String> addParams,String fileName) {
	  try {
	     String content = get(url, addParams);
	     System.out.println(content);
	     ByteArrayOutputStream baos = new ByteArrayOutputStream();
	     baos.write(content.getBytes("UTF-8"));
	     toClientForHtml(baos,fileName,response);
	  } catch (IOException e) {
	   e.printStackTrace();
	  }

	}


/** 返回客户端信息
 * @throws UnsupportedEncodingException 
 * */
private static void toClientForReport(ByteArrayOutputStream baos,String name,HttpServletResponse response) throws UnsupportedEncodingException{
	response.setHeader("Content-Type", "application/msword");
	response.setHeader("Content-Disposition", "attachment;filename=\""+new String(name.getBytes("gbk"),"ISO8859-1")+".doc\"");
	response.setCharacterEncoding("UTF-8");
	try {
		ServletOutputStream out = response.getOutputStream();
		baos.writeTo(out);
		baos.flush();
		baos.close();
	} catch (IOException e) {
		e.printStackTrace();
	}
}


/** 返回客户端信息
 * @throws UnsupportedEncodingException 
 * */
private static void toClientForHtml(ByteArrayOutputStream baos,String name,HttpServletResponse response) throws UnsupportedEncodingException{
	response.setHeader("Content-Type", "application/html");
	response.setHeader("Content-Disposition", "attachment;filename=\""+new String(name.getBytes("gbk"),"ISO8859-1")+".html\"");
	response.setCharacterEncoding("UTF-8");
	try {
		ServletOutputStream out = response.getOutputStream();
		baos.writeTo(out);
		baos.flush();
		baos.close();
	} catch (IOException e) {
		e.printStackTrace();
	}
}




public static void deleteUploadFile(String deleteFilePath){
		File file=new File(deleteFilePath);
		if(file.exists()){
			if(file.delete()){
				file.getParentFile().delete();
				System.out.println("文件被成功删除");
			}
		}
}



public static String get(String url, Map<String, String> addParams) {
	String result = "";
	
	HttpURLConnection huc = null;
	BufferedReader br = null;
	try
	{
		if (addParams != null && addParams.size() > 0)
		{
			if (url.indexOf("?") == -1)
			{
				url = url + "?";
			}
			Set<Entry<String, String>> paramSet = addParams.entrySet();
			for (Entry<String, String> param : paramSet)
			{
				url = url + "&" + param.getKey() + "=" + param.getValue();
			}
		}
		
		URL u = new URL(url);
		
		huc = (HttpURLConnection) u.openConnection();
		huc.setConnectTimeout(2000);
		
		br  = new BufferedReader(new InputStreamReader(huc.getInputStream(), "UTF-8"));
		
		StringBuilder sb = new StringBuilder();
		String tmp = null;
		while ((tmp = br.readLine()) != null) {
			sb.append(tmp);
		}
		result = sb.toString();
	}
	catch (Exception e)
	{
		e.printStackTrace();
	}
	finally
	{
		if (br != null)
		{
			try {
				br.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
		if (huc != null)
		{
			huc.disconnect();
		}
	}
	
	return result;
}

}