@pattern-stack/codegen 0.6.6 → 0.6.8

package/examples/auth-integrations/runtime/integrations/oauth/use-cases/list-user-integrations.use-case.ts

@@ -0,0 +1,21 @@
+import { Injectable } from '@nestjs/common';
+import { IntegrationService } from '../../integration.service';
+import type { Integration } from '../../integration.entity';
+
+/**
+ * Lists a user's integrations newest-first. Used by the settings page
+ * (`GET /integrations`) and any "which providers are connected?" UI.
+ *
+ * Returns rows with ciphertexts intact — callers should NOT pass these
+ * to the frontend. Use `IntegrationsService.listByUser` if you need
+ * the consumer-facing facade behavior (which strips ciphertexts before
+ * returning).
+ */
+@Injectable()
+export class ListUserIntegrationsUseCase {
+  constructor(private readonly integrations: IntegrationService) {}
+
+  async execute(userId: string): Promise<Integration[]> {
+    return this.integrations.findByUserId(userId);
+  }
+}

package/examples/auth-integrations/runtime/integrations/oauth/use-cases/mark-integration-requires-reauth.use-case.ts

@@ -0,0 +1,21 @@
+import { Injectable } from '@nestjs/common';
+import { IntegrationService } from '../../integration.service';
+import type { Integration } from '../../integration.entity';
+
+/**
+ * Flips an integration's status to `requires_reauth`. Called when the
+ * refresh path raises `IntegrationBrokenError` (refresh token rejected,
+ * scopes revoked, etc.) — see `OAuth2RefreshStrategy` + `withAuthRetry`.
+ *
+ * Idempotent: calling on an already-broken row is a no-op write.
+ */
+@Injectable()
+export class MarkIntegrationRequiresReauthUseCase {
+  constructor(private readonly integrations: IntegrationService) {}
+
+  async execute(integrationId: string): Promise<Integration> {
+    return this.integrations.update(integrationId, {
+      status: 'requires_reauth',
+    });
+  }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pattern-stack/codegen",
-  "version": "0.6.6",
+  "version": "0.6.8",
   "description": "Entity-driven code generation for full-stack TypeScript applications",
   "license": "MIT",
   "type": "module",
@@ -40,6 +40,7 @@
     "dist",
     "runtime",
     "templates",
+    "examples/auth-integrations/**",
     "src/config/*.mjs",
     "src/schema/naming-config.schema.mjs",
     "src/patterns/registry.ts",

package/templates/entity/new/clean-lite-ps/entity.ejs.t

@@ -18,6 +18,12 @@ import { type InferSelectModel } from 'drizzle-orm';
 import { <%= rel.relatedTable %> } from '<%= rel.importPath %>';
 <%_ } _%>
 <%_ }) _%>
