3 files changed, 26 insertions, 0 deletions
diff --git a/src/server/repositories/card.test.ts b/src/server/repositories/card.test.ts
index b959709..8630e19 100644
--- a/src/server/repositories/card.test.ts
+++ b/src/server/repositories/card.test.ts
@@ -116,6 +116,8 @@ function createMockCardRepo(): CardRepository {
 		findDueCardsWithNoteData: vi.fn(),
 		findDueCardsForStudy: vi.fn(),
 		updateFSRSFields: vi.fn(),
+		countTotalCards: vi.fn(),
+		countReviewStateCards: vi.fn(),
 	};
 }
 
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,
diff --git a/src/server/repositories/types.ts b/src/server/repositories/types.ts
index b70b247..7e0819a 100644
--- a/src/server/repositories/types.ts
+++ b/src/server/repositories/types.ts
@@ -146,6 +146,8 @@ export interface CardRepository {
 	findDueCards(deckId: string, now: Date): Promise<Card[]>;
 	countDueCards(deckId: string, now: Date): Promise<number>;
 	countNewCards(deckId: string): Promise<number>;
+	countTotalCards(deckId: string): Promise<number>;
+	countReviewStateCards(deckId: string): Promise<number>;
 	findDueCardsWithNoteData(
 		deckId: string,
 		now: Date,