com.fhs.fileService.ueditor.hunter.FileManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fhs_file Show documentation
文件服务
The newest version!
package com.fhs.fileService.ueditor.hunter;

import com.fhs.common.spring.SpringContextUtil;
import com.fhs.fileService.bean.ServiceFile;
import com.fhs.fileService.service.ServiceFileService;
import com.fhs.fileService.ueditor.define.AppInfo;
import com.fhs.fileService.ueditor.define.BaseState;
import com.fhs.fileService.ueditor.define.MultiState;
import com.fhs.fileService.ueditor.define.State;
import org.apache.commons.io.FileUtils;

import java.io.File;
import java.util.Arrays;
import java.util.Collection;
import java.util.Map;


public class FileManager {

	private String dir = null;
	private String rootPath = null;
	private String[] allowFiles = null;
	private int count = 0;

	/**
	 * 文件服务域名
	 */
	public static String basePath;

	/**
	 * 文件储存路径
	 */
	public static String fileSavePath;


	public FileManager ( Map conf ) {

		this.rootPath = (String)conf.get( "rootPath" );
		this.dir = this.rootPath + (String)conf.get( "dir" );
		this.allowFiles = this.getAllowFiles( conf.get("allowFiles") );
		this.count = (Integer)conf.get( "count" );

	}

	public State listFile (int index ) {
		File dir = new File( fileSavePath );
		State state = null;

		if ( !dir.exists() ) {
			return new BaseState( false, AppInfo.NOT_EXIST );
		}

		if ( !dir.isDirectory() ) {
			return new BaseState( false, AppInfo.NOT_DIRECTORY );
		}

		Collection list = FileUtils.listFiles( dir, this.allowFiles, true );

		if ( index < 0 || index > list.size() ) {
			state = new MultiState( true );
		} else {
			Object[] fileList = Arrays.copyOfRange( list.toArray(), index, index + this.count );
			state = this.getState( fileList );
		}

		state.putInfo( "start", index );
		state.putInfo( "total", list.size() );

		return state;

	}

	private State getState ( Object[] files ) {

		MultiState state = new MultiState( true );
		BaseState fileState = null;

		File file = null;
		ServiceFileService service = (ServiceFileService) SpringContextUtil.getBeansByClass(ServiceFileService.class);
		for ( Object obj : files ) {
			if ( obj == null ) {
				break;
			}
			file = (File)obj;
			fileState = new BaseState( true );
			String fileId = file.getName ().substring (0, file.getName ().indexOf ("."));
			ServiceFile serviceFile = service.selectById (fileId);
			fileState.putInfo( "url", basePath + "downLoad/file?fileId=" + fileId);
			fileState.putInfo ("original", serviceFile.getFileName ());
			fileState.putInfo("type", serviceFile.getFileSuffix ().replace (".",""));
			state.addState( fileState );
		}

		return state;

	}

	private String getPath ( File file ) {
//		String path = file.getAbsolutePath();
//		return "/"+path.substring(this.rootPath.length() , path.length());
		return "";
	}

	private String[] getAllowFiles ( Object fileExt ) {

		String[] exts = null;
		String ext = null;

		if ( fileExt == null ) {
			return new String[ 0 ];
		}

		exts = (String[])fileExt;

		for ( int i = 0, len = exts.length; i < len; i++ ) {

			ext = exts[ i ];
			exts[ i ] = ext.replace( ".", "" );

		}

		return exts;

	}

}