com.daphne.es.maintain.staticresource.web.controller.utils.YuiCompressorUtils.java Source code

Introduction

Here is the source code for com.daphne.es.maintain.staticresource.web.controller.utils.YuiCompressorUtils.java

Source

/**
 * 
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 */
package com.daphne.es.maintain.staticresource.web.controller.utils;

import com.daphne.es.common.Constants;
import com.yahoo.platform.yui.compressor.CssCompressor;
import com.yahoo.platform.yui.compressor.JavaScriptCompressor;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.mozilla.javascript.ErrorReporter;
import org.mozilla.javascript.EvaluatorException;

import java.io.*;

/**
 * <p>User: 
 * <p>Date: 13-6-9 ?11:50
 * <p>Version: 1.0
 */
public class YuiCompressorUtils {

    /**
     * js/css  ??.min.js/css
     * @param fileName
     * @return ??
     */
    public static String compress(final String fileName) {
        String minFileName = null;
        final String extension = FilenameUtils.getExtension(fileName);
        if ("js".equalsIgnoreCase(extension)) {
            minFileName = fileName.substring(0, fileName.length() - 3) + ".min.js";
        } else if ("css".equals(extension)) {
            minFileName = fileName.substring(0, fileName.length() - 4) + ".min.css";
        } else {
            throw new RuntimeException(
                    "file type only is js/css, but was fileName:" + fileName + ", extension:" + extension);
        }
        Reader in = null;
        Writer out = null;
        try {
            in = new BufferedReader(new InputStreamReader(new FileInputStream(fileName), Constants.ENCODING));
            out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(minFileName), Constants.ENCODING));

            if ("js".equals(extension)) {

                CustomErrorReporter errorReporter = new CustomErrorReporter();

                JavaScriptCompressor compressor = new JavaScriptCompressor(in, errorReporter);
                compressor.compress(out, 10, true, false, false, false);

                if (errorReporter.hasError()) {
                    throw new RuntimeException(errorReporter.getErrorMessage());
                }

            } else if ("css".equals(extension)) {

                CssCompressor compressor = new CssCompressor(in);
                compressor.compress(out, 10);
            }
        } catch (Exception e) {
            //?js/css
            try {
                FileUtils.copyFile(new File(fileName), new File(minFileName));
            } catch (IOException ioException) {
                throw new RuntimeException("compress error:" + ioException.getMessage(), ioException);
            }
            throw new RuntimeException("compress error:" + e.getMessage(), e);
        } finally {
            if (in != null) {
                try {
                    in.close();
                } catch (IOException e) {
                }
            }
            if (out != null) {
                try {
                    out.close();
                } catch (IOException e) {
                }
            }
        }

        if (FileUtils.sizeOf(new File(minFileName)) == 0) {
            try {
                FileUtils.copyFile(new File(fileName), new File(minFileName));
            } catch (IOException ioException) {
                throw new RuntimeException("compress error:" + ioException.getMessage(), ioException);
            }
        }

        return minFileName;
    }

    public static boolean hasCompress(String fileName) {
        return fileName.toLowerCase().endsWith(".min.js") || fileName.endsWith(".min.css");
    }

    public static String getCompressFileName(String fileName) {
        if (hasCompress(fileName)) {
            return fileName;
        }
        String compressFileName = null;
        final String extension = FilenameUtils.getExtension(fileName);
        if ("js".equalsIgnoreCase(extension)) {
            compressFileName = fileName.substring(0, fileName.length() - 3) + ".min.js";
        } else if ("css".equals(extension)) {
            compressFileName = fileName.substring(0, fileName.length() - 4) + ".min.css";
        }
        return compressFileName;
    }

    public static String getNoneCompressFileName(String fileName) {
        if (!hasCompress(fileName)) {
            return fileName;
        }
        String noneCompressFileName = null;
        final String extension = FilenameUtils.getExtension(fileName);
        if ("js".equalsIgnoreCase(extension)) {
            noneCompressFileName = fileName.substring(0, fileName.length() - 7) + ".js";
        } else if ("css".equals(extension)) {
            noneCompressFileName = fileName.substring(0, fileName.length() - 8) + ".css";
        }
        return noneCompressFileName;
    }

    private static class CustomErrorReporter implements ErrorReporter {
        private StringBuilder error = new StringBuilder();

        public boolean hasError() {
            return error.length() > 0;
        }

        public String getErrorMessage() {
            return error.toString();
        }

        public void warning(String message, String sourceName, int line, String lineSource, int lineOffset) {
            if (line < 0) {
                error.append("\n[WARNING] " + message);
            } else {
                error.append("\n[WARNING] " + line + ':' + lineOffset + ':' + message);
            }
        }

        public void error(String message, String sourceName, int line, String lineSource, int lineOffset) {
            if (line < 0) {
                error.append("\n[ERROR] " + message);
            } else {
                error.append("\n[ERROR] " + line + ':' + lineOffset + ':' + message);
            }
        }

        public EvaluatorException runtimeError(String message, String sourceName, int line, String lineSource,
                int lineOffset) {
            error(message, sourceName, line, lineSource, lineOffset);
            return new EvaluatorException(message);
        }
    }

}