birdclaw 0.5.1 → 0.6.0

package/src/lib/db.ts

@@ -131,6 +131,7 @@ export interface DmConversationsTable {
 	account_id: string;
 	participant_profile_id: string;
 	title: string;
+	inbox_kind: "accepted" | "request";
 	last_message_at: string;
 	unread_count: number;
 	needs_reply: number;
@@ -429,6 +430,7 @@ const BASE_SCHEMA_SQL = `
     account_id text not null,
     participant_profile_id text not null,
     title text not null,
+    inbox_kind text not null default 'accepted',
     last_message_at text not null,
     unread_count integer not null default 0,
     needs_reply integer not null default 0
@@ -678,6 +680,15 @@ function ensureAccountExternalUserIdColumn(db: Database) {
 	}
 }
 
+function ensureDmConversationInboxColumns(db: Database) {
+	const columnNames = getColumnNames(db, "dm_conversations");
+	if (!columnNames.has("inbox_kind")) {
+		db.exec(
+			"alter table dm_conversations add column inbox_kind text not null default 'accepted'",
+		);
+	}
+}
+
 function ensureTweetCollectionsTable(db: Database) {
 	db.exec(`
     create table if not exists tweet_collections (
@@ -941,6 +952,7 @@ function initDatabase(options: InitDatabaseOptions = {}) {
 		nativeDb = new NativeSqliteDatabase(dbPath);
 		nativeDb.exec(BASE_SCHEMA_SQL);
 		ensureAccountExternalUserIdColumn(nativeDb);
+		ensureDmConversationInboxColumns(nativeDb);
 		ensureTweetMetadataColumns(nativeDb);
 		ensureProfileAvatarColumns(nativeDb);
 		ensureTweetCollectionsTable(nativeDb);
@@ -981,11 +993,29 @@ export function getDb() {
 	return kyselyDb as Kysely<BirdclawDatabase>;
 }
 
-export function resetDatabaseForTests() {
-	kyselyDb?.destroy();
+export async function closeDatabase() {
+	const db = kyselyDb;
+	const native = nativeDb;
 	kyselyDb = undefined;
+	nativeDb = undefined;
+	demoSeedAttempted = false;
+
+	if (db) {
+		await db.destroy();
+	} else {
+		native?.close();
+	}
+}
 
-	nativeDb?.close();
+export function resetDatabaseForTests() {
+	const db = kyselyDb;
+	const native = nativeDb;
+	kyselyDb = undefined;
 	nativeDb = undefined;
 	demoSeedAttempted = false;
+	if (db) {
+		void db.destroy();
+	} else {
+		native?.close();
+	}
 }