zinfer 0.0.1 → 0.1.2

package/dist/cli.js

@@ -0,0 +1,486 @@
+#!/usr/bin/env node
+import { Command } from "commander";
+import { resolve, dirname, basename, relative, parse as parsePath } from "pathe";
+import { existsSync, writeFileSync, mkdirSync } from "fs";
+import { ZodTypeExtractor, generateDeclarationFile, NameMapper, FileResolver, DescriptionExtractor, ConfigLoader, formatError, NoFilesMatchedError, NoSchemasFoundError, InvalidOptionError, generateTypeTests, generateImportPrefix, setVerbose, logVerbose, logProgress, } from "./core/index.js";
+const program = new Command();
+program.name("zinfer").description("Extract input/output types from Zod schemas").version("0.1.0");
+program
+    .argument("[files...]", "File paths or glob patterns")
+    .option("-c, --config <path>", "Path to config file")
+    .option("-p, --project <path>", "Path to tsconfig.json")
+    .option("--schemas <names>", "Comma-separated schema names to extract")
+    .option("--input-only", "Output only input types")
+    .option("--output-only", "Output only output types")
+    .option("--merge-same", "Single type if input===output")
+    .option("--suffix <suffix>", "Remove suffix from schema names (e.g., 'Schema')")
+    .option("--input-suffix <suffix>", "Suffix for input type names (default: 'Input')")
+    .option("--output-suffix <suffix>", "Suffix for output type names (default: 'Output')")
+    .option("--map <mappings>", "Custom name mappings (e.g., 'UserSchema:User')")
+    .option("--outDir <dir>", "Output directory for generated files")
+    .option("--outFile <file>", "Single output file for all types")
+    .option("--outPattern <pattern>", "Output file naming pattern (e.g., '[name].types.ts')")
+    .option("-d, --declaration", "Generate .d.ts files")
+    .option("--dry-run", "Preview without writing files")
+    .option("--with-descriptions", "Include Zod .describe() as TSDoc comments")
+    .option("--generate-tests", "Generate vitest type equality tests alongside type files")
+    .option("-v, --verbose", "Enable verbose output")
+    .action(async (files, options) => {
+    try {
+        await runCLI(files, options);
+    }
+    catch (error) {
+        console.error(formatError(error));
+        process.exit(1);
+    }
+});
+program.parse();
+/**
+ * Main CLI execution logic.
+ */
+async function runCLI(files, options) {
+    const cwd = process.cwd();
+    // Enable verbose mode if requested
+    if (options.verbose) {
+        setVerbose(true);
+        logVerbose("Verbose mode enabled");
+    }
+    // Load config file
+    logVerbose("Loading configuration...");
+    const configLoader = new ConfigLoader();
+    const { config: fileConfig } = await configLoader.load(cwd);
+    // Merge CLI options with config file (CLI takes precedence)
+    const config = mergeCliWithConfig(options, fileConfig);
+    // Validate options
+    validateOptions(config);
+    logVerbose("Configuration validated");
+    // Use files from CLI args, or from config, or fail
+    const inputFiles = files.length > 0 ? files : config.include || [];
+    if (inputFiles.length === 0) {
+        throw new NoFilesMatchedError(["(no files specified)"]);
+    }
+    // Resolve file paths (support glob patterns)
+    logVerbose("Resolving input files...");
+    const fileResolver = new FileResolver();
+    const resolvedFiles = await fileResolver.resolveInputFiles(inputFiles, cwd);
+    if (resolvedFiles.length === 0) {
+        throw new NoFilesMatchedError(inputFiles);
+    }
+    logVerbose(`Found ${resolvedFiles.length} file(s) to process`);
+    // Find tsconfig
+    const tsconfigPath = config.project ? resolve(cwd, config.project) : findTsConfig(cwd);
+    logVerbose(`Using tsconfig: ${tsconfigPath || "(none)"}`);
+    // Validate --generate-tests requires file output
+    if (config.generateTests && !config.outDir && !config.outFile) {
+        throw new Error("--generate-tests requires --outDir or --outFile");
+    }
+    // Create extractor and name mapper
+    const extractor = new ZodTypeExtractor(tsconfigPath);
+    const nameMapper = createNameMapper(config);
+    const descriptionExtractor = config.withDescriptions ? new DescriptionExtractor() : null;
+    // Parse schema names if specified
+    const schemaFilter = config.schemas;
+    // Output options
+    const outputOptions = {
+        outDir: config.outDir,
+        outFile: config.outFile,
+        outPattern: config.outPattern,
+        declaration: config.declaration,
+    };
+    // Declaration options
+    const declOptions = {
+        inputOnly: config.inputOnly,
+        outputOnly: config.outputOnly,
+        mergeSame: config.mergeSame,
+    };
+    // Single output file mode
+    if (config.outFile) {
+        logVerbose("Processing files for single output...");
+        const allResults = [];
+        const fileResultsMap = new Map();
+        for (let i = 0; i < resolvedFiles.length; i++) {
+            const filePath = resolvedFiles[i];
+            logProgress(i + 1, resolvedFiles.length, `Processing ${basename(filePath)}`);
+            let results = getFilteredResults(extractor, filePath, schemaFilter);
+            // Add descriptions if enabled
+            if (descriptionExtractor) {
+                results = await addDescriptionsToResults(descriptionExtractor, filePath, results);
+            }
+            if (results.length > 0) {
+                allResults.push(...results);
+                fileResultsMap.set(filePath, results);
+            }
+        }
+        if (allResults.length === 0) {
+            throw new NoSchemasFoundError(resolvedFiles, schemaFilter);
+        }
+        const content = generateDeclarationFile(allResults, nameMapper.createMapFunction(), declOptions);
+        const outputPath = resolve(cwd, config.outFile);
+        if (options.dryRun) {
+            console.log(`Would write to: ${outputPath}`);
+            console.log("---");
+            console.log(content);
+        }
+        else {
+            ensureDir(dirname(outputPath));
+            writeFile(outputPath, content);
+            console.log(`Generated: ${outputPath} (${allResults.length} types)`);
+        }
+        // Generate test file if requested
+        if (config.generateTests) {
+            const testPath = outputPath.replace(/\.ts$/, ".test.ts");
+            const testContent = generateTestFileForSingleOutput(fileResultsMap, outputPath, testPath, nameMapper);
+            if (options.dryRun) {
+                console.log(`Would write to: ${testPath}`);
+                console.log("---");
+                console.log(testContent);
+            }
+            else {
+                writeFile(testPath, testContent);
+                console.log(`Generated: ${testPath} (${allResults.length * 2} test cases)`);
+            }
+        }
+        return;
+    }
+    // Per-file output mode or console output
+    logVerbose("Processing files...");
+    let totalResults = 0;
+    for (let i = 0; i < resolvedFiles.length; i++) {
+        const filePath = resolvedFiles[i];
+        logProgress(i + 1, resolvedFiles.length, `Processing ${basename(filePath)}`);
+        let results = getFilteredResults(extractor, filePath, schemaFilter);
+        if (results.length === 0) {
+            logVerbose(`  No schemas found in ${basename(filePath)}`);
+            continue;
+        }
+        totalResults += results.length;
+        // Add descriptions if enabled
+        if (descriptionExtractor) {
+            results = await addDescriptionsToResults(descriptionExtractor, filePath, results);
+        }
+        // File output mode
+        if (config.outDir || config.outPattern) {
+            const content = generateDeclarationFile(results, nameMapper.createMapFunction(), declOptions);
+            const outputPath = fileResolver.resolveOutputPath(filePath, outputOptions, cwd);
+            if (options.dryRun) {
+                console.log(`Would write to: ${outputPath}`);
+                console.log("---");
+                console.log(content);
+                console.log("");
+            }
+            else {
+                ensureDir(dirname(outputPath));
+                writeFile(outputPath, content);
+                console.log(`Generated: ${outputPath} (${results.length} types)`);
+            }
+            // Generate test file if requested
+            if (config.generateTests) {
+                const testPath = outputPath.replace(/\.ts$/, ".test.ts");
+                const testContent = generateTestFileForPerFile(filePath, outputPath, testPath, results, nameMapper);
+                if (options.dryRun) {
+                    console.log(`Would write to: ${testPath}`);
+                    console.log("---");
+                    console.log(testContent);
+                    console.log("");
+                }
+                else {
+                    writeFile(testPath, testContent);
+                    console.log(`Generated: ${testPath} (${results.length * 2} test cases)`);
+                }
+            }
+        }
+        else {
+            // Console output mode
+            if (resolvedFiles.length > 1) {
+                console.log(`// File: ${filePath}`);
+            }
+            const content = generateDeclarationFile(results, nameMapper.createMapFunction(), declOptions);
+            console.log(content);
+        }
+    }
+    // Error if no schemas were found
+    if (totalResults === 0) {
+        throw new NoSchemasFoundError(resolvedFiles, schemaFilter);
+    }
+}
+/**
+ * Adds descriptions from Zod schemas to extraction results.
+ */
+async function addDescriptionsToResults(descriptionExtractor, filePath, results) {
+    const schemaNames = results.map((r) => r.schemaName);
+    const descriptions = await descriptionExtractor.extractDescriptions(filePath, schemaNames);
+    return results.map((result) => {
+        const desc = descriptions.get(result.schemaName);
+        if (!desc) {
+            return result;
+        }
+        return {
+            ...result,
+            description: desc.description,
+            fieldDescriptions: desc.fields,
+        };
+    });
+}
+/**
+ * Gets extraction results, filtering by schema names if specified.
+ * Only extracts schemas that actually exist in the file.
+ */
+function getFilteredResults(extractor, filePath, schemaFilter) {
+    if (!schemaFilter) {
+        return extractor.extractAll(filePath);
+    }
+    const existingSchemas = extractor.getSchemaNames(filePath);
+    const schemasToExtract = schemaFilter.filter((name) => existingSchemas.includes(name));
+    if (schemasToExtract.length === 0) {
+        return [];
+    }
+    return extractor.extractMultiple(filePath, schemasToExtract);
+}
+/**
+ * Merges CLI options with config file options.
+ * CLI options take precedence.
+ */
+function mergeCliWithConfig(cliOptions, fileConfig) {
+    const merged = { ...fileConfig };
+    // Merge CLI options (only non-undefined values)
+    if (cliOptions.project !== undefined)
+        merged.project = cliOptions.project;
+    if (cliOptions.schemas !== undefined) {
+        merged.schemas = parseSchemaNames(cliOptions.schemas);
+    }
+    if (cliOptions.inputOnly !== undefined)
+        merged.inputOnly = cliOptions.inputOnly;
+    if (cliOptions.outputOnly !== undefined)
+        merged.outputOnly = cliOptions.outputOnly;
+    if (cliOptions.mergeSame !== undefined)
+        merged.mergeSame = cliOptions.mergeSame;
+    if (cliOptions.suffix !== undefined)
+        merged.suffix = cliOptions.suffix;
+    if (cliOptions.inputSuffix !== undefined)
+        merged.inputSuffix = cliOptions.inputSuffix;
+    if (cliOptions.outputSuffix !== undefined)
+        merged.outputSuffix = cliOptions.outputSuffix;
+    if (cliOptions.map !== undefined) {
+        merged.map = parseCustomMap(cliOptions.map);
+    }
+    if (cliOptions.outDir !== undefined)
+        merged.outDir = cliOptions.outDir;
+    if (cliOptions.outFile !== undefined)
+        merged.outFile = cliOptions.outFile;
+    if (cliOptions.outPattern !== undefined)
+        merged.outPattern = cliOptions.outPattern;
+    if (cliOptions.declaration !== undefined)
+        merged.declaration = cliOptions.declaration;
+    if (cliOptions.withDescriptions !== undefined)
+        merged.withDescriptions = cliOptions.withDescriptions;
+    if (cliOptions.generateTests !== undefined)
+        merged.generateTests = cliOptions.generateTests;
+    return merged;
+}
+/**
+ * Validates CLI options for conflicts and invalid values.
+ * @throws InvalidOptionError if any validation fails
+ */
+function validateOptions(config) {
+    // Check mutually exclusive options
+    if (config.inputOnly && config.outputOnly) {
+        throw new InvalidOptionError("--input-only / --output-only", "Cannot use both options together", "Use only one of --input-only or --output-only");
+    }
+    // Check conflicting output options
+    if (config.outFile && (config.outDir || config.outPattern)) {
+        throw new InvalidOptionError("--outFile", "Cannot use with --outDir or --outPattern", "Use --outFile for single output, or --outDir/--outPattern for multiple outputs");
+    }
+    // Check empty suffix
+    if (config.suffix === "") {
+        throw new InvalidOptionError("--suffix", "Empty suffix is not allowed", "Provide a non-empty suffix value or omit the option");
+    }
+}
+/**
+ * Creates a NameMapper from config.
+ */
+function createNameMapper(config) {
+    const mappingOptions = {};
+    if (config.suffix) {
+        mappingOptions.removeSuffix = config.suffix;
+    }
+    if (config.inputSuffix != null) {
+        mappingOptions.inputSuffix = config.inputSuffix;
+    }
+    if (config.outputSuffix != null) {
+        mappingOptions.outputSuffix = config.outputSuffix;
+    }
+    if (config.map) {
+        mappingOptions.customMap = config.map;
+    }
+    return new NameMapper(mappingOptions);
+}
+/**
+ * Parses schema names from a comma-separated string.
+ * Filters out empty strings and validates identifier format.
+ *
+ * @param schemasStr - Comma-separated schema names
+ * @returns Array of valid schema names
+ * @throws Error if any schema name is invalid
+ */
+function parseSchemaNames(schemasStr) {
+    const schemas = schemasStr
+        .split(",")
+        .map((s) => s.trim())
+        .filter((s) => s.length > 0);
+    // Validate that each schema name is a valid identifier
+    const identifierPattern = /^[a-zA-Z_$][a-zA-Z0-9_$]*$/;
+    for (const schema of schemas) {
+        if (!identifierPattern.test(schema)) {
+            throw new Error(`Invalid schema name: "${schema}". Schema names must be valid TypeScript identifiers.`);
+        }
+    }
+    return schemas;
+}
+/**
+ * Parses custom mapping string: "Schema1:Type1,Schema2:Type2"
+ * Validates both schema names and type names.
+ *
+ * @param mapStr - Custom mapping string
+ * @returns Map of schema names to type names
+ * @throws Error if any mapping is invalid
+ */
+function parseCustomMap(mapStr) {
+    const result = {};
+    const identifierPattern = /^[a-zA-Z_$][a-zA-Z0-9_$]*$/;
+    for (const pair of mapStr.split(",")) {
+        const trimmedPair = pair.trim();
+        if (!trimmedPair)
+            continue;
+        const colonIndex = trimmedPair.indexOf(":");
+        if (colonIndex === -1) {
+            throw new Error(`Invalid mapping format: "${trimmedPair}". Expected "SchemaName:TypeName".`);
+        }
+        const from = trimmedPair.substring(0, colonIndex).trim();
+        const to = trimmedPair.substring(colonIndex + 1).trim();
+        if (!from || !to) {
+            throw new Error(`Invalid mapping: "${trimmedPair}". Both schema name and type name are required.`);
+        }
+        if (!identifierPattern.test(from)) {
+            throw new Error(`Invalid schema name in mapping: "${from}". Must be a valid TypeScript identifier.`);
+        }
+        if (!identifierPattern.test(to)) {
+            throw new Error(`Invalid type name in mapping: "${to}". Must be a valid TypeScript identifier.`);
+        }
+        if (result[from] !== undefined) {
+            console.warn(`Warning: Duplicate mapping for "${from}". Using "${to}" (overwriting "${result[from]}").`);
+        }
+        result[from] = to;
+    }
+    return result;
+}
+/**
+ * Finds tsconfig.json starting from the given directory.
+ */
+function findTsConfig(startDir) {
+    let currentDir = startDir;
+    // Get the root directory in a cross-platform way (e.g., "/" on Unix, "C:\" on Windows)
+    const root = parsePath(resolve(startDir)).root;
+    while (currentDir !== root) {
+        const tsconfigPath = resolve(currentDir, "tsconfig.json");
+        if (existsSync(tsconfigPath)) {
+            return tsconfigPath;
+        }
+        const parentDir = resolve(currentDir, "..");
+        // Prevent infinite loop at root
+        if (parentDir === currentDir) {
+            break;
+        }
+        currentDir = parentDir;
+    }
+    // Check root directory as well
+    const rootTsconfig = resolve(root, "tsconfig.json");
+    if (existsSync(rootTsconfig)) {
+        return rootTsconfig;
+    }
+    return undefined;
+}
+/**
+ * Ensures a directory exists.
+ * @throws Error if the directory cannot be created
+ */
+function ensureDir(dirPath) {
+    if (!existsSync(dirPath)) {
+        try {
+            mkdirSync(dirPath, { recursive: true });
+        }
+        catch (error) {
+            const err = error;
+            throw new Error(`Failed to create directory "${dirPath}": ${err.message}`);
+        }
+    }
+}
+/**
+ * Writes content to a file with proper error handling.
+ * @throws Error if the file cannot be written
+ */
+function writeFile(filePath, content) {
+    try {
+        writeFileSync(filePath, content, "utf-8");
+    }
+    catch (error) {
+        const err = error;
+        throw new Error(`Failed to write file "${filePath}": ${err.message}`);
+    }
+}
+/**
+ * Creates test schema info from extraction results.
+ */
+function createTestSchemas(results, nameMapper) {
+    return results
+        .filter((r) => r.isExported)
+        .map((result) => ({
+        schemaName: result.schemaName,
+        inputTypeName: nameMapper.map(result.schemaName).inputName,
+        outputTypeName: nameMapper.map(result.schemaName).outputName,
+    }));
+}
+/**
+ * Generates test file content for single output mode (--outFile).
+ */
+function generateTestFileForSingleOutput(fileResultsMap, typesPath, testPath, nameMapper) {
+    const testFiles = [];
+    const testDir = dirname(testPath);
+    for (const [schemaFile, results] of fileResultsMap) {
+        const schemas = createTestSchemas(results, nameMapper);
+        if (schemas.length === 0)
+            continue;
+        testFiles.push({
+            schemaFilePath: getRelativePath(testDir, schemaFile),
+            typesFilePath: getRelativePath(testDir, typesPath),
+            importPrefix: generateImportPrefix(basename(schemaFile, ".ts")),
+            schemas,
+        });
+    }
+    return generateTypeTests(testFiles);
+}
+/**
+ * Generates test file content for per-file output mode (--outDir).
+ */
+function generateTestFileForPerFile(schemaFile, typesPath, testPath, results, nameMapper) {
+    const schemas = createTestSchemas(results, nameMapper);
+    if (schemas.length === 0) {
+        return "";
+    }
+    const testDir = dirname(testPath);
+    return generateTypeTests([
+        {
+            schemaFilePath: getRelativePath(testDir, schemaFile),
+            typesFilePath: getRelativePath(testDir, typesPath),
+            importPrefix: generateImportPrefix(basename(schemaFile, ".ts")),
+            schemas,
+        },
+    ]);
+}
+/**
+ * Gets relative path from one file to another, ensuring it starts with ./
+ */
+function getRelativePath(from, to) {
+    const rel = relative(from, to);
+    return rel.startsWith(".") ? rel : "./" + rel;
+}
+//# sourceMappingURL=cli.js.map

