100 lines
2.7 KiB
TypeScript
100 lines
2.7 KiB
TypeScript
import { db } from "@/db/postgre.js";
|
|
import { Page } from "@/models/entities/page.js";
|
|
import type { Packed } from "@/misc/schema.js";
|
|
import { awaitAll } from "@/prelude/await-all.js";
|
|
import type { DriveFile } from "@/models/entities/drive-file.js";
|
|
import type { User } from "@/models/entities/user.js";
|
|
import { Users, DriveFiles, PageLikes } from "../index.js";
|
|
|
|
export const PageRepository = db.getRepository(Page).extend({
|
|
async pack(
|
|
src: Page["id"] | Page,
|
|
me?: { id: User["id"] } | null | undefined,
|
|
): Promise<Packed<"Page">> {
|
|
const meId = me ? me.id : null;
|
|
const page =
|
|
typeof src === "object" ? src : await this.findOneByOrFail({ id: src });
|
|
|
|
const attachedFiles: Promise<DriveFile | null>[] = [];
|
|
const collectFile = (xs: any[]) => {
|
|
for (const x of xs) {
|
|
if (x.type === "image") {
|
|
attachedFiles.push(
|
|
DriveFiles.findOneBy({
|
|
id: x.fileId,
|
|
userId: page.userId,
|
|
}),
|
|
);
|
|
}
|
|
if (x.children) {
|
|
collectFile(x.children);
|
|
}
|
|
}
|
|
};
|
|
collectFile(page.content);
|
|
|
|
// 後方互換性のため
|
|
let migrated = false;
|
|
const migrate = (xs: any[]) => {
|
|
for (const x of xs) {
|
|
if (x.type === "input") {
|
|
if (x.inputType === "text") {
|
|
x.type = "textInput";
|
|
}
|
|
if (x.inputType === "number") {
|
|
x.type = "numberInput";
|
|
if (x.default) x.default = parseInt(x.default, 10);
|
|
}
|
|
migrated = true;
|
|
}
|
|
if (x.children) {
|
|
migrate(x.children);
|
|
}
|
|
}
|
|
};
|
|
migrate(page.content);
|
|
if (migrated) {
|
|
this.update(page.id, {
|
|
content: page.content,
|
|
});
|
|
}
|
|
|
|
return await awaitAll({
|
|
id: page.id,
|
|
createdAt: page.createdAt.toISOString(),
|
|
updatedAt: page.updatedAt.toISOString(),
|
|
userId: page.userId,
|
|
user: Users.pack(page.user || page.userId, me), // { detail: true } すると無限ループするので注意
|
|
content: page.content,
|
|
variables: page.variables,
|
|
title: page.title,
|
|
isPublic: page.isPublic,
|
|
name: page.name,
|
|
summary: page.summary,
|
|
hideTitleWhenPinned: page.hideTitleWhenPinned,
|
|
alignCenter: page.alignCenter,
|
|
font: page.font,
|
|
script: page.script,
|
|
eyeCatchingImageId: page.eyeCatchingImageId,
|
|
eyeCatchingImage: page.eyeCatchingImageId
|
|
? await DriveFiles.pack(page.eyeCatchingImageId)
|
|
: null,
|
|
attachedFiles: DriveFiles.packMany(
|
|
(
|
|
await Promise.all(attachedFiles)
|
|
).filter((x): x is DriveFile => x != null),
|
|
),
|
|
likedCount: page.likedCount,
|
|
isLiked: meId
|
|
? await PageLikes.findOneBy({ pageId: page.id, userId: meId }).then(
|
|
(x) => x != null,
|
|
)
|
|
: undefined,
|
|
});
|
|
},
|
|
|
|
packMany(pages: Page[], me?: { id: User["id"] } | null | undefined) {
|
|
return Promise.all(pages.map((x) => this.pack(x, me)));
|
|
},
|
|
});
|