mirror of
https://github.com/kmc7468/arkvault.git
synced 2025-12-12 21:08:46 +00:00
/api/file/[id]/delete, /api/file/[id]/rename, /api/directory/[id]/delete, /api/directory/[id]/rename Endpoint 구현
This commit is contained in:
@@ -69,6 +69,47 @@ export const getDirectory = async (userId: number, directoryId: number) => {
|
||||
return res[0] ?? null;
|
||||
};
|
||||
|
||||
export const setDirectoryEncName = async (
|
||||
userId: number,
|
||||
directoryId: number,
|
||||
encName: string,
|
||||
encNameIv: string,
|
||||
) => {
|
||||
await db
|
||||
.update(directory)
|
||||
.set({ encName: { ciphertext: encName, iv: encNameIv } })
|
||||
.where(and(eq(directory.userId, userId), eq(directory.id, directoryId)))
|
||||
.execute();
|
||||
};
|
||||
|
||||
export const unregisterDirectory = async (userId: number, directoryId: number) => {
|
||||
return await db.transaction(async (tx) => {
|
||||
const getFilePaths = async (parentId: number) => {
|
||||
const files = await tx
|
||||
.select({ path: file.path })
|
||||
.from(file)
|
||||
.where(and(eq(file.userId, userId), eq(file.parentId, parentId)));
|
||||
return files.map(({ path }) => path);
|
||||
};
|
||||
const unregisterSubDirectoriesRecursively = async (directoryId: number): Promise<string[]> => {
|
||||
const subDirectories = await tx
|
||||
.select({ id: directory.id })
|
||||
.from(directory)
|
||||
.where(and(eq(directory.userId, userId), eq(directory.parentId, directoryId)));
|
||||
const subDirectoryFilePaths = await Promise.all(
|
||||
subDirectories.map(async ({ id }) => await unregisterSubDirectoriesRecursively(id)),
|
||||
);
|
||||
const filePaths = await getFilePaths(directoryId);
|
||||
|
||||
await tx.delete(file).where(eq(file.parentId, directoryId));
|
||||
await tx.delete(directory).where(eq(directory.id, directoryId));
|
||||
|
||||
return filePaths.concat(...subDirectoryFilePaths);
|
||||
};
|
||||
return await unregisterSubDirectoriesRecursively(directoryId);
|
||||
});
|
||||
};
|
||||
|
||||
export const registerNewFile = async (params: NewFileParams) => {
|
||||
await db.transaction(async (tx) => {
|
||||
const meks = await tx
|
||||
@@ -115,3 +156,25 @@ export const getFile = async (userId: number, fileId: number) => {
|
||||
.execute();
|
||||
return res[0] ?? null;
|
||||
};
|
||||
|
||||
export const setFileEncName = async (
|
||||
userId: number,
|
||||
fileId: number,
|
||||
encName: string,
|
||||
encNameIv: string,
|
||||
) => {
|
||||
await db
|
||||
.update(file)
|
||||
.set({ encName: { ciphertext: encName, iv: encNameIv } })
|
||||
.where(and(eq(file.userId, userId), eq(file.id, fileId)))
|
||||
.execute();
|
||||
};
|
||||
|
||||
export const unregisterFile = async (userId: number, fileId: number) => {
|
||||
const res = await db
|
||||
.delete(file)
|
||||
.where(and(eq(file.userId, userId), eq(file.id, fileId)))
|
||||
.returning({ path: file.path })
|
||||
.execute();
|
||||
return res[0]?.path ?? null;
|
||||
};
|
||||
|
||||
@@ -1,5 +1,11 @@
|
||||
import { z } from "zod";
|
||||
|
||||
export const directoryRenameRequest = z.object({
|
||||
name: z.string().base64().nonempty(),
|
||||
nameIv: z.string().base64().nonempty(),
|
||||
});
|
||||
export type DirectoryRenameRequest = z.infer<typeof directoryRenameRequest>;
|
||||
|
||||
export const directroyInfoResponse = z.object({
|
||||
metadata: z
|
||||
.object({
|
||||
|
||||
@@ -1,5 +1,11 @@
|
||||
import { z } from "zod";
|
||||
|
||||
export const fileRenameRequest = z.object({
|
||||
name: z.string().base64().nonempty(),
|
||||
nameIv: z.string().base64().nonempty(),
|
||||
});
|
||||
export type FileRenameRequest = z.infer<typeof fileRenameRequest>;
|
||||
|
||||
export const fileInfoResponse = z.object({
|
||||
createdAt: z.date(),
|
||||
mekVersion: z.number().int().positive(),
|
||||
|
||||
68
src/lib/server/services/directory.ts
Normal file
68
src/lib/server/services/directory.ts
Normal file
@@ -0,0 +1,68 @@
|
||||
import { error } from "@sveltejs/kit";
|
||||
import { unlink } from "fs/promises";
|
||||
import {
|
||||
getAllDirectoriesByParent,
|
||||
registerNewDirectory,
|
||||
getDirectory,
|
||||
setDirectoryEncName,
|
||||
unregisterDirectory,
|
||||
getAllFilesByParent,
|
||||
type NewDirectroyParams,
|
||||
} from "$lib/server/db/file";
|
||||
import { getActiveMekVersion } from "$lib/server/db/mek";
|
||||
|
||||
export const deleteDirectory = async (userId: number, directoryId: number) => {
|
||||
const directory = await getDirectory(userId, directoryId);
|
||||
if (!directory) {
|
||||
error(404, "Invalid directory id");
|
||||
}
|
||||
|
||||
const filePaths = await unregisterDirectory(userId, directoryId);
|
||||
filePaths.map((path) => unlink(path)); // Intended
|
||||
};
|
||||
|
||||
export const renameDirectory = async (
|
||||
userId: number,
|
||||
directoryId: number,
|
||||
newEncName: string,
|
||||
newEncNameIv: string,
|
||||
) => {
|
||||
const directory = await getDirectory(userId, directoryId);
|
||||
if (!directory) {
|
||||
error(404, "Invalid directory id");
|
||||
}
|
||||
|
||||
await setDirectoryEncName(userId, directoryId, newEncName, newEncNameIv);
|
||||
};
|
||||
|
||||
export const getDirectroyInformation = async (userId: number, directroyId: "root" | number) => {
|
||||
const directory = directroyId !== "root" ? await getDirectory(userId, directroyId) : undefined;
|
||||
if (directory === null) {
|
||||
error(404, "Invalid directory id");
|
||||
}
|
||||
|
||||
const directories = await getAllDirectoriesByParent(userId, directroyId);
|
||||
const files = await getAllFilesByParent(userId, directroyId);
|
||||
|
||||
return {
|
||||
metadata: directory && {
|
||||
createdAt: directory.createdAt,
|
||||
mekVersion: directory.mekVersion,
|
||||
encDek: directory.encDek,
|
||||
encName: directory.encName,
|
||||
},
|
||||
directories: directories.map(({ id }) => id),
|
||||
files: files.map(({ id }) => id),
|
||||
};
|
||||
};
|
||||
|
||||
export const createDirectory = async (params: NewDirectroyParams) => {
|
||||
const activeMekVersion = await getActiveMekVersion(params.userId);
|
||||
if (activeMekVersion === null) {
|
||||
error(500, "Invalid MEK version");
|
||||
} else if (activeMekVersion !== params.mekVersion) {
|
||||
error(400, "Invalid MEK version");
|
||||
}
|
||||
|
||||
await registerNewDirectory(params);
|
||||
};
|
||||
@@ -5,48 +5,27 @@ import { mkdir, stat, unlink } from "fs/promises";
|
||||
import { dirname } from "path";
|
||||
import { v4 as uuidv4 } from "uuid";
|
||||
import {
|
||||
getAllDirectoriesByParent,
|
||||
registerNewDirectory,
|
||||
getDirectory,
|
||||
registerNewFile,
|
||||
getAllFilesByParent,
|
||||
getFile,
|
||||
type NewDirectroyParams,
|
||||
setFileEncName,
|
||||
unregisterFile,
|
||||
type NewFileParams,
|
||||
} from "$lib/server/db/file";
|
||||
import { getActiveMekVersion } from "$lib/server/db/mek";
|
||||
import env from "$lib/server/loadenv";
|
||||
|
||||
export const getDirectroyInformation = async (userId: number, directroyId: "root" | number) => {
|
||||
const directory = directroyId !== "root" ? await getDirectory(userId, directroyId) : undefined;
|
||||
if (directory === null) {
|
||||
error(404, "Invalid directory id");
|
||||
export const deleteFile = async (userId: number, fileId: number) => {
|
||||
const file = await getFile(userId, fileId);
|
||||
if (!file) {
|
||||
error(404, "Invalid file id");
|
||||
}
|
||||
|
||||
const directories = await getAllDirectoriesByParent(userId, directroyId);
|
||||
const files = await getAllFilesByParent(userId, directroyId);
|
||||
|
||||
return {
|
||||
metadata: directory && {
|
||||
createdAt: directory.createdAt,
|
||||
mekVersion: directory.mekVersion,
|
||||
encDek: directory.encDek,
|
||||
encName: directory.encName,
|
||||
},
|
||||
directories: directories.map(({ id }) => id),
|
||||
files: files.map(({ id }) => id),
|
||||
};
|
||||
};
|
||||
|
||||
export const createDirectory = async (params: NewDirectroyParams) => {
|
||||
const activeMekVersion = await getActiveMekVersion(params.userId);
|
||||
if (activeMekVersion === null) {
|
||||
error(500, "Invalid MEK version");
|
||||
} else if (activeMekVersion !== params.mekVersion) {
|
||||
error(400, "Invalid MEK version");
|
||||
const path = await unregisterFile(userId, fileId);
|
||||
if (!path) {
|
||||
error(500, "Invalid file id");
|
||||
}
|
||||
|
||||
await registerNewDirectory(params);
|
||||
unlink(path); // Intended
|
||||
};
|
||||
|
||||
const convertToReadableStream = (readStream: ReadStream) => {
|
||||
@@ -75,6 +54,20 @@ export const getFileStream = async (userId: number, fileId: number) => {
|
||||
};
|
||||
};
|
||||
|
||||
export const renameFile = async (
|
||||
userId: number,
|
||||
fileId: number,
|
||||
newEncName: string,
|
||||
newEncNameIv: string,
|
||||
) => {
|
||||
const file = await getFile(userId, fileId);
|
||||
if (!file) {
|
||||
error(404, "Invalid file id");
|
||||
}
|
||||
|
||||
await setFileEncName(userId, fileId, newEncName, newEncNameIv);
|
||||
};
|
||||
|
||||
export const getFileInformation = async (userId: number, fileId: number) => {
|
||||
const file = await getFile(userId, fileId);
|
||||
if (!file) {
|
||||
|
||||
@@ -2,7 +2,7 @@ import { error, json } from "@sveltejs/kit";
|
||||
import { z } from "zod";
|
||||
import { authorize } from "$lib/server/modules/auth";
|
||||
import { directroyInfoResponse, type DirectroyInfoResponse } from "$lib/server/schemas";
|
||||
import { getDirectroyInformation } from "$lib/server/services/file";
|
||||
import { getDirectroyInformation } from "$lib/server/services/directory";
|
||||
import type { RequestHandler } from "./$types";
|
||||
|
||||
export const GET: RequestHandler = async ({ cookies, params }) => {
|
||||
|
||||
20
src/routes/api/directory/[id]/delete/+server.ts
Normal file
20
src/routes/api/directory/[id]/delete/+server.ts
Normal file
@@ -0,0 +1,20 @@
|
||||
import { error, text } from "@sveltejs/kit";
|
||||
import { z } from "zod";
|
||||
import { authorize } from "$lib/server/modules/auth";
|
||||
import { deleteDirectory } from "$lib/server/services/directory";
|
||||
import type { RequestHandler } from "./$types";
|
||||
|
||||
export const POST: RequestHandler = async ({ cookies, params }) => {
|
||||
const { userId } = await authorize(cookies, "activeClient");
|
||||
|
||||
const zodRes = z
|
||||
.object({
|
||||
id: z.coerce.number().int().positive(),
|
||||
})
|
||||
.safeParse(params);
|
||||
if (!zodRes.success) error(400, "Invalid path parameters");
|
||||
const { id } = zodRes.data;
|
||||
|
||||
await deleteDirectory(userId, id);
|
||||
return text("Directory deleted", { headers: { "Content-Type": "text/plain" } });
|
||||
};
|
||||
27
src/routes/api/directory/[id]/rename/+server.ts
Normal file
27
src/routes/api/directory/[id]/rename/+server.ts
Normal file
@@ -0,0 +1,27 @@
|
||||
import { error, text } from "@sveltejs/kit";
|
||||
import { z } from "zod";
|
||||
import { authorize } from "$lib/server/modules/auth";
|
||||
import { parseSignedRequest } from "$lib/server/modules/crypto";
|
||||
import { directoryRenameRequest } from "$lib/server/schemas";
|
||||
import { renameDirectory } from "$lib/server/services/directory";
|
||||
import type { RequestHandler } from "./$types";
|
||||
|
||||
export const POST: RequestHandler = async ({ request, cookies, params }) => {
|
||||
const { userId, clientId } = await authorize(cookies, "activeClient");
|
||||
|
||||
const zodRes = z
|
||||
.object({
|
||||
id: z.coerce.number().int().positive(),
|
||||
})
|
||||
.safeParse(params);
|
||||
if (!zodRes.success) error(400, "Invalid path parameters");
|
||||
const { id } = zodRes.data;
|
||||
const { name, nameIv } = await parseSignedRequest(
|
||||
clientId,
|
||||
await request.json(),
|
||||
directoryRenameRequest,
|
||||
);
|
||||
|
||||
await renameDirectory(userId, id, name, nameIv);
|
||||
return text("Directory renamed", { headers: { "Content-Type": "text/plain" } });
|
||||
};
|
||||
@@ -2,7 +2,7 @@ import { text } from "@sveltejs/kit";
|
||||
import { authorize } from "$lib/server/modules/auth";
|
||||
import { parseSignedRequest } from "$lib/server/modules/crypto";
|
||||
import { directoryCreateRequest } from "$lib/server/schemas";
|
||||
import { createDirectory } from "$lib/server/services/file";
|
||||
import { createDirectory } from "$lib/server/services/directory";
|
||||
import type { RequestHandler } from "./$types";
|
||||
|
||||
export const POST: RequestHandler = async ({ request, cookies }) => {
|
||||
|
||||
20
src/routes/api/file/[id]/delete/+server.ts
Normal file
20
src/routes/api/file/[id]/delete/+server.ts
Normal file
@@ -0,0 +1,20 @@
|
||||
import { error, text } from "@sveltejs/kit";
|
||||
import { z } from "zod";
|
||||
import { authorize } from "$lib/server/modules/auth";
|
||||
import { deleteFile } from "$lib/server/services/file";
|
||||
import type { RequestHandler } from "./$types";
|
||||
|
||||
export const POST: RequestHandler = async ({ cookies, params }) => {
|
||||
const { userId } = await authorize(cookies, "activeClient");
|
||||
|
||||
const zodRes = z
|
||||
.object({
|
||||
id: z.coerce.number().int().positive(),
|
||||
})
|
||||
.safeParse(params);
|
||||
if (!zodRes.success) error(400, "Invalid path parameters");
|
||||
const { id } = zodRes.data;
|
||||
|
||||
await deleteFile(userId, id);
|
||||
return text("File deleted", { headers: { "Content-Type": "text/plain" } });
|
||||
};
|
||||
27
src/routes/api/file/[id]/rename/+server.ts
Normal file
27
src/routes/api/file/[id]/rename/+server.ts
Normal file
@@ -0,0 +1,27 @@
|
||||
import { error, text } from "@sveltejs/kit";
|
||||
import { z } from "zod";
|
||||
import { authorize } from "$lib/server/modules/auth";
|
||||
import { parseSignedRequest } from "$lib/server/modules/crypto";
|
||||
import { fileRenameRequest } from "$lib/server/schemas";
|
||||
import { renameFile } from "$lib/server/services/file";
|
||||
import type { RequestHandler } from "./$types";
|
||||
|
||||
export const POST: RequestHandler = async ({ request, cookies, params }) => {
|
||||
const { userId, clientId } = await authorize(cookies, "activeClient");
|
||||
|
||||
const zodRes = z
|
||||
.object({
|
||||
id: z.coerce.number().int().positive(),
|
||||
})
|
||||
.safeParse(params);
|
||||
if (!zodRes.success) error(400, "Invalid path parameters");
|
||||
const { id } = zodRes.data;
|
||||
const { name, nameIv } = await parseSignedRequest(
|
||||
clientId,
|
||||
await request.json(),
|
||||
fileRenameRequest,
|
||||
);
|
||||
|
||||
await renameFile(userId, id, name, nameIv);
|
||||
return text("File renamed", { headers: { "Content-Type": "text/plain" } });
|
||||
};
|
||||
Reference in New Issue
Block a user