@workos/oagen-emitters 0.2.1 → 0.4.0

package/src/go/enums.ts

@@ -0,0 +1,196 @@
+import type { Enum, EmitterContext, GeneratedFile, Service } from '@workos/oagen';
+import { walkTypeRef } from '@workos/oagen';
+import { className } from './naming.js';
+
+/**
+ * Generate Go typed string enum constants from IR Enum definitions.
+ *
+ * Each enum becomes a named string type + const block:
+ *   type Status string
+ *   const (
+ *     StatusActive   Status = "active"
+ *     StatusInactive Status = "inactive"
+ *   )
+ */
+export function generateEnums(enums: Enum[], ctx: EmitterContext): GeneratedFile[] {
+  if (enums.length === 0) return [];
+
+  const aliasOf = collectEnumAliasOf(enums);
+  const files: GeneratedFile[] = [];
+
+  // Group all enums into a single file per SDK
+  const lines: string[] = [];
+  lines.push(`package ${ctx.namespace}`);
+  lines.push('');
+
+  for (const enumDef of enums) {
+    // If this enum is an alias, emit a simple type alias
+    const canonicalName = aliasOf.get(enumDef.name);
+    if (canonicalName) {
+      const aliasType = className(enumDef.name);
+      const canonicalType = className(canonicalName);
+      lines.push(`// ${aliasType} is an alias for ${canonicalType}.`);
+      lines.push(`type ${aliasType} = ${canonicalType}`);
+      lines.push('');
+      continue;
+    }
+
+    const typeName = className(enumDef.name);
+
+    if (enumDef.values.length === 0) {
+      const humanized = humanize(enumDef.name);
+      lines.push(`// ${typeName} represents ${humanized} values.`);
+      lines.push(`type ${typeName} = string`);
+      lines.push('');
+      continue;
+    }
+
+    // Deduplicate values
+    const seenValues = new Set<string>();
+    const uniqueValues: typeof enumDef.values = [];
+    for (const v of enumDef.values) {
+      const vs = String(v.value);
+      if (!seenValues.has(vs)) {
+        seenValues.add(vs);
+        uniqueValues.push(v);
+      }
+    }
+
+    const humanized = humanize(enumDef.name);
+    lines.push(`// ${typeName} represents ${humanized} values.`);
+    lines.push(`type ${typeName} string`);
+    lines.push('');
+    lines.push('const (');
+
+    const usedNames = new Set<string>();
+    for (const v of uniqueValues) {
+      let constSuffix = className(String(v.value));
+      // Avoid collision with the type itself
+      if (usedNames.has(`${typeName}${constSuffix}`)) {
+        let suffix = 2;
+        while (usedNames.has(`${typeName}${constSuffix}${suffix}`)) suffix++;
+        constSuffix = `${constSuffix}${suffix}`;
+      }
+      const constName = `${typeName}${constSuffix}`;
+      usedNames.add(constName);
+      const valueStr = typeof v.value === 'string' ? `"${v.value}"` : String(v.value);
+      if (v.description) {
+        lines.push(`\t// ${constName} is ${v.description}.`);
+      }
+      if (v.deprecated) {
+        if (v.description) lines.push(`\t//`);
+        lines.push(`\t// Deprecated: this value is deprecated.`);
+      }
+      lines.push(`\t${constName} ${typeName} = ${valueStr}`);
+    }
+    lines.push(')');
+    lines.push('');
+  }
+
+  files.push({
+    path: 'enums.go',
+    content: lines.join('\n'),
+    overwriteExisting: true,
+  });
+
+  return files;
+}
+
+/** Known acronyms to preserve as single tokens during humanization. */
+const HUMANIZE_ACRONYMS: [RegExp, string][] = [
+  [/OAuth/g, 'OAUTH_ACRN'],
+  [/URN/g, 'URN_ACRN'],
+  [/IETF/g, 'IETF_ACRN'],
+  [/API/g, 'API_ACRN'],
+  [/SSO/g, 'SSO_ACRN'],
+  [/PKCE/g, 'PKCE_ACRN'],
+  [/JWT/g, 'JWT_ACRN'],
+  [/MFA/g, 'MFA_ACRN'],
+  [/TOTP/g, 'TOTP_ACRN'],
+  [/SAML/g, 'SAML_ACRN'],
+  [/SCIM/g, 'SCIM_ACRN'],
+  [/OIDC/g, 'OIDC_ACRN'],
+  [/CORS/g, 'CORS_ACRN'],
+  [/RBAC/g, 'RBAC_ACRN'],
+];
+
+const HUMANIZE_RESTORE: [RegExp, string][] = [
+  [/oauth_acrn/g, 'OAuth'],
+  [/urn_acrn/g, 'URN'],
+  [/ietf_acrn/g, 'IETF'],
+  [/api_acrn/g, 'API'],
+  [/sso_acrn/g, 'SSO'],
+  [/pkce_acrn/g, 'PKCE'],
+  [/jwt_acrn/g, 'JWT'],
+  [/mfa_acrn/g, 'MFA'],
+  [/totp_acrn/g, 'TOTP'],
+  [/saml_acrn/g, 'SAML'],
+  [/scim_acrn/g, 'SCIM'],
+  [/oidc_acrn/g, 'OIDC'],
+  [/cors_acrn/g, 'CORS'],
+  [/rbac_acrn/g, 'RBAC'],
+];
+
+function humanize(name: string): string {
+  // Replace known acronyms with placeholders before splitting
+  let s = name;
+  for (const [pattern, replacement] of HUMANIZE_ACRONYMS) {
+    s = s.replace(pattern, replacement);
+  }
+  let result = s.replace(/([a-z])([A-Z])/g, '$1 $2');
+  result = result.replace(/([A-Z]+)([A-Z][a-z])/g, '$1 $2');
+  result = result.toLowerCase();
+  for (const [pattern, replacement] of HUMANIZE_RESTORE) {
+    result = result.replace(pattern, replacement);
+  }
+  return result;
+}
+
+function collectEnumAliasOf(enums: Enum[]): Map<string, string> {
+  const hashGroups = new Map<string, string[]>();
+  for (const enumDef of enums) {
+    const hash = [...enumDef.values]
+      .map((v) => String(v.value))
+      .sort()
+      .join('|');
+    if (!hashGroups.has(hash)) hashGroups.set(hash, []);
+    hashGroups.get(hash)!.push(enumDef.name);
+  }
+
+  const aliasOf = new Map<string, string>();
+  for (const [, names] of hashGroups) {
+    if (names.length <= 1) continue;
+    const sorted = [...names].sort();
+    const canonical = sorted[0];
+    for (let i = 1; i < sorted.length; i++) {
+      aliasOf.set(sorted[i], canonical);
+    }
+  }
+  return aliasOf;
+}
+
+export function assignEnumsToServices(enums: Enum[], services: Service[]): Map<string, string> {
+  const enumToService = new Map<string, string>();
+  const enumNames = new Set(enums.map((e) => e.name));
+
+  for (const service of services) {
+    for (const op of service.operations) {
+      const refs = new Set<string>();
+      const collect = (ref: any) => {
+        walkTypeRef(ref, { enum: (r: any) => refs.add(r.name) });
+      };
+      if (op.requestBody) collect(op.requestBody);
+      collect(op.response);
+      for (const p of [...op.pathParams, ...op.queryParams, ...op.headerParams, ...(op.cookieParams ?? [])]) {
+        collect(p.type);
+      }
+      for (const name of refs) {
+        if (enumNames.has(name) && !enumToService.has(name)) {
+          enumToService.set(name, service.name);
+        }
+      }
+    }
+  }
+
+  return enumToService;
+}

