Add moddatetime extension support

docs/repl/allExtensions.ts

@@ -40,3 +40,4 @@ export { unaccent } from '@electric-sql/pglite/contrib/unaccent'
 export { uuid_ossp } from '@electric-sql/pglite/contrib/uuid_ossp'
 export { vector } from '@electric-sql/pglite-pgvector'
 export { pgmq } from '@electric-sql/pglite-pgmq'
+export { moddatetime } from '@electric-sql/pglite/contrib/moddatetime'

packages/pglite/package.json

@@ -110,7 +110,6 @@
     "!dist/pglite.html",
     "!dist/pgrowlocks.tar.gz",
     "!dist/pgstattuple.tar.gz",
-    "!dist/spi.tar.gz",
     "!dist/test_decoding.tar.gz",
     "!dist/vacuumlo.tar.gz",
     "!dist/xml2.tar.gz",

packages/pglite/src/contrib/moddatetime.ts

@@ -0,0 +1,16 @@
+import type {
+  Extension,
+  ExtensionSetupResult,
+  PGliteInterface,
+} from '../interface'
+
+const setup = async (_pg: PGliteInterface, _emscriptenOpts: any) => {
+  return {
+    bundlePath: new URL('../../release/spi.tar.gz', import.meta.url),
+  } satisfies ExtensionSetupResult
+}
+
+export const moddatetime = {
+  name: 'moddatetime',
+  setup,
+} satisfies Extension

packages/pglite/tests/contrib/moddatetime.test.ts

@@ -0,0 +1,61 @@
+import { describe, it, expect, beforeEach, afterEach } from 'vitest'
+import { PGlite } from '../../dist/index.js'
+import { moddatetime } from '../../dist/contrib/moddatetime.js'
+
+describe('moddatetime', () => {
+  let pg: PGlite
+  let dataDirArchive: File | Blob
+  beforeEach(async () => {
+    if (!dataDirArchive) {
+      pg = await PGlite.create({
+        extensions: { moddatetime },
+      })
+      dataDirArchive = await pg.dumpDataDir('gzip')
+    } else {
+      pg = await PGlite.create({
+        extensions: { moddatetime },
+        loadDataDir: dataDirArchive,
+      })
+    }
+    await pg.exec('CREATE EXTENSION IF NOT EXISTS "moddatetime";')
+  })
+  afterEach(async () => {
+    if (!pg.closed) {
+      await pg.close()
+    }
+  })
+
+  it('moddatetime', async () => {
+    await pg.exec(`
+      CREATE OR REPLACE FUNCTION manage_updated_at(_tbl regclass) RETURNS VOID AS $$
+      BEGIN
+          EXECUTE format('CREATE TRIGGER set_updated_at BEFORE UPDATE ON %s
+                          FOR EACH ROW EXECUTE PROCEDURE moddatetime("updatedAt")', _tbl);
+      END;
+      $$ LANGUAGE plpgsql;
+    `)
+
+    await pg.exec(`
+      CREATE TABLE "users" (
+        "id" SERIAL PRIMARY KEY,
+        "name" TEXT NOT NULL,
+        "createdAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW(),
+        "updatedAt" TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW()
+      );
+
+      SELECT manage_updated_at('users');
+    `)
+
+    await pg.exec(`INSERT INTO "users" ("name") VALUES ('test');`)
+    const res = await pg.query(`SELECT * FROM "users"`)
+    expect(res.rows[0].updatedAt.getTime()).toBe(
+      res.rows[0].createdAt.getTime(),
+    )
+
+    await pg.exec(`UPDATE "users" SET "name" = 'test2' WHERE "name" = 'test';`)
+    const res2 = await pg.query(`SELECT * FROM "users"`)
+    expect(res2.rows[0].updatedAt.getTime()).toBeGreaterThan(
+      res2.rows[0].createdAt.getTime(),
+    )
+  })
+})