package/dist/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../src/cli.ts"],"names":[],"mappings":";AACA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,IAAI,SAAS,EAAE,MAAM,OAAO,CAAC;AACjF,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC1D,OAAO,EACL,gBAAgB,EAChB,uBAAuB,EACvB,UAAU,EACV,YAAY,EACZ,oBAAoB,EACpB,YAAY,EACZ,WAAW,EACX,mBAAmB,EACnB,mBAAmB,EACnB,kBAAkB,EAClB,iBAAiB,EACjB,oBAAoB,EACpB,UAAU,EACV,UAAU,EACV,WAAW,GAQZ,MAAM,iBAAiB,CAAC;AAuBzB,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;AAE9B,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,WAAW,CAAC,6CAA6C,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;AAEnG,OAAO;KACJ,QAAQ,CAAC,YAAY,EAAE,6BAA6B,CAAC;KACrD,MAAM,CAAC,qBAAqB,EAAE,qBAAqB,CAAC;KACpD,MAAM,CAAC,sBAAsB,EAAE,uBAAuB,CAAC;KACvD,MAAM,CAAC,mBAAmB,EAAE,yCAAyC,CAAC;KACtE,MAAM,CAAC,cAAc,EAAE,yBAAyB,CAAC;KACjD,MAAM,CAAC,eAAe,EAAE,0BAA0B,CAAC;KACnD,MAAM,CAAC,cAAc,EAAE,+BAA+B,CAAC;KACvD,MAAM,CAAC,mBAAmB,EAAE,kDAAkD,CAAC;KAC/E,MAAM,CAAC,yBAAyB,EAAE,gDAAgD,CAAC;KACnF,MAAM,CAAC,0BAA0B,EAAE,kDAAkD,CAAC;KACtF,MAAM,CAAC,kBAAkB,EAAE,gDAAgD,CAAC;KAC5E,MAAM,CAAC,gBAAgB,EAAE,sCAAsC,CAAC;KAChE,MAAM,CAAC,kBAAkB,EAAE,kCAAkC,CAAC;KAC9D,MAAM,CAAC,wBAAwB,EAAE,sDAAsD,CAAC;KACxF,MAAM,CAAC,mBAAmB,EAAE,sBAAsB,CAAC;KACnD,MAAM,CAAC,WAAW,EAAE,+BAA+B,CAAC;KACpD,MAAM,CAAC,qBAAqB,EAAE,2CAA2C,CAAC;KAC1E,MAAM,CAAC,kBAAkB,EAAE,0DAA0D,CAAC;KACtF,MAAM,CAAC,eAAe,EAAE,uBAAuB,CAAC;KAChD,MAAM,CAAC,KAAK,EAAE,KAAe,EAAE,OAAmB,EAAE,EAAE;IACrD,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAC/B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;QAClC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC,CAAC,CAAC;AAEL,OAAO,CAAC,KAAK,EAAE,CAAC;AAEhB;;GAEG;AACH,KAAK,UAAU,MAAM,CAAC,KAAe,EAAE,OAAmB;IACxD,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAE1B,mCAAmC;IACnC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,UAAU,CAAC,IAAI,CAAC,CAAC;QACjB,UAAU,CAAC,sBAAsB,CAAC,CAAC;IACrC,CAAC;IAED,mBAAmB;IACnB,UAAU,CAAC,0BAA0B,CAAC,CAAC;IACvC,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;IACxC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAE5D,4DAA4D;IAC5D,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IAEvD,mBAAmB;IACnB,eAAe,CAAC,MAAM,CAAC,CAAC;IACxB,UAAU,CAAC,yBAAyB,CAAC,CAAC;IAEtC,mDAAmD;IACnD,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;IAEnE,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,mBAAmB,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC;IAC1D,CAAC;IAED,6CAA6C;IAC7C,UAAU,CAAC,0BAA0B,CAAC,CAAC;IACvC,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;IACxC,MAAM,aAAa,GAAG,MAAM,YAAY,CAAC,iBAAiB,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;IAE5E,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,MAAM,IAAI,mBAAmB,CAAC,UAAU,CAAC,CAAC;IAC5C,CAAC;IACD,UAAU,CAAC,SAAS,aAAa,CAAC,MAAM,qBAAqB,CAAC,CAAC;IAE/D,gBAAgB;IAChB,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IACvF,UAAU,CAAC,mBAAmB,YAAY,IAAI,QAAQ,EAAE,CAAC,CAAC;IAE1D,iDAAiD;IACjD,IAAI,MAAM,CAAC,aAAa,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QAC9D,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;IACrE,CAAC;IAED,mCAAmC;IACnC,MAAM,SAAS,GAAG,IAAI,gBAAgB,CAAC,YAAY,CAAC,CAAC;IACrD,MAAM,UAAU,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAC5C,MAAM,oBAAoB,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,oBAAoB,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;IAEzF,kCAAkC;IAClC,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC;IAEpC,iBAAiB;IACjB,MAAM,aAAa,GAAkB;QACnC,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,WAAW,EAAE,MAAM,CAAC,WAAW;KAChC,CAAC;IAEF,sBAAsB;IACtB,MAAM,WAAW,GAAuB;QACtC,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,SAAS,EAAE,MAAM,CAAC,SAAS;KAC5B,CAAC;IAEF,0BAA0B;IAC1B,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,UAAU,CAAC,uCAAuC,CAAC,CAAC;QACpD,MAAM,UAAU,GAAoB,EAAE,CAAC;QACvC,MAAM,cAAc,GAAiC,IAAI,GAAG,EAAE,CAAC;QAE/D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9C,MAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;YAClC,WAAW,CAAC,CAAC,GAAG,CAAC,EAAE,aAAa,CAAC,MAAM,EAAE,cAAc,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC7E,IAAI,OAAO,GAAG,kBAAkB,CAAC,SAAS,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;YAEpE,8BAA8B;YAC9B,IAAI,oBAAoB,EAAE,CAAC;gBACzB,OAAO,GAAG,MAAM,wBAAwB,CAAC,oBAAoB,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;YACpF,CAAC;YAED,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvB,UAAU,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC;gBAC5B,cAAc,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QAED,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,mBAAmB,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,OAAO,GAAG,uBAAuB,CACrC,UAAU,EACV,UAAU,CAAC,iBAAiB,EAAE,EAC9B,WAAW,CACZ,CAAC;QAEF,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QAEhD,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,OAAO,CAAC,GAAG,CAAC,mBAAmB,UAAU,EAAE,CAAC,CAAC;YAC7C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACnB,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;aAAM,CAAC;YACN,SAAS,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;YAC/B,SAAS,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,cAAc,UAAU,KAAK,UAAU,CAAC,MAAM,SAAS,CAAC,CAAC;QACvE,CAAC;QAED,kCAAkC;QAClC,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YACzD,MAAM,WAAW,GAAG,+BAA+B,CACjD,cAAc,EACd,UAAU,EACV,QAAQ,EACR,UAAU,CACX,CAAC;YAEF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,OAAO,CAAC,GAAG,CAAC,mBAAmB,QAAQ,EAAE,CAAC,CAAC;gBAC3C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBACnB,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAC3B,CAAC;iBAAM,CAAC;gBACN,SAAS,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;gBACjC,OAAO,CAAC,GAAG,CAAC,cAAc,QAAQ,KAAK,UAAU,CAAC,MAAM,GAAG,CAAC,cAAc,CAAC,CAAC;YAC9E,CAAC;QACH,CAAC;QAED,OAAO;IACT,CAAC;IAED,yCAAyC;IACzC,UAAU,CAAC,qBAAqB,CAAC,CAAC;IAClC,IAAI,YAAY,GAAG,CAAC,CAAC;IAErB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,MAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QAClC,WAAW,CAAC,CAAC,GAAG,CAAC,EAAE,aAAa,CAAC,MAAM,EAAE,cAAc,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC7E,IAAI,OAAO,GAAG,kBAAkB,CAAC,SAAS,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;QAEpE,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzB,UAAU,CAAC,yBAAyB,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC1D,SAAS;QACX,CAAC;QAED,YAAY,IAAI,OAAO,CAAC,MAAM,CAAC;QAE/B,8BAA8B;QAC9B,IAAI,oBAAoB,EAAE,CAAC;YACzB,OAAO,GAAG,MAAM,wBAAwB,CAAC,oBAAoB,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;QACpF,CAAC;QAED,mBAAmB;QACnB,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;YACvC,MAAM,OAAO,GAAG,uBAAuB,CAAC,OAAO,EAAE,UAAU,CAAC,iBAAiB,EAAE,EAAE,WAAW,CAAC,CAAC;YAE9F,MAAM,UAAU,GAAG,YAAY,CAAC,iBAAiB,CAAC,QAAQ,EAAE,aAAa,EAAE,GAAG,CAAC,CAAC;YAEhF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnB,OAAO,CAAC,GAAG,CAAC,mBAAmB,UAAU,EAAE,CAAC,CAAC;gBAC7C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBACnB,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrB,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAClB,CAAC;iBAAM,CAAC;gBACN,SAAS,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;gBAC/B,SAAS,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;gBAC/B,OAAO,CAAC,GAAG,CAAC,cAAc,UAAU,KAAK,OAAO,CAAC,MAAM,SAAS,CAAC,CAAC;YACpE,CAAC;YAED,kCAAkC;YAClC,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBACzB,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;gBACzD,MAAM,WAAW,GAAG,0BAA0B,CAC5C,QAAQ,EACR,UAAU,EACV,QAAQ,EACR,OAAO,EACP,UAAU,CACX,CAAC;gBAEF,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;oBACnB,OAAO,CAAC,GAAG,CAAC,mBAAmB,QAAQ,EAAE,CAAC,CAAC;oBAC3C,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;oBACnB,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;oBACzB,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAClB,CAAC;qBAAM,CAAC;oBACN,SAAS,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;oBACjC,OAAO,CAAC,GAAG,CAAC,cAAc,QAAQ,KAAK,OAAO,CAAC,MAAM,GAAG,CAAC,cAAc,CAAC,CAAC;gBAC3E,CAAC;YACH,CAAC;QACH,CAAC;aAAM,CAAC;YACN,sBAAsB;YACtB,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC7B,OAAO,CAAC,GAAG,CAAC,YAAY,QAAQ,EAAE,CAAC,CAAC;YACtC,CAAC;YAED,MAAM,OAAO,GAAG,uBAAuB,CAAC,OAAO,EAAE,UAAU,CAAC,iBAAiB,EAAE,EAAE,WAAW,CAAC,CAAC;YAC9F,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAED,iCAAiC;IACjC,IAAI,YAAY,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,mBAAmB,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;IAC7D,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,wBAAwB,CACrC,oBAA0C,EAC1C,QAAgB,EAChB,OAAwB;IAExB,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;IACrD,MAAM,YAAY,GAAG,MAAM,oBAAoB,CAAC,mBAAmB,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAE3F,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;QAC5B,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACjD,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,OAAO;YACL,GAAG,MAAM;YACT,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,iBAAiB,EAAE,IAAI,CAAC,MAAM;SAC/B,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,SAAS,kBAAkB,CACzB,SAA2B,EAC3B,QAAgB,EAChB,YAAuB;IAEvB,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED,MAAM,eAAe,GAAG,SAAS,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC3D,MAAM,gBAAgB,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;IAEvF,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAClC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,SAAS,CAAC,eAAe,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC;AAC/D,CAAC;AAED;;;GAGG;AACH,SAAS,kBAAkB,CAAC,UAAsB,EAAE,UAAwB;IAC1E,MAAM,MAAM,GAAiB,EAAE,GAAG,UAAU,EAAE,CAAC;IAE/C,gDAAgD;IAChD,IAAI,UAAU,CAAC,OAAO,KAAK,SAAS;QAAE,MAAM,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;IAC1E,IAAI,UAAU,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QACrC,MAAM,CAAC,OAAO,GAAG,gBAAgB,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IACxD,CAAC;IACD,IAAI,UAAU,CAAC,SAAS,KAAK,SAAS;QAAE,MAAM,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;IAChF,IAAI,UAAU,CAAC,UAAU,KAAK,SAAS;QAAE,MAAM,CAAC,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC;IACnF,IAAI,UAAU,CAAC,SAAS,KAAK,SAAS;QAAE,MAAM,CAAC,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC;IAChF,IAAI,UAAU,CAAC,MAAM,KAAK,SAAS;QAAE,MAAM,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;IACvE,IAAI,UAAU,CAAC,WAAW,KAAK,SAAS;QAAE,MAAM,CAAC,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;IACtF,IAAI,UAAU,CAAC,YAAY,KAAK,SAAS;QAAE,MAAM,CAAC,YAAY,GAAG,UAAU,CAAC,YAAY,CAAC;IACzF,IAAI,UAAU,CAAC,GAAG,KAAK,SAAS,EAAE,CAAC;QACjC,MAAM,CAAC,GAAG,GAAG,cAAc,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IAC9C,CAAC;IACD,IAAI,UAAU,CAAC,MAAM,KAAK,SAAS;QAAE,MAAM,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;IACvE,IAAI,UAAU,CAAC,OAAO,KAAK,SAAS;QAAE,MAAM,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;IAC1E,IAAI,UAAU,CAAC,UAAU,KAAK,SAAS;QAAE,MAAM,CAAC,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC;IACnF,IAAI,UAAU,CAAC,WAAW,KAAK,SAAS;QAAE,MAAM,CAAC,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;IACtF,IAAI,UAAU,CAAC,gBAAgB,KAAK,SAAS;QAC3C,MAAM,CAAC,gBAAgB,GAAG,UAAU,CAAC,gBAAgB,CAAC;IACxD,IAAI,UAAU,CAAC,aAAa,KAAK,SAAS;QAAE,MAAM,CAAC,aAAa,GAAG,UAAU,CAAC,aAAa,CAAC;IAE5F,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,SAAS,eAAe,CAAC,MAAoB;IAC3C,mCAAmC;IACnC,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QAC1C,MAAM,IAAI,kBAAkB,CAC1B,8BAA8B,EAC9B,kCAAkC,EAClC,+CAA+C,CAChD,CAAC;IACJ,CAAC;IAED,mCAAmC;IACnC,IAAI,MAAM,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;QAC3D,MAAM,IAAI,kBAAkB,CAC1B,WAAW,EACX,0CAA0C,EAC1C,gFAAgF,CACjF,CAAC;IACJ,CAAC;IAED,qBAAqB;IACrB,IAAI,MAAM,CAAC,MAAM,KAAK,EAAE,EAAE,CAAC;QACzB,MAAM,IAAI,kBAAkB,CAC1B,UAAU,EACV,6BAA6B,EAC7B,qDAAqD,CACtD,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,gBAAgB,CAAC,MAAoB;IAC5C,MAAM,cAAc,GAAuB,EAAE,CAAC;IAE9C,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QAClB,cAAc,CAAC,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC;IAC9C,CAAC;IACD,IAAI,MAAM,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;QAC/B,cAAc,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;IAClD,CAAC;IACD,IAAI,MAAM,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;QAChC,cAAc,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;IACpD,CAAC;IACD,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC;QACf,cAAc,CAAC,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC;IACxC,CAAC;IAED,OAAO,IAAI,UAAU,CAAC,cAAc,CAAC,CAAC;AACxC,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,gBAAgB,CAAC,UAAkB;IAC1C,MAAM,OAAO,GAAG,UAAU;SACvB,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;SACpB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAE/B,uDAAuD;IACvD,MAAM,iBAAiB,GAAG,4BAA4B,CAAC;IACvD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CACb,yBAAyB,MAAM,uDAAuD,CACvF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;;;GAOG;AACH,SAAS,cAAc,CAAC,MAAc;IACpC,MAAM,MAAM,GAA2B,EAAE,CAAC;IAC1C,MAAM,iBAAiB,GAAG,4BAA4B,CAAC;IAEvD,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QACrC,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAChC,IAAI,CAAC,WAAW;YAAE,SAAS;QAE3B,MAAM,UAAU,GAAG,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC5C,IAAI,UAAU,KAAK,CAAC,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,4BAA4B,WAAW,oCAAoC,CAAC,CAAC;QAC/F,CAAC;QAED,MAAM,IAAI,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;QACzD,MAAM,EAAE,GAAG,WAAW,CAAC,SAAS,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAExD,IAAI,CAAC,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CACb,qBAAqB,WAAW,iDAAiD,CAClF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CACb,oCAAoC,IAAI,2CAA2C,CACpF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CACb,kCAAkC,EAAE,2CAA2C,CAChF,CAAC;QACJ,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;YAC/B,OAAO,CAAC,IAAI,CACV,mCAAmC,IAAI,aAAa,EAAE,mBAAmB,MAAM,CAAC,IAAI,CAAC,KAAK,CAC3F,CAAC;QACJ,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;IACpB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,QAAgB;IACpC,IAAI,UAAU,GAAG,QAAQ,CAAC;IAC1B,uFAAuF;IACvF,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC;IAE/C,OAAO,UAAU,KAAK,IAAI,EAAE,CAAC;QAC3B,MAAM,YAAY,GAAG,OAAO,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;QAC1D,IAAI,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;YAC7B,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;QAC5C,gCAAgC;QAChC,IAAI,SAAS,KAAK,UAAU,EAAE,CAAC;YAC7B,MAAM;QACR,CAAC;QACD,UAAU,GAAG,SAAS,CAAC;IACzB,CAAC;IAED,+BAA+B;IAC/B,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;IACpD,IAAI,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;QAC7B,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;GAGG;AACH,SAAS,SAAS,CAAC,OAAe;IAChC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,SAAS,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,GAAG,KAA8B,CAAC;YAC3C,MAAM,IAAI,KAAK,CAAC,+BAA+B,OAAO,MAAM,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7E,CAAC;IACH,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,SAAS,SAAS,CAAC,QAAgB,EAAE,OAAe;IAClD,IAAI,CAAC;QACH,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,GAAG,GAAG,KAA8B,CAAC;QAC3C,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,MAAM,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;IACxE,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,iBAAiB,CAAC,OAAwB,EAAE,UAAsB;IACzE,OAAO,OAAO;SACX,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC;SAC3B,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAChB,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,aAAa,EAAE,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS;QAC1D,cAAc,EAAE,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,UAAU;KAC7D,CAAC,CAAC,CAAC;AACR,CAAC;AAED;;GAEG;AACH,SAAS,+BAA+B,CACtC,cAA4C,EAC5C,SAAiB,EACjB,QAAgB,EAChB,UAAsB;IAEtB,MAAM,SAAS,GAAmB,EAAE,CAAC;IACrC,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;IAElC,KAAK,MAAM,CAAC,UAAU,EAAE,OAAO,CAAC,IAAI,cAAc,EAAE,CAAC;QACnD,MAAM,OAAO,GAAG,iBAAiB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QACvD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QAEnC,SAAS,CAAC,IAAI,CAAC;YACb,cAAc,EAAE,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC;YACpD,aAAa,EAAE,eAAe,CAAC,OAAO,EAAE,SAAS,CAAC;YAClD,YAAY,EAAE,oBAAoB,CAAC,QAAQ,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAC/D,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED,OAAO,iBAAiB,CAAC,SAAS,CAAC,CAAC;AACtC,CAAC;AAED;;GAEG;AACH,SAAS,0BAA0B,CACjC,UAAkB,EAClB,SAAiB,EACjB,QAAgB,EAChB,OAAwB,EACxB,UAAsB;IAEtB,MAAM,OAAO,GAAG,iBAAiB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACvD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;IAElC,OAAO,iBAAiB,CAAC;QACvB;YACE,cAAc,EAAE,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC;YACpD,aAAa,EAAE,eAAe,CAAC,OAAO,EAAE,SAAS,CAAC;YAClD,YAAY,EAAE,oBAAoB,CAAC,QAAQ,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAC/D,OAAO;SACR;KACF,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,IAAY,EAAE,EAAU;IAC/C,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAC/B,OAAO,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,GAAG,GAAG,CAAC;AAChD,CAAC"}

