1 files changed, 22 insertions, 0 deletions
diff --git a/src/server/repositories/card.ts b/src/server/repositories/card.ts
index 223050a..a2b0972 100644
--- a/src/server/repositories/card.ts
+++ b/src/server/repositories/card.ts
@@ -230,6 +230,28 @@ export const cardRepository: CardRepository = {
 		return result[0]?.count ?? 0;
 	},
 
+	async countTotalCards(deckId: string): Promise<number> {
+		const result = await db
+			.select({ count: sql<number>`count(*)::int` })
+			.from(cards)
+			.where(and(eq(cards.deckId, deckId), isNull(cards.deletedAt)));
+		return result[0]?.count ?? 0;
+	},
+
+	async countReviewStateCards(deckId: string): Promise<number> {
+		const result = await db
+			.select({ count: sql<number>`count(*)::int` })
+			.from(cards)
+			.where(
+				and(
+					eq(cards.deckId, deckId),
+					isNull(cards.deletedAt),
+					eq(cards.state, CardState.Review),
+				),
+			);
+		return result[0]?.count ?? 0;
+	},
+
 	async findDueCardsWithNoteData(
 		deckId: string,
 		now: Date,