db-flush.ts RawHistoryFolder

1import { createClient } from '@libsql/client';
2import * as dotenv from 'dotenv';
3
4// Load environment variables from .env file
5dotenv.config();
6
7const { TURSO_DATABASE_URL, TURSO_AUTH_TOKEN } = process.env;
8
9if (!TURSO_DATABASE_URL || !TURSO_AUTH_TOKEN) {
10    throw new Error('Missing required environment variables TURSO_DATABASE_URL and/or TURSO_AUTH_TOKEN');
11}
12
13async function flushDatabase() {
14    console.log('Starting database content flush...');
15    
16    const client = createClient({ 
17        url: TURSO_DATABASE_URL as string, 
18        authToken: TURSO_AUTH_TOKEN as string 
19    });
20
21    try {
22        // Clear tables in correct order to respect foreign key constraints
23        console.log('Clearing messages table...');
24        await client.execute('DELETE FROM messages');
25        
26        console.log('Clearing sessions table...');
27        await client.execute('DELETE FROM sessions');
28        
29        console.log('Clearing chat rooms table...');
30        await client.execute('DELETE FROM chat_rooms');
31        
32        console.log('Clearing users table...');
33        await client.execute('DELETE FROM users');
34
35        console.log('Database content flush completed successfully');
36    } catch (error) {
37        console.error('Error flushing database content:', error);
38        throw error;
39    } finally {
40        await client.close();
41    }
42}
43
44// Execute if this file is run directly
45if (new URL(import.meta.url).pathname === process.argv[1]) {
46    flushDatabase()
47        .then(() => {
48            console.debug('[db-flush] Exiting process with code 0');
49            process.exit(0);
50        })
51        .catch((error) => {
52            console.error('[db-flush] Failed to flush database content:', error);
53            process.exit(1);
54        });
55}