birdclaw 0.5.0 → 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
@@ -574,11 +576,13 @@ const BASE_SCHEMA_SQL = `
 
 const INDEX_SQL = `
   create index if not exists idx_tweets_kind_created on tweets(kind, created_at desc);
+  create index if not exists idx_tweets_created on tweets(created_at desc);
   create index if not exists idx_tweets_account_created on tweets(account_id, created_at desc);
   create index if not exists idx_tweets_quoted on tweets(quoted_tweet_id);
   create index if not exists idx_tweet_collections_kind_account on tweet_collections(kind, account_id, collected_at desc, tweet_id);
   create index if not exists idx_tweet_collections_tweet on tweet_collections(tweet_id);
   create index if not exists idx_tweet_account_edges_kind_account on tweet_account_edges(kind, account_id, last_seen_at desc, tweet_id);
+  create index if not exists idx_tweet_account_edges_kind_tweet on tweet_account_edges(kind, tweet_id, account_id);
   create index if not exists idx_tweet_account_edges_tweet on tweet_account_edges(tweet_id);
   create index if not exists idx_dm_conversations_account on dm_conversations(account_id, last_message_at desc);
   create index if not exists idx_dm_messages_conversation on dm_messages(conversation_id, created_at asc);
@@ -676,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 (
@@ -939,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);
@@ -979,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();
+	}
 }