Drizzle migrations

Simple tool that ads up and down and down-one migration capability for drizzle projects.

Automatic migration generation for drizzle was adapted from the PayloadCMS repository.

Warning! This tool is work in progress and was only tested on pg databases, but should be working on sqlite and mysql too. This is also something that drizzle will probably have built-in inside drizzle-kit in the future.

Installation

npm install @aegon_targaryen/drizzle-migrations # or yarn,pnpm,bun

Configuration

To make this work you have to make small changes inside your drizzle.config.ts file

import { defineConfig } from '@aegon_targaryen/drizzle-migrations'
import { drizzle } from 'drizzle-orm/postgres-js'
import postgres from 'postgres'

// notice how we are not using `defineConfig` from drizzle but from `@aegon_targaryen/drizzle-migrations`
export default defineConfig({
  schema: './src/schema.ts',
  dialect: 'postgresql',
  dbCredentials: {  ...  },
  out: './src/migrations',

  // this is default if not provided. schema is for now supported only for postgres,
  migrations: {
    schema: 'public',
    table: 'drizzle_migrations',
  },

  // define your own client using adapter of your choice
  getMigrator: async () => {
    const migrationClient = postgres('postgres://postgres:postgres@localhost/postgres-db', {
      max: 1,
    })

    return drizzle(migrationClient)
  },

  // only needed if you want to use seed commands
  seed: { dirPath: './src/seeders', defaultSeeder: 'db-seeder' }
})

Usage

Generate migration

npm run drizzle-migrations generate --name <migration-name>

Run migrations

npm run drizzle-migrations up

Rollback migrations

npm run drizzle-migrations down

Rollback one migration

npm run drizzle-migrations down-one

Rollback up to specific batch

npm run drizzle-migrations down --bach <batch-number>

Get status of migrations

npm run drizzle-migrations status

Rollback all migrations

npm run drizzle-migrations fresh

Rollback all migrations and run them again

npm run drizzle-migrations refresh

Create seeder

npm run drizzle-migrations seed:create --name <seeder-name>

Run seeders

npm run drizzle-migrations seed:run --name <seeder-name> // default seeder is db-seeder

Example of generated migration

import { sql } from "drizzle-orm";
import type { MigrationArgs } from "@aegon_targaryen/drizzle-migrations";

export async function up({ db }: MigrationArgs<"postgresql">): Promise<void> {
  await db.execute(sql`
        CREATE TABLE IF NOT EXISTS "posts" (
                "id" serial PRIMARY KEY NOT NULL,
                "title" varchar(255),
                "content" varchar,
                "user_id" integer
        );

        CREATE TABLE IF NOT EXISTS "users" (
                "id" serial PRIMARY KEY NOT NULL,
                "name" varchar,
                "slug" uuid
        );

        DO $$ BEGIN
        ALTER TABLE "posts" ADD CONSTRAINT "posts_user_id_users_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."users"("id") ON DELETE no action ON UPDATE no action;
        EXCEPTION
        WHEN duplicate_object THEN null;
        END $$;

        `);
}

export async function down({ db }: MigrationArgs<"postgresql">): Promise<void> {
  await db.execute(sql`
        DROP TABLE "posts";
        DROP TABLE "users";
`);
}