driftdetect-core 0.1.3 → 0.3.0

package/dist/parsers/tree-sitter/pydantic/inheritance-resolver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"inheritance-resolver.js","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/inheritance-resolver.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAIH,OAAO,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AAEnD,+CAA+C;AAC/C,6BAA6B;AAC7B,+CAA+C;AAE/C;;;;;;;;GAQG;AACH,MAAM,OAAO,mBAAmB;IAC9B;;;;;;OAMG;IACH,aAAa,CACX,KAAwB,EACxB,SAAyC;QAEzC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;QAClC,OAAO,IAAI,CAAC,sBAAsB,CAAC,KAAK,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAChE,CAAC;IAED;;OAEG;IACK,sBAAsB,CAC5B,KAAwB,EACxB,SAAyC,EACzC,OAAoB;QAEpB,6BAA6B;QAC7B,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5B,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAExB,MAAM,eAAe,GAAwB,EAAE,CAAC;QAEhD,sCAAsC;QACtC,KAAK,MAAM,QAAQ,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;YAE/C,6BAA6B;YAC7B,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;gBAC5B,SAAS;YACX,CAAC;YAED,kBAAkB;YAClB,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;gBACvB,SAAS;YACX,CAAC;YAED,wBAAwB;YACxB,MAAM,WAAW,GAAG,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACjD,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,SAAS;YACX,CAAC;YAED,gCAAgC;YAChC,MAAM,YAAY,GAAG,IAAI,CAAC,sBAAsB,CAC9C,WAAW,EACX,SAAS,EACT,OAAO,CACR,CAAC;YAEF,4CAA4C;YAC5C,MAAM,iBAAiB,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC;gBACpD,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,YAAY,EAAE,WAAW,EAAE,QAAQ,CAAC,QAAQ,CAAC;gBAC3E,CAAC,CAAC,YAAY,CAAC;YAEjB,0CAA0C;YAC1C,KAAK,MAAM,KAAK,IAAI,iBAAiB,EAAE,CAAC;gBACtC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;oBACxD,eAAe,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC;QACH,CAAC;QAED,sCAAsC;QACtC,MAAM,SAAS,GAAG,CAAC,GAAG,eAAe,CAAC,CAAC;QACvC,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;YACjC,MAAM,aAAa,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI,CAAC,CAAC;YACxE,IAAI,aAAa,IAAI,CAAC,EAAE,CAAC;gBACvB,SAAS,CAAC,aAAa,CAAC,GAAG,KAAK,CAAC;YACnC,CAAC;iBAAM,CAAC;gBACN,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,OAAe;QAC5B,kCAAkC;QAClC,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAEnE,IAAI,YAAY,EAAE,CAAC;YACjB,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,GAAG,YAAY,CAAC;YACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;YAEnD,OAAO;gBACL,IAAI,EAAE,IAAI,IAAI,EAAE;gBAChB,QAAQ,EAAE,IAAI;gBACd,QAAQ;gBACR,cAAc,EAAE,qBAAqB,CAAC,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC;gBACrD,SAAS,EAAE,IAAI,KAAK,SAAS;aAC9B,CAAC;QACJ,CAAC;QAED,oBAAoB;QACpB,OAAO;YACL,IAAI,EAAE,OAAO;YACb,QAAQ,EAAE,IAAI;YACd,QAAQ,EAAE,EAAE;YACZ,cAAc,EAAE,qBAAqB,CAAC,GAAG,CAAC,OAAO,CAAC;YAClD,SAAS,EAAE,OAAO,KAAK,SAAS;SACjC,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,OAAe;QACnC,MAAM,IAAI,GAAa,EAAE,CAAC;QAC1B,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;YAC3B,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjC,KAAK,EAAE,CAAC;gBACR,OAAO,IAAI,IAAI,CAAC;YAClB,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACxC,KAAK,EAAE,CAAC;gBACR,OAAO,IAAI,IAAI,CAAC;YAClB,CAAC;iBAAM,IAAI,IAAI,KAAK,GAAG,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;gBACvC,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;oBACnB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC5B,CAAC;gBACD,OAAO,GAAG,EAAE,CAAC;YACf,CAAC;iBAAM,CAAC;gBACN,OAAO,IAAI,IAAI,CAAC;YAClB,CAAC;QACH,CAAC;QAED,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;YACnB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC5B,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,sBAAsB,CAC5B,MAA2B,EAC3B,WAA8B,EAC9B,QAAkB;QAElB,sEAAsE;QACtE,MAAM,aAAa,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAE1D,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,IAAI,aAAa,CAAC,MAAM,KAAK,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC3E,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,0BAA0B;QAC1B,MAAM,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAC;QAChD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9C,aAAa,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAE,EAAE,QAAQ,CAAC,CAAC,CAAE,CAAC,CAAC;QACrD,CAAC;QAED,gCAAgC;QAChC,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAC5B,GAAG,KAAK;YACR,IAAI,EAAE,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,aAAa,CAAC;SACrD,CAAC,CAAC,CAAC;IACN,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,KAAwB;QAChD,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YAC/B,IAAI,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;gBAChC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;gBAC9C,IAAI,KAAK,EAAE,CAAC;oBACV,OAAO,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED;;OAEG;IACK,cAAc,CACpB,IAAc,EACd,aAAkC;QAElC,2CAA2C;QAC3C,MAAM,WAAW,GAAG,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjD,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO;gBACL,GAAG,IAAI;gBACP,IAAI,EAAE,WAAW;gBACjB,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,WAAW,CAAC;aAC9C,CAAC;QACJ,CAAC;QAED,2CAA2C;QAC3C,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,OAAO;gBACL,GAAG,IAAI;gBACP,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,aAAa,CAAC,CAAC;aACtE,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACH,mBAAmB,CACjB,KAAwB,EACxB,SAAyC;QAEzC,MAAM,KAAK,GAAwB,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;QAElC,IAAI,CAAC,qBAAqB,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QAE7D,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACK,qBAAqB,CAC3B,KAAwB,EACxB,SAAyC,EACzC,KAA0B,EAC1B,OAAoB;QAEpB,IAAI,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAExB,kCAAkC;QAClC,KAAK,MAAM,QAAQ,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;YAE/C,IAAI,QAAQ,CAAC,cAAc,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;gBAClD,SAAS;YACX,CAAC;YAED,MAAM,WAAW,GAAG,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACjD,IAAI,WAAW,EAAE,CAAC;gBAChB,IAAI,CAAC,qBAAqB,CAAC,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;YACrE,CAAC;QACH,CAAC;QAED,iBAAiB;QACjB,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,YAAY,CACV,KAAwB,EACxB,YAAoB,EACpB,SAAyC;QAEzC,MAAM,KAAK,GAAG,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACzD,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,eAAe,CACb,YAAoB,EACpB,SAAyC;QAEzC,MAAM,WAAW,GAAwB,EAAE,CAAC;QAE5C,KAAK,MAAM,KAAK,IAAI,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC;YACvC,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY;gBAC3B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,YAAY,EAAE,SAAS,CAAC,EAAE,CAAC;gBACtD,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;CACF"}

