[1078] 完善录像下载

src/main/java/com/genersoft/iot/vmp/conf/ftpServer/FileCallback.java

@@ -0,0 +1,8 @@
+package com.genersoft.iot.vmp.conf.ftpServer;
+
+import java.io.OutputStream;
+
+public interface FileCallback {
+
+    OutputStream run(String path);
+}

src/main/java/com/genersoft/iot/vmp/conf/ftpServer/FtpFileSystemFactory.java

@@ -0,0 +1,33 @@
+package com.genersoft.iot.vmp.conf.ftpServer;
+
+import org.apache.ftpserver.ftplet.FileSystemFactory;
+import org.apache.ftpserver.ftplet.FileSystemView;
+import org.apache.ftpserver.ftplet.FtpException;
+import org.apache.ftpserver.ftplet.User;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.io.OutputStream;
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
+@Component
+public class FtpFileSystemFactory implements FileSystemFactory {
+
+    private final Map<String, OutputStream> outputStreamMap = new ConcurrentHashMap<>();
+
+    @Override
+    public FileSystemView createFileSystemView(User user) throws FtpException {
+        return new FtpFileSystemView(user, path -> {
+            return outputStreamMap.get(path);
+        });
+    }
+
+    public void addOutputStream(String filePath, OutputStream outputStream) {
+        outputStreamMap.put(filePath, outputStream);
+    }
+
+    public void removeOutputStream(String filePath) {
+        outputStreamMap.remove(filePath);
+    }
+}

src/main/java/com/genersoft/iot/vmp/conf/ftpServer/FtpFileSystemView.java

@@ -0,0 +1,63 @@
+package com.genersoft.iot.vmp.conf.ftpServer;
+
+import org.apache.ftpserver.ftplet.FileSystemView;
+import org.apache.ftpserver.ftplet.FtpException;
+import org.apache.ftpserver.ftplet.FtpFile;
+import org.apache.ftpserver.ftplet.User;
+
+import java.io.OutputStream;
+
+public class FtpFileSystemView implements FileSystemView {
+
+    private User user;
+
+    private FileCallback fileCallback;
+
+    public FtpFileSystemView(User user, FileCallback fileCallback) {
+        this.user = user;
+        this.fileCallback = fileCallback;
+    }
+
+    public static String HOME_PATH = "root";
+
+    public FtpFile workDir = VirtualFtpFile.getDir(HOME_PATH);
+
+    @Override
+    public FtpFile getHomeDirectory() throws FtpException {
+        return VirtualFtpFile.getDir(HOME_PATH);
+    }
+
+    @Override
+    public FtpFile getWorkingDirectory() throws FtpException {
+        return workDir;
+    }
+
+    @Override
+    public boolean changeWorkingDirectory(String dir) throws FtpException {
+        workDir = VirtualFtpFile.getDir(dir);
+        return true;
+    }
+
+    @Override
+    public FtpFile getFile(String file) throws FtpException {
+        VirtualFtpFile ftpFile = VirtualFtpFile.getFile(file);
+        if (fileCallback != null) {
+            OutputStream outputStream = fileCallback.run(workDir.getName());
+            if (outputStream != null) {
+                ftpFile.setOutputStream(outputStream);
+            }
+        }
+        return ftpFile;
+    }
+
+    @Override
+    public boolean isRandomAccessible() throws FtpException {
+        return true;
+    }
+
+    @Override
+    public void dispose() {
+    }
+
+
+}

src/main/java/com/genersoft/iot/vmp/conf/ftpServer/FtpServerConfig.java

