531 lines
18 KiB
TypeScript
531 lines
18 KiB
TypeScript
/**
|
|
* File Management API Tests
|
|
*
|
|
* Tests the /api/workspaces/:workspaceId/files/* endpoints
|
|
* for virtual storage mode (filesystem-backed).
|
|
*/
|
|
import { describe, it, expect, beforeAll, afterAll } from 'vitest';
|
|
import fs from 'node:fs';
|
|
import path from 'node:path';
|
|
import os from 'node:os';
|
|
import { MindDB, SessionStore, FrameStore } from '@waggle/core';
|
|
import { buildLocalServer } from '../../src/local/index.js';
|
|
import type { FastifyInstance } from 'fastify';
|
|
import type { FileEntry } from '../../src/local/storage/types.js';
|
|
import { injectWithAuth } from '../test-utils.js';
|
|
|
|
describe('File Management API', () => {
|
|
let server: FastifyInstance;
|
|
let tmpDir: string;
|
|
let workspaceId: string;
|
|
let targetWorkspaceId: string;
|
|
|
|
beforeAll(async () => {
|
|
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'waggle-files-'));
|
|
|
|
// Create personal.mind (required by buildLocalServer)
|
|
const personalPath = path.join(tmpDir, 'personal.mind');
|
|
const mind = new MindDB(personalPath);
|
|
const sessions = new SessionStore(mind);
|
|
const frames = new FrameStore(mind);
|
|
const s1 = sessions.create('files-test');
|
|
frames.createIFrame(s1.gop_id, 'File management test', 'normal');
|
|
mind.close();
|
|
|
|
server = await buildLocalServer({ dataDir: tmpDir });
|
|
|
|
// Create a test workspace
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: '/api/workspaces',
|
|
payload: { name: 'File Test Workspace', group: 'Test' },
|
|
});
|
|
workspaceId = res.json().id;
|
|
|
|
const targetRes = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: '/api/workspaces',
|
|
payload: { name: 'Cross-workspace Target', group: 'Test' },
|
|
});
|
|
targetWorkspaceId = targetRes.json().id;
|
|
});
|
|
|
|
afterAll(async () => {
|
|
await server.close();
|
|
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
});
|
|
|
|
const prefix = () => `/api/workspaces/${workspaceId}/files`;
|
|
|
|
// ── List ─────────────────────────────────────────────────────
|
|
|
|
describe('GET /list', () => {
|
|
it('lists root directory with standard dirs', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/list?path=/`,
|
|
});
|
|
expect(res.statusCode).toBe(200);
|
|
const entries = res.json();
|
|
expect(Array.isArray(entries)).toBe(true);
|
|
|
|
const names = entries.map((e: FileEntry) => e.name);
|
|
expect(names).toContain('attachments');
|
|
expect(names).toContain('exports');
|
|
expect(names).toContain('notes');
|
|
});
|
|
|
|
it('returns empty array for non-existent subdirectory', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/list?path=/nonexistent`,
|
|
});
|
|
expect(res.statusCode).toBe(200);
|
|
expect(res.json()).toEqual([]);
|
|
});
|
|
|
|
it('rejects path traversal', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/list?path=/../../../etc`,
|
|
});
|
|
expect(res.statusCode).toBe(400);
|
|
expect(res.json().error).toContain('Invalid path');
|
|
});
|
|
});
|
|
|
|
// ── Upload ───────────────────────────────────────────────────
|
|
|
|
describe('POST /upload', () => {
|
|
it('uploads a file via JSON base64', async () => {
|
|
const content = 'Hello, Waggle!';
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/upload`,
|
|
payload: {
|
|
path: '/attachments',
|
|
name: 'hello.txt',
|
|
data: Buffer.from(content).toString('base64'),
|
|
},
|
|
});
|
|
expect(res.statusCode).toBe(201);
|
|
const entry = res.json();
|
|
expect(entry.name).toBe('hello.txt');
|
|
expect(entry.path).toBe('/attachments/hello.txt');
|
|
expect(entry.type).toBe('file');
|
|
expect(entry.size).toBe(content.length);
|
|
expect(entry.mimeType).toBe('text/plain');
|
|
});
|
|
|
|
it('uploaded file appears in list', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/list?path=/attachments`,
|
|
});
|
|
const entries = res.json();
|
|
const names = entries.map((e: FileEntry) => e.name);
|
|
expect(names).toContain('hello.txt');
|
|
});
|
|
|
|
it('accepts empty file upload (0 bytes)', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/upload`,
|
|
payload: { path: '/', name: 'empty.bin', data: '' },
|
|
});
|
|
expect(res.statusCode).toBe(201);
|
|
expect(res.json().size).toBe(0);
|
|
});
|
|
});
|
|
|
|
// ── Download ─────────────────────────────────────────────────
|
|
|
|
describe('GET /download', () => {
|
|
it('downloads an uploaded file', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/download?path=/attachments/hello.txt`,
|
|
});
|
|
expect(res.statusCode).toBe(200);
|
|
expect(res.body).toBe('Hello, Waggle!');
|
|
expect(res.headers['content-type']).toContain('text/plain');
|
|
});
|
|
|
|
it('returns 404 for non-existent file', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/download?path=/nonexistent.txt`,
|
|
});
|
|
expect(res.statusCode).toBe(404);
|
|
});
|
|
|
|
it('returns 400 when path is missing', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/download`,
|
|
});
|
|
expect(res.statusCode).toBe(400);
|
|
});
|
|
});
|
|
|
|
// ── Mkdir ────────────────────────────────────────────────────
|
|
|
|
describe('POST /mkdir', () => {
|
|
it('creates a new directory', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/mkdir`,
|
|
payload: { path: '/attachments/screenshots' },
|
|
});
|
|
expect(res.statusCode).toBe(201);
|
|
const entry = res.json();
|
|
expect(entry.name).toBe('screenshots');
|
|
expect(entry.path).toBe('/attachments/screenshots');
|
|
expect(entry.type).toBe('directory');
|
|
});
|
|
|
|
it('created directory appears in list', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/list?path=/attachments`,
|
|
});
|
|
const names = res.json().map((e: FileEntry) => e.name);
|
|
expect(names).toContain('screenshots');
|
|
});
|
|
|
|
it('returns 400 when path is missing', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/mkdir`,
|
|
payload: {},
|
|
});
|
|
expect(res.statusCode).toBe(400);
|
|
});
|
|
});
|
|
|
|
// ── Move / Rename ────────────────────────────────────────────
|
|
|
|
describe('POST /move', () => {
|
|
it('moves a file to a different directory', async () => {
|
|
// First upload a file
|
|
await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/upload`,
|
|
payload: {
|
|
path: '/',
|
|
name: 'moveme.txt',
|
|
data: Buffer.from('move this').toString('base64'),
|
|
},
|
|
});
|
|
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/move`,
|
|
payload: { from: '/moveme.txt', to: '/exports/moved.txt' },
|
|
});
|
|
expect(res.statusCode).toBe(200);
|
|
const entry = res.json();
|
|
expect(entry.name).toBe('moved.txt');
|
|
expect(entry.path).toBe('/exports/moved.txt');
|
|
|
|
// Original should be gone
|
|
const origList = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/list?path=/`,
|
|
});
|
|
const rootNames = origList.json().map((e: FileEntry) => e.name);
|
|
expect(rootNames).not.toContain('moveme.txt');
|
|
});
|
|
|
|
it('renames a file within the same directory', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/move`,
|
|
payload: { from: '/exports/moved.txt', to: '/exports/renamed.txt' },
|
|
});
|
|
expect(res.statusCode).toBe(200);
|
|
expect(res.json().name).toBe('renamed.txt');
|
|
});
|
|
|
|
it('returns 400 for non-existent source', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/move`,
|
|
payload: { from: '/ghost.txt', to: '/exports/ghost.txt' },
|
|
});
|
|
expect(res.statusCode).toBe(400);
|
|
});
|
|
});
|
|
|
|
// ── Copy ─────────────────────────────────────────────────────
|
|
|
|
describe('POST /copy', () => {
|
|
it('copies a file', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/copy`,
|
|
payload: { from: '/exports/renamed.txt', to: '/notes/copy.txt' },
|
|
});
|
|
expect(res.statusCode).toBe(200);
|
|
const entry = res.json();
|
|
expect(entry.name).toBe('copy.txt');
|
|
expect(entry.path).toBe('/notes/copy.txt');
|
|
|
|
// Original should still exist
|
|
const origRes = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/download?path=/exports/renamed.txt`,
|
|
});
|
|
expect(origRes.statusCode).toBe(200);
|
|
});
|
|
|
|
it('copies a file from another workspace without reading from the target store', async () => {
|
|
const source = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/upload`,
|
|
payload: {
|
|
path: '/exports',
|
|
name: 'cross-workspace.txt',
|
|
data: Buffer.from('kept in the source workspace').toString('base64'),
|
|
},
|
|
});
|
|
expect(source.statusCode).toBe(201);
|
|
|
|
const copied = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `/api/workspaces/${targetWorkspaceId}/files/copy`,
|
|
payload: {
|
|
sourceWorkspaceId: workspaceId,
|
|
from: '/exports/cross-workspace.txt',
|
|
to: '/notes/cross-workspace.txt',
|
|
},
|
|
});
|
|
expect(copied.statusCode).toBe(200);
|
|
expect(copied.json()).toMatchObject({ name: 'cross-workspace.txt', path: '/notes/cross-workspace.txt', type: 'file' });
|
|
|
|
const targetDownload = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `/api/workspaces/${targetWorkspaceId}/files/download?path=/notes/cross-workspace.txt`,
|
|
});
|
|
expect(targetDownload.statusCode).toBe(200);
|
|
expect(targetDownload.body).toBe('kept in the source workspace');
|
|
|
|
const sourceDownload = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/download?path=/exports/cross-workspace.txt`,
|
|
});
|
|
expect(sourceDownload.statusCode).toBe(200);
|
|
});
|
|
|
|
it('rejects cross-workspace directory copies explicitly', async () => {
|
|
const response = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `/api/workspaces/${targetWorkspaceId}/files/copy`,
|
|
payload: {
|
|
sourceWorkspaceId: workspaceId,
|
|
from: '/attachments',
|
|
to: '/attachments',
|
|
},
|
|
});
|
|
expect(response.statusCode).toBe(400);
|
|
expect(response.json().error).toContain('files only');
|
|
});
|
|
});
|
|
|
|
// ── Delete ───────────────────────────────────────────────────
|
|
|
|
describe('POST /delete', () => {
|
|
it('deletes a file', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/delete`,
|
|
payload: { path: '/notes/copy.txt' },
|
|
});
|
|
expect(res.statusCode).toBe(204);
|
|
|
|
// Confirm it's gone
|
|
const dl = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/download?path=/notes/copy.txt`,
|
|
});
|
|
expect(dl.statusCode).toBe(404);
|
|
});
|
|
|
|
it('deletes a directory recursively', async () => {
|
|
// Upload a file inside the screenshots dir
|
|
await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/upload`,
|
|
payload: {
|
|
path: '/attachments/screenshots',
|
|
name: 'screen1.png',
|
|
data: Buffer.from('fakepng').toString('base64'),
|
|
},
|
|
});
|
|
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/delete`,
|
|
payload: { path: '/attachments/screenshots' },
|
|
});
|
|
expect(res.statusCode).toBe(204);
|
|
|
|
// Confirm directory is gone
|
|
const list = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/list?path=/attachments`,
|
|
});
|
|
const names = list.json().map((e: FileEntry) => e.name);
|
|
expect(names).not.toContain('screenshots');
|
|
});
|
|
|
|
it('returns 400 when path is missing', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/delete`,
|
|
payload: {},
|
|
});
|
|
expect(res.statusCode).toBe(400);
|
|
});
|
|
});
|
|
|
|
// ── Security ─────────────────────────────────────────────────
|
|
|
|
describe('Path traversal prevention', () => {
|
|
it('rejects .. in upload path', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/upload`,
|
|
payload: {
|
|
path: '/../../../tmp',
|
|
name: 'evil.txt',
|
|
data: Buffer.from('pwned').toString('base64'),
|
|
},
|
|
});
|
|
expect(res.statusCode).toBe(400);
|
|
});
|
|
|
|
it('rejects .. in download path', async () => {
|
|
// Fastify URL-normalizes the path, so we encode the dots
|
|
const res = await injectWithAuth(server, {
|
|
method: 'GET',
|
|
url: `${prefix()}/download?path=/../../../etc/passwd`,
|
|
});
|
|
// Fastify strips .. during URL parsing → becomes /etc/passwd → 404 (not found)
|
|
// Either 400 (safePath catches it) or 404 (file doesn't exist) is acceptable
|
|
expect([400, 404]).toContain(res.statusCode);
|
|
});
|
|
|
|
it('rejects .. in delete path', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/delete`,
|
|
payload: { path: '/../../../tmp' },
|
|
});
|
|
expect(res.statusCode).toBe(400);
|
|
});
|
|
|
|
it('rejects .. in move source', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/move`,
|
|
payload: { from: '/../../../etc/passwd', to: '/stolen.txt' },
|
|
});
|
|
expect(res.statusCode).toBe(400);
|
|
});
|
|
|
|
it('rejects .. in mkdir', async () => {
|
|
const res = await injectWithAuth(server, {
|
|
method: 'POST',
|
|
url: `${prefix()}/mkdir`,
|
|
payload: { path: '/../../../tmp/evil' },
|
|
});
|
|
expect(res.statusCode).toBe(400);
|
|
});
|
|
});
|
|
});
|
|
|
|
// ── Storage Provider Unit Tests ────────────────────────────────
|
|
|
|
describe('FsStorageProvider', () => {
|
|
let tmpRoot: string;
|
|
|
|
beforeAll(() => {
|
|
tmpRoot = fs.mkdtempSync(path.join(os.tmpdir(), 'waggle-fsprovider-'));
|
|
});
|
|
|
|
afterAll(() => {
|
|
fs.rmSync(tmpRoot, { recursive: true, force: true });
|
|
});
|
|
|
|
it('ensureStructure creates standard directories', async () => {
|
|
const { FsStorageProvider } = await import('../../src/local/storage/fs-provider.js');
|
|
const provider = new FsStorageProvider(tmpRoot);
|
|
provider.ensureStructure();
|
|
|
|
expect(fs.existsSync(path.join(tmpRoot, 'attachments'))).toBe(true);
|
|
expect(fs.existsSync(path.join(tmpRoot, 'exports'))).toBe(true);
|
|
expect(fs.existsSync(path.join(tmpRoot, 'notes'))).toBe(true);
|
|
});
|
|
|
|
it('write + read roundtrip', async () => {
|
|
const { FsStorageProvider } = await import('../../src/local/storage/fs-provider.js');
|
|
const provider = new FsStorageProvider(tmpRoot);
|
|
|
|
const entry = await provider.write('/test.txt', Buffer.from('hello'));
|
|
expect(entry.name).toBe('test.txt');
|
|
expect(entry.size).toBe(5);
|
|
|
|
const data = await provider.read('/test.txt');
|
|
expect(data.toString()).toBe('hello');
|
|
});
|
|
|
|
it('exists returns correct values', async () => {
|
|
const { FsStorageProvider } = await import('../../src/local/storage/fs-provider.js');
|
|
const provider = new FsStorageProvider(tmpRoot);
|
|
|
|
expect(await provider.exists('/test.txt')).toBe(true);
|
|
expect(await provider.exists('/nope.txt')).toBe(false);
|
|
});
|
|
|
|
it('list returns sorted entries (dirs first)', async () => {
|
|
const { FsStorageProvider } = await import('../../src/local/storage/fs-provider.js');
|
|
const provider = new FsStorageProvider(tmpRoot);
|
|
|
|
const entries = await provider.list('/');
|
|
expect(entries.length).toBeGreaterThan(0);
|
|
// First entries should be directories
|
|
const firstDir = entries.findIndex(e => e.type === 'directory');
|
|
const firstFile = entries.findIndex(e => e.type === 'file');
|
|
if (firstDir >= 0 && firstFile >= 0) {
|
|
expect(firstDir).toBeLessThan(firstFile);
|
|
}
|
|
});
|
|
});
|
|
|
|
describe('Path Security', () => {
|
|
it('safePath rejects traversal', async () => {
|
|
const { safePath } = await import('../../src/local/storage/security.js');
|
|
expect(() => safePath('/root', '../etc/passwd')).toThrow('Invalid path');
|
|
expect(() => safePath('/root', '../../etc')).toThrow('Invalid path');
|
|
expect(() => safePath('/root', './../../etc')).toThrow('Invalid path');
|
|
});
|
|
|
|
it('safePath allows valid paths', async () => {
|
|
const { safePath } = await import('../../src/local/storage/security.js');
|
|
const tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'waggle-safe-'));
|
|
fs.mkdirSync(path.join(tmpDir, 'subdir'), { recursive: true });
|
|
|
|
const result = safePath(tmpDir, 'subdir');
|
|
expect(result).toBe(path.resolve(tmpDir, 'subdir'));
|
|
|
|
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
});
|
|
|
|
it('toRelativePath converts correctly', async () => {
|
|
const { toRelativePath } = await import('../../src/local/storage/security.js');
|
|
const result = toRelativePath('/root/data', '/root/data/attachments/file.pdf');
|
|
expect(result).toBe('/attachments/file.pdf');
|
|
});
|
|
});
|