feat: Support Rename and Edit Staitc Files (#573)

src/main/java/run/halo/app/controller/admin/api/StaticStorageController.java

@@ -7,6 +7,7 @@
 import run.halo.app.service.StaticStorageService;
 
 import java.util.List;
+import java.util.Map;
 
 /**
  * Static storage controller.
@@ -49,4 +50,17 @@ public void upload(String basePath,
                        @RequestPart("file") MultipartFile file) {
         staticStorageService.upload(basePath, file);
     }
+
+    @PostMapping("rename")
+    @ApiOperation("Renames static file")
+    public void rename(String basePath,
+                       String newName) {
+        staticStorageService.rename(basePath, newName);
+    }
+
+    @PutMapping("save")
+    @ApiOperation("Save static file")
+    public void save(@RequestBody Map<String, String> data) {
+        staticStorageService.save(data.get("basePath"), data.get("content"));
+    }
 }

src/main/java/run/halo/app/service/StaticStorageService.java

@@ -50,4 +50,20 @@ public interface StaticStorageService {
      * @param file     file must not be null.
      */
     void upload(String basePath, @NonNull MultipartFile file);
+
+    /**
+     * Rename static file or folder.
+     *
+     * @param basePath base path must not be null
+     * @param newName new name must not be null
+     */
+    void rename(@NonNull String basePath, @NonNull String newName);
+
+    /**
+     * Save static file.
+     *
+     * @param basePath base path must not be null
+     * @param content saved content
+     */
+    void save(@NonNull String basePath, String content);
 }

src/main/java/run/halo/app/service/impl/StaticStorageServiceImpl.java

@@ -21,9 +21,8 @@
 
 import javax.activation.MimetypesFileTypeMap;
 import java.io.IOException;
-import java.nio.file.Files;
-import java.nio.file.Path;
-import java.nio.file.Paths;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.*;
 import java.util.LinkedList;
 import java.util.List;
 import java.util.stream.Stream;
@@ -167,6 +166,42 @@ public void upload(String basePath, MultipartFile file) {
         }
     }
 
+    @Override
+    public void rename(String basePath, String newName) {
+        Assert.notNull(basePath, "Base path must not be null");
+        Assert.notNull(newName, "New name must not be null");
+
+        Path pathToRename;
+
+        if (StringUtils.startsWith(newName, API_FOLDER_NAME)) {
+            throw new FileOperationException("重命名名称 " + newName + " 不合法");
+        }
+
+        pathToRename = Paths.get(staticDir.toString(), basePath);
+
+        try {
+            FileUtils.rename(pathToRename, newName);
+            onChange();
+        } catch (FileAlreadyExistsException e) {
+            throw new FileOperationException("该路径下名称 " + newName + " 已存在");
+        } catch (IOException e) {
+            throw new FileOperationException("重命名 " + pathToRename.toString() + " 失败");
+        }
+    }
+
+    @Override
+    public void save(String basePath, String content) {
+        Assert.notNull(basePath, "Base path must not be null");
+
+        Path path = Paths.get(staticDir.toString(), basePath);
+
+        try {
+            Files.write(path, content.getBytes(StandardCharsets.UTF_8));
+        } catch (IOException e) {
+            throw new ServiceException("保存内容失败 " + path, e);
+        }
+    }
+
     private void onChange() {
         eventPublisher.publishEvent(new StaticStorageChangedEvent(this, staticDir));
     }

src/main/java/run/halo/app/utils/FileUtils.java

@@ -77,6 +77,24 @@ public static void deleteFolder(@NonNull Path deletingPath) throws IOException {
         log.info("Deleted [{}] successfully", deletingPath);
     }
 