package/dist/core/brand-detector.d.ts

@@ -0,0 +1,32 @@
+import { SourceFile } from "ts-morph";
+import type { BrandInfo } from "./types.js";
+/**
+ * Map of schema name to its brand information.
+ */
+export type SchemaBrandMap = Map<string, BrandInfo[]>;
+/**
+ * Detects .brand<...>() calls in Zod schemas.
+ */
+export declare class BrandDetector {
+    /**
+     * Analyzes a source file to find all branded types in schemas.
+     *
+     * @param sourceFile - The source file to analyze
+     * @param schemaNames - Set of schema names to analyze
+     * @returns Map of schema name to brand information
+     */
+    detectBrands(sourceFile: SourceFile, schemaNames: Set<string>): SchemaBrandMap;
+    /**
+     * Recursively finds brand calls in a node.
+     */
+    private findBrandsInNode;
+    /**
+     * Extracts brand information from a .brand<...>() call.
+     */
+    private extractBrandFromCall;
+    /**
+     * Extracts the brand name from a type argument.
+     */
+    private extractBrandName;
+}
+//# sourceMappingURL=brand-detector.d.ts.map

package/dist/core/brand-detector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"brand-detector.d.ts","sourceRoot":"","sources":["../../src/core/brand-detector.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAwB,MAAM,UAAU,CAAC;AAC5D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAE5C;;GAEG;AACH,MAAM,MAAM,cAAc,GAAG,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;AAEtD;;GAEG;AACH,qBAAa,aAAa;IACxB;;;;;;OAMG;IACH,YAAY,CAAC,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,GAAG,CAAC,MAAM,CAAC,GAAG,cAAc;IAsB9E;;OAEG;IACH,OAAO,CAAC,gBAAgB;IA8CxB;;OAEG;IACH,OAAO,CAAC,oBAAoB;IA6B5B;;OAEG;IACH,OAAO,CAAC,gBAAgB;CAkBzB"}

