summaryrefslogtreecommitdiff
path: root/packages/backend/src/models/RepositoryModule.ts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/backend/src/models/RepositoryModule.ts')
-rw-r--r--packages/backend/src/models/RepositoryModule.ts9
1 files changed, 9 insertions, 0 deletions
diff --git a/packages/backend/src/models/RepositoryModule.ts b/packages/backend/src/models/RepositoryModule.ts
index b7142d91bf..146dbbc3b8 100644
--- a/packages/backend/src/models/RepositoryModule.ts
+++ b/packages/backend/src/models/RepositoryModule.ts
@@ -42,6 +42,7 @@ import {
MiNoteFavorite,
MiNoteReaction,
MiNoteThreadMuting,
+ MiNoteDraft,
MiPage,
MiPageLike,
MiPasswordResetRequest,
@@ -140,6 +141,12 @@ const $noteReactionsRepository: Provider = {
inject: [DI.db],
};
+const $noteDraftsRepository: Provider = {
+ provide: DI.noteDraftsRepository,
+ useFactory: (db: DataSource) => db.getRepository(MiNoteDraft).extend(miRepository as MiRepository<MiNoteDraft>),
+ inject: [DI.db],
+};
+
const $pollsRepository: Provider = {
provide: DI.pollsRepository,
useFactory: (db: DataSource) => db.getRepository(MiPoll).extend(miRepository as MiRepository<MiPoll>),
@@ -542,6 +549,7 @@ const $reversiGamesRepository: Provider = {
$noteFavoritesRepository,
$noteThreadMutingsRepository,
$noteReactionsRepository,
+ $noteDraftsRepository,
$pollsRepository,
$pollVotesRepository,
$userProfilesRepository,
@@ -618,6 +626,7 @@ const $reversiGamesRepository: Provider = {
$noteFavoritesRepository,
$noteThreadMutingsRepository,
$noteReactionsRepository,
+ $noteDraftsRepository,
$pollsRepository,
$pollVotesRepository,
$userProfilesRepository,