fosrl/pangolin
1
0
Fork 0
mirror of https://github.com/fosrl/pangolin.git synced 2025-05-13 05:40:38 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Drizzle the db

Browse source
This commit is contained in:
Owen Schwartz 2024-09-28 12:42:38 -04:00
parent 584a4f28dc
commit db3ce357df
No known key found for this signature in database
GPG key ID: 8271FDFFD9E0CCBD
2 changed files with 26 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
server/index.ts
View file

@ -7,6 +7,7 @@ import helmet from "helmet";
import cors from "cors";
import unauth from "@server/routers/unauth";
import Database from 'better-sqlite3';
import { drizzle } from 'drizzle-orm/better-sqlite3';
const dev = environment.ENVIRONMENT !== "prod";
const app = next({ dev });
@ -17,22 +18,29 @@ let db: Database.Database;
app.prepare().then(() => {
// Open the SQLite database connection
db = new Database(`${environment.CONFIG_PATH}/db/db.sqlite`, { verbose: console.log });
const sqlite = new Database(`${environment.CONFIG_PATH}/db/db.sqlite`, { verbose: console.log });
const db = drizzle(sqlite);
const server = express();
server.use(helmet());
server.use(cors());
const prefix = `/api/${environment.API_VERSION}`;
// Middleware to attach db to req object
server.use((req: Request & { db?: Database.Database }, res: Response, next) => {
req.db = db;
next();
// Run migrations (if you're using Drizzle's migration system)
// migrate(db, { migrationsFolder: './drizzle' });
// Middleware to attach the database to the request
server.use((req, res, next) => {
(req as any).db = db;
next();
});
const prefix = `/api/${environment.API_VERSION}`;
server.use(prefix, express.json(), unauth);
// We are using NEXT from here on
server.all("*", (req: Request, res: Response) => {
const parsedUrl = parse(req.url!, true);
handle(req, res, parsedUrl);

18
server/routers/badger/getConfig.ts
View file

@ -1,11 +1,13 @@
import { Request, Response, NextFunction } from 'express';
import { Database } from 'better-sqlite3';
import { DrizzleError } from 'drizzle-orm';
import { BetterSQLite3Database } from 'drizzle-orm/better-sqlite3';
import { sites, Site } from '../../db/schema';
interface CustomRequest extends Request {
db?: Database;
db?: BetterSQLite3Database;
}
export const getConfig = (req: Request, res: Response, next: NextFunction): void => {
export const getConfig = async (req: Request, res: Response, next: NextFunction): Promise<void> => {
try {
const customReq = req as CustomRequest;
const db = customReq.db;
@ -14,13 +16,15 @@ export const getConfig = (req: Request, res: Response, next: NextFunction): void
throw new Error('Database is not attached to the request');
}
const query = 'SELECT * FROM sites';
const statement = db.prepare(query);
const results = statement.all();
const results: Site[] = db.select().from(sites).all();
res.json(results);
} catch (error) {
console.error('Error querying database:', error);
next(error);
if (error instanceof DrizzleError) {
res.status(500).json({ error: 'Database query error', message: error.message });
} else {
next(error);
}
}
};