File size: 1,485 Bytes
5c5b371
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import type sqlite3 from "better-sqlite3";

type Migration = {
  name: string;
  version: number;
  up: (db: sqlite3.Database) => void;
  down: (db: sqlite3.Database) => void;
};

export const migrations = [
  {
    name: "create db",
    version: 1,
    up: () => {},
    down: () => {},
  },
  {
    name: "add events table",
    version: 2,
    up: (db) => {
      db.exec(
        `CREATE TABLE IF NOT EXISTS events
         (
             id           INTEGER PRIMARY KEY AUTOINCREMENT,
             type         TEXT    NOT NULL,
             ip           TEXT    NOT NULL,
             date         TEXT    NOT NULL,
             model        TEXT    NOT NULL,
             family       TEXT    NOT NULL,
             hashes       TEXT    NOT NULL,
             userToken    TEXT    NOT NULL,
             inputTokens  INTEGER NOT NULL,
             outputTokens INTEGER NOT NULL
         )`
      );
    },
    down: (db) => db.exec("DROP TABLE events"),
  },
  {
    name: "add events indexes",
    version: 3,
    up: (db) => {
      // language=SQLite
      db.exec(
        `BEGIN;
        CREATE INDEX IF NOT EXISTS idx_events_userToken ON events (userToken);
        CREATE INDEX IF NOT EXISTS idx_events_ip ON events (ip);
        COMMIT;`
      );
    },
    down: (db) => {
      // language=SQLite
      db.exec(
        `BEGIN;
        DROP INDEX idx_events_userToken;
        DROP INDEX idx_events_ip;
        COMMIT;`
      );
    },
  },
] satisfies Migration[];