tiny.smg.scan.clike

1const std = @import("std");2const smg = @import("../root.zig");3const core = @import("core/root.zig");4const scan_metrics = @import("metrics.zig");56const graph_mod = smg.graph;7const lines_mod = smg.lines;8const model = smg.model;9const spans = smg.span;10const tree = smg.tree;1112const SourceLines = lines_mod.SourceLines;1314const Context = struct {15    allocator: std.mem.Allocator,16    scratch: std.mem.Allocator,17    metric_inputs: std.mem.Allocator,18    graph: *graph_mod.Graph,19    path: []const u8,20    source: []const u8,21    is_cpp: bool,22};2324pub fn enhanceMetadata(allocator: std.mem.Allocator, scratch: std.mem.Allocator, metric_inputs: std.mem.Allocator, graph: *graph_mod.Graph, path: []const u8, source: []const u8, source_lines: SourceLines, lang: []const u8, limits: smg.ParserLimits) !void {25    const is_cpp = usesCppParser(source, lang);26    if (!treeParseCandidate(source)) return try enhanceMetadataFromText(allocator, metric_inputs, graph, path, source_lines, is_cpp);27    const parsed = if (is_cpp)28        try tree.parseCpp(scratch, source, limits)29    else30        try tree.parseC(scratch, source, limits);31    const parsed_value = parsed orelse return try enhanceMetadataFromText(allocator, metric_inputs, graph, path, source_lines, is_cpp);32    defer tree.deinit(parsed_value);33    const context: Context = .{ .allocator = allocator, .scratch = scratch, .metric_inputs = metric_inputs, .graph = graph, .path = path, .source = source, .is_cpp = is_cpp };34    try enhanceNode(context, tree.root(parsed_value));35}3637pub fn usesCppParser(source: []const u8, lang: []const u8) bool {38    if (std.mem.eql(u8, lang, "cpp") or std.mem.eql(u8, lang, "cpp_header")) return true;39    if (std.mem.eql(u8, lang, "c_header")) return cHeaderLooksLikeCpp(source);40    return false;41}4243pub fn treeParseCandidate(source: []const u8) bool {44    return core.compactParseCandidateAny(source, &.{"int "});45}4647fn enhanceMetadataFromText(allocator: std.mem.Allocator, metric_inputs: std.mem.Allocator, graph: *graph_mod.Graph, path: []const u8, source_lines: SourceLines, is_cpp: bool) !void {48    for (graph.nodes.items) |*node| {49        const file = node.file orelse continue;50        if (!std.mem.eql(u8, file, path)) continue;51        const line = node.line orelse continue;52        const end_line = node.end_line orelse line;53        if (std.mem.eql(u8, node.type, model.NodeType.function) or std.mem.eql(u8, node.type, model.NodeType.method)) {54            if (spans.startsWith(source_lines, line, end_line, "#define ")) {55                node.metadata = try model.mergePairs(allocator, node.metadata, try scan_metrics.textMacroMetadata(allocator, source_lines, line, end_line));56            } else {57                node.metadata = try model.mergePairs(allocator, node.metadata, try scan_metrics.textFunctionMetadata(allocator, metric_inputs, source_lines, line, end_line, .c));58            }59        } else if (std.mem.eql(u8, node.type, model.NodeType.class) or std.mem.eql(u8, node.type, model.NodeType.type)) {60            const c_kind = if (is_cpp) null else model.pairValue(node.metadata, "c_kind");61            node.metadata = try model.mergePairs(allocator, node.metadata, try scan_metrics.textSpanMetadata(allocator, source_lines, line, end_line, c_kind));62        }63    }64}6566fn cHeaderLooksLikeCpp(source: []const u8) bool {67    const markers = [_][]const u8{ "class ", "namespace ", "template ", "public:", "private:", "protected:", "virtual ", "override", "::" };68    for (markers) |marker| if (std.mem.indexOf(u8, source, marker) != null) return true;69    return false;70}7172fn enhanceNode(context: Context, node: tree.Node) anyerror!void {73    if (try enhanceNodeMetadata(context, node)) return;74    const children = tree.childCount(node);75    for (0..children) |raw_index| try enhanceNode(context, tree.child(node, @intCast(raw_index)));76}7778fn enhanceNodeMetadata(context: Context, node: tree.Node) !bool {79    const node_kind = tree.kind(node);80    if (std.mem.eql(u8, node_kind, "type_definition")) {81        try enhanceTypeDefinition(context, node);82        return true;83    }84    if (try enhanceFunctionNode(context, node_kind, node)) return false;85    if (try enhanceRecordNode(context, node_kind, node)) return false;86    try enhanceMethodDeclarationNode(context, node_kind, node);87    return false;88}8990fn enhanceFunctionNode(context: Context, node_kind: []const u8, node: tree.Node) !bool {91    if (std.mem.eql(u8, node_kind, "function_definition")) {92        try enhanceFunctionDefinition(context, node);93        return true;94    }95    if (std.mem.eql(u8, node_kind, "preproc_function_def")) {96        try enhanceMacroDefinition(context, node);97        return true;98    }99    return false;100}101102fn enhanceFunctionDefinition(context: Context, node: tree.Node) !void {103    const types = [_][]const u8{ model.NodeType.function, model.NodeType.method };104    try replaceNodeMetadata(context, node, &types, try scan_metrics.cFunctionMetadata(context.allocator, context.scratch, context.metric_inputs, context.source, node));105}106107fn enhanceMacroDefinition(context: Context, node: tree.Node) !void {108    const types = [_][]const u8{model.NodeType.function};109    try replaceNodeMetadata(context, node, &types, try scan_metrics.cMacroMetadata(context.allocator, context.scratch, context.source, node));110}111112fn enhanceRecordNode(context: Context, node_kind: []const u8, node: tree.Node) !bool {113    if (std.mem.eql(u8, node_kind, "class_specifier")) {114        try enhanceClassSpecifier(context, node);115        return true;116    }117    if (std.mem.eql(u8, node_kind, "struct_specifier")) {118        try enhanceAggregateSpecifier(context, node, model.NodeType.class, "struct");119        return true;120    }121    if (std.mem.eql(u8, node_kind, "union_specifier")) {122        try enhanceAggregateSpecifier(context, node, model.NodeType.class, "union");123        return true;124    }125    if (std.mem.eql(u8, node_kind, "enum_specifier")) {126        try enhanceAggregateSpecifier(context, node, model.NodeType.type, "enum");127        return true;128    }129    return false;130}131132fn enhanceClassSpecifier(context: Context, node: tree.Node) !void {133    const types = [_][]const u8{model.NodeType.class};134    try replaceNodeMetadata(context, node, &types, try scan_metrics.cDeclarationMetadata(context.allocator, context.scratch, context.source, node));135}136137fn enhanceAggregateSpecifier(context: Context, node: tree.Node, node_type: []const u8, c_kind: []const u8) !void {138    const types = [_][]const u8{node_type};139    const metadata = if (context.is_cpp and !std.mem.eql(u8, c_kind, "enum")) try scan_metrics.cDeclarationMetadata(context.allocator, context.scratch, context.source, node) else try scan_metrics.cRecordMetadata(context.allocator, context.scratch, context.source, node, c_kind);140    try replaceNodeMetadata(context, node, &types, metadata);141}142143fn enhanceMethodDeclarationNode(context: Context, node_kind: []const u8, node: tree.Node) !void {144    if (!std.mem.eql(u8, node_kind, "declaration") and !std.mem.eql(u8, node_kind, "field_declaration")) return;145    if (!core.hasDescendant(node, "function_declarator")) return;146    const types = [_][]const u8{model.NodeType.method};147    try replaceNodeMetadata(context, node, &types, try scan_metrics.cDeclarationMetadata(context.allocator, context.scratch, context.source, node));148}149150fn enhanceTypeDefinition(context: Context, node: tree.Node) !void {151    const type_node = core.findChild(node, "struct_specifier") orelse core.findChild(node, "union_specifier") orelse core.findChild(node, "enum_specifier") orelse return;152    const type_kind = tree.kind(type_node);153    const node_type = if (std.mem.eql(u8, type_kind, "enum_specifier")) model.NodeType.type else model.NodeType.class;154    const c_kind = if (std.mem.eql(u8, type_kind, "struct_specifier"))155        "struct"156    else if (std.mem.eql(u8, type_kind, "union_specifier"))157        "union"158    else159        "enum";160    const types = [_][]const u8{node_type};161    try replaceNodeMetadata(context, node, &types, try scan_metrics.cRecordMetadata(context.allocator, context.scratch, context.source, node, c_kind));162}163164fn replaceNodeMetadata(context: Context, node: tree.Node, node_types: []const []const u8, metadata: []const model.Pair) !void {165    replaceUniqueNodeMetadata(context.graph, context.path, tree.startLine(node), tree.endLine(node), node_types, metadata);166}167168fn replaceUniqueNodeMetadata(graph: *graph_mod.Graph, path: []const u8, line: i64, end_line: i64, node_types: []const []const u8, metadata: []const model.Pair) void {169    var found: ?usize = null;170    var matches: usize = 0;171    for (graph.nodes.items, 0..) |node, index| {172        if (!nodeMatchesRange(node, path, line, end_line, node_types)) continue;173        found = index;174        matches += 1;175    }176    if (matches != 1) return;177    graph.nodes.items[found.?].metadata = metadata;178}179180fn nodeMatchesRange(node: model.Node, path: []const u8, line: i64, end_line: i64, node_types: []const []const u8) bool {181    const file = node.file orelse return false;182    if (!std.mem.eql(u8, file, path)) return false;183    if (node.line == null or node.line.? != line) return false;184    if (node.end_line == null or node.end_line.? != end_line) return false;185    for (node_types) |node_type| if (std.mem.eql(u8, node.type, node_type)) return true;186    return false;187}

1pub const clike = @import("clike.zig");