From 04aa4a5e2c8c4b290e7e4bd5c5b89d17b6a8e007 Mon Sep 17 00:00:00 2001
From: tamaina <tamaina@hotmail.co.jp>
Date: Sun, 12 Sep 2021 15:35:59 +0900
Subject: [PATCH] add Packed type

---
 src/misc/schema.ts              | 4 +++-
 src/models/repositories/note.ts | 4 ++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/misc/schema.ts b/src/misc/schema.ts
index d27c9eff99..52fbbca43b 100644
--- a/src/misc/schema.ts
+++ b/src/misc/schema.ts
@@ -47,6 +47,8 @@ export const refs = {
 	GalleryPost: packedGalleryPostSchema,
 };
 
+export type Packed<x extends keyof typeof refs> = SchemaType<typeof refs[x]>
+
 export interface Schema extends SimpleSchema {
 	items?: Schema;
 	properties?: Obj;
@@ -92,7 +94,7 @@ export type SchemaType<p extends Schema> =
 	p['type'] extends 'array' ? NullOrUndefined<p, MyType<NonNullable<p['items']>>[]> :
 	p['type'] extends 'object' ? (
 		p['ref'] extends keyof typeof refs
-			? NullOrUndefined<p, SchemaType<typeof refs[p['ref']]>>
+			? NullOrUndefined<p, Packed<p['ref']>>
 			: NullOrUndefined<p, ObjType<NonNullable<p['properties']>>>
 	) :
 	p['type'] extends 'any' ? NullOrUndefined<p, any> :
diff --git a/src/models/repositories/note.ts b/src/models/repositories/note.ts
index 376a09d0c6..3a821e9d22 100644
--- a/src/models/repositories/note.ts
+++ b/src/models/repositories/note.ts
@@ -3,14 +3,14 @@ import * as mfm from 'mfm-js';
 import { Note } from '@/models/entities/note';
 import { User } from '@/models/entities/user';
 import { Users, PollVotes, DriveFiles, NoteReactions, Followings, Polls, Channels } from '../index';
-import { SchemaType } from '@/misc/schema';
+import { Packed } from '@/misc/schema';
 import { nyaize } from '@/misc/nyaize';
 import { awaitAll } from '@/prelude/await-all';
 import { convertLegacyReaction, convertLegacyReactions, decodeReaction } from '@/misc/reaction-lib';
 import { NoteReaction } from '@/models/entities/note-reaction';
 import { aggregateNoteEmojis, populateEmojis, prefetchEmojis } from '@/misc/populate-emojis';
 
-export type PackedNote = SchemaType<typeof packedNoteSchema>;
+export type PackedNote = Packed<'Note'>;
 
 @EntityRepository(Note)
 export class NoteRepository extends Repository<Note> {