如何使用TypeScript编译器API提取数组类型

npm install -g typescript

let myArray: number[] = [1, 2, 3, 4, 5];

import * as ts from 'typescript';

// 读取TypeScript文件
const fileName = 'example.ts';
const sourceCode = fs.readFileSync(fileName, 'utf-8');

// 创建编译器选项
const compilerOptions: ts.CompilerOptions = {
  target: ts.ScriptTarget.ES5,
  module: ts.ModuleKind.CommonJS
};

// 创建编译器Host
const compilerHost: ts.CompilerHost = {
  getSourceFile: (fileName) => {
    return ts.createSourceFile(fileName, sourceCode, ts.ScriptTarget.ES5);
  },
  getDefaultLibFileName: () => ts.getDefaultLibFilePath(compilerOptions),
  writeFile: () => {},
  getCurrentDirectory: () => process.cwd(),
  getDirectories: () => [],
  fileExists: (fileName) => fs.existsSync(fileName),
  readFile: (fileName) => fs.readFileSync(fileName, 'utf-8'),
  getCanonicalFileName: (fileName) => fileName,
  useCaseSensitiveFileNames: () => true,
  getNewLine: () => os.EOL
};

// 创建编译器
const program = ts.createProgram([fileName], compilerOptions, compilerHost);
const typeChecker = program.getTypeChecker();

// 获取语法树
const sourceFile = program.getSourceFile(fileName);

// 遍历语法树
ts.forEachChild(sourceFile, (node) => {
  if (ts.isArrayLiteralExpression(node)) {
    const type = typeChecker.getTypeAtLocation(node);
    const elementType = typeChecker.getElementTypeOfArrayType(type);
    console.log('数组类型:', elementType);
  }
});