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