Function extractEntityFrameworkModels

src/ast/extract-csharp.ts:111–167

(
  _filePath: string,
  content: string
)

// ─── Entity Framework models ───────────────────────────────────────────────

export function extractEntityFrameworkModels(
  _filePath: string,
  content: string
): SchemaModel[] {
  const models: SchemaModel[] = [];

  // Find DbContext subclass
  if (
    !content.includes("DbContext") &&
    !content.includes("DbSet<")
  ) {
    return models;
  }

  // Extract each DbSet<ModelName> property
  const dbSetPattern = /DbSet\s*<\s*(\w+)\s*>/g;
  const modelNames = new Set<string>();
  let m: RegExpExecArray | null;
  while ((m = dbSetPattern.exec(content)) !== null) {
    modelNames.add(m[1]);
  }

  if (modelNames.size === 0) return models;

  // For each model name, try to find class definition in same file
  for (const modelName of modelNames) {
    const classPattern = new RegExp(
      `class\\s+${modelName}\\s*(?::\\s*[\\w<>, ]+)?\\s*\\{([\\s\\S]*?)\\n\\s*\\}`,
      "m"
    );
    const classMatch = content.match(classPattern);

    if (classMatch) {
      const body = classMatch[1];
      const fields = extractCSharpProperties(body);
      if (fields.length > 0) {
        models.push({
          name: modelName,
          fields,
          relations: extractCSharpRelations(body),
          orm: "entity-framework",
        });
        continue;
      }
    }

    // Model class not in this file — just record the name from DbSet
    models.push({
      name: modelName,
      fields: [],
      relations: [],
      orm: "entity-framework",
    });
  }

  return models;
}