/**
* Copyright (c) 2005-2012 https://github.com/zhangkaitao
*
* Licensed under the Apache License, Version 2.0 (the "License");
*/
package com.sishuok.es.maintain.staticresource.web.controller.utils;
import com.sishuok.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: Zhang Kaitao
* <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);
}
}
}