c-next 0.1.0

package/src/symbols/CNextSymbolCollector.ts

@@ -0,0 +1,326 @@
+/**
+ * C-Next Symbol Collector
+ * Extracts symbols from C-Next parse trees for the unified symbol table
+ */
+
+import * as Parser from "../parser/grammar/CNextParser";
+import ISymbol from "../types/ISymbol";
+import ESymbolKind from "../types/ESymbolKind";
+import ESourceLanguage from "../types/ESourceLanguage";
+
+/**
+ * Collects symbols from a C-Next parse tree
+ */
+class CNextSymbolCollector {
+  private sourceFile: string;
+
+  private symbols: ISymbol[] = [];
+
+  constructor(sourceFile: string) {
+    this.sourceFile = sourceFile;
+  }
+
+  /**
+   * Collect all symbols from a C-Next program
+   */
+  collect(tree: Parser.ProgramContext): ISymbol[] {
+    this.symbols = [];
+
+    for (const decl of tree.declaration()) {
+      this.collectDeclaration(decl);
+    }
+
+    return this.symbols;
+  }
+
+  private collectDeclaration(decl: Parser.DeclarationContext): void {
+    // ADR-016: Handle scope declarations (renamed from namespace)
+    if (decl.scopeDeclaration()) {
+      this.collectScope(decl.scopeDeclaration()!);
+    }
+
+    if (decl.structDeclaration()) {
+      this.collectStruct(decl.structDeclaration()!);
+    }
+
+    if (decl.registerDeclaration()) {
+      this.collectRegister(decl.registerDeclaration()!);
+    }
+
+    if (decl.functionDeclaration()) {
+      this.collectFunction(decl.functionDeclaration()!, undefined);
+    }
+
+    if (decl.variableDeclaration()) {
+      this.collectVariable(decl.variableDeclaration()!, undefined);
+    }
+
+    // ADR-034: Handle bitmap declarations
+    if (decl.bitmapDeclaration()) {
+      this.collectBitmap(decl.bitmapDeclaration()!, undefined);
+    }
+  }
+
+  // ADR-016: Collect scope declarations (renamed from namespace)
+  private collectScope(scopeDecl: Parser.ScopeDeclarationContext): void {
+    const name = scopeDecl.IDENTIFIER().getText();
+    const line = scopeDecl.start?.line ?? 0;
+
+    this.symbols.push({
+      name,
+      kind: ESymbolKind.Namespace, // Keep as Namespace kind for backwards compat
+      sourceFile: this.sourceFile,
+      sourceLine: line,
+      sourceLanguage: ESourceLanguage.CNext,
+      isExported: true,
+    });
+
+    // Collect scope members
+    for (const member of scopeDecl.scopeMember()) {
+      if (member.variableDeclaration()) {
+        this.collectVariable(member.variableDeclaration()!, name);
+      }
+      if (member.functionDeclaration()) {
+        this.collectFunction(member.functionDeclaration()!, name);
+      }
+      // ADR-034: Handle bitmap declarations in scopes
+      if (member.bitmapDeclaration()) {
+        this.collectBitmap(member.bitmapDeclaration()!, name);
+      }
+      // Handle register declarations in scopes
+      if (member.registerDeclaration()) {
+        this.collectScopedRegister(member.registerDeclaration()!, name);
+      }
+    }
+  }
+
+  // Collect scoped register declarations (e.g., scope.register -> Scope_Register)
+  private collectScopedRegister(
+    reg: Parser.RegisterDeclarationContext,
+    scopeName: string,
+  ): void {
+    const regName = reg.IDENTIFIER().getText();
+    const fullName = `${scopeName}_${regName}`;
+    const line = reg.start?.line ?? 0;
+
+    this.symbols.push({
+      name: fullName,
+      kind: ESymbolKind.Register,
+      sourceFile: this.sourceFile,
+      sourceLine: line,
+      sourceLanguage: ESourceLanguage.CNext,
+      isExported: true,
+      parent: scopeName,
+    });
+
+    // Collect register members with full scoped prefix
+    for (const member of reg.registerMember()) {
+      const memberName = member.IDENTIFIER().getText();
+      const memberLine = member.start?.line ?? 0;
+      const accessMod = member.accessModifier().getText();
+      const memberType = member.type()?.getText() ?? "u32";
+
+      this.symbols.push({
+        name: `${fullName}_${memberName}`,
+        kind: ESymbolKind.RegisterMember,
+        type: memberType,
+        sourceFile: this.sourceFile,
+        sourceLine: memberLine,
+        sourceLanguage: ESourceLanguage.CNext,
+        isExported: true,
+        parent: fullName,
+        accessModifier: accessMod,
+      });
+    }
+  }
+
+  private collectStruct(struct: Parser.StructDeclarationContext): void {
+    const name = struct.IDENTIFIER().getText();
+    const line = struct.start?.line ?? 0;
+
+    this.symbols.push({
+      name,
+      kind: ESymbolKind.Struct,
+      sourceFile: this.sourceFile,
+      sourceLine: line,
+      sourceLanguage: ESourceLanguage.CNext,
+      isExported: true,
+    });
+  }
+
+  private collectRegister(reg: Parser.RegisterDeclarationContext): void {
+    const name = reg.IDENTIFIER().getText();
+    const line = reg.start?.line ?? 0;
+
+    this.symbols.push({
+      name,
+      kind: ESymbolKind.Register,
+      sourceFile: this.sourceFile,
+      sourceLine: line,
+      sourceLanguage: ESourceLanguage.CNext,
+      isExported: true,
+    });
+
+    // Collect register members
+    for (const member of reg.registerMember()) {
+      const memberName = member.IDENTIFIER().getText();
+      const memberLine = member.start?.line ?? 0;
+      const accessMod = member.accessModifier().getText();
+      const memberType = member.type()?.getText() ?? "u32";
+
+      this.symbols.push({
+        name: `${name}_${memberName}`,
+        kind: ESymbolKind.RegisterMember,
+        type: memberType,
+        sourceFile: this.sourceFile,
+        sourceLine: memberLine,
+        sourceLanguage: ESourceLanguage.CNext,
+        isExported: true,
+        parent: name,
+        accessModifier: accessMod,
+      });
+    }
+  }
+
+  // ADR-034: Collect bitmap declarations
+  private collectBitmap(
+    bitmap: Parser.BitmapDeclarationContext,
+    parent: string | undefined,
+  ): void {
+    const name = bitmap.IDENTIFIER().getText();
+    const line = bitmap.start?.line ?? 0;
+    const bitmapType = bitmap.bitmapType().getText();
+    const fullName = parent ? `${parent}_${name}` : name;
+
+    // Add the bitmap type symbol
+    this.symbols.push({
+      name: fullName,
+      kind: ESymbolKind.Bitmap,
+      type: bitmapType,
+      sourceFile: this.sourceFile,
+      sourceLine: line,
+      sourceLanguage: ESourceLanguage.CNext,
+      isExported: true,
+      parent,
+    });
+
+    // Collect bitmap fields
+    let bitOffset = 0;
+    for (const member of bitmap.bitmapMember()) {
+      const fieldName = member.IDENTIFIER().getText();
+      const fieldLine = member.start?.line ?? line;
+
+      // Get field width: default 1 bit, or explicit [N]
+      let width = 1;
+      const intLiteral = member.INTEGER_LITERAL();
+      if (intLiteral) {
+        width = parseInt(intLiteral.getText(), 10);
+      }
+
+      // Calculate bit range for display
+      const bitEnd = bitOffset + width - 1;
+      const bitRange =
+        width === 1 ? `bit ${bitOffset}` : `bits ${bitOffset}-${bitEnd}`;
+
+      this.symbols.push({
+        name: `${fullName}_${fieldName}`,
+        kind: ESymbolKind.BitmapField,
+        type:
+          width === 1 ? "bool" : `u${width <= 8 ? 8 : width <= 16 ? 16 : 32}`,
+        sourceFile: this.sourceFile,
+        sourceLine: fieldLine,
+        sourceLanguage: ESourceLanguage.CNext,
+        isExported: true,
+        parent: fullName,
+        signature: `${bitRange} (${width} bit${width > 1 ? "s" : ""})`,
+      });
+
+      bitOffset += width;
+    }
+  }
+
+  private collectFunction(
+    func: Parser.FunctionDeclarationContext,
+    parent: string | undefined,
+  ): void {
+    const name = func.IDENTIFIER().getText();
+    const line = func.start?.line ?? 0;
+    const returnType = func.type()?.getText() ?? "void";
+    const fullName = parent ? `${parent}_${name}` : name;
+
+    // Build signature for overload detection
+    const params = func.parameterList()?.parameter() ?? [];
+    const paramTypes = params.map((p) => p.type()?.getText() ?? "unknown");
+    const signature = `${returnType} ${fullName}(${paramTypes.join(", ")})`;
+
+    this.symbols.push({
+      name: fullName,
+      kind: ESymbolKind.Function,
+      type: returnType,
+      sourceFile: this.sourceFile,
+      sourceLine: line,
+      sourceLanguage: ESourceLanguage.CNext,
+      isExported: true,
+      parent,
+      signature,
+    });
+
+    // Collect function parameters as symbols for hover support
+    for (const param of params) {
+      const paramName = param.IDENTIFIER().getText();
+      const paramLine = param.start?.line ?? line;
+      const paramType = param.type()?.getText() ?? "unknown";
+      const arrayDims = param.arrayDimension();
+      const isArray = arrayDims.length > 0;
+      const displayType = isArray ? `${paramType}[]` : paramType;
+
+      this.symbols.push({
+        name: paramName,
+        kind: ESymbolKind.Variable,
+        type: displayType,
+        sourceFile: this.sourceFile,
+        sourceLine: paramLine,
+        sourceLanguage: ESourceLanguage.CNext,
+        isExported: false,
+        parent: fullName, // Parent is the function
+      });
+    }
+  }
+
+  private collectVariable(
+    varDecl: Parser.VariableDeclarationContext,
+    parent: string | undefined,
+  ): void {
+    const name = varDecl.IDENTIFIER().getText();
+    const line = varDecl.start?.line ?? 0;
+    const typeCtx = varDecl.type();
+    const varType = typeCtx ? typeCtx.getText() : "unknown";
+    const fullName = parent ? `${parent}_${name}` : name;
+
+    // Check for array (ADR-036: arrayDimension() now returns array for multi-dim)
+    const arrayDims = varDecl.arrayDimension();
+    let size: number | undefined;
+    if (arrayDims.length > 0) {
+      // Get first dimension size for symbol tracking
+      const dimText = arrayDims[0].getText();
+      const match = dimText.match(/\[(\d+)\]/);
+      if (match) {
+        size = parseInt(match[1], 10);
+      }
+    }
+
+    this.symbols.push({
+      name: fullName,
+      kind: ESymbolKind.Variable,
+      type: varType,
+      sourceFile: this.sourceFile,
+      sourceLine: line,
+      sourceLanguage: ESourceLanguage.CNext,
+      isExported: true,
+      parent,
+      size,
+    });
+  }
+}
+
+export default CNextSymbolCollector;