1b: *std.Build,
  2step: *std.Build.Step,
  3test_index: usize,
  4test_filters: []const []const u8,
  5test_target_filters: []const []const u8,
  6
  7const TestCase = struct {
  8    name: []const u8,
  9    sources: std.array_list.Managed(SourceFile),
 10    expected_lines: std.array_list.Managed([]const u8),
 11    allow_warnings: bool,
 12    target: std.Target.Query = .{},
 13
 14    const SourceFile = struct {
 15        filename: []const u8,
 16        source: []const u8,
 17    };
 18
 19    pub fn addSourceFile(self: *TestCase, filename: []const u8, source: []const u8) void {
 20        self.sources.append(SourceFile{
 21            .filename = filename,
 22            .source = source,
 23        }) catch unreachable;
 24    }
 25
 26    pub fn addExpectedLine(self: *TestCase, text: []const u8) void {
 27        self.expected_lines.append(text) catch unreachable;
 28    }
 29};
 30
 31pub fn create(
 32    self: *TranslateCContext,
 33    allow_warnings: bool,
 34    filename: []const u8,
 35    name: []const u8,
 36    source: []const u8,
 37    expected_lines: []const []const u8,
 38) *TestCase {
 39    const tc = self.b.allocator.create(TestCase) catch unreachable;
 40    tc.* = TestCase{
 41        .name = name,
 42        .sources = std.array_list.Managed(TestCase.SourceFile).init(self.b.allocator),
 43        .expected_lines = std.array_list.Managed([]const u8).init(self.b.allocator),
 44        .allow_warnings = allow_warnings,
 45    };
 46
 47    tc.addSourceFile(filename, source);
 48    var arg_i: usize = 0;
 49    while (arg_i < expected_lines.len) : (arg_i += 1) {
 50        tc.addExpectedLine(expected_lines[arg_i]);
 51    }
 52    return tc;
 53}
 54
 55pub fn add(
 56    self: *TranslateCContext,
 57    name: []const u8,
 58    source: []const u8,
 59    expected_lines: []const []const u8,
 60) void {
 61    const tc = self.create(false, "source.h", name, source, expected_lines);
 62    self.addCase(tc);
 63}
 64
 65pub fn addWithTarget(
 66    self: *TranslateCContext,
 67    name: []const u8,
 68    target: std.Target.Query,
 69    source: []const u8,
 70    expected_lines: []const []const u8,
 71) void {
 72    const tc = self.create(false, "source.h", name, source, expected_lines);
 73    tc.target = target;
 74    self.addCase(tc);
 75}
 76
 77pub fn addAllowWarnings(
 78    self: *TranslateCContext,
 79    name: []const u8,
 80    source: []const u8,
 81    expected_lines: []const []const u8,
 82) void {
 83    const tc = self.create(true, "source.h", name, source, expected_lines);
 84    self.addCase(tc);
 85}
 86
 87pub fn addCase(self: *TranslateCContext, case: *const TestCase) void {
 88    const b = self.b;
 89
 90    const translate_c_cmd = "translate-c";
 91    const annotated_case_name = fmt.allocPrint(self.b.allocator, "{s} {s}", .{ translate_c_cmd, case.name }) catch unreachable;
 92    for (self.test_filters) |test_filter| {
 93        if (mem.indexOf(u8, annotated_case_name, test_filter)) |_| break;
 94    } else if (self.test_filters.len > 0) return;
 95
 96    const target = b.resolveTargetQuery(case.target);
 97
 98    if (self.test_target_filters.len > 0) {
 99        const triple_txt = target.query.zigTriple(b.allocator) catch @panic("OOM");
100
101        for (self.test_target_filters) |filter| {
102            if (std.mem.indexOf(u8, triple_txt, filter) != null) break;
103        } else return;
104    }
105
106    const write_src = b.addWriteFiles();
107    const first_src = case.sources.items[0];
108    const root_source_file = write_src.add(first_src.filename, first_src.source);
109    for (case.sources.items[1..]) |src_file| {
110        _ = write_src.add(src_file.filename, src_file.source);
111    }
112
113    const translate_c = b.addTranslateC(.{
114        .root_source_file = root_source_file,
115        .target = target,
116        .optimize = .Debug,
117    });
118
119    translate_c.step.name = annotated_case_name;
120
121    const check_file = translate_c.addCheckFile(case.expected_lines.items);
122
123    self.step.dependOn(&check_file.step);
124}
125
126const TranslateCContext = @This();
127const std = @import("std");
128const fmt = std.fmt;
129const mem = std.mem;
130const fs = std.fs;