summaryrefslogtreecommitdiff
path: root/app/lib/db.server.ts
blob: cd8d198f9b21e1a6b60c6e392a34b47f4a8d9c36 (plain)
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
import Database from "better-sqlite3";
import path from "path";

let db: Database.Database | null = null;

export function getDb(): Database.Database {
  if (!db) {
    const dbPath = process.env.DB_PATH ?? path.resolve("microblog.db");
    db = new Database(dbPath);
    db.pragma("journal_mode = WAL");
    db.pragma("foreign_keys = ON");
    initSchema(db);
  }
  return db;
}

function initSchema(db: Database.Database) {
  db.exec(`
    CREATE TABLE IF NOT EXISTS posts (
      id         TEXT PRIMARY KEY,
      content    TEXT NOT NULL,
      created_at TEXT NOT NULL DEFAULT (datetime('now'))
    );
    CREATE INDEX IF NOT EXISTS idx_posts_created_at ON posts(created_at DESC);

    CREATE TABLE IF NOT EXISTS credentials (
      id         TEXT PRIMARY KEY,
      public_key TEXT NOT NULL,
      counter    INTEGER NOT NULL DEFAULT 0,
      transports TEXT
    );
  `);
}

/* ── Posts ── */

export interface Post {
  id: string;
  content: string;
  created_at: string;
}

export interface CreatePostInput {
  id: string;
  content: string;
}

export function listPosts(): Post[] {
  return getDb().prepare("SELECT * FROM posts ORDER BY created_at DESC").all() as Post[];
}

export function getPostById(id: string): Post | null {
  return getDb().prepare("SELECT * FROM posts WHERE id = ?").get(id) as Post | null;
}

export function createPost(input: CreatePostInput): Post {
  getDb().prepare("INSERT INTO posts (id, content) VALUES (?, ?)").run(input.id, input.content);
  return getPostById(input.id)!;
}

export function deletePost(id: string): void {
  getDb().prepare("DELETE FROM posts WHERE id = ?").run(id);
}

/* ── Credentials (passkey) ── */

export interface Credential {
  id: string;
  public_key: string;
  counter: number;
  transports: string | null;
}

export function listCredentials(): Credential[] {
  return getDb().prepare("SELECT * FROM credentials").all() as Credential[];
}

export function getCredentialById(id: string): Credential | null {
  return getDb().prepare("SELECT * FROM credentials WHERE id = ?").get(id) as Credential | null;
}

export function saveCredential(input: {
  id: string;
  publicKey: Uint8Array;
  counter: number;
  transports?: string[];
}): void {
  getDb()
    .prepare(
      "INSERT INTO credentials (id, public_key, counter, transports) VALUES (?, ?, ?, ?)"
    )
    .run(
      input.id,
      Buffer.from(input.publicKey).toString("base64url"),
      input.counter,
      input.transports ? JSON.stringify(input.transports) : null
    );
}

export function updateCredentialCounter(id: string, counter: number): void {
  getDb().prepare("UPDATE credentials SET counter = ? WHERE id = ?").run(counter, id);
}