mirror of
https://github.com/kmc7468/arkvault.git
synced 2025-12-16 15:08:46 +00:00
/api/directory/[id], /api/directory/create Endpoint 구현
This commit is contained in:
73
src/lib/server/db/file.ts
Normal file
73
src/lib/server/db/file.ts
Normal file
@@ -0,0 +1,73 @@
|
||||
import { and, eq, isNull } from "drizzle-orm";
|
||||
import db from "./drizzle";
|
||||
import { directory, file, mek } from "./schema";
|
||||
|
||||
type DirectroyId = "root" | number;
|
||||
|
||||
export interface NewDirectroyParams {
|
||||
userId: number;
|
||||
parentId: DirectroyId;
|
||||
mekVersion: number;
|
||||
encDek: string;
|
||||
encDekIv: string;
|
||||
encName: string;
|
||||
encNameIv: string;
|
||||
}
|
||||
|
||||
export const registerNewDirectory = async (params: NewDirectroyParams) => {
|
||||
return await db.transaction(async (tx) => {
|
||||
const meks = await tx
|
||||
.select()
|
||||
.from(mek)
|
||||
.where(and(eq(mek.userId, params.userId), eq(mek.state, "active")));
|
||||
if (meks[0]?.version !== params.mekVersion) {
|
||||
throw new Error("Invalid MEK version");
|
||||
}
|
||||
|
||||
const now = new Date();
|
||||
await tx.insert(directory).values({
|
||||
createdAt: now,
|
||||
parentId: params.parentId === "root" ? null : params.parentId,
|
||||
userId: params.userId,
|
||||
mekVersion: params.mekVersion,
|
||||
encDek: { ciphertext: params.encDek, iv: params.encDekIv },
|
||||
encryptedAt: now,
|
||||
encName: { ciphertext: params.encName, iv: params.encNameIv },
|
||||
});
|
||||
});
|
||||
};
|
||||
|
||||
export const getAllDirectoriesByParent = async (userId: number, directoryId: DirectroyId) => {
|
||||
return await db
|
||||
.select()
|
||||
.from(directory)
|
||||
.where(
|
||||
and(
|
||||
eq(directory.userId, userId),
|
||||
directoryId === "root" ? isNull(directory.parentId) : eq(directory.parentId, directoryId),
|
||||
),
|
||||
)
|
||||
.execute();
|
||||
};
|
||||
|
||||
export const getDirectory = async (userId: number, directoryId: number) => {
|
||||
const res = await db
|
||||
.select()
|
||||
.from(directory)
|
||||
.where(and(eq(directory.userId, userId), eq(directory.id, directoryId)))
|
||||
.execute();
|
||||
return res[0] ?? null;
|
||||
};
|
||||
|
||||
export const getAllFilesByParent = async (userId: number, parentId: DirectroyId) => {
|
||||
return await db
|
||||
.select()
|
||||
.from(file)
|
||||
.where(
|
||||
and(
|
||||
eq(file.userId, userId),
|
||||
parentId === "root" ? isNull(file.parentId) : eq(file.parentId, parentId),
|
||||
),
|
||||
)
|
||||
.execute();
|
||||
};
|
||||
@@ -35,6 +35,15 @@ export const getInitialMek = async (userId: number) => {
|
||||
return meks[0] ?? null;
|
||||
};
|
||||
|
||||
export const getActiveMekVersion = async (userId: number) => {
|
||||
const meks = await db
|
||||
.select({ version: mek.version })
|
||||
.from(mek)
|
||||
.where(and(eq(mek.userId, userId), eq(mek.state, "active")))
|
||||
.execute();
|
||||
return meks[0]?.version ?? null;
|
||||
};
|
||||
|
||||
export const getAllValidClientMeks = async (userId: number, clientId: number) => {
|
||||
return await db
|
||||
.select()
|
||||
|
||||
58
src/lib/server/db/schema/file.ts
Normal file
58
src/lib/server/db/schema/file.ts
Normal file
@@ -0,0 +1,58 @@
|
||||
import { sqliteTable, text, integer, foreignKey } from "drizzle-orm/sqlite-core";
|
||||
import { mek } from "./mek";
|
||||
import { user } from "./user";
|
||||
|
||||
const ciphertext = (name: string) =>
|
||||
text(name, { mode: "json" }).$type<{
|
||||
ciphertext: string;
|
||||
iv: string;
|
||||
}>();
|
||||
|
||||
export const directory = sqliteTable(
|
||||
"directory",
|
||||
{
|
||||
id: integer("id").primaryKey({ autoIncrement: true }),
|
||||
createdAt: integer("created_at", { mode: "timestamp_ms" }).notNull(),
|
||||
parentId: integer("parent_id"),
|
||||
userId: integer("user_id")
|
||||
.notNull()
|
||||
.references(() => user.id),
|
||||
mekVersion: integer("master_encryption_key_version").notNull(),
|
||||
encDek: ciphertext("encrypted_data_encryption_key").notNull().unique(),
|
||||
encryptedAt: integer("encrypted_at", { mode: "timestamp_ms" }).notNull(),
|
||||
encName: ciphertext("encrypted_name").notNull(),
|
||||
},
|
||||
(t) => ({
|
||||
ref1: foreignKey({
|
||||
columns: [t.parentId],
|
||||
foreignColumns: [t.id],
|
||||
}),
|
||||
ref2: foreignKey({
|
||||
columns: [t.userId, t.mekVersion],
|
||||
foreignColumns: [mek.userId, mek.version],
|
||||
}),
|
||||
}),
|
||||
);
|
||||
|
||||
export const file = sqliteTable(
|
||||
"file",
|
||||
{
|
||||
id: integer("id").primaryKey({ autoIncrement: true }),
|
||||
path: text("path").notNull().unique(),
|
||||
parentId: integer("parent_id").references(() => directory.id),
|
||||
createdAt: integer("created_at", { mode: "timestamp_ms" }).notNull(),
|
||||
userId: integer("user_id")
|
||||
.notNull()
|
||||
.references(() => user.id),
|
||||
mekVersion: integer("master_encryption_key_version").notNull(),
|
||||
encDek: ciphertext("encrypted_data_encryption_key").notNull().unique(),
|
||||
encryptedAt: integer("encrypted_at", { mode: "timestamp_ms" }).notNull(),
|
||||
encName: ciphertext("encrypted_name").notNull(),
|
||||
},
|
||||
(t) => ({
|
||||
ref: foreignKey({
|
||||
columns: [t.userId, t.mekVersion],
|
||||
foreignColumns: [mek.userId, mek.version],
|
||||
}),
|
||||
}),
|
||||
);
|
||||
@@ -1,4 +1,5 @@
|
||||
export * from "./client";
|
||||
export * from "./file";
|
||||
export * from "./mek";
|
||||
export * from "./token";
|
||||
export * from "./user";
|
||||
|
||||
Reference in New Issue
Block a user