package/src/go/fixtures.ts

@@ -0,0 +1,212 @@
+import type { Model, TypeRef, Enum } from '@workos/oagen';
+import { fileName, fieldName } from './naming.js';
+import { isListMetadataModel, isListWrapperModel } from './models.js';
+
+/**
+ * Prefix mapping for generating realistic ID fixture values.
+ */
+export const ID_PREFIXES: Record<string, string> = {
+  Connection: 'conn_',
+  Organization: 'org_',
+  OrganizationMembership: 'om_',
+  User: 'user_',
+  Directory: 'directory_',
+  DirectoryGroup: 'dir_grp_',
+  DirectoryUser: 'dir_usr_',
+  Invitation: 'inv_',
+  Session: 'session_',
+  AuthenticationFactor: 'auth_factor_',
+  EmailVerification: 'email_verification_',
+  MagicAuth: 'magic_auth_',
+  PasswordReset: 'password_reset_',
+};
+
+/**
+ * Generate JSON fixture files for test data.
+ */
+export function generateFixtures(spec: {
+  models: Model[];
+  enums: Enum[];
+  services: any[];
+}): { path: string; content: string }[] {
+  if (spec.models.length === 0) return [];
+
+  const modelMap = new Map(spec.models.map((m) => [m.name, m]));
+  const enumMap = new Map(spec.enums.map((e) => [e.name, e]));
+  const files: { path: string; content: string }[] = [];
+
+  for (const model of spec.models) {
+    if (isListMetadataModel(model)) continue;
+    if (isListWrapperModel(model)) continue;
+
+    const fixture = model.fields.length === 0 ? {} : generateModelFixture(model, modelMap, enumMap);
+
+    files.push({
+      path: `testdata/${fileName(model.name)}.json`,
+      content: JSON.stringify(fixture, null, 2),
+    });
+  }
+
+  // Generate list fixtures for paginated responses
+  for (const service of spec.services) {
+    for (const op of service.operations) {
+      if (op.pagination) {
+        let itemModel = op.pagination.itemType.kind === 'model' ? modelMap.get(op.pagination.itemType.name) : null;
+        if (itemModel) {
+          const unwrapped = unwrapListModel(itemModel, modelMap);
+          if (unwrapped) itemModel = unwrapped;
+          if (itemModel.fields.length === 0) continue;
+          const fixture = generateModelFixture(itemModel, modelMap, enumMap);
+          const listFixture = {
+            data: [fixture],
+            list_metadata: {
+              before: null,
+              after: null,
+            },
+          };
+          files.push({
+            path: `testdata/list_${fileName(itemModel.name)}.json`,
+            content: JSON.stringify(listFixture, null, 2),
+          });
+        }
+      }
+    }
+  }
+
+  // Deduplicate fixtures with identical content.
+  // When multiple fixtures have the same content, emit one shared file and
+  // rewrite the others as references to the shared path.
+  const contentGroups = new Map<string, string[]>();
+  for (const f of files) {
+    if (!contentGroups.has(f.content)) contentGroups.set(f.content, []);
+    contentGroups.get(f.content)!.push(f.path);
+  }
+
+  const pathRewrites = new Map<string, string>();
+  for (const [_content, paths] of contentGroups) {
+    if (paths.length < 3) continue; // only dedup when 3+ files are identical
+    // Use the shortest path as the canonical shared fixture
+    const sorted = [...paths].sort((a, b) => a.length - b.length);
+    const canonical = sorted[0];
+    for (let i = 1; i < sorted.length; i++) {
+      pathRewrites.set(sorted[i], canonical);
+    }
+  }
+
+  // Remove duplicate files (they'll reference the canonical)
+  const deduped = files.filter((f) => !pathRewrites.has(f.path));
+
+  return deduped;
+}
+
+function unwrapListModel(model: Model, modelMap: Map<string, Model>): Model | null {
+  const dataField = model.fields.find((f) => f.name === 'data');
+  const hasListMetadata = model.fields.some((f) => f.name === 'list_metadata' || f.name === 'listMetadata');
+  if (dataField && hasListMetadata && dataField.type.kind === 'array') {
+    const itemType = dataField.type.items;
+    if (itemType.kind === 'model') {
+      return modelMap.get(itemType.name) ?? null;
+    }
+  }
+  return null;
+}
+
+export function generateModelFixture(
+  model: Model,
+  modelMap: Map<string, Model>,
+  enumMap: Map<string, Enum>,
+): Record<string, any> {
+  const fixture: Record<string, any> = {};
+
+  const seenFieldNames = new Set<string>();
+  const deduplicatedFields = model.fields.filter((f) => {
+    const goName = fieldName(f.name);
+    if (seenFieldNames.has(goName)) return false;
+    seenFieldNames.add(goName);
+    return true;
+  });
+
+  for (const field of deduplicatedFields) {
+    const wireName = field.name;
+    if (field.example !== undefined) {
+      fixture[wireName] = field.example;
+    } else {
+      fixture[wireName] = generateFieldValue(field.type, field.name, model.name, modelMap, enumMap);
+    }
+  }
+
+  return fixture;
+}
+
+function generateFieldValue(
+  ref: TypeRef,
+  fName: string,
+  modelName: string,
+  modelMap: Map<string, Model>,
+  enumMap: Map<string, Enum>,
+): any {
+  switch (ref.kind) {
+    case 'primitive':
+      return generatePrimitiveValue(ref.type, ref.format, fName, modelName);
+    case 'literal':
+      return ref.value;
+    case 'enum': {
+      const e = enumMap.get(ref.name);
+      return e?.values[0]?.value ?? 'unknown';
+    }
+    case 'model': {
+      const nested = modelMap.get(ref.name);
+      if (nested) return generateModelFixture(nested, modelMap, enumMap);
+      return {};
+    }
+    case 'array': {
+      if (ref.items.kind === 'enum') {
+        const e = enumMap.get(ref.items.name);
+        if (e && e.values.length > 0) {
+          return e.values.map((v) => v.value);
+        }
+      }
+      const item = generateFieldValue(ref.items, fName, modelName, modelMap, enumMap);
+      return [item];
+    }
+    case 'nullable':
+      return generateFieldValue(ref.inner, fName, modelName, modelMap, enumMap);
+    case 'union':
+      if (ref.variants.length > 0) {
+        return generateFieldValue(ref.variants[0], fName, modelName, modelMap, enumMap);
+      }
+      return null;
+    case 'map':
+      return {
+        key: generateFieldValue(ref.valueType, 'value', modelName, modelMap, enumMap),
+      };
+  }
+}
+
+function generatePrimitiveValue(type: string, format: string | undefined, name: string, modelName: string): any {
+  switch (type) {
+    case 'string':
+      if (format === 'date-time') return '2023-01-01T00:00:00.000Z';
+      if (format === 'date') return '2023-01-01';
+      if (format === 'uuid') return '00000000-0000-0000-0000-000000000000';
+      if (name === 'id') {
+        const prefix = ID_PREFIXES[modelName] ?? '';
+        return `${prefix}01234`;
+      }
+      if (name.includes('id')) return `${name}_01234`;
+      if (name.includes('email')) return 'test@example.com';
+      if (name.includes('url') || name.includes('uri')) return 'https://example.com';
+      if (name.includes('name')) return 'Test';
+      return `test_${name}`;
+    case 'integer':
+      return 1;
+    case 'number':
+      return 1.0;
+    case 'boolean':
+      return true;
+    case 'unknown':
+      return {};
+    default:
+      return null;
+  }
+}

