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;
|
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) => {
|
export const registerNewFile = async (params: NewFileParams) => {
|
||||||
await db.transaction(async (tx) => {
|
await db.transaction(async (tx) => {
|
||||||
const meks = await tx
|
const meks = await tx
|
||||||
@@ -115,3 +156,25 @@ export const getFile = async (userId: number, fileId: number) => {
|
|||||||
.execute();
|
.execute();
|
||||||
return res[0] ?? null;
|
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";
|
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({
|
export const directroyInfoResponse = z.object({
|
||||||
metadata: z
|
metadata: z
|
||||||
.object({
|
.object({
|
||||||
|
|||||||
@@ -1,5 +1,11 @@
|
|||||||
import { z } from "zod";
|
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({
|
export const fileInfoResponse = z.object({
|
||||||
createdAt: z.date(),
|
createdAt: z.date(),
|
||||||
mekVersion: z.number().int().positive(),
|
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 { dirname } from "path";
|
||||||
import { v4 as uuidv4 } from "uuid";
|
import { v4 as uuidv4 } from "uuid";
|
||||||
import {
|
import {
|
||||||
getAllDirectoriesByParent,
|
|
||||||
registerNewDirectory,
|
|
||||||
getDirectory,
|
|
||||||
registerNewFile,
|
registerNewFile,
|
||||||
getAllFilesByParent,
|
|
||||||
getFile,
|
getFile,
|
||||||
type NewDirectroyParams,
|
setFileEncName,
|
||||||
|
unregisterFile,
|
||||||
type NewFileParams,
|
type NewFileParams,
|
||||||
} from "$lib/server/db/file";
|
} from "$lib/server/db/file";
|
||||||
import { getActiveMekVersion } from "$lib/server/db/mek";
|
import { getActiveMekVersion } from "$lib/server/db/mek";
|
||||||
import env from "$lib/server/loadenv";
|
import env from "$lib/server/loadenv";
|
||||||
|
|
||||||
export const getDirectroyInformation = async (userId: number, directroyId: "root" | number) => {
|
export const deleteFile = async (userId: number, fileId: number) => {
|
||||||
const directory = directroyId !== "root" ? await getDirectory(userId, directroyId) : undefined;
|
const file = await getFile(userId, fileId);
|
||||||
if (directory === null) {
|
if (!file) {
|
||||||
error(404, "Invalid directory id");
|
error(404, "Invalid file id");
|
||||||
}
|
}
|
||||||
|
|
||||||
const directories = await getAllDirectoriesByParent(userId, directroyId);
|
const path = await unregisterFile(userId, fileId);
|
||||||
const files = await getAllFilesByParent(userId, directroyId);
|
if (!path) {
|
||||||
|
error(500, "Invalid file id");
|
||||||
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);
|
unlink(path); // Intended
|
||||||
};
|
};
|
||||||
|
|
||||||
const convertToReadableStream = (readStream: ReadStream) => {
|
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) => {
|
export const getFileInformation = async (userId: number, fileId: number) => {
|
||||||
const file = await getFile(userId, fileId);
|
const file = await getFile(userId, fileId);
|
||||||
if (!file) {
|
if (!file) {
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ import { error, json } from "@sveltejs/kit";
|
|||||||
import { z } from "zod";
|
import { z } from "zod";
|
||||||
import { authorize } from "$lib/server/modules/auth";
|
import { authorize } from "$lib/server/modules/auth";
|
||||||
import { directroyInfoResponse, type DirectroyInfoResponse } from "$lib/server/schemas";
|
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";
|
import type { RequestHandler } from "./$types";
|
||||||
|
|
||||||
export const GET: RequestHandler = async ({ cookies, params }) => {
|
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 { authorize } from "$lib/server/modules/auth";
|
||||||
import { parseSignedRequest } from "$lib/server/modules/crypto";
|
import { parseSignedRequest } from "$lib/server/modules/crypto";
|
||||||
import { directoryCreateRequest } from "$lib/server/schemas";
|
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";
|
import type { RequestHandler } from "./$types";
|
||||||
|
|
||||||
export const POST: RequestHandler = async ({ request, cookies }) => {
|
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