summaryrefslogtreecommitdiff
path: root/app/lib/db.server.ts
blob: b076ec7b148d35276f6e355231a6d26822ba981a (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
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);
  `);
}

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)!;
}