-
Notifications
You must be signed in to change notification settings - Fork 4
/
extract.ts
39 lines (30 loc) · 1.42 KB
/
extract.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import * as ts from 'typescript';
import { generateSchemaFromType } from './jsonSchemaGenerator';
export function extractFunctionArgumentsAndReturn(node: ts.FunctionDeclaration, typeChecker: ts.TypeChecker, program: ts.Program): void {
if (!node.name || !node.type) return;
//console.log(`Function: ${node.name.getText()}`);
// Extracting argument types
node.parameters.forEach((parameter) => {
// console.log(`Parameter: ${parameter.name.getText()} - Type: ${parameter.type?.getText()}`);
});
// Extracting return type
const returnTypeText = node.type.getText();
// console.log(`Return type: ${returnTypeText}`);
// Check if the return type is a Promise
const returnType = typeChecker.getTypeFromTypeNode(node.type);
const typeArgs = (returnType as ts.TypeReference).typeArguments;
let schema;
if (typeArgs && returnTypeText.startsWith('Promise')) {
const nestedType = typeArgs[0];
// console.log(`Nested type: ${nestedType.toString()}`);
// Generate JSON schema for nested type (Person)
const nestedSchema = generateSchemaFromType(typeChecker, nestedType, program);
// console.log('JSON Schema:', JSON.stringify(nestedSchema, null, 2));
schema = nestedSchema;
} else {
// Generate JSON schema for type (Person)
schema = generateSchemaFromType(typeChecker, returnType, program);
// console.log('JSON Schema:', JSON.stringify(schema, null, 2));
}
return schema;
}