package org.nutz.mvc.upload;
import java.util.regex.Pattern;
import org.nutz.filepool.FilePool;
import org.nutz.filepool.NutFilePool;
import org.nutz.lang.Encoding;
import org.nutz.lang.Strings;
import org.nutz.log.Log;
import org.nutz.log.Logs;
/**
* 执行上传时一些必要的配置信息
*
* @author zozoh(zozohtnt@gmail.com)
*/
public class UploadingContext {
private static final Log log = Logs.get();
public static UploadingContext create(String poolPath) {
return create(new NutFilePool(poolPath));
}
public static UploadingContext create(FilePool pool) {
return new UploadingContext(pool);
}
public UploadingContext(String poolPath) {
this(new NutFilePool(poolPath, 2000));
}
public UploadingContext(FilePool pool) {
charset = Encoding.UTF8;
bufferSize = 8192;
this.filePool = pool;
}
/**
* 默认为 UTF-8，上传字节流的编码
*/
private String charset;
/**
* 临时文件池
*/
private FilePool filePool;
/**
* 缓冲，默认 8192
*/
private int bufferSize;
/**
* 是否忽略空文件，默认为 false
*/
private boolean ignoreNull;
/**
* 如果大于0，对于每个上传的文件，都判断，如果超过了这个大小，则拒绝继续上传
* <p>
* 单位为字节
*/
private int maxFileSize;
/**
* 一个正则表达式，描述了可以允许的文件名
*/
private String nameFilter;
/**
* 一个正则表达式，描述了可以允许的文件内容类型
*/
private String contentTypeFilter;
public String getCharset() {
return charset;
}
public UploadingContext setCharset(String charset) {
this.charset = charset;
return this;
}
public FilePool getFilePool() {
return filePool;
}
public UploadingContext setFilePool(FilePool pool) {
this.filePool = pool;
return this;
}
public int getBufferSize() {
return bufferSize;
}
public UploadingContext setBufferSize(int bufferSize) {
this.bufferSize = bufferSize;
if (bufferSize < 128 && log.isWarnEnabled()) {
log.warn("Uploading buffer is less than 128!! Auto-fix to 128!! 8192 will be much better!!");
this.bufferSize = 128;
}
return this;
}
public boolean isIgnoreNull() {
return ignoreNull;
}
public UploadingContext setIgnoreNull(boolean ignoreNull) {
this.ignoreNull = ignoreNull;
return this;
}
public int getMaxFileSize() {
return maxFileSize;
}
public UploadingContext setMaxFileSize(int maxFileSize) {
this.maxFileSize = maxFileSize;
return this;
}
public String getNameFilter() {
return nameFilter;
}
public UploadingContext setNameFilter(String nameFilter) {
this.nameFilter = nameFilter;
return this;
}
public boolean isNameAccepted(String name) {
if (null == nameFilter || Strings.isBlank(name)
|| "\"\"".equals(name)) //用户不选择文件时,文件名会是"" 两个双引号
return true;
return Pattern.matches(nameFilter, name.toLowerCase());
}
public String getContentTypeFilter() {
return contentTypeFilter;
}
public UploadingContext setContentTypeFilter(String contentTypeFilter) {
this.contentTypeFilter = contentTypeFilter;
return this;
}
public boolean isContentTypeAccepted(String contentType) {
if (null == contentTypeFilter || Strings.isBlank(contentType))
return true;
return Pattern.matches(contentTypeFilter, contentType.toLowerCase());
}
}