+<%_ if (typeof clpEnumFields !== 'undefined' && clpEnumFields.length > 0) { _%>
+
+<%_ clpEnumFields.forEach(ef => { _%>
+export const <%= ef.enumName %> = pgEnum('<%= ef.dbName %>', [<%- ef.choices.map(c => `'${c}'`).join(', ') %>]);
+<%_ }) _%>
+<%_ } _%>
 
 export const <%= entityNamePlural %> = pgTable(
   '<%= entityNamePlural %>',

package/templates/entity/new/clean-lite-ps/prompt-extension.js

@@ -201,7 +201,7 @@ const EXTERNAL_ID_TRACKING_FIELDS = new Set([
 /**
  * Build a Drizzle column chain for a field
  */
-function buildDrizzleChain(fieldName, field, drizzleType) {
+function buildDrizzleChain(fieldName, field, drizzleType, enumName) {
   const nullable = field.nullable ?? false;
   const required = field.required ?? false;
   const hasDefault = field.default !== undefined && field.default !== null;
@@ -211,7 +211,11 @@ function buildDrizzleChain(fieldName, field, drizzleType) {
   // schemas using z.coerce.date() align with the entity type.
   // `timestamp` already defaults to Date — no mode override needed.
   let chain;
-  if (drizzleType === 'date') {
+  if (drizzleType === 'enum' && enumName) {
+    // Reference the pgEnum declaration emitted at the top of the entity file.
+    // The column name argument keeps the snake_case YAML field name.
+    chain = `${enumName}('${fieldName}')`;
+  } else if (drizzleType === 'date') {
     chain = `${drizzleType}('${fieldName}', { mode: 'date' })`;
   } else {
     chain = `${drizzleType}('${fieldName}')`;
@@ -247,7 +251,15 @@ function processFields(fields) {
     const choices = field.choices;
     const hasChoices = Array.isArray(choices) && choices.length > 0;
 
-    const drizzleType = DRIZZLE_TYPE_MAP[type] || 'text';
+    // Enum-typed fields (or any field with a `choices` list) emit a
+    // Postgres-native pgEnum declaration + column reference, so the
+    // generated `InferSelectModel` type narrows to the literal union
+    // instead of falling back to `string`. Matches the backend pipeline
+    // (templates/entity/new/backend/database/schema.ejs.t:66-104).
+    const drizzleType = hasChoices
+      ? 'enum'
+      : (DRIZZLE_TYPE_MAP[type] || 'text');
+    const enumName = hasChoices ? camelCase(fieldName) + 'Enum' : null;
     const tsType = hasChoices
       ? choices.map((c) => `'${c}'`).join(' | ')
       : (TS_TYPE_MAP[type] || 'unknown');
@@ -255,7 +267,7 @@ function processFields(fields) {
       ? `z.enum([${choices.map((c) => `'${c}'`).join(', ')}])`
       : (ZOD_TYPE_MAP[type] || 'z.unknown()');
 
-    const drizzleChain = buildDrizzleChain(fieldName, field, drizzleType);
+    const drizzleChain = buildDrizzleChain(fieldName, field, drizzleType, enumName);
 
     processed.push({
       name: fieldName,
@@ -271,6 +283,7 @@ function processFields(fields) {
       drizzleChain,
       choices,
       hasChoices,
+      enumName,
     });
   }
 
@@ -355,6 +368,12 @@ function collectDrizzleImports(processedFields, belongsTo, hasTimestamps, hasSof
   const imports = new Set(['pgTable', 'uuid']);
 
   for (const field of processedFields) {
+    if (field.drizzleType === 'enum') {
+      // Enum columns reference a `pgEnum` declaration emitted at the top
+      // of the entity file; the helper itself comes from drizzle-orm/pg-core.
+      imports.add('pgEnum');
+      continue;
+    }
     const importName = DRIZZLE_IMPORT_MAP[field.drizzleType];
     if (importName) imports.add(importName);
   }
@@ -783,6 +802,18 @@ export function buildCleanLitePsLocals(definition, baseLocals) {
   const fkFieldNames = new Set(belongsTo.map((r) => r.field));
   const nonFkFields = processedFields.filter((f) => !fkFieldNames.has(f.name));
 
+  // Enum field declarations — surface a separate collection so the entity
+  // template can emit `export const xEnum = pgEnum('x', [...])` ahead of
+  // the `pgTable(...)` block. Both FK-filtered and unfiltered processing
+  // include the same enum fields; they're never FKs.
+  const clpEnumFields = processedFields
+    .filter((f) => f.hasChoices && f.enumName)
+    .map((f) => ({
+      enumName: f.enumName,
+      dbName: f.name,
+      choices: f.choices,
+    }));
+
   // Drizzle imports needed
   const drizzleEntityImports = collectDrizzleImports(processedFields, belongsTo, hasTimestamps, hasSoftDelete, hasExternalIdTracking);
   // Whether relations() import is needed
@@ -980,6 +1011,7 @@ export function buildCleanLitePsLocals(definition, baseLocals) {
     // Drizzle
     clpDrizzleImports: drizzleEntityImports,
     clpHasRelationsBlock: hasRelationsBlock,
+    clpEnumFields,
 
     // Declarative queries
     processedQueries,