package apache.poi;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;

import org.apache.poi.hwpf.HWPFDocument;
import org.apache.poi.hwpf.usermodel.Range;
import org.apache.poi.poifs.filesystem.DirectoryEntry;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;


public class ExportDocTest {
  
  public static void main(String[] args) {
    String destFile=\"D:\\\\11.doc\";
    //#####################根据自定义内容导出Word文档#################################################
    StringBuffer fileCon=new StringBuffer();
    fileCon.append(\"        张大炮      男       317258963215223\\n\" +
        \"2011   09    2013   07    3\\n\" +
        \"  二炮研究       成人\\n\" +
        \"2013000001               2013   07   08\");
    fileCon.append(\"\\n\\r\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\\n\");
    
    new ExportDocTest().exportDoc(destFile, fileCon.toString());
    
    //##################根据Word模板导出单个Word文档###################################################
    Map<String, String> map=new HashMap<String, String>();
    
    map.put(\"name\", \"Zues\");
    map.put(\"sex\", \"男\");
    map.put(\"idCard\", \"200010\");
    map.put(\"year1\", \"2000\");
    map.put(\"month1\", \"07\");
    map.put(\"year2\", \"2008\");
    map.put(\"month2\", \"07\");
    map.put(\"gap\", \"2\");
    map.put(\"zhuanye\", \"计算机科学与技术\");
    map.put(\"type\", \"研究生\");
    map.put(\"bianhao\", \"2011020301\");
    map.put(\"nowy\", \"2011\");
    map.put(\"nowm\", \"01\");
    map.put(\"nowd\", \"20220301\");
    //注意biyezheng_moban.doc文档位置,此例中为应用根目录
    HWPFDocument document=new ExportDocTest().replaceDoc(\"biyezheng_moban.doc\", map);
    ByteArrayOutputStream ostream = new ByteArrayOutputStream();
    try {
      document.write(ostream);
      //输出word文件
      OutputStream outs=new FileOutputStream(destFile);
      outs.write(ostream.toByteArray());
      outs.close();
    } catch (IOException e) {
      e.printStackTrace();
    }
    
  }
  
  
  /**
   * 
   * @param destFile
   * @param fileCon
   */
  public void exportDoc(String destFile,String fileCon){
    try {
      //doc content
      ByteArrayInputStream bais = new ByteArrayInputStream(fileCon.getBytes());
      POIFSFileSystem fs = new POIFSFileSystem();
      DirectoryEntry directory = fs.getRoot(); 
      directory.createDocument(\"WordDocument\", bais);
      FileOutputStream ostream = new FileOutputStream(destFile);
      fs.writeFilesystem(ostream);
      bais.close();
      ostream.close();
      
    } catch (IOException e) {
      e.printStackTrace();
    }
  }
  
  
  /**
   * 读取word模板并替换变量
   * @param srcPath
   * @param map
   * @return
   */
  public HWPFDocument replaceDoc(String srcPath, Map<String, String> map) {
    try {
      // 读取word模板
      FileInputStream fis = new FileInputStream(new File(srcPath));
      HWPFDocument doc = new HWPFDocument(fis);
      // 读取word文本内容
      Range bodyRange = doc.getRange();
      // 替换文本内容
      for (Map.Entry<String, String> entry : map.entrySet()) {
        bodyRange.replaceText(\"${\" + entry.getKey() + \"}\", entry
            .getValue());
      }
      return doc;
    } catch (Exception e) {
      e.printStackTrace();
      return null;
    }
  }

}