package/src/go/index.ts

@@ -0,0 +1,84 @@
+import type {
+  Emitter,
+  EmitterContext,
+  FormatCommand,
+  GeneratedFile,
+  ApiSpec,
+  Model,
+  Enum,
+  Service,
+} from '@workos/oagen';
+
+import { generateModels } from './models.js';
+import { enrichModelsFromSpec, getSyntheticEnums } from '../shared/model-utils.js';
+import { generateEnums } from './enums.js';
+import { generateResources } from './resources.js';
+import { generateClient } from './client.js';
+import { generateTests } from './tests.js';
+import { generateManifest } from './manifest.js';
+
+/** Ensure every generated file's content ends with a trailing newline. */
+function ensureTrailingNewlines(files: GeneratedFile[]): GeneratedFile[] {
+  for (const f of files) {
+    if (f.content && !f.content.endsWith('\n')) {
+      f.content += '\n';
+    }
+  }
+  return files;
+}
+
+export const goEmitter: Emitter = {
+  language: 'go',
+
+  generateModels(models: Model[], ctx: EmitterContext): GeneratedFile[] {
+    // Enrich models by flattening oneOf/allOf+oneOf variant fields from the raw spec
+    const enriched = enrichModelsFromSpec(models);
+    return ensureTrailingNewlines(generateModels(enriched, ctx));
+  },
+
+  generateEnums(enums: Enum[], ctx: EmitterContext): GeneratedFile[] {
+    // Merge synthetic enums produced during model enrichment (inline oneOf
+    // definitions) so they get proper type definitions in enums.go.
+    const syntheticEnums = getSyntheticEnums();
+    return ensureTrailingNewlines(generateEnums([...enums, ...syntheticEnums], ctx));
+  },
+
+  generateResources(services: Service[], ctx: EmitterContext): GeneratedFile[] {
+    return ensureTrailingNewlines(generateResources(services, ctx));
+  },
+
+  generateClient(spec: ApiSpec, ctx: EmitterContext): GeneratedFile[] {
+    return ensureTrailingNewlines(generateClient(spec, ctx));
+  },
+
+  generateErrors(): GeneratedFile[] {
+    return [];
+  },
+
+  generateTypeSignatures(): GeneratedFile[] {
+    // Go uses inline type annotations -- no separate type signature files needed
+    return [];
+  },
+
+  generateTests(spec: ApiSpec, ctx: EmitterContext): GeneratedFile[] {
+    return ensureTrailingNewlines(generateTests(spec, ctx));
+  },
+
+  generateManifest(spec: ApiSpec, ctx: EmitterContext): GeneratedFile[] {
+    return ensureTrailingNewlines(generateManifest(spec, ctx));
+  },
+
+  fileHeader(): string {
+    return '// Code generated by oagen. DO NOT EDIT.';
+  },
+
+  formatCommand(targetDir: string): FormatCommand | null {
+    // Pass targetDir as the first path so gofmt formats the entire directory
+    // (including hand-maintained files), not just the generated file list.
+    return {
+      cmd: 'gofmt',
+      args: ['-w', targetDir],
+      batchSize: 999999,
+    };
+  },
+};