@@ -22,7 +22,10 @@ public class FtpServerConfig {
     private UserManager userManager;
 
     @Autowired
-    private Ftplet ftpPlet;
+    private FtpFileSystemFactory fileSystemFactory;
+
+    @Autowired
+    private Ftplet ftplet;
 
     @Autowired
     private FtpSetting ftpSetting;
@@ -45,11 +48,12 @@ public class FtpServerConfig {
         serverFactory.addListener("default", listener);
         // 5、配置自定义用户事件
         Map<String, org.apache.ftpserver.ftplet.Ftplet> ftpLets = new HashMap<>();
-        ftpLets.put("ftpService", ftpPlet);
+        ftpLets.put("ftpService", ftplet);
         serverFactory.setFtplets(ftpLets);
         // 6、读取用户的配置信息
         // 6.2、设置用信息
         serverFactory.setUserManager(userManager);
+        serverFactory.setFileSystem(fileSystemFactory);
         // 7、实例化FTP Server
         FtpServer server = serverFactory.createServer();
         try {

src/main/java/com/genersoft/iot/vmp/conf/ftpServer/Ftplet.java

@@ -42,6 +42,8 @@ public class Ftplet extends DefaultFtplet {
         return super.onUploadUniqueEnd(session, request);
     }
 
+
+
     @Override
     public FtpletResult onAppendEnd(FtpSession session, FtpRequest request) throws FtpException, IOException {
         FtpFile file = session.getFileSystemView().getFile(request.getArgument());

src/main/java/com/genersoft/iot/vmp/conf/ftpServer/VirtualFtpFile.java

@@ -0,0 +1,167 @@
+package com.genersoft.iot.vmp.conf.ftpServer;
+
+import lombok.Setter;
+import org.apache.ftpserver.ftplet.FtpFile;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.util.Collections;
+import java.util.List;
+
+public class VirtualFtpFile implements FtpFile {
+
+    @Setter
+    private String name;
+
+    @Setter
+    private boolean hidden = false;
+
+    @Setter
+    private boolean directory = false;
+
+    @Setter
+    private String ownerName;
+
+    private Long lastModified = null;
+
+    @Setter
+    private long size = 0;
+
+    @Setter
+    private OutputStream outputStream;
+
+    public static VirtualFtpFile getFile(String name) {
+        VirtualFtpFile virtualFtpFile = new VirtualFtpFile();
+        virtualFtpFile.setName(name);
+        return virtualFtpFile;
+    }
+
+    public static VirtualFtpFile getDir(String name) {
+        if (name.endsWith("/")) {
+            name = name.replaceAll("/", "");
+        }
+        VirtualFtpFile virtualFtpFile = new VirtualFtpFile();
+        virtualFtpFile.setName(name);
+        virtualFtpFile.setDirectory(true);
+        return virtualFtpFile;
+    }
+
+    @Override
+    public String getAbsolutePath() {
+        return FtpFileSystemView.HOME_PATH + "/" + name;
+    }
+
+    @Override
+    public String getName() {
+        return name;
+    }
+
+    @Override
+    public boolean isHidden() {
+        return hidden;
+    }
+
+    @Override
+    public boolean isDirectory() {
+        return directory;
+    }
+
+    @Override
+    public boolean isFile() {
+        return !directory;
+    }
+
+    @Override
+    public boolean doesExist() {
+        return false;
+    }
+
+    @Override
+    public boolean isReadable() {
+        return true;
+    }
+
+    @Override
+    public boolean isWritable() {
+        return true;
+    }
+
+    @Override
+    public boolean isRemovable() {
+        return true;
+    }
+
+    @Override
+    public String getOwnerName() {
+        return ownerName;
+    }
+
+    @Override
+    public String getGroupName() {
+        return "root";
+    }
+
+    @Override
+    public int getLinkCount() {
+        return 0;
+    }
+
+    @Override
+    public long getLastModified() {
+        if (lastModified == null) {
+            lastModified = System.currentTimeMillis();
+        }
+        return lastModified;
+    }
+
+    @Override
+    public boolean setLastModified(long time) {
+        lastModified = time;
+        return true;
+    }
+
+    @Override
+    public long getSize() {
+        return size;
+    }
+
+    @Override
+    public Object getPhysicalFile() {
+        System.err.println("getPhysicalFile");
+        return null;
+    }
+
+    @Override
+    public boolean mkdir() {
+        return true;
+    }
+
+    @Override
+    public boolean delete() {
+        return true;
+    }
+
+    @Override
+    public boolean move(FtpFile destination) {
+        this.name = destination.getName();
+        return true;
+    }
+
+    @Override
+    public List<? extends FtpFile> listFiles() {
+        return Collections.emptyList();
+    }
+
+    @Override
+    public OutputStream createOutputStream(long offset) throws IOException {
+        System.out.println("createOutputStream++++");
+        return outputStream;
+    }
+
+    @Override
+    public InputStream createInputStream(long offset) throws IOException {
+        System.out.println("createInputStream----");
+        return null;
+    }
+}

src/main/java/com/genersoft/iot/vmp/conf/security/WebSecurityConfig.java

@@ -100,6 +100,7 @@ public class WebSecurityConfig extends WebSecurityConfigurerAdapter {
         defaultExcludes.add("/index/hook/**");
         defaultExcludes.add("/api/device/query/snap/**");
         defaultExcludes.add("/index/hook/abl/**");
+        defaultExcludes.add("/api/jt1078/playback/download");