package/dist/parsers/tree-sitter/pydantic/pydantic-extractor.d.ts

@@ -0,0 +1,102 @@
+/**
+ * Pydantic Model Extractor
+ *
+ * Main orchestrator for extracting Pydantic model information.
+ * Coordinates field, validator, config, and inheritance extraction.
+ *
+ * @module pydantic/pydantic-extractor
+ */
+import type { TreeSitterTree } from '../types.js';
+import type { PydanticModelInfo, PydanticFieldInfo } from '../types.js';
+import type { ExtractionContext } from './types.js';
+/**
+ * Main Pydantic model extractor.
+ *
+ * Orchestrates extraction of:
+ * - Model definitions
+ * - Field definitions with types and constraints
+ * - Validators (v1 and v2)
+ * - Model configuration
+ * - Inheritance resolution
+ */
+export declare class PydanticExtractor {
+    private readonly fieldExtractor;
+    private readonly validatorExtractor;
+    private readonly configExtractor;
+    private readonly inheritanceResolver;
+    constructor();
+    /**
+     * Extract all Pydantic models from a tree-sitter tree.
+     *
+     * @param tree - The parsed tree-sitter tree
+     * @param source - Original source code
+     * @param options - Extraction options
+     * @returns Array of extracted model information
+     */
+    extractModels(tree: TreeSitterTree, source: string, options?: Partial<ExtractionContext>): PydanticModelInfo[];
+    /**
+     * Extract a single model with resolved inheritance.
+     *
+     * @param modelName - Name of the model to extract
+     * @param allModels - Map of all models
+     * @returns Model with resolved inherited fields
+     */
+    resolveModel(modelName: string, allModels: Map<string, PydanticModelInfo>): PydanticModelInfo | null;
+    /**
+     * Get all models as a map for easy lookup.
+     */
+    modelsToMap(models: PydanticModelInfo[]): Map<string, PydanticModelInfo>;
+    /**
+     * Extract a single Pydantic model from a class definition.
+     */
+    private extractModel;
+    /**
+     * Find all model names in the tree (first pass).
+     */
+    private findModelNames;
+    /**
+     * Find all class definitions in the tree.
+     */
+    private findClassDefinitions;
+    /**
+     * Extract type aliases from the tree.
+     */
+    private extractTypeAliases;
+    /**
+     * Extract base classes from a class definition.
+     */
+    private extractBaseClasses;
+    /**
+     * Check if base classes indicate a Pydantic model.
+     */
+    private isPydanticModel;
+    /**
+     * Extract decorators from a class.
+     */
+    private extractDecorators;
+    /**
+     * Extract docstring from a class body.
+     */
+    private extractDocstring;
+    /**
+     * Detect Pydantic version from model structure.
+     */
+    private detectPydanticVersion;
+    /**
+     * Walk the tree and call visitor for each node.
+     */
+    private walkTree;
+}
+/**
+ * Convert extracted models to a flat field list for contracts.
+ */
+export declare function modelsToContractFields(models: PydanticModelInfo[], modelName: string): PydanticFieldInfo[] | null;
+/**
+ * Find a model by name in the extracted models.
+ */
+export declare function findModel(models: PydanticModelInfo[], name: string): PydanticModelInfo | null;
+/**
+ * Get all model names from extracted models.
+ */
+export declare function getModelNames(models: PydanticModelInfo[]): string[];
+//# sourceMappingURL=pydantic-extractor.d.ts.map

