tiny.profiling.iteration.model

1const std = @import("std");23const profiling = @import("../root.zig");4const scenario = profiling.scenario;56pub const schema = "tiny.profiling.iteration/v1";7pub const receipt_schema = "tiny.profiling.iteration-receipt/v1";8pub const test_receipt_schema = "tiny.build.test-run-receipt/v1";9pub const marker_schema = "tiny.build.iteration-generation/v1";10pub const maximum_plan_bytes: usize = 1024 * 1024;11pub const maximum_source_bytes: usize = 16 * 1024 * 1024;12pub const maximum_patch_bytes: usize = 512 * 1024;13pub const maximum_timeout_ms: u64 = 30 * 60 * 1000;14pub const default_timeout_ms: u64 = 5 * 60 * 1000;1516pub const Patch = struct {17    path: []const u8,18    before: []const u8,19    after: []const u8,20};2122pub const Plan = struct {23    name: []const u8,24    step: []const u8,25    args: []const []const u8 = &.{},26    patch: Patch,27    timeout_ms: u64 = default_timeout_ms,2829    pub fn validate(self: Plan) !void {30        try validateString(self.name);31        try validateStep(self.step);32        if (self.args.len > scenario.maximum_arguments) {33            return error.TooManyIterationArguments;34        }35        for (self.args) |argument| {36            try validateString(argument);37            if (reservedArgument(argument)) {38                return error.ReservedIterationArgument;39            }40        }41        if (std.fs.path.isAbsolute(self.patch.path)) {42            return error.AbsoluteIterationPatchPath;43        }44        try validateString(self.patch.path);45        if (self.patch.before.len == 0 or46            self.patch.before.len > maximum_patch_bytes or47            self.patch.after.len == 0 or48            self.patch.after.len > maximum_patch_bytes or49            std.mem.indexOfScalar(u8, self.patch.before, 0) != null or50            std.mem.indexOfScalar(u8, self.patch.after, 0) != null)51        {52            return error.InvalidIterationPatch;53        }54        if (whitespaceEqual(self.patch.before, self.patch.after)) {55            return error.WhitespaceOnlyIterationPatch;56        }57        if (self.timeout_ms < 1000 or self.timeout_ms > maximum_timeout_ms) {58            return error.InvalidIterationTimeout;59        }60    }61};6263fn validateString(value: []const u8) !void {64    if (value.len == 0 or65        value.len > scenario.maximum_string_bytes or66        std.mem.indexOfScalar(u8, value, 0) != null)67    {68        return error.InvalidIterationString;69    }70}7172fn validateStep(value: []const u8) !void {73    try validateString(value);74    if (value[0] == '-') return error.InvalidIterationStep;75    for (value) |byte| {76        if (!std.ascii.isAlphanumeric(byte) and77            byte != '.' and byte != '_' and byte != '-')78        {79            return error.InvalidIterationStep;80        }81    }82    if (std.mem.eql(u8, value, "watch") or83        std.mem.eql(u8, value, "iteration-observe"))84    {85        return error.InvalidIterationStep;86    }87}8889fn reservedArgument(value: []const u8) bool {90    const reserved = [_][]const u8{91        "-Diteration-step",92        "-Diteration-receipt-dir",93        "-Diteration-receipt-token",94        "-Dtiming-receipt-dir",95        "-Dtiming-receipt-token",96        "-Dtiming-runner",97        "-fincremental",98        "-fno-incremental",99        "--watch",100    };101    for (reserved) |prefix| {102        if (std.mem.startsWith(u8, value, prefix) and103            (value.len == prefix.len or value[prefix.len] == '='))104        {105            return true;106        }107    }108    return false;109}110111fn whitespaceEqual(left: []const u8, right: []const u8) bool {112    var left_index: usize = 0;113    var right_index: usize = 0;114    while (true) {115        while (left_index < left.len and116            std.ascii.isWhitespace(left[left_index]))117        {118            left_index += 1;119        }120        while (right_index < right.len and121            std.ascii.isWhitespace(right[right_index]))122        {123            right_index += 1;124        }125        if (left_index == left.len or right_index == right.len) {126            return left_index == left.len and right_index == right.len;127        }128        if (left[left_index] != right[right_index]) return false;129        left_index += 1;130        right_index += 1;131    }132}133134test "iteration plans reject formatting-only patches and receipt overrides" {135    const base = Plan{136        .name = "focused compile",137        .step = "profile-harness-test",138        .patch = .{139            .path = "src/profiling/test.zig",140            .before = "const x = 1;",141            .after = "const x = 2;",142        },143    };144    try base.validate();145    var formatting = base;146    formatting.patch.after = " const   x = 1; ";147    try std.testing.expectError(148        error.WhitespaceOnlyIterationPatch,149        formatting.validate(),150    );151    var override = base;152    override.args = &.{"-Dtiming-runner=false"};153    try std.testing.expectError(154        error.ReservedIterationArgument,155        override.validate(),156    );157}

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