package/src/go/manifest.ts

@@ -0,0 +1,36 @@
+import type { ApiSpec, EmitterContext, GeneratedFile } from '@workos/oagen';
+import { resolveMethodName } from './naming.js';
+import { buildServiceAccessPaths } from './client.js';
+import { getMountTarget } from '../shared/resolved-ops.js';
+
+/**
+ * Generate smoke test manifest mapping HTTP operations to SDK methods.
+ */
+export function generateManifest(spec: ApiSpec, ctx: EmitterContext): GeneratedFile[] {
+  const manifest: Record<string, { sdkMethod: string; service: string }> = {};
+  const accessPaths = buildServiceAccessPaths(spec.services, ctx);
+
+  for (const service of spec.services) {
+    let propName = accessPaths.get(service.name);
+    if (!propName) {
+      const mountTarget = getMountTarget(service, ctx);
+      propName = accessPaths.get(mountTarget);
+    }
+    if (!propName) {
+      throw new Error(`Missing public client access path for service ${service.name}`);
+    }
+    for (const op of service.operations) {
+      const httpKey = `${op.httpMethod.toUpperCase()} ${op.path}`;
+      const method = resolveMethodName(op, service, ctx);
+      manifest[httpKey] = { sdkMethod: method, service: propName };
+    }
+  }
+
+  return [
+    {
+      path: 'smoke-manifest.json',
+      content: JSON.stringify(manifest, null, 2),
+      integrateTarget: false,
+    },
+  ];
+}