package/dist/parsers/tree-sitter/pydantic/pydantic-extractor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pydantic-extractor.d.ts","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/pydantic-extractor.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAkB,cAAc,EAAE,MAAM,aAAa,CAAC;AAClE,OAAO,KAAK,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AACxE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAWpD;;;;;;;;;GASG;AACH,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAiB;IAChD,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAqB;IACxD,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAkB;IAClD,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAsB;;IAS1D;;;;;;;OAOG;IACH,aAAa,CACX,IAAI,EAAE,cAAc,EACpB,MAAM,EAAE,MAAM,EACd,OAAO,GAAE,OAAO,CAAC,iBAAiB,CAAM,GACvC,iBAAiB,EAAE;IA2BtB;;;;;;OAMG;IACH,YAAY,CACV,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,GACxC,iBAAiB,GAAG,IAAI;IAe3B;;OAEG;IACH,WAAW,CAAC,MAAM,EAAE,iBAAiB,EAAE,GAAG,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC;IAYxE;;OAEG;IACH,OAAO,CAAC,YAAY;IAqEpB;;OAEG;IACH,OAAO,CAAC,cAAc;IAiBtB;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAkB5B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IA+B1B;;OAEG;IACH,OAAO,CAAC,kBAAkB;IA+B1B;;OAEG;IACH,OAAO,CAAC,eAAe;IAsBvB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAkBzB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAyBxB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAuD7B;;OAEG;IACH,OAAO,CAAC,QAAQ;CASjB;AAMD;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,MAAM,EAAE,iBAAiB,EAAE,EAC3B,SAAS,EAAE,MAAM,GAChB,iBAAiB,EAAE,GAAG,IAAI,CAG5B;AAED;;GAEG;AACH,wBAAgB,SAAS,CACvB,MAAM,EAAE,iBAAiB,EAAE,EAC3B,IAAI,EAAE,MAAM,GACX,iBAAiB,GAAG,IAAI,CAE1B;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,MAAM,EAAE,iBAAiB,EAAE,GAAG,MAAM,EAAE,CAEnE"}

package/dist/parsers/tree-sitter/pydantic/pydantic-extractor.js