package/dist/core/brand-detector.js

@@ -0,0 +1,121 @@
+import { Node } from "ts-morph";
+/**
+ * Detects .brand<...>() calls in Zod schemas.
+ */
+export class BrandDetector {
+    /**
+     * Analyzes a source file to find all branded types in schemas.
+     *
+     * @param sourceFile - The source file to analyze
+     * @param schemaNames - Set of schema names to analyze
+     * @returns Map of schema name to brand information
+     */
+    detectBrands(sourceFile, schemaNames) {
+        const result = new Map();
+        const statements = sourceFile.getVariableStatements();
+        for (const stmt of statements) {
+            for (const decl of stmt.getDeclarations()) {
+                const schemaName = decl.getName();
+                if (!schemaNames.has(schemaName))
+                    continue;
+                const init = decl.getInitializer();
+                if (!init)
+                    continue;
+                const brands = this.findBrandsInNode(init, "");
+                if (brands.length > 0) {
+                    result.set(schemaName, brands);
+                }
+            }
+        }
+        return result;
+    }
+    /**
+     * Recursively finds brand calls in a node.
+     */
+    findBrandsInNode(node, currentPath) {
+        const brands = [];
+        // Check if this node is a .brand<...>() call
+        if (Node.isCallExpression(node)) {
+            const brandInfo = this.extractBrandFromCall(node, currentPath);
+            if (brandInfo) {
+                brands.push(brandInfo);
+            }
+            // Check the expression being called (for method chains)
+            const expr = node.getExpression();
+            if (Node.isPropertyAccessExpression(expr)) {
+                const base = expr.getExpression();
+                brands.push(...this.findBrandsInNode(base, currentPath));
+            }
+            // Check arguments (for z.object({ field: schema }))
+            for (const arg of node.getArguments()) {
+                brands.push(...this.findBrandsInNode(arg, currentPath));
+            }
+        }
+        // Check object literals for field definitions
+        if (Node.isObjectLiteralExpression(node)) {
+            for (const prop of node.getProperties()) {
+                if (Node.isPropertyAssignment(prop)) {
+                    const fieldName = prop.getName();
+                    const fieldPath = currentPath ? `${currentPath}.${fieldName}` : fieldName;
+                    const initializer = prop.getInitializer();
+                    if (initializer) {
+                        brands.push(...this.findBrandsInNode(initializer, fieldPath));
+                    }
+                }
+            }
+        }
+        // Check property access expressions (method chains)
+        if (Node.isPropertyAccessExpression(node)) {
+            const base = node.getExpression();
+            brands.push(...this.findBrandsInNode(base, currentPath));
+        }
+        return brands;
+    }
+    /**
+     * Extracts brand information from a .brand<...>() call.
+     */
+    extractBrandFromCall(node, fieldPath) {
+        const expr = node.getExpression();
+        if (!Node.isPropertyAccessExpression(expr)) {
+            return null;
+        }
+        const methodName = expr.getName();
+        if (methodName !== "brand") {
+            return null;
+        }
+        // Get the type argument: .brand<"UserId">()
+        const typeArgs = node.getTypeArguments();
+        if (typeArgs.length === 0) {
+            return null;
+        }
+        const typeArg = typeArgs[0];
+        const brandName = this.extractBrandName(typeArg);
+        if (!brandName) {
+            return null;
+        }
+        return {
+            brandName,
+            fieldPath,
+        };
+    }
+    /**
+     * Extracts the brand name from a type argument.
+     */
+    extractBrandName(node) {
+        // Handle string literal type: "UserId"
+        if (Node.isLiteralTypeNode(node)) {
+            const literal = node.getLiteral();
+            if (Node.isStringLiteral(literal)) {
+                return literal.getLiteralText();
+            }
+        }
+        // Handle direct string literal (some TS versions)
+        const text = node.getText();
+        const match = text.match(/^["'](.+)["']$/);
+        if (match) {
+            return match[1];
+        }
+        return null;
+    }
+}
+//# sourceMappingURL=brand-detector.js.map