mirror of
https://github.com/Drop-OSS/drop.git
synced 2025-11-18 18:51:13 +10:00
feat: basic screenshot manager
This commit is contained in:
58
server/internal/screenshots/index.ts
Normal file
58
server/internal/screenshots/index.ts
Normal file
@ -0,0 +1,58 @@
|
||||
import { randomUUID } from "node:crypto";
|
||||
import type { IncomingMessage } from "node:http";
|
||||
import objectHandler from "../objects";
|
||||
import stream from "node:stream/promises";
|
||||
import prisma from "../db/database";
|
||||
|
||||
class ScreenshotManager {
|
||||
async get(id: string) {
|
||||
return await prisma.screenshot.findUnique({
|
||||
where: {
|
||||
id,
|
||||
},
|
||||
});
|
||||
}
|
||||
|
||||
async getAllByGame(gameId: string, userId: string) {
|
||||
const results = await prisma.screenshot.findMany({
|
||||
where: {
|
||||
gameId,
|
||||
userId,
|
||||
},
|
||||
});
|
||||
return results;
|
||||
}
|
||||
|
||||
async delete(id: string) {
|
||||
await prisma.screenshot.delete({
|
||||
where: {
|
||||
id,
|
||||
},
|
||||
});
|
||||
}
|
||||
|
||||
async upload(gameId: string, userId: string, inputStream: IncomingMessage) {
|
||||
const objectId = randomUUID();
|
||||
const saveStream = await objectHandler.createWithStream(objectId, {}, []);
|
||||
if (!saveStream)
|
||||
throw createError({
|
||||
statusCode: 500,
|
||||
statusMessage: "Failed to create writing stream to storage backend.",
|
||||
});
|
||||
|
||||
// pipe into object store
|
||||
await stream.pipeline(inputStream, saveStream);
|
||||
|
||||
// TODO: set createAt to the time screenshot was taken
|
||||
await prisma.screenshot.create({
|
||||
data: {
|
||||
gameId,
|
||||
userId,
|
||||
objectId,
|
||||
},
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
export const screenshotManager = new ScreenshotManager();
|
||||
export default screenshotManager;
|
||||
Reference in New Issue
Block a user