@@ -0,0 +1,399 @@
+/**
+ * Pydantic Model Extractor
+ *
+ * Main orchestrator for extracting Pydantic model information.
+ * Coordinates field, validator, config, and inheritance extraction.
+ *
+ * @module pydantic/pydantic-extractor
+ */
+import { createExtractionContext, PYDANTIC_BASE_CLASSES } from './types.js';
+import { FieldExtractor } from './field-extractor.js';
+import { ValidatorExtractor } from './validator-extractor.js';
+import { ConfigExtractor } from './config-extractor.js';
+import { InheritanceResolver } from './inheritance-resolver.js';
+// ============================================
+// Pydantic Extractor Class
+// ============================================
+/**
+ * Main Pydantic model extractor.
+ *
+ * Orchestrates extraction of:
+ * - Model definitions
+ * - Field definitions with types and constraints
+ * - Validators (v1 and v2)
+ * - Model configuration
+ * - Inheritance resolution
+ */
+export class PydanticExtractor {
+    fieldExtractor;
+    validatorExtractor;
+    configExtractor;
+    inheritanceResolver;
+    constructor() {
+        this.fieldExtractor = new FieldExtractor();
+        this.validatorExtractor = new ValidatorExtractor();
+        this.configExtractor = new ConfigExtractor();
+        this.inheritanceResolver = new InheritanceResolver();
+    }
+    /**
+     * Extract all Pydantic models from a tree-sitter tree.
+     *
+     * @param tree - The parsed tree-sitter tree
+     * @param source - Original source code
+     * @param options - Extraction options
+     * @returns Array of extracted model information
+     */
+    extractModels(tree, source, options = {}) {
+        // First pass: find all model names
+        const modelNames = this.findModelNames(tree.rootNode);
+        // Create context with known models
+        const context = createExtractionContext(source, {
+            ...options,
+            knownModels: new Set(modelNames),
+        });
+        // Extract type aliases
+        this.extractTypeAliases(tree.rootNode, context);
+        // Second pass: extract full model info
+        const models = [];
+        const classNodes = this.findClassDefinitions(tree.rootNode);
+        for (const classNode of classNodes) {
+            const model = this.extractModel(classNode, context);
+            if (model) {
+                models.push(model);
+            }
+        }
+        return models;
+    }
+    /**
+     * Extract a single model with resolved inheritance.
+     *
+     * @param modelName - Name of the model to extract
+     * @param allModels - Map of all models
+     * @returns Model with resolved inherited fields
+     */
+    resolveModel(modelName, allModels) {
+        const model = allModels.get(modelName);
+        if (!model)
+            return null;
+        const resolvedFields = this.inheritanceResolver.resolveFields(model, allModels);
+        return {
+            ...model,
+            fields: resolvedFields,
+        };
+    }
+    /**
+     * Get all models as a map for easy lookup.
+     */
+    modelsToMap(models) {
+        const map = new Map();
+        for (const model of models) {
+            map.set(model.name, model);
+        }
+        return map;
+    }
+    // ============================================
+    // Model Extraction
+    // ============================================
+    /**
+     * Extract a single Pydantic model from a class definition.
+     */
+    extractModel(classNode, context) {
+        // Get class name
+        const nameNode = classNode.childForFieldName('name');
+        if (!nameNode)
+            return null;
+        const name = nameNode.text;
+        // Get base classes
+        const bases = this.extractBaseClasses(classNode);
+        // Check if this is a Pydantic model
+        if (!this.isPydanticModel(bases)) {
+            return null;
+        }
+        // Get class body
+        const bodyNode = classNode.childForFieldName('body');
+        if (!bodyNode)
+            return null;
+        // Extract components
+        const fields = this.fieldExtractor.extractFields(bodyNode, context);
+        const validators = this.validatorExtractor.extractValidators(bodyNode, context);
+        const config = this.configExtractor.extractConfig(bodyNode, context);
+        // Extract decorators
+        const decorators = this.extractDecorators(classNode);
+        // Extract docstring
+        const docstring = this.extractDocstring(bodyNode);
+        // Detect Pydantic version
+        const version = this.detectPydanticVersion(bodyNode, bases);
+        // Build position info
+        const positions = context.includePositions
+            ? {
+                startPosition: {
+                    row: classNode.startPosition.row,
+                    column: classNode.startPosition.column,
+                },
+                endPosition: {
+                    row: classNode.endPosition.row,
+                    column: classNode.endPosition.column,
+                },
+            }
+            : {
+                startPosition: { row: 0, column: 0 },
+                endPosition: { row: 0, column: 0 },
+            };
+        return {
+            name,
+            bases,
+            fields,
+            validators,
+            config,
+            decorators,
+            docstring,
+            version,
+            ...positions,
+        };
+    }
+    // ============================================
+    // Discovery Methods
+    // ============================================
+    /**
+     * Find all model names in the tree (first pass).
+     */
+    findModelNames(rootNode) {
+        const names = [];
+        const classNodes = this.findClassDefinitions(rootNode);
+        for (const classNode of classNodes) {
+            const nameNode = classNode.childForFieldName('name');
+            if (!nameNode)
+                continue;
+            const bases = this.extractBaseClasses(classNode);
+            if (this.isPydanticModel(bases)) {
+                names.push(nameNode.text);
+            }
+        }
+        return names;
+    }
+    /**
+     * Find all class definitions in the tree.
+     */
+    findClassDefinitions(node) {
+        const classes = [];
+        this.walkTree(node, (n) => {
+            if (n.type === 'class_definition') {
+                classes.push(n);
+            }
+            // Also check decorated definitions
+            if (n.type === 'decorated_definition') {
+                for (const child of n.namedChildren) {
+                    if (child.type === 'class_definition') {
+                        classes.push(child);
+                    }
+                }
+            }
+        });
+        return classes;
+    }
+    /**
+     * Extract type aliases from the tree.
+     */
+    extractTypeAliases(rootNode, context) {
+        this.walkTree(rootNode, (node) => {
+            if (node.type !== 'expression_statement')
+                return;
+            const inner = node.namedChildren[0];
+            if (!inner || inner.type !== 'assignment')
+                return;
+            const left = inner.childForFieldName('left');
+            const right = inner.childForFieldName('right');
+            if (!left || !right)
+                return;
+            if (left.type !== 'identifier')
+                return;
+            // Check if this looks like a type alias
+            const name = left.text;
+            const value = right.text;
+            // Simple heuristic: uppercase first letter or ends with common type suffixes
+            if (/^[A-Z]/.test(name) || name.endsWith('Type') || name.endsWith('T')) {
+                context.typeAliases.set(name, value);
+            }
+        });
+    }
+    // ============================================
+    // Base Class Handling
+    // ============================================
+    /**
+     * Extract base classes from a class definition.
+     */
+    extractBaseClasses(classNode) {
+        const bases = [];
+        const argumentList = classNode.childForFieldName('superclasses');
+        if (!argumentList)
+            return bases;
+        for (const child of argumentList.namedChildren) {
+            // Handle simple identifiers
+            if (child.type === 'identifier') {
+                bases.push(child.text);
+            }
+            // Handle attributes (e.g., pydantic.BaseModel)
+            else if (child.type === 'attribute') {
+                bases.push(child.text);
+            }
+            // Handle subscripts (e.g., Generic[T])
+            else if (child.type === 'subscript') {
+                bases.push(child.text);
+            }
+            // Handle calls (e.g., some_factory())
+            else if (child.type === 'call') {
+                const func = child.childForFieldName('function');
+                if (func) {
+                    bases.push(func.text);
+                }
+            }
+        }
+        return bases;
+    }
+    /**
+     * Check if base classes indicate a Pydantic model.
+     */
+    isPydanticModel(bases) {
+        return bases.some((base) => {
+            // Direct match
+            if (PYDANTIC_BASE_CLASSES.has(base))
+                return true;
+            // Check without generic args
+            const baseName = base.split('[')[0];
+            if (baseName && PYDANTIC_BASE_CLASSES.has(baseName))
+                return true;
+            // Check for common patterns
+            if (base.includes('BaseModel') || base.includes('BaseSettings')) {
+                return true;
+            }
+            return false;
+        });
+    }
+    // ============================================
+    // Decorator & Docstring Extraction
+    // ============================================
+    /**
+     * Extract decorators from a class.
+     */
+    extractDecorators(classNode) {
+        const decorators = [];
+        const parent = classNode.parent;
+        if (parent?.type === 'decorated_definition') {
+            for (const child of parent.namedChildren) {
+                if (child.type === 'decorator') {
+                    const content = child.namedChildren[0];
+                    if (content) {
+                        decorators.push(content.text);
+                    }
+                }
+            }
+        }
+        return decorators;
+    }
+    /**
+     * Extract docstring from a class body.
+     */
+    extractDocstring(bodyNode) {
+        if (bodyNode.namedChildCount === 0)
+            return null;
+        const firstChild = bodyNode.namedChildren[0];
+        if (firstChild?.type !== 'expression_statement')
+            return null;
+        const expr = firstChild.namedChildren[0];
+        if (expr?.type !== 'string')
+            return null;
+        // Remove quotes
+        const text = expr.text;
+        if (text.startsWith('"""') || text.startsWith("'''")) {
+            return text.slice(3, -3).trim();
+        }
+        if (text.startsWith('"') || text.startsWith("'")) {
+            return text.slice(1, -1).trim();
+        }
+        return null;
+    }
+    // ============================================
+    // Version Detection
+    // ============================================
+    /**
+     * Detect Pydantic version from model structure.
+     */
+    detectPydanticVersion(bodyNode, bases) {
+        // Check for v2 indicators
+        for (const child of bodyNode.namedChildren) {
+            // model_config is v2
+            if (child.type === 'expression_statement') {
+                const inner = child.namedChildren[0];
+                if (inner?.type === 'assignment') {
+                    const left = inner.childForFieldName('left');
+                    if (left?.text === 'model_config') {
+                        return 2;
+                    }
+                }
+            }
+            // Config class is v1
+            if (child.type === 'class_definition') {
+                const name = child.childForFieldName('name');
+                if (name?.text === 'Config') {
+                    return 1;
+                }
+            }
+            // Check for v2 validators
+            if (child.type === 'decorated_definition') {
+                for (const decorator of child.namedChildren) {
+                    if (decorator.type === 'decorator') {
+                        const content = decorator.namedChildren[0];
+                        const text = content?.text ?? '';
+                        if (text.includes('field_validator') || text.includes('model_validator')) {
+                            return 2;
+                        }
+                        if (text.includes('validator') || text.includes('root_validator')) {
+                            return 1;
+                        }
+                    }
+                }
+            }
+        }
+        // Check base classes for hints
+        if (bases.some((b) => b.includes('BaseSettings'))) {
+            // BaseSettings exists in both, but v2 has different import
+            return 'unknown';
+        }
+        return 'unknown';
+    }
+    // ============================================
+    // Tree Walking
+    // ============================================
+    /**
+     * Walk the tree and call visitor for each node.
+     */
+    walkTree(node, visitor) {
+        visitor(node);
+        for (const child of node.children) {
+            this.walkTree(child, visitor);
+        }
+    }
+}
+// ============================================
+// Utility Functions
+// ============================================
+/**
+ * Convert extracted models to a flat field list for contracts.
+ */
+export function modelsToContractFields(models, modelName) {
+    const model = models.find((m) => m.name === modelName);
+    return model?.fields ?? null;
+}
+/**
+ * Find a model by name in the extracted models.
+ */
+export function findModel(models, name) {
+    return models.find((m) => m.name === name) ?? null;
+}
+/**
+ * Get all model names from extracted models.
+ */
+export function getModelNames(models) {
+    return models.map((m) => m.name);
+}
+//# sourceMappingURL=pydantic-extractor.js.map