+    /**
+     * Renames file or folder.
+     *
+     * @param pathToRename file path to rename must not be null
+     * @param newName new name must not be null
+     */
+    public static void rename(@NonNull Path pathToRename, @NonNull String newName) throws IOException {
+        Assert.notNull(pathToRename, "File path to rename must not be null");
+        Assert.notNull(newName, "New name must not be null");
+
+        Path newPath = pathToRename.resolveSibling(newName);
+        log.info("Rename [{}] to [{}]", pathToRename.toString(), newPath.toString());
+
+        Files.move(pathToRename, newPath);
+
+        log.info("Rename [{}] successfully", pathToRename.toString());
+    }
+
     /**
      * Unzips content to the target path.
      *

src/test/java/run/halo/app/utils/FileUtilsTest.java

@@ -8,6 +8,7 @@
 
 import java.io.IOException;
 import java.io.RandomAccessFile;
+import java.nio.file.FileAlreadyExistsException;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
@@ -108,4 +109,90 @@ public void dbFileReadTest() throws IOException {
             log.debug("Buffer String: [{}]", bufString);
         }
     }
+
+    @Test
+    public void testRenameFile() throws IOException {
+        // Create a temp folder
+        Path tempDirectory = Files.createTempDirectory("halo-test");
+
+        Path testPath = tempDirectory.resolve("test/test");
+        Path filePath = tempDirectory.resolve("test/test/test.file");
+
+        // Create a temp file and folder
+        Files.createDirectories(testPath);
+        Files.createFile(filePath);
+
+        // Write content to the temp file
+        String content = "Test Content!\n";
+        Files.write(filePath, content.getBytes());
+
+        // Rename temp file
+        FileUtils.rename(filePath, "newName");
+        Path newPath = filePath.resolveSibling("newName");
+
+        Assert.assertFalse(Files.exists(filePath));
+        Assert.assertTrue(Files.isRegularFile(newPath));
+        Assert.assertEquals(new String(Files.readAllBytes(newPath)), content);
+
+        FileUtils.deleteFolder(tempDirectory);
+    }
+
+    @Test
+    public void testRenameFolder() throws IOException {
+        // Create a temp folder
+        Path tempDirectory = Files.createTempDirectory("halo-test");
+
+        Path testPath = tempDirectory.resolve("test/test");
+        Path filePath = tempDirectory.resolve("test/test.file");
+
+        // Create a temp file and folder
+        Files.createDirectories(testPath);
+        Files.createFile(filePath);
+
+        // Rename temp folder
+        FileUtils.rename(tempDirectory.resolve("test"), "newName");
+        Path newPath = tempDirectory.resolve("newName");
+
+        Assert.assertTrue(Files.isDirectory(newPath));
+        Assert.assertTrue(Files.isRegularFile(newPath.resolve("test.file")));
+
+        FileUtils.deleteFolder(tempDirectory);
+    }
+
+    @Test
+    public void testRenameRepeat() throws IOException {
+        // Create a temp folder
+        Path tempDirectory = Files.createTempDirectory("halo-test");
+
+        Path testPathOne = tempDirectory.resolve("test/testOne");
+        Path testPathTwo = tempDirectory.resolve("test/testTwo");
+        Path filePathOne = tempDirectory.resolve("test/testOne.file");
+        Path filePathTwo = tempDirectory.resolve("test/testTwo.file");
+
+        // Create temp files and folders
+        Files.createDirectories(testPathOne);
+        Files.createDirectories(testPathTwo);
+        Files.createFile(filePathOne);
+        Files.createFile(filePathTwo);
+
+        try {
+            FileUtils.rename(testPathOne, "testTwo");
+        } catch (Exception e) {
+            Assert.assertTrue(e instanceof FileAlreadyExistsException);
+        }
+
+        try {
+            FileUtils.rename(filePathOne, "testTwo.file");
+        } catch (Exception e) {
+            Assert.assertTrue(e instanceof FileAlreadyExistsException);
+        }
+
+        try {
+            FileUtils.rename(filePathOne, "testOne");
+        } catch (Exception e) {
+            Assert.assertTrue(e instanceof FileAlreadyExistsException);
+        }
+
+        FileUtils.deleteFolder(tempDirectory);
+    }
 }