fix

Helius

2021-07-20 12bdaf99a9abf9e3930b449c65dd427c861b0ca4

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package com.matrix.component.ueditor.upload;
import com.matrix.component.ueditor.define.State;
 
import com.matrix.component.ueditor.PathFormat;
import com.matrix.component.ueditor.define.AppInfo;
import com.matrix.component.ueditor.define.BaseState;
import com.matrix.component.ueditor.define.FileType;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
 
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
 
public class BinaryUploader {
 
    public static final State save(HttpServletRequest request, Map<String, Object> conf) {
        // FileItemStream fileStream = null;
        // boolean isAjaxUpload = request.getHeader( "X_Requested_With" ) != null;
 
        if (!ServletFileUpload.isMultipartContent(request)) {
            return new BaseState(false, AppInfo.NOT_MULTIPART_CONTENT);
        }
 
        // ServletFileUpload upload = new ServletFileUpload(
        // new DiskFileItemFactory());
        //
        // if ( isAjaxUpload ) {
        // upload.setHeaderEncoding( "UTF-8" );
        // }
 
        try {
            // FileItemIterator iterator = upload.getItemIterator(request);
            //
            // while (iterator.hasNext()) {
            // fileStream = iterator.next();
            //
            // if (!fileStream.isFormField())
            // break;
            // fileStream = null;
            // }
            //
            // if (fileStream == null) {
            // return new BaseState(false, AppInfo.NOTFOUND_UPLOAD_DATA);
            // }
            MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
            MultipartFile multipartFile = multipartRequest.getFile(conf.get("fieldName").toString());
            if (multipartFile == null) {
                return new BaseState(false, AppInfo.NOTFOUND_UPLOAD_DATA);
            }
 
            String savePath = (String) conf.get("savePath");
            // String originFileName = fileStream.getName();
            String originFileName = multipartFile.getOriginalFilename();
            String suffix = FileType.getSuffixByFilename(originFileName);
 
            originFileName = originFileName.substring(0, originFileName.length() - suffix.length());
            savePath = savePath + suffix;
 
            long maxSize = ((Long) conf.get("maxSize")).longValue();
 
            if (!validType(suffix, (String[]) conf.get("allowFiles"))) {
                return new BaseState(false, AppInfo.NOT_ALLOW_FILE_TYPE);
            }
 
            savePath = PathFormat.parse(savePath, originFileName);
 
            String physicalPath = (String) conf.get("rootPath") + savePath;
 
            // InputStream is = fileStream.openStream();
            InputStream is = multipartFile.getInputStream();
            State storageState = StorageManager.saveFileByInputStream(is, physicalPath, maxSize);
            is.close();
 
            if (storageState.isSuccess()) {
                storageState.putInfo("url", PathFormat.format(savePath));
                storageState.putInfo("type", suffix);
                storageState.putInfo("original", originFileName + suffix);
            }
 
            return storageState;
            // } catch (FileUploadException e) {
            // return new BaseState(false, AppInfo.PARSE_REQUEST_ERROR);
        } catch (IOException e) {
        }
        return new BaseState(false, AppInfo.IO_ERROR);
    }
 
    private static boolean validType(String type, String[] allowTypes) {
        List<String> list = Arrays.asList(allowTypes);
 
        return list.contains(type);
    }
}