package/dist/parsers/tree-sitter/pydantic/pydantic-extractor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pydantic-extractor.js","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/pydantic-extractor.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAKH,OAAO,EAAE,uBAAuB,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AAC5E,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAEhE,+CAA+C;AAC/C,2BAA2B;AAC3B,+CAA+C;AAE/C;;;;;;;;;GASG;AACH,MAAM,OAAO,iBAAiB;IACX,cAAc,CAAiB;IAC/B,kBAAkB,CAAqB;IACvC,eAAe,CAAkB;IACjC,mBAAmB,CAAsB;IAE1D;QACE,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;QAC3C,IAAI,CAAC,kBAAkB,GAAG,IAAI,kBAAkB,EAAE,CAAC;QACnD,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC7C,IAAI,CAAC,mBAAmB,GAAG,IAAI,mBAAmB,EAAE,CAAC;IACvD,CAAC;IAED;;;;;;;OAOG;IACH,aAAa,CACX,IAAoB,EACpB,MAAc,EACd,UAAsC,EAAE;QAExC,mCAAmC;QACnC,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAEtD,mCAAmC;QACnC,MAAM,OAAO,GAAG,uBAAuB,CAAC,MAAM,EAAE;YAC9C,GAAG,OAAO;YACV,WAAW,EAAE,IAAI,GAAG,CAAC,UAAU,CAAC;SACjC,CAAC,CAAC;QAEH,uBAAuB;QACvB,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAEhD,uCAAuC;QACvC,MAAM,MAAM,GAAwB,EAAE,CAAC;QACvC,MAAM,UAAU,GAAG,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE5D,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACpD,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;OAMG;IACH,YAAY,CACV,SAAiB,EACjB,SAAyC;QAEzC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QAExB,MAAM,cAAc,GAAG,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAC3D,KAAK,EACL,SAAS,CACV,CAAC;QAEF,OAAO;YACL,GAAG,KAAK;YACR,MAAM,EAAE,cAAc;SACvB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,MAA2B;QACrC,MAAM,GAAG,GAAG,IAAI,GAAG,EAA6B,CAAC;QACjD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,+CAA+C;IAC/C,mBAAmB;IACnB,+CAA+C;IAE/C;;OAEG;IACK,YAAY,CAClB,SAAyB,EACzB,OAA0B;QAE1B,iBAAiB;QACjB,MAAM,QAAQ,GAAG,SAAS,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACrD,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC;QAC3B,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;QAE3B,mBAAmB;QACnB,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAEjD,oCAAoC;QACpC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;YACjC,OAAO,IAAI,CAAC;QACd,CAAC;QAED,iBAAiB;QACjB,MAAM,QAAQ,GAAG,SAAS,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACrD,IAAI,CAAC,QAAQ;YAAE,OAAO,IAAI,CAAC;QAE3B,qBAAqB;QACrB,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACpE,MAAM,UAAU,GAAG,IAAI,CAAC,kBAAkB,CAAC,iBAAiB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAChF,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAErE,qBAAqB;QACrB,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAErD,oBAAoB;QACpB,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAElD,0BAA0B;QAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,qBAAqB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAE5D,sBAAsB;QACtB,MAAM,SAAS,GAAG,OAAO,CAAC,gBAAgB;YACxC,CAAC,CAAC;gBACE,aAAa,EAAE;oBACb,GAAG,EAAE,SAAS,CAAC,aAAa,CAAC,GAAG;oBAChC,MAAM,EAAE,SAAS,CAAC,aAAa,CAAC,MAAM;iBACvC;gBACD,WAAW,EAAE;oBACX,GAAG,EAAE,SAAS,CAAC,WAAW,CAAC,GAAG;oBAC9B,MAAM,EAAE,SAAS,CAAC,WAAW,CAAC,MAAM;iBACrC;aACF;YACH,CAAC,CAAC;gBACE,aAAa,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;gBACpC,WAAW,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;aACnC,CAAC;QAEN,OAAO;YACL,IAAI;YACJ,KAAK;YACL,MAAM;YACN,UAAU;YACV,MAAM;YACN,UAAU;YACV,SAAS;YACT,OAAO;YACP,GAAG,SAAS;SACb,CAAC;IACJ,CAAC;IAED,+CAA+C;IAC/C,oBAAoB;IACpB,+CAA+C;IAE/C;;OAEG;IACK,cAAc,CAAC,QAAwB;QAC7C,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,MAAM,UAAU,GAAG,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;QAEvD,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,SAAS,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YACrD,IAAI,CAAC,QAAQ;gBAAE,SAAS;YAExB,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;YACjD,IAAI,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;gBAChC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,IAAoB;QAC/C,MAAM,OAAO,GAAqB,EAAE,CAAC;QACrC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;YACxB,IAAI,CAAC,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClB,CAAC;YACD,mCAAmC;YACnC,IAAI,CAAC,CAAC,IAAI,KAAK,sBAAsB,EAAE,CAAC;gBACtC,KAAK,MAAM,KAAK,IAAI,CAAC,CAAC,aAAa,EAAE,CAAC;oBACpC,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;wBACtC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;oBACtB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QACH,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACK,kBAAkB,CACxB,QAAwB,EACxB,OAA0B;QAE1B,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE;YAC/B,IAAI,IAAI,CAAC,IAAI,KAAK,sBAAsB;gBAAE,OAAO;YAEjD,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;YACpC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY;gBAAE,OAAO;YAElD,MAAM,IAAI,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,KAAK,GAAG,KAAK,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAE/C,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO;YAC5B,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY;gBAAE,OAAO;YAEvC,wCAAwC;YACxC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;YACvB,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC;YAEzB,6EAA6E;YAC7E,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACvE,OAAO,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACvC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,+CAA+C;IAC/C,sBAAsB;IACtB,+CAA+C;IAE/C;;OAEG;IACK,kBAAkB,CAAC,SAAyB;QAClD,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,MAAM,YAAY,GAAG,SAAS,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QAEjE,IAAI,CAAC,YAAY;YAAE,OAAO,KAAK,CAAC;QAEhC,KAAK,MAAM,KAAK,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;YAC/C,4BAA4B;YAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAChC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC;YACD,+CAA+C;iBAC1C,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACpC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC;YACD,uCAAuC;iBAClC,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACpC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC;YACD,sCAAsC;iBACjC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC/B,MAAM,IAAI,GAAG,KAAK,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;gBACjD,IAAI,IAAI,EAAE,CAAC;oBACT,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACxB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,KAAe;QACrC,OAAO,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;YACzB,eAAe;YACf,IAAI,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC;gBAAE,OAAO,IAAI,CAAC;YAEjD,6BAA6B;YAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACpC,IAAI,QAAQ,IAAI,qBAAqB,CAAC,GAAG,CAAC,QAAQ,CAAC;gBAAE,OAAO,IAAI,CAAC;YAEjE,4BAA4B;YAC5B,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;gBAChE,OAAO,IAAI,CAAC;YACd,CAAC;YAED,OAAO,KAAK,CAAC;QACf,CAAC,CAAC,CAAC;IACL,CAAC;IAED,+CAA+C;IAC/C,mCAAmC;IACnC,+CAA+C;IAE/C;;OAEG;IACK,iBAAiB,CAAC,SAAyB;QACjD,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC;QAEhC,IAAI,MAAM,EAAE,IAAI,KAAK,sBAAsB,EAAE,CAAC;YAC5C,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBACzC,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC/B,MAAM,OAAO,GAAG,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;oBACvC,IAAI,OAAO,EAAE,CAAC;wBACZ,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;oBAChC,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,QAAwB;QAC/C,IAAI,QAAQ,CAAC,eAAe,KAAK,CAAC;YAAE,OAAO,IAAI,CAAC;QAEhD,MAAM,UAAU,GAAG,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;QAC7C,IAAI,UAAU,EAAE,IAAI,KAAK,sBAAsB;YAAE,OAAO,IAAI,CAAC;QAE7D,MAAM,IAAI,GAAG,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;QACzC,IAAI,IAAI,EAAE,IAAI,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAC;QAEzC,gBAAgB;QAChB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACvB,IAAI,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YACrD,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAClC,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YACjD,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAClC,CAAC;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED,+CAA+C;IAC/C,oBAAoB;IACpB,+CAA+C;IAE/C;;OAEG;IACK,qBAAqB,CAC3B,QAAwB,EACxB,KAAe;QAEf,0BAA0B;QAC1B,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC;YAC3C,qBAAqB;YACrB,IAAI,KAAK,CAAC,IAAI,KAAK,sBAAsB,EAAE,CAAC;gBAC1C,MAAM,KAAK,GAAG,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;gBACrC,IAAI,KAAK,EAAE,IAAI,KAAK,YAAY,EAAE,CAAC;oBACjC,MAAM,IAAI,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;oBAC7C,IAAI,IAAI,EAAE,IAAI,KAAK,cAAc,EAAE,CAAC;wBAClC,OAAO,CAAC,CAAC;oBACX,CAAC;gBACH,CAAC;YACH,CAAC;YAED,qBAAqB;YACrB,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;gBACtC,MAAM,IAAI,GAAG,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAC7C,IAAI,IAAI,EAAE,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAC5B,OAAO,CAAC,CAAC;gBACX,CAAC;YACH,CAAC;YAED,0BAA0B;YAC1B,IAAI,KAAK,CAAC,IAAI,KAAK,sBAAsB,EAAE,CAAC;gBAC1C,KAAK,MAAM,SAAS,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;oBAC5C,IAAI,SAAS,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBACnC,MAAM,OAAO,GAAG,SAAS,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;wBAC3C,MAAM,IAAI,GAAG,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC;wBACjC,IAAI,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC;4BACzE,OAAO,CAAC,CAAC;wBACX,CAAC;wBACD,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;4BAClE,OAAO,CAAC,CAAC;wBACX,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,+BAA+B;QAC/B,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC;YAClD,2DAA2D;YAC3D,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,+CAA+C;IAC/C,eAAe;IACf,+CAA+C;IAE/C;;OAEG;IACK,QAAQ,CACd,IAAoB,EACpB,OAAoC;QAEpC,OAAO,CAAC,IAAI,CAAC,CAAC;QACd,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClC,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;CACF;AAED,+CAA+C;AAC/C,oBAAoB;AACpB,+CAA+C;AAE/C;;GAEG;AACH,MAAM,UAAU,sBAAsB,CACpC,MAA2B,EAC3B,SAAiB;IAEjB,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;IACvD,OAAO,KAAK,EAAE,MAAM,IAAI,IAAI,CAAC;AAC/B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,SAAS,CACvB,MAA2B,EAC3B,IAAY;IAEZ,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,IAAI,CAAC;AACrD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,aAAa,CAAC,MAA2B;IACvD,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;AACnC,CAAC"}

