forumuploadddxc.java

来自「cwbbs 云网论坛源码」· Java 代码 · 共 74 行

package com.redmoon.forum;

import cn.js.fan.util.ErrMsgException;
import com.redmoon.kit.util.UploadDdxc;
import com.redmoon.kit.util.FileUpload;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.ServletContext;
import com.redmoon.forum.person.UserDb;
import com.redmoon.kit.util.UploadFileInfo;
import java.io.File;
import cn.js.fan.web.SkinUtil;
import cn.js.fan.web.Global;


public class ForumUploadDdxc extends UploadDdxc {

    public ForumUploadDdxc() {
        maxFileSize = 1024 * 10; 
    }

    public String receive(ServletContext application,
                          HttpServletRequest request) throws ErrMsgException {
        return super.receive(application, request);
    }

    public String ReceiveUploadFileHeader(HttpServletRequest request, FileUpload fu) throws ErrMsgException {
        
        Privilege privilege = new Privilege();
        if (!privilege.canUploadAttachment(request)) {
            UserDb ud = new UserDb();
            ud = ud.getUser(privilege.getUser(request));
            String str = SkinUtil.LoadString(request, "res.forum.ForumUploadDdxc", "err_space");
            str = str.replaceFirst("\\$allowed", "" + ud.getDiskSpaceAllowed());
            str = str.replaceFirst("\\$used", "" + ud.getDiskSpaceUsed());
            throw new ErrMsgException(str); 
        }
        visualPath = Config.getInstance().getAttachmentPath();
        return super.ReceiveUploadFileHeader(request, fu);
    }

    public String ReceiveUploadFileFinished(HttpServletRequest request, FileUpload fu) throws
            ErrMsgException {
        
        String fileId = fu.getFieldValue("fileId");
        if (fileId == null)
            throw new ErrMsgException("Want fileId");
        UploadFileInfo ufi = (UploadFileInfo) getUploadFileInfos().get(fileId);
        
        if (ufi == null)
                throw new ErrMsgException("Want thread header");

        String re = super.ReceiveUploadFileFinished(request, fu);
        
        if (!re.equals(file_finished_ok))
            return re;

        
        String fullSavePath = ufi.getFullSavePath(Global.getRealPath());

        File file = new File(fullSavePath);
        
        if (file.exists()) {
            
            Privilege privilege = new Privilege();
            UserDb ud = new UserDb();
            ud = ud.getUser(privilege.getUser(request));
            ud.setDiskSpaceUsed(ud.getDiskSpaceUsed() + file.length());
            ud.save();
        }

        return re;
    }
}