1. 程式人生 > >java word檔案轉html (轉換後可線上預覽)

java word檔案轉html (轉換後可線上預覽)

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStreamWriter;
import java.util.List;

import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;

import org.apache.commons.io.output.ByteArrayOutputStream;
import org.apache.poi.hwpf.HWPFDocument;
import org.apache.poi.hwpf.converter.PicturesManager;
import org.apache.poi.hwpf.converter.WordToHtmlConverter;
import org.apache.poi.hwpf.usermodel.Picture;
import org.apache.poi.hwpf.usermodel.PictureType;
import org.apache.poi.xwpf.converter.core.FileImageExtractor;
import org.apache.poi.xwpf.converter.core.FileURIResolver;
import org.apache.poi.xwpf.converter.core.IURIResolver;
import org.apache.poi.xwpf.converter.xhtml.XHTMLConverter;
import org.apache.poi.xwpf.converter.xhtml.XHTMLOptions;
import org.apache.poi.xwpf.usermodel.XWPFDocument;
import org.apache.poi.xwpf.usermodel.XWPFPictureData;
import org.w3c.dom.Document;



public class WordToHtml {
	public static void main(String[] args) {
		try {
			wordToHtml("d:\\12.docx", "d:\\", "123.html");
			wordToHtml("d:\\2.doc", "d:\\", "12.html");
		} catch (TransformerException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (ParserConfigurationException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
	
	public static void wordToHtml(String wordPath,String htmlPath,String newFilename) throws TransformerException, IOException, ParserConfigurationException {    
			convert2Html(wordPath, htmlPath, newFilename);
    }    
    
	public static void writeFile(String content, String path) {    
        FileOutputStream fos = null;    
        BufferedWriter bw = null;
        try {    
            File file = new File(path);
            if(!file.exists()){
            	
            }
            fos = new FileOutputStream(file);    
            bw = new BufferedWriter(new OutputStreamWriter(fos));    
            bw.write(content);  
        } catch (FileNotFoundException fnfe) {    
            fnfe.printStackTrace();    
        } catch (IOException ioe) {    
            ioe.printStackTrace();    
        } finally {    
            try {    
                if (bw != null)    
                    bw.close();    
                if (fos != null)    
                    fos.close();    
            } catch (IOException ie) {    
            }    
        }    
    }    
    
    /**
     * 將word轉換成html
     * 支援 .doc and .docx
     * @param fileName word檔名
     * @param outPutFilePath html儲存路徑
     * @param newFileName html名
     * @throws TransformerException
     * @throws IOException
     * @throws ParserConfigurationException
     */
    public static void convert2Html(String fileName, String outPutFilePath,String newFileName)    
            throws TransformerException, IOException,    
            ParserConfigurationException {
    	String substring = fileName.substring(fileName.lastIndexOf(".")+1);
    	ByteArrayOutputStream out = new ByteArrayOutputStream();
    	
    	/**
    	 * word2007和word2003的構建方式不同,
    	 * 前者的構建方式是xml,後者的構建方式是dom樹。
    	 * 檔案的字尾也不同,前者字尾為.docx,後者字尾為.doc
    	 * 相應的,apache.poi提供了不同的實現類。
    	 */
    	if("docx".equals(substring)){
//    		writeFile(new String("<html><head>  <meta http-equiv=\"content-type\" content=\"text/html\" charset=\"utf-8\"/></head>對不起,.docx格式的word文件,暫時不能生成預覽</html>".getBytes("utf-8")), outPutFilePath+newFileName); 
    		
    		//step 1 : load DOCX into XWPFDocument
    		InputStream inputStream = new FileInputStream(new File(fileName));
    		XWPFDocument document = new XWPFDocument(inputStream);
    		
    		//step 2 : prepare XHTML options
    		final String imageUrl = "";
    		
    		XHTMLOptions options = XHTMLOptions.create();
    		options.setExtractor(new FileImageExtractor(new File(outPutFilePath + imageUrl)));
    		options.setIgnoreStylesIfUnused(false);
    		options.setFragment(true);
    		options.URIResolver(new IURIResolver() {
//    			@Override 重寫的方法,加上這個報錯,你看看是啥問題
    			public String resolve(String uri) {
    				return imageUrl + uri;
    			}
    		});
    		
    		//step 3 : convert XWPFDocument to XHTML
    		XHTMLConverter.getInstance().convert(document, out, options);
    	}else{
        HWPFDocument wordDocument = new HWPFDocument(new FileInputStream(fileName));//WordToHtmlUtils.loadDoc(new FileInputStream(inputFile));    
        WordToHtmlConverter wordToHtmlConverter = new WordToHtmlConverter(    
                DocumentBuilderFactory.newInstance().newDocumentBuilder()    
                        .newDocument());    
         wordToHtmlConverter.setPicturesManager( new PicturesManager()    
         {    
             public String savePicture( byte[] content,    
                     PictureType pictureType, String suggestedName,    
                     float widthInches, float heightInches )    
             {    
                 return suggestedName;    
             }    
         } );    
        wordToHtmlConverter.processDocument(wordDocument);    
        //save pictures    
        List pics=wordDocument.getPicturesTable().getAllPictures();    
        if(pics!=null){    
            for(int i=0;i<pics.size();i++){    
                Picture pic = (Picture)pics.get(i);    
                System.out.println();    
                try {    
                    pic.writeImageContent(new FileOutputStream(outPutFilePath    
                            + pic.suggestFullFileName()));    
                } catch (FileNotFoundException e) {    
                    e.printStackTrace();    
                }      
            }    
        }    
        Document htmlDocument = wordToHtmlConverter.getDocument();    
        DOMSource domSource = new DOMSource(htmlDocument);    
        StreamResult streamResult = new StreamResult(out);    
    
        TransformerFactory tf = TransformerFactory.newInstance();    //這個應該是轉換成xml的
        Transformer serializer = tf.newTransformer();    
        serializer.setOutputProperty(OutputKeys.ENCODING, "utf-8");    
        serializer.setOutputProperty(OutputKeys.INDENT, "yes");    
        serializer.setOutputProperty(OutputKeys.METHOD, "html");    
        serializer.transform(domSource, streamResult);    
    	}    
    	
    	out.close();    
    	writeFile(new String(out.toByteArray()), outPutFilePath+newFileName);    
    }
    
}