package/dist/parsers/tree-sitter/pydantic/type-resolver.d.ts

@@ -0,0 +1,89 @@
+/**
+ * Pydantic Type Resolver
+ *
+ * Resolves Python type annotations to TypeInfo structures.
+ * Handles Optional, Union, List, Dict, Literal, and nested generics.
+ *
+ * @module pydantic/type-resolver
+ */
+import type { TypeInfo } from '../types.js';
+import type { ExtractionContext } from './types.js';
+/**
+ * Resolves Python type annotations to TypeInfo.
+ *
+ * Handles:
+ * - Simple types: `str`, `int`, `bool`
+ * - Optional: `Optional[str]`, `str | None`
+ * - Union: `Union[str, int]`, `str | int`
+ * - List/Dict: `List[str]`, `Dict[str, int]`
+ * - Literal: `Literal['a', 'b', 'c']`
+ * - Nested: `List[Dict[str, Optional[int]]]`
+ */
+export declare class TypeResolver {
+    /**
+     * Resolve a type annotation string to TypeInfo.
+     *
+     * @param typeStr - The type annotation string
+     * @param context - Extraction context
+     * @param depth - Current recursion depth
+     * @returns Resolved TypeInfo
+     */
+    resolve(typeStr: string, context: ExtractionContext, depth?: number): TypeInfo;
+    /**
+     * Create an unknown type placeholder.
+     */
+    createUnknownType(raw?: string): TypeInfo;
+    /**
+     * Parse a type string into a structured format.
+     */
+    private parseTypeString;
+    /**
+     * Check if type string contains union operator (|) outside brackets.
+     */
+    private containsUnionOperator;
+    /**
+     * Parse a union type (str | int | None).
+     */
+    private parseUnionType;
+    /**
+     * Split union members respecting bracket depth.
+     */
+    private splitUnionMembers;
+    /**
+     * Parse generic type arguments.
+     */
+    private parseGenericArgs;
+    /**
+     * Convert ParsedType to TypeInfo.
+     */
+    private parsedToTypeInfo;
+    /**
+     * Handle Union type conversion.
+     */
+    private handleUnionType;
+    /**
+     * Handle Literal type conversion.
+     */
+    private handleLiteralType;
+    /**
+     * Handle List/Sequence type conversion.
+     */
+    private handleListType;
+    /**
+     * Handle Dict/Mapping type conversion.
+     */
+    private handleDictType;
+    /**
+     * Handle Tuple type conversion.
+     */
+    private handleTupleType;
+    /**
+     * Check if a type name is a list-like type.
+     */
+    private isListType;
+    /**
+     * Check if a type name is a dict-like type.
+     */
+    private isDictType;
+}
+//# sourceMappingURL=type-resolver.d.ts.map

package/dist/parsers/tree-sitter/pydantic/type-resolver.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"type-resolver.d.ts","sourceRoot":"","sources":["../../../../src/parsers/tree-sitter/pydantic/type-resolver.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,KAAK,EAAE,iBAAiB,EAAc,MAAM,YAAY,CAAC;AAOhE;;;;;;;;;;GAUG;AACH,qBAAa,YAAY;IACvB;;;;;;;OAOG;IACH,OAAO,CACL,OAAO,EAAE,MAAM,EACf,OAAO,EAAE,iBAAiB,EAC1B,KAAK,GAAE,MAAU,GAChB,QAAQ;IAsBX;;OAEG;IACH,iBAAiB,CAAC,GAAG,GAAE,MAAkB,GAAG,QAAQ;IAoBpD;;OAEG;IACH,OAAO,CAAC,eAAe;IAgCvB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAU7B;;OAEG;IACH,OAAO,CAAC,cAAc;IAgCtB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA2BzB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAiCxB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAkExB;;OAEG;IACH,OAAO,CAAC,eAAe;IAwCvB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IAmCzB;;OAEG;IACH,OAAO,CAAC,cAAc;IAyBtB;;OAEG;IACH,OAAO,CAAC,cAAc;IA8BtB;;OAEG;IACH,OAAO,CAAC,eAAe;IAwBvB;;OAEG;IACH,OAAO,CAAC,UAAU;IAUlB;;OAEG;IACH,OAAO,CAAC,UAAU;CAQnB"}