backend/index.ts

177 lines
4.5 KiB
TypeScript

import express, { Request, Response } from "express";
import { Database } from "sqlite3";
import bcrypt from "bcrypt";
import session from "express-session";
const db = new Database("db.sqlite");
declare module "express-session" {
// noinspection JSUnusedGlobalSymbols
interface SessionData {
user?: number;
}
}
(async () => {
await dbRun("CREATE TABLE IF NOT EXISTS version ( version INT );");
const dbVersion: { version: number } | undefined
= await dbGet("SELECT version FROM version;");
if (dbVersion) {
if (dbVersion.version === 2) {
console.log("skip db setup")
return;
} else if (dbVersion.version === 1) {
await dbRun("ALTER TABLE highScores ADD COLUMN created TIMESTAMP DEFAULT CURRENT_TIMESTAMP")
await dbRun("UPDATE version SET version = 2");
return;
}
}
await dbRun(`CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
pass TEXT NOT NULL
)`);
await dbRun("CREATE UNIQUE INDEX IF NOT EXISTS idx_users_name ON users (name);");
await dbRun(`CREATE TABLE IF NOT EXISTS highScores (
id INTEGER PRIMARY KEY AUTOINCREMENT,
user INTEGER,
score BIGINT NOT NULL,
FOREIGN KEY (user) REFERENCES users(id)
)`);
await dbRun("INSERT INTO version VALUES (1)");
})();
const app = express();
app.use(express.json());
app.use(session({
secret: "your mom",
resave: true,
}));
app.get("/", (_req: Request, res: Response) => {
res.send("Hello World!");
});
app.get("/leaderboard", async (req: Request, res: Response) => {
const { limit = 100, offset = 0 } = req.query;
const leaders = await dbAll(`
select h.score, u.name, h.created
from highScores h
join main.users u on u.id = h.user
order by score desc
limit ? offset ?`, limit, offset);
res.json(leaders);
});
app.put("/leaderboard", async (req: Request, res: Response) => {
const { score } = req.body;
if (!req.session.user) {
res.status(403).json({ success: false, message: "unauthorised" });
return;
}
if (!score) {
res.status(400).json({ success: false, message: "provide score" });
return;
}
await dbRun("insert into highScores (user, score) VALUES (?, ?)", req.session.user, score);
res.status(200).json({ success: true });
});
app.post("/login", async (req: Request, res: Response) => {
await sleep(Math.random() * 50 + 50);
const { username, password } = req.body;
const user: {
name: string,
pass: string,
id: number,
} = await dbGet("SELECT * FROM users WHERE name = ?", username);
if (!user) {
res.status(403).json({ success: false, message: "invalid credentials" });
return;
}
if (!await bcrypt.compare(password, user.pass)) {
res.status(403).json({ success: false, message: "invalid credentials" });
return;
}
await new Promise(res => req.session.regenerate(res));
req.session.user = user.id;
res.status(200).json({ success: true });
});
app.post("/register", async (req: Request, res: Response) => {
await sleep(Math.random() * 50 + 50);
const { username, password } = req.body;
if (!username || !password) {
res.status(400).json({ success: false, message: "invalid credentials" });
return;
}
const pass = await bcrypt.hash(password, 12);
const success = await dbRun("INSERT INTO users (name, pass) VALUES (?, ?)", [ username, pass ])
.then(() => true, () => false);
if (!success) {
res.status(400).json({ success: false, message: "invalid credentials" });
} else {
res.status(200).json({ success: true });
}
});
app.post("/logout", async (req: Request, res: Response) => {
await new Promise(res => req.session.destroy(res));
res.send(200).json({ success: true });
});
app.listen(3000, () => {
console.log("Server running on port 3000");
});
function dbGet<T>(query: string, ...args: any[]): Promise<T> {
return new Promise((res, rej) => {
db.get(query, ...args, (err: unknown, result: T) => {
if (err) rej(err);
else res(result);
});
});
}
function dbAll<T>(query: string, ...args: any[]): Promise<T[]> {
return new Promise((res, rej) => {
db.all(query, ...args, (err: unknown, result: T[]) => {
if (err) rej(err);
else res(result);
});
});
}
function dbRun(query: string, ...args: any[]): Promise<void> {
return new Promise((res, rej) => {
db.run(query, ...args, (err: unknown) => {
if (err) rej(err);
else res();
});
})
}
function sleep(ms: number) {
return new Promise(resolve => setTimeout(resolve, ms));
}