Add tree multiselection, batch actions and Markdown export

internal/api/export_test.go

@@ -0,0 +1,73 @@
+package api
+
+import (
+	"archive/zip"
+	"bytes"
+	"io"
+	"os"
+	"path/filepath"
+	"strings"
+	"testing"
+)
+
+func TestExportMarkdownAndReferencedAssets(t *testing.T) {
+	f := newFixture(t)
+	f.write("资料/文章.md", "# 标题\n![图片](../assets/图.png)\n`![](不存在.png)`\n[其他笔记](未选中.md)")
+	if err := os.MkdirAll(filepath.Join(f.root, "assets"), 0755); err != nil {
+		t.Fatal(err)
+	}
+	if err := os.WriteFile(filepath.Join(f.root, "assets/图.png"), []byte("图片内容"), 0644); err != nil {
+		t.Fatal(err)
+	}
+	response := f.do(f.admin, "POST", "/api/v1/export", `{"paths":["资料/文章.md"]}`, nil)
+	data, _ := io.ReadAll(response.Body)
+	response.Body.Close()
+	if response.StatusCode != 200 {
+		t.Fatalf("导出失败：%d %s", response.StatusCode, data)
+	}
+	reader, err := zip.NewReader(bytes.NewReader(data), int64(len(data)))
+	if err != nil {
+		t.Fatal(err)
+	}
+	files := map[string]string{}
+	for _, file := range reader.File {
+		r, err := file.Open()
+		if err != nil {
+			t.Fatal(err)
+		}
+		content, _ := io.ReadAll(r)
+		r.Close()
+		files[file.Name] = string(content)
+	}
+	if files["assets/图.png"] != "图片内容" || !strings.Contains(files["资料/文章.md"], "../assets/图.png") || len(files) != 2 {
+		t.Fatalf("附件或目录不符：%v", files)
+	}
+	f.write("单篇.md", "原始 Markdown")
+	response = f.do(f.admin, "POST", "/api/v1/export", `{"paths":["单篇.md"]}`, nil)
+	defer response.Body.Close()
+	data, _ = io.ReadAll(response.Body)
+	if response.StatusCode != 200 || string(data) != "原始 Markdown" || !strings.HasPrefix(response.Header.Get("Content-Type"), "text/markdown") {
+		t.Fatalf("单篇导出不符：%d %s", response.StatusCode, data)
+	}
+}
+
+func TestExportRejectsMissingAssetsAndSymlinks(t *testing.T) {
+	f := newFixture(t)
+	f.write("文章.md", "![](missing.png)")
+	for _, body := range []string{`{"paths":["文章.md"]}`, `{"paths":[".rawmark/app.db"]}`, `{"paths":[]}`} {
+		response := f.do(f.admin, "POST", "/api/v1/export", body, nil)
+		response.Body.Close()
+		if response.StatusCode == 200 {
+			t.Fatalf("不应成功：%s", body)
+		}
+	}
+	f.write("真实.md", "内容")
+	if err := os.Symlink(filepath.Join(f.root, "真实.md"), filepath.Join(f.root, "别名.md")); err != nil {
+		t.Fatal(err)
+	}
+	response := f.do(f.admin, "POST", "/api/v1/export", `{"paths":["别名.md"]}`, nil)
+	response.Body.Close()
+	if response.StatusCode != 400 {
+		t.Fatalf("应拒绝符号链接：%d", response.StatusCode)
+	}
+}