feat(database): do database migrate

packages/cli/package.json

@@ -47,11 +47,12 @@
     "typecheck": "tsc --noEmit"
   },
   "dependencies": {
-    "@animegarden/scraper": "workspace:*",
     "@animegarden/database": "workspace:*",
+    "@animegarden/scraper": "workspace:*",
     "@breadc/color": "^0.9.7",
     "animegarden": "workspace:*",
     "breadc": "^0.9.7",
+    "dotenv": "^16.3.1",
     "fs-extra": "^11.2.0",
     "simptrad": "^0.1.0",
     "undici": "^6.3.0",

packages/cli/src/cli.ts

@@ -18,4 +18,16 @@ cli
     }
   });
 
+cli
+  .command('database migrate', 'Migrate database')
+  .alias('db migrate')
+  .action(async () => {
+    // TODO: pass options
+    await import('dotenv/config');
+    const { connectDatabase, migrateDatabase } = await import('@animegarden/database');
+    const { connection, db } = connectDatabase(`postgres://root:[email protected]:5432/animegarden`);
+    await migrateDatabase(db);
+    await connection.end();
+  });
+
 cli.run(process.argv.slice(2)).catch((err) => console.error(err));

packages/database/README.md

@@ -5,6 +5,14 @@
 
 The database for [AnimeGarden](https://garden.onekuma.cn/).
 
+## Local Development
+
+Start dev postgres and redis with `docker-compose.yml`
+
+```bash
+docker compose up --file packages/database/docker-compose.yml
+```
+
 ## Credits
 
 + [動漫花園](https://share.dmhy.org/)

packages/database/docker-compose.yml

@@ -0,0 +1,29 @@
+version: '3.9'
+
+services:
+  postgresql:
+    image: postgres
+    container_name: animegarden_postgres
+    restart: always
+    ports:
+      - '5432:5432'
+    # volumes:
+    #   - ./.animegarden/database:/var/lib/postgresql/data
+    environment:
+      TZ: Asia/Shanghai
+      PGTZ: Asia/Shanghai
+      POSTGRES_DB: animegarden
+      POSTGRES_USER: root
+      POSTGRES_PASSWORD: example
+      # postgres://root:[email protected]:5432/animegarden
+
+  redis:
+    image: redis
+    container_name: animegarden_redis
+    restart: always
+    ports:
+      - '6379:6379'
+    # volumes:
+    #   - ./.animegarden/redis:/data
+    environment:
+      TZ: Asia/Shanghai

packages/database/drizzle.config.ts

@@ -3,13 +3,7 @@ import type { Config } from 'drizzle-kit';
 import 'dotenv/config';
 
 export default {
-  schema: './src/schema/',
-  out: './drizzle',
   driver: 'pg',
-  dbCredentials: {
-    host: process.env.DB_HOST!,
-    database: process.env.DB_NAME!,
-    user: process.env.DB_USER,
-    password: process.env.DB_PASSWORD
-  }
+  out: './drizzle',
+  schema: './src/schema/'
 } satisfies Config;

packages/database/package.json

@@ -32,7 +32,8 @@
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
   "files": [
-    "dist"
+    "dist",
+    "drizzle"
   ],
   "scripts": {
     "build": "unbuild",
@@ -44,10 +45,12 @@
   },
   "dependencies": {
     "animegarden": "workspace:*",
-    "dotenv": "^16.3.1",
     "drizzle-orm": "^0.29.3",
     "postgres": "^3.4.3"
   },
+  "devDependencies": {
+    "dotenv": "^16.3.1"
+  },
   "engines": {
     "node": ">=v18.16.0"
   }

packages/database/src/connection.ts

@@ -0,0 +1,16 @@
+import postgres from 'postgres';
+import { drizzle } from 'drizzle-orm/postgres-js';
+
+import { users } from './schema/user';
+import { teams } from './schema/team';
+import { resources } from './schema/resource';
+
+export interface DatabaseConnectionConfig extends postgres.Options<{}> {}
+
+export const connectDatabase = (...args: Parameters<typeof postgres>) => {
+  const queryClient = postgres(...args);
+  return {
+    connection: queryClient,
+    db: drizzle(queryClient, { schema: { resources, users, teams } })
+  };
+};

packages/database/src/index.ts

@@ -1 +1,5 @@
 export * from './schema';
+
+export * from './connection';
+
+export * from './migrate';

packages/database/src/migrate.ts

@@ -0,0 +1,6 @@
+import { migrate } from 'drizzle-orm/postgres-js/migrator';
+
+export async function migrateDatabase(db: Parameters<typeof migrate>[0]) {
+  const migrationsFolder = new URL('../drizzle', import.meta.url).pathname;
+  await migrate(db, { migrationsFolder });
+}

packages/server/package.json

@@ -45,6 +45,7 @@
     "@animegarden/database": "workspace:*",
     "@animegarden/scraper": "workspace:*",
     "animegarden": "workspace:*",
+    "dotenv": "^16.3.1",
     "simptrad": "^0.1.0",
     "zod": "^3.22.4"
   },

tsconfig.json

@@ -22,6 +22,9 @@
       "@animegarden/scraper": [
         "./packages/scraper/src/index.ts"
       ],
+      "@animegarden/database": [
+        "./packages/database/src/index.ts"
+      ],
       "@animegarden/cli": [
         "./packages/cli/src/index.ts"
       ]