实现快传，完善快传和网盘的功能，实现文件的互传等一系列功能

2026-03-20 14:16:18 +08:00
parent 944ab6dbf8
commit 43358e29d7
109 changed files with 5237 additions and 2465 deletions
--- a/backend/src/test/java/com/yoyuzh/files/FileServiceTest.java
+++ b/backend/src/test/java/com/yoyuzh/files/FileServiceTest.java
@@ -27,6 +27,7 @@ import java.util.zip.ZipInputStream;

 import static org.assertj.core.api.Assertions.assertThat;
 import static org.assertj.core.api.Assertions.assertThatThrownBy;
+import static org.mockito.AdditionalMatchers.aryEq;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.never;
@@ -43,13 +44,16 @@ class FileServiceTest {
    @Mock
    private FileContentStorage fileContentStorage;

+    @Mock
+    private FileShareLinkRepository fileShareLinkRepository;
+
    private FileService fileService;

    @BeforeEach
    void setUp() {
        FileStorageProperties properties = new FileStorageProperties();
        properties.setMaxFileSize(500L * 1024 * 1024);
-        fileService = new FileService(storedFileRepository, fileContentStorage, properties);
+        fileService = new FileService(storedFileRepository, fileContentStorage, fileShareLinkRepository, properties);
    }

    @Test
@@ -167,6 +171,140 @@ class FileServiceTest {
        verify(fileContentStorage).renameDirectory(7L, "/docs/archive", "/docs/renamed-archive", List.of(childFile));
    }

+    @Test
+    void shouldMoveFileToAnotherDirectory() {
+        User user = createUser(7L);
+        StoredFile file = createFile(10L, user, "/docs", "notes.txt");
+        StoredFile targetDirectory = createDirectory(11L, user, "/", "下载");
+        when(storedFileRepository.findById(10L)).thenReturn(Optional.of(file));
+        when(storedFileRepository.findByUserIdAndPathAndFilename(7L, "/", "下载")).thenReturn(Optional.of(targetDirectory));
+        when(storedFileRepository.existsByUserIdAndPathAndFilename(7L, "/下载", "notes.txt")).thenReturn(false);
+        when(storedFileRepository.save(any(StoredFile.class))).thenAnswer(invocation -> invocation.getArgument(0));
+
+        FileMetadataResponse response = fileService.move(user, 10L, "/下载");
+
+        assertThat(response.path()).isEqualTo("/下载");
+        assertThat(file.getPath()).isEqualTo("/下载");
+        verify(fileContentStorage).moveFile(7L, "/docs", "/下载", "notes.txt");
+    }
+
+    @Test
+    void shouldMoveDirectoryAndUpdateDescendantPaths() {
+        User user = createUser(7L);
+        StoredFile directory = createDirectory(10L, user, "/docs", "archive");
+        StoredFile targetDirectory = createDirectory(11L, user, "/", "图片");
+        StoredFile childFile = createFile(12L, user, "/docs/archive", "nested.txt");
+        when(storedFileRepository.findById(10L)).thenReturn(Optional.of(directory));
+        when(storedFileRepository.findByUserIdAndPathAndFilename(7L, "/", "图片")).thenReturn(Optional.of(targetDirectory));
+        when(storedFileRepository.existsByUserIdAndPathAndFilename(7L, "/图片", "archive")).thenReturn(false);
+        when(storedFileRepository.findByUserIdAndPathEqualsOrDescendant(7L, "/docs/archive")).thenReturn(List.of(childFile));
+        when(storedFileRepository.save(any(StoredFile.class))).thenAnswer(invocation -> invocation.getArgument(0));
+        when(storedFileRepository.saveAll(List.of(childFile))).thenReturn(List.of(childFile));
+
+        FileMetadataResponse response = fileService.move(user, 10L, "/图片");
+
+        assertThat(response.path()).isEqualTo("/图片/archive");
+        assertThat(directory.getPath()).isEqualTo("/图片");
+        assertThat(childFile.getPath()).isEqualTo("/图片/archive");
+        verify(fileContentStorage).renameDirectory(7L, "/docs/archive", "/图片/archive", List.of(childFile));
+    }
+
+    @Test
+    void shouldRejectMovingDirectoryIntoItsOwnDescendant() {
+        User user = createUser(7L);
+        StoredFile directory = createDirectory(10L, user, "/docs", "archive");
+        StoredFile docsDirectory = createDirectory(11L, user, "/", "docs");
+        StoredFile archiveDirectory = createDirectory(12L, user, "/docs", "archive");
+        StoredFile descendantDirectory = createDirectory(13L, user, "/docs/archive", "nested");
+        when(storedFileRepository.findById(10L)).thenReturn(Optional.of(directory));
+        when(storedFileRepository.findByUserIdAndPathAndFilename(7L, "/", "docs"))
+                .thenReturn(Optional.of(docsDirectory));
+        when(storedFileRepository.findByUserIdAndPathAndFilename(7L, "/docs", "archive"))
+                .thenReturn(Optional.of(archiveDirectory));
+        when(storedFileRepository.findByUserIdAndPathAndFilename(7L, "/docs/archive", "nested"))
+                .thenReturn(Optional.of(descendantDirectory));
+
+        assertThatThrownBy(() -> fileService.move(user, 10L, "/docs/archive/nested"))
+                .isInstanceOf(BusinessException.class)
+                .hasMessageContaining("不能移动到当前目录或其子目录");
+    }
+
+    @Test
+    void shouldCopyFileToAnotherDirectory() {
+        User user = createUser(7L);
+        StoredFile file = createFile(10L, user, "/docs", "notes.txt");
+        StoredFile targetDirectory = createDirectory(11L, user, "/", "下载");
+        when(storedFileRepository.findById(10L)).thenReturn(Optional.of(file));
+        when(storedFileRepository.findByUserIdAndPathAndFilename(7L, "/", "下载")).thenReturn(Optional.of(targetDirectory));
+        when(storedFileRepository.existsByUserIdAndPathAndFilename(7L, "/下载", "notes.txt")).thenReturn(false);
+        when(storedFileRepository.save(any(StoredFile.class))).thenAnswer(invocation -> {
+            StoredFile storedFile = invocation.getArgument(0);
+            if (storedFile.getId() == null) {
+                storedFile.setId(20L);
+            }
+            return storedFile;
+        });
+
+        FileMetadataResponse response = fileService.copy(user, 10L, "/下载");
+
+        assertThat(response.id()).isEqualTo(20L);
+        assertThat(response.path()).isEqualTo("/下载");
+        verify(fileContentStorage).copyFile(7L, "/docs", "/下载", "notes.txt");
+    }
+
+    @Test
+    void shouldCopyDirectoryAndDescendants() {
+        User user = createUser(7L);
+        StoredFile directory = createDirectory(10L, user, "/docs", "archive");
+        StoredFile targetDirectory = createDirectory(11L, user, "/", "图片");
+        StoredFile childDirectory = createDirectory(12L, user, "/docs/archive", "nested");
+        StoredFile childFile = createFile(13L, user, "/docs/archive", "notes.txt");
+        StoredFile nestedFile = createFile(14L, user, "/docs/archive/nested", "todo.txt");
+        when(storedFileRepository.findById(10L)).thenReturn(Optional.of(directory));
+        when(storedFileRepository.findByUserIdAndPathAndFilename(7L, "/", "图片")).thenReturn(Optional.of(targetDirectory));
+        when(storedFileRepository.existsByUserIdAndPathAndFilename(7L, "/图片", "archive")).thenReturn(false);
+        when(storedFileRepository.findByUserIdAndPathEqualsOrDescendant(7L, "/docs/archive"))
+                .thenReturn(List.of(childDirectory, childFile, nestedFile));
+        when(storedFileRepository.existsByUserIdAndPathAndFilename(7L, "/图片/archive", "nested")).thenReturn(false);
+        when(storedFileRepository.existsByUserIdAndPathAndFilename(7L, "/图片/archive", "notes.txt")).thenReturn(false);
+        when(storedFileRepository.existsByUserIdAndPathAndFilename(7L, "/图片/archive/nested", "todo.txt")).thenReturn(false);
+        when(storedFileRepository.save(any(StoredFile.class))).thenAnswer(invocation -> {
+            StoredFile storedFile = invocation.getArgument(0);
+            if (storedFile.getId() == null) {
+                storedFile.setId(100L + storedFile.getFilename().length());
+            }
+            return storedFile;
+        });
+
+        FileMetadataResponse response = fileService.copy(user, 10L, "/图片");
+
+        assertThat(response.path()).isEqualTo("/图片/archive");
+        verify(fileContentStorage).ensureDirectory(7L, "/图片/archive");
+        verify(fileContentStorage).ensureDirectory(7L, "/图片/archive/nested");
+        verify(fileContentStorage).copyFile(7L, "/docs/archive", "/图片/archive", "notes.txt");
+        verify(fileContentStorage).copyFile(7L, "/docs/archive/nested", "/图片/archive/nested", "todo.txt");
+    }
+
+    @Test
+    void shouldRejectCopyingDirectoryIntoItsOwnDescendant() {
+        User user = createUser(7L);
+        StoredFile directory = createDirectory(10L, user, "/docs", "archive");
+        StoredFile docsDirectory = createDirectory(11L, user, "/", "docs");
+        StoredFile archiveDirectory = createDirectory(12L, user, "/docs", "archive");
+        StoredFile descendantDirectory = createDirectory(13L, user, "/docs/archive", "nested");
+        when(storedFileRepository.findById(10L)).thenReturn(Optional.of(directory));
+        when(storedFileRepository.findByUserIdAndPathAndFilename(7L, "/", "docs"))
+                .thenReturn(Optional.of(docsDirectory));
+        when(storedFileRepository.findByUserIdAndPathAndFilename(7L, "/docs", "archive"))
+                .thenReturn(Optional.of(archiveDirectory));
+        when(storedFileRepository.findByUserIdAndPathAndFilename(7L, "/docs/archive", "nested"))
+                .thenReturn(Optional.of(descendantDirectory));
+
+        assertThatThrownBy(() -> fileService.copy(user, 10L, "/docs/archive/nested"))
+                .isInstanceOf(BusinessException.class)
+                .hasMessageContaining("不能复制到当前目录或其子目录");
+    }
+
    @Test
    void shouldRejectDeletingOtherUsersFile() {
        User owner = createUser(1L);
@@ -293,6 +431,60 @@ class FileServiceTest {
        verify(fileContentStorage).readFile(7L, "/docs/archive/nested", "todo.txt");
    }

+    @Test
+    void shouldCreateShareLinkForOwnedFile() {
+        User user = createUser(7L);
+        StoredFile file = createFile(22L, user, "/docs", "notes.txt");
+        when(storedFileRepository.findById(22L)).thenReturn(Optional.of(file));
+        when(fileShareLinkRepository.save(any(FileShareLink.class))).thenAnswer(invocation -> {
+            FileShareLink shareLink = invocation.getArgument(0);
+            shareLink.setId(100L);
+            shareLink.setToken("share-token-1");
+            return shareLink;
+        });
+
+        CreateFileShareLinkResponse response = fileService.createShareLink(user, 22L);
+
+        assertThat(response.token()).isEqualTo("share-token-1");
+        assertThat(response.filename()).isEqualTo("notes.txt");
+        verify(fileShareLinkRepository).save(any(FileShareLink.class));
+    }
+
+    @Test
+    void shouldImportSharedFileIntoRecipientWorkspace() {
+        User owner = createUser(7L);
+        User recipient = createUser(8L);
+        StoredFile sourceFile = createFile(22L, owner, "/docs", "notes.txt");
+        FileShareLink shareLink = new FileShareLink();
+        shareLink.setId(100L);
+        shareLink.setToken("share-token-1");
+        shareLink.setOwner(owner);
+        shareLink.setFile(sourceFile);
+        shareLink.setCreatedAt(LocalDateTime.now());
+        when(fileShareLinkRepository.findByToken("share-token-1")).thenReturn(Optional.of(shareLink));
+        when(storedFileRepository.existsByUserIdAndPathAndFilename(8L, "/下载", "notes.txt")).thenReturn(false);
+        when(storedFileRepository.save(any(StoredFile.class))).thenAnswer(invocation -> {
+            StoredFile file = invocation.getArgument(0);
+            file.setId(200L);
+            return file;
+        });
+        when(fileContentStorage.readFile(7L, "/docs", "notes.txt"))
+                .thenReturn("hello".getBytes(StandardCharsets.UTF_8));
+
+        FileMetadataResponse response = fileService.importSharedFile(recipient, "share-token-1", "/下载");
+
+        assertThat(response.id()).isEqualTo(200L);
+        assertThat(response.path()).isEqualTo("/下载");
+        assertThat(response.filename()).isEqualTo("notes.txt");
+        verify(fileContentStorage).storeImportedFile(
+                eq(8L),
+                eq("/下载"),
+                eq("notes.txt"),
+                eq(sourceFile.getContentType()),
+                aryEq("hello".getBytes(StandardCharsets.UTF_8))
+        );
+    }
+
    private User createUser(Long id) {
        User user = new User();
        user.setId(id);
--- a/backend/src/test/java/com/yoyuzh/files/FileShareControllerIntegrationTest.java
+++ b/backend/src/test/java/com/yoyuzh/files/FileShareControllerIntegrationTest.java
@@ -0,0 +1,228 @@
+package com.yoyuzh.files;
+
+import com.yoyuzh.PortalBackendApplication;
+import com.yoyuzh.auth.User;
+import com.yoyuzh.auth.UserRepository;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.http.MediaType;
+import org.springframework.test.web.servlet.MockMvc;
+
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.time.LocalDateTime;
+
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.patch;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
+import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.anonymous;
+import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.user;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+@SpringBootTest(
+        classes = PortalBackendApplication.class,
+        properties = {
+                "spring.datasource.url=jdbc:h2:mem:file_share_api_test;MODE=MySQL;DB_CLOSE_DELAY=-1",
+                "spring.datasource.driver-class-name=org.h2.Driver",
+                "spring.datasource.username=sa",
+                "spring.datasource.password=",
+                "spring.jpa.hibernate.ddl-auto=create-drop",
+                "app.jwt.secret=0123456789abcdef0123456789abcdef",
+                "app.storage.root-dir=./target/test-storage-file-share"
+        }
+)
+@AutoConfigureMockMvc
+class FileShareControllerIntegrationTest {
+
+    private static final Path STORAGE_ROOT = Path.of("./target/test-storage-file-share").toAbsolutePath().normalize();
+    private Long sharedFileId;
+
+    @Autowired
+    private MockMvc mockMvc;
+
+    @Autowired
+    private UserRepository userRepository;
+
+    @Autowired
+    private StoredFileRepository storedFileRepository;
+
+    @Autowired
+    private FileShareLinkRepository fileShareLinkRepository;
+
+    @BeforeEach
+    void setUp() throws Exception {
+        fileShareLinkRepository.deleteAll();
+        storedFileRepository.deleteAll();
+        userRepository.deleteAll();
+        if (Files.exists(STORAGE_ROOT)) {
+            try (var paths = Files.walk(STORAGE_ROOT)) {
+                paths.sorted((left, right) -> right.compareTo(left)).forEach(path -> {
+                    try {
+                        Files.deleteIfExists(path);
+                    } catch (Exception ex) {
+                        throw new RuntimeException(ex);
+                    }
+                });
+            }
+        }
+        Files.createDirectories(STORAGE_ROOT);
+
+        User owner = new User();
+        owner.setUsername("alice");
+        owner.setEmail("alice@example.com");
+        owner.setPhoneNumber("13800138000");
+        owner.setPasswordHash("encoded-password");
+        owner.setCreatedAt(LocalDateTime.now());
+        owner = userRepository.save(owner);
+
+        User recipient = new User();
+        recipient.setUsername("bob");
+        recipient.setEmail("bob@example.com");
+        recipient.setPhoneNumber("13800138001");
+        recipient.setPasswordHash("encoded-password");
+        recipient.setCreatedAt(LocalDateTime.now());
+        recipient = userRepository.save(recipient);
+
+        StoredFile file = new StoredFile();
+        file.setUser(owner);
+        file.setFilename("notes.txt");
+        file.setPath("/docs");
+        file.setStorageName("notes.txt");
+        file.setContentType("text/plain");
+        file.setSize(5L);
+        file.setDirectory(false);
+        sharedFileId = storedFileRepository.save(file).getId();
+
+        Path ownerDir = STORAGE_ROOT.resolve(owner.getId().toString()).resolve("docs");
+        Files.createDirectories(ownerDir);
+        Files.writeString(ownerDir.resolve("notes.txt"), "hello", StandardCharsets.UTF_8);
+    }
+
+    @Test
+    void shouldCreateInspectAndImportSharedFile() throws Exception {
+        String response = mockMvc.perform(post("/api/files/{fileId}/share-links", sharedFileId)
+                        .with(user("alice")))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value(0))
+                .andExpect(jsonPath("$.data.token").isNotEmpty())
+                .andExpect(jsonPath("$.data.filename").value("notes.txt"))
+                .andReturn()
+                .getResponse()
+                .getContentAsString();
+
+        String token = com.jayway.jsonpath.JsonPath.read(response, "$.data.token");
+
+        mockMvc.perform(get("/api/files/share-links/{token}", token))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.data.token").value(token))
+                .andExpect(jsonPath("$.data.filename").value("notes.txt"))
+                .andExpect(jsonPath("$.data.ownerUsername").value("alice"));
+
+        mockMvc.perform(post("/api/files/share-links/{token}/import", token)
+                        .with(anonymous())
+                        .contentType(MediaType.APPLICATION_JSON)
+                        .content("""
+                                {
+                                  "path": "/下载"
+                                }
+                                """))
+                .andExpect(status().isUnauthorized());
+
+        mockMvc.perform(post("/api/files/share-links/{token}/import", token)
+                        .with(user("bob"))
+                        .contentType(MediaType.APPLICATION_JSON)
+                        .content("""
+                                {
+                                  "path": "/下载"
+                                }
+                                """))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.data.filename").value("notes.txt"))
+                .andExpect(jsonPath("$.data.path").value("/下载"));
+
+        mockMvc.perform(get("/api/files/list")
+                        .with(user("bob"))
+                        .param("path", "/下载")
+                        .param("page", "0")
+                        .param("size", "20"))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.data.items[0].filename").value("notes.txt"));
+    }
+
+    @Test
+    void shouldMoveFileIntoAnotherDirectoryThroughApi() throws Exception {
+        User owner = userRepository.findByUsername("alice").orElseThrow();
+
+        StoredFile downloadDirectory = new StoredFile();
+        downloadDirectory.setUser(owner);
+        downloadDirectory.setFilename("下载");
+        downloadDirectory.setPath("/");
+        downloadDirectory.setStorageName("下载");
+        downloadDirectory.setContentType("directory");
+        downloadDirectory.setSize(0L);
+        downloadDirectory.setDirectory(true);
+        storedFileRepository.save(downloadDirectory);
+        Files.createDirectories(STORAGE_ROOT.resolve(owner.getId().toString()).resolve("下载"));
+
+        mockMvc.perform(patch("/api/files/{fileId}/move", sharedFileId)
+                        .with(user("alice"))
+                        .contentType(MediaType.APPLICATION_JSON)
+                        .content("""
+                                {
+                                  "path": "/下载"
+                                }
+                                """))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.data.filename").value("notes.txt"))
+                .andExpect(jsonPath("$.data.path").value("/下载"));
+
+        mockMvc.perform(get("/api/files/list")
+                        .with(user("alice"))
+                        .param("path", "/下载")
+                        .param("page", "0")
+                        .param("size", "20"))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.data.items[0].filename").value("notes.txt"));
+    }
+
+    @Test
+    void shouldCopyFileIntoAnotherDirectoryThroughApi() throws Exception {
+        User owner = userRepository.findByUsername("alice").orElseThrow();
+
+        StoredFile downloadDirectory = new StoredFile();
+        downloadDirectory.setUser(owner);
+        downloadDirectory.setFilename("下载");
+        downloadDirectory.setPath("/");
+        downloadDirectory.setStorageName("下载");
+        downloadDirectory.setContentType("directory");
+        downloadDirectory.setSize(0L);
+        downloadDirectory.setDirectory(true);
+        storedFileRepository.save(downloadDirectory);
+        Files.createDirectories(STORAGE_ROOT.resolve(owner.getId().toString()).resolve("下载"));
+
+        mockMvc.perform(post("/api/files/{fileId}/copy", sharedFileId)
+                        .with(user("alice"))
+                        .contentType(MediaType.APPLICATION_JSON)
+                        .content("""
+                                {
+                                  "path": "/下载"
+                                }
+                                """))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.data.filename").value("notes.txt"))
+                .andExpect(jsonPath("$.data.path").value("/下载"));
+
+        mockMvc.perform(get("/api/files/list")
+                        .with(user("alice"))
+                        .param("path", "/下载")
+                        .param("page", "0")
+                        .param("size", "20"))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.data.items[0].filename").value("notes.txt"));
+    }
+}