const std = @import("std"); const Builder = std.build.Builder; const Step = std.build.Step; const assert = std.debug.assert; const print = std.debug.print; const Exercise = struct { /// main_file must have the format key_name.zig. /// The key will be used as a shorthand to build /// just one example. main_file: []const u8, /// This is the desired output of the program. /// A program passes if its output ends with this string. output: []const u8, /// This is an optional hint to give if the program does not succeed. hint: []const u8 = "", /// By default, we verify output against stderr. /// Set this to true to check stdout instead. check_stdout: bool = false, /// Returns the name of the main file with .zig stripped. pub fn baseName(self: Exercise) []const u8 { assert(std.mem.endsWith(u8, self.main_file, ".zig")); return self.main_file[0 .. self.main_file.len - 4]; } /// Returns the key of the main file, which is the text before the _. /// For example, "01_hello.zig" has the key "01". pub fn key(self: Exercise) []const u8 { const end_index = std.mem.indexOfScalar(u8, self.main_file, '_'); assert(end_index != null); // main file must be key_description.zig return self.main_file[0..end_index.?]; } }; const exercises = [_]Exercise{ .{ .main_file = "01_hello.zig", .output = "Hello world", .hint = "Note the error: the source file has a hint for fixing 'main'.", }, .{ .main_file = "02_std.zig", .output = "Standard Library", }, .{ .main_file = "03_assignment.zig", .output = "55 314159 -11", .hint = "There are three mistakes in this one!", }, .{ .main_file = "04_arrays.zig", .output = "Fourth: 7, Length: 8", .hint = "There are two things to complete here.", }, .{ .main_file = "05_arrays2.zig", .output = "LEET: 1337, Bits: 100110011001", .hint = "Fill in the two arrays.", }, .{ .main_file = "06_strings.zig", .output = "d=d ha ha ha Major Tom", .hint = "Each '???' needs something filled in.", }, .{ .main_file = "07_strings2.zig", .output = "Ziggy", .hint = "Please fix the lyrics!", }, .{ .main_file = "08_quiz.zig", .output = "Program in Zig", .hint = "See if you can fix the program!", }, .{ .main_file = "09_if.zig", .output = "Foo is 1!", }, .{ .main_file = "10_if2.zig", .output = "price is $17", }, .{ .main_file = "11_while.zig", .output = "n=1024", .hint = "You probably want a 'less than' condition.", }, .{ .main_file = "12_while2.zig", .output = "n=1024", .hint = "It might help to look back at the previous exercise.", }, .{ .main_file = "13_while3.zig", .output = "1 2 4 7 8 11 13 14 16 17 19", }, .{ .main_file = "14_while4.zig", .output = "n=4", }, .{ .main_file = "15_for.zig", .output = "A Dramatic Story: :-) :-) :-( :-| :-) The End.", }, .{ .main_file = "16_for2.zig", .output = "13", }, .{ .main_file = "17_quiz2.zig", .output = "8, Fizz, Buzz, 11, Fizz, 13, 14, FizzBuzz, 16", .hint = "This is a famous game!", }, .{ .main_file = "18_functions.zig", .output = "Question: 42", .hint = "Can you help write the function?", }, .{ .main_file = "19_functions2.zig", .output = "2 4 8 16", }, .{ .main_file = "20_quiz3.zig", .output = "32 64 128 256", .hint = "Unexpected pop quiz! Help!", }, .{ .main_file = "21_errors.zig", .output = "2<4. 3<4. 4=4. 5>4. 6>4.", .hint = "What's the deal with fours?", }, .{ .main_file = "22_errors2.zig", .output = "I compiled", .hint = "Get the error union type right to allow this to compile.", }, .{ .main_file = "23_errors3.zig", .output = "a=64, b=22", }, .{ .main_file = "24_errors4.zig", .output = "a=20, b=14, c=10", }, .{ .main_file = "25_errors5.zig", .output = "a=0, b=19, c=0", }, .{ .main_file = "26_hello2.zig", .output = "Hello world", .hint = "Try using a try!", .check_stdout = true, }, .{ .main_file = "27_defer.zig", .output = "One Two", }, .{ .main_file = "28_defer2.zig", .output = "(Goat) (Cat) (Dog) (Dog) (Goat) (Unknown) done.", }, .{ .main_file = "29_errdefer.zig", .output = "Getting number...got 5. Getting number...failed!", }, .{ .main_file = "30_switch.zig", .output = "ZIG?", }, .{ .main_file = "31_switch2.zig", .output = "ZIG!", }, .{ .main_file = "32_unreachable.zig", .output = "1 2 3 9 8 7", }, .{ .main_file = "33_iferror.zig", .output = "2<4. 3<4. 4=4. 5>4. 6>4.", .hint = "Seriously, what's the deal with fours?", }, .{ .main_file = "34_quiz4.zig", .output = "my_num=42", .hint = "Can you make this work?", .check_stdout = true, }, .{ .main_file = "35_enums.zig", .output = "1 2 3 9 8 7", .hint = "This problem seems familiar...", }, .{ .main_file = "36_enums2.zig", .output = "#0000ff", .hint = "I'm feeling blue about this.", }, .{ .main_file = "37_structs.zig", .output = "Your wizard has 90 health and 25 gold.", }, .{ .main_file = "38_structs2.zig", .output = "Character 2 - G:10 H:100 XP:20", }, .{ .main_file = "39_pointers.zig", .output = "num1: 5, num2: 5", .hint = "Pointers aren't so bad.", }, .{ .main_file = "40_pointers2.zig", .output = "a: 12, b: 12", }, .{ .main_file = "41_pointers3.zig", .output = "foo=6, bar=11", }, .{ .main_file = "42_pointers4.zig", .output = "num: 5, more_nums: 1 1 5 1", }, .{ .main_file = "43_pointers5.zig", .output = "Wizard (G:10 H:100 XP:20)", }, .{ .main_file = "44_quiz5.zig", .output = "Elephant A. Elephant B. Elephant C.", .hint = "Oh no! We forgot Elephant B!", }, }; /// Check the zig version to make sure it can compile the examples properly. /// This will compile with Zig 0.6.0 and later. fn checkVersion() bool { if (!@hasDecl(std.builtin, "zig_version")) { return false; } const needed_version = std.SemanticVersion.parse("0.8.0-dev.1065") catch unreachable; const version = std.builtin.zig_version; const order = version.order(needed_version); return order != .lt; } pub fn build(b: *Builder) void { // Use a comptime branch for the version check. // If this fails, code after this block is not compiled. // It is parsed though, so versions of zig from before 0.6.0 // cannot do the version check and will just fail to compile. // We could fix this by moving the ziglings code to a separate file, // but 0.5.0 was a long time ago, it is unlikely that anyone who // attempts these exercises is still using it. if (comptime !checkVersion()) { // very old versions of Zig used warn instead of print. const stderrPrintFn = if (@hasDecl(std.debug, "print")) std.debug.print else std.debug.warn; stderrPrintFn( \\Error: Your version of zig is too old. Please download a master build from \\https://ziglang.org/download/ \\ , .{}); return; } use_color_escapes = false; switch (b.color) { .on => use_color_escapes = true, .off => use_color_escapes = false, .auto => { if (std.io.getStdErr().supportsAnsiEscapeCodes()) { use_color_escapes = true; } else if (std.builtin.os.tag == .windows) { const w32 = struct { const WINAPI = std.os.windows.WINAPI; const DWORD = std.os.windows.DWORD; const ENABLE_VIRTUAL_TERMINAL_PROCESSING = 0x0004; const STD_ERROR_HANDLE = @bitCast(DWORD, @as(i32, -12)); extern "kernel32" fn GetStdHandle(id: DWORD) callconv(WINAPI) ?*c_void; extern "kernel32" fn GetConsoleMode(console: ?*c_void, out_mode: *DWORD) callconv(WINAPI) u32; extern "kernel32" fn SetConsoleMode(console: ?*c_void, mode: DWORD) callconv(WINAPI) u32; }; const handle = w32.GetStdHandle(w32.STD_ERROR_HANDLE); var mode: w32.DWORD = 0; if (w32.GetConsoleMode(handle, &mode) != 0) { mode |= w32.ENABLE_VIRTUAL_TERMINAL_PROCESSING; use_color_escapes = w32.SetConsoleMode(handle, mode) != 0; } } }, } if (use_color_escapes) { red_text = "\x1b[31m"; green_text = "\x1b[32m"; bold_text = "\x1b[1m"; reset_text = "\x1b[0m"; } const header_step = b.addLog( \\ \\ _ _ _ \\ ___(_) __ _| (_)_ __ __ _ ___ \\ |_ | |/ _' | | | '_ \ / _' / __| \\ / /| | (_| | | | | | | (_| \__ \ \\ /___|_|\__, |_|_|_| |_|\__, |___/ \\ |___/ |___/ \\ \\ , .{}); const verify_all = b.step("ziglings", "Verify all ziglings"); verify_all.dependOn(&header_step.step); b.default_step = verify_all; var prev_chain_verify = verify_all; for (exercises) |ex| { const base_name = ex.baseName(); const file_path = std.fs.path.join(b.allocator, &[_][]const u8{ "exercises", ex.main_file, }) catch unreachable; const build_step = b.addExecutable(base_name, file_path); build_step.install(); const verify_step = ZiglingStep.create(b, ex); const key = ex.key(); const named_test = b.step(b.fmt("{s}_test", .{key}), b.fmt("Run {s} without verifying output", .{ex.main_file})); const run_step = build_step.run(); named_test.dependOn(&run_step.step); const named_install = b.step(b.fmt("{s}_install", .{key}), b.fmt("Install {s} to zig-cache/bin", .{ex.main_file})); named_install.dependOn(&build_step.install_step.?.step); const named_verify = b.step(b.fmt("{s}_only", .{key}), b.fmt("Verify {s} only", .{ex.main_file})); named_verify.dependOn(&verify_step.step); const chain_verify = b.allocator.create(Step) catch unreachable; chain_verify.* = Step.initNoOp(.Custom, b.fmt("chain {s}", .{key}), b.allocator); chain_verify.dependOn(&verify_step.step); const named_chain = b.step(key, b.fmt("Verify all solutions starting at {s}", .{ex.main_file})); named_chain.dependOn(&header_step.step); named_chain.dependOn(chain_verify); prev_chain_verify.dependOn(chain_verify); prev_chain_verify = chain_verify; } } var use_color_escapes = false; var red_text: []const u8 = ""; var green_text: []const u8 = ""; var bold_text: []const u8 = ""; var reset_text: []const u8 = ""; const ZiglingStep = struct { step: Step, exercise: Exercise, builder: *Builder, pub fn create(builder: *Builder, exercise: Exercise) *@This() { const self = builder.allocator.create(@This()) catch unreachable; self.* = .{ .step = Step.init(.Custom, exercise.main_file, builder.allocator, make), .exercise = exercise, .builder = builder, }; return self; } fn make(step: *Step) anyerror!void { const self = @fieldParentPtr(@This(), "step", step); self.makeInternal() catch { if (self.exercise.hint.len > 0) { print("\n{s}hint: {s}{s}", .{ bold_text, self.exercise.hint, reset_text }); } print("\n{s}Edit exercises/{s} and run this again.{s}", .{ red_text, self.exercise.main_file, reset_text }); print("\n{s}To continue from this zigling, use this command:{s}\n {s}zig build {s}{s}\n", .{ red_text, reset_text, bold_text, self.exercise.key(), reset_text }); std.os.exit(0); }; } fn makeInternal(self: *@This()) !void { print("Compiling {s}...\n", .{self.exercise.main_file}); const exe_file = try self.doCompile(); print("Verifying {s}...\n", .{self.exercise.main_file}); const cwd = self.builder.build_root; const argv = [_][]const u8{exe_file}; const child = std.ChildProcess.init(&argv, self.builder.allocator) catch unreachable; defer child.deinit(); child.cwd = cwd; child.env_map = self.builder.env_map; child.stdin_behavior = .Inherit; if (self.exercise.check_stdout) { child.stdout_behavior = .Pipe; child.stderr_behavior = .Inherit; } else { child.stdout_behavior = .Inherit; child.stderr_behavior = .Pipe; } child.spawn() catch |err| { print("{s}Unable to spawn {s}: {s}{s}\n", .{ red_text, argv[0], @errorName(err), reset_text }); return err; }; // Allow up to 1 MB of stdout capture const max_output_len = 1 * 1024 * 1024; const output = if (self.exercise.check_stdout) try child.stdout.?.reader().readAllAlloc(self.builder.allocator, max_output_len) else try child.stderr.?.reader().readAllAlloc(self.builder.allocator, max_output_len); // at this point stdout is closed, wait for the process to terminate const term = child.wait() catch |err| { print("{s}Unable to spawn {s}: {s}{s}\n", .{ red_text, argv[0], @errorName(err), reset_text }); return err; }; // make sure it exited cleanly. switch (term) { .Exited => |code| { if (code != 0) { print("{s}{s} exited with error code {d} (expected {d}){s}\n", .{ red_text, self.exercise.main_file, code, 0, reset_text }); return error.BadExitCode; } }, else => { print("{s}{s} terminated unexpectedly{s}\n", .{ red_text, self.exercise.main_file, reset_text }); return error.UnexpectedTermination; }, } // validate the output if (std.mem.indexOf(u8, output, self.exercise.output) == null) { print( \\ \\{s}----------- Expected this output -----------{s} \\{s} \\{s}----------- but found -----------{s} \\{s} \\{s}-----------{s} \\ , .{ red_text, reset_text, self.exercise.output, red_text, reset_text, output, red_text, reset_text }); return error.InvalidOutput; } print("{s}{s}{s}\n", .{ green_text, output, reset_text }); } // The normal compile step calls os.exit, so we can't use it as a library :( // This is a stripped down copy of std.build.LibExeObjStep.make. fn doCompile(self: *@This()) ![]const u8 { const builder = self.builder; var zig_args = std.ArrayList([]const u8).init(builder.allocator); defer zig_args.deinit(); zig_args.append(builder.zig_exe) catch unreachable; zig_args.append("build-exe") catch unreachable; if (builder.color != .auto) { zig_args.append("--color") catch unreachable; zig_args.append(@tagName(builder.color)) catch unreachable; } const zig_file = std.fs.path.join(builder.allocator, &[_][]const u8{ "exercises", self.exercise.main_file }) catch unreachable; zig_args.append(builder.pathFromRoot(zig_file)) catch unreachable; zig_args.append("--cache-dir") catch unreachable; zig_args.append(builder.pathFromRoot(builder.cache_root)) catch unreachable; zig_args.append("--enable-cache") catch unreachable; const argv = zig_args.items; var code: u8 = undefined; const output_dir_nl = builder.execAllowFail(argv, &code, .Inherit) catch |err| { switch (err) { error.FileNotFound => { print("{s}{s}: Unable to spawn the following command: file not found{s}\n", .{ red_text, self.exercise.main_file, reset_text }); for (argv) |v| print("{s} ", .{v}); print("\n", .{}); }, error.ExitCodeFailure => { print("{s}{s}: The following command exited with error code {}:{s}\n", .{ red_text, self.exercise.main_file, code, reset_text }); for (argv) |v| print("{s} ", .{v}); print("\n", .{}); }, error.ProcessTerminated => { print("{s}{s}: The following command terminated unexpectedly:{s}\n", .{ red_text, self.exercise.main_file, reset_text }); for (argv) |v| print("{s} ", .{v}); print("\n", .{}); }, else => {}, } return err; }; const build_output_dir = std.mem.trimRight(u8, output_dir_nl, "\r\n"); const target_info = std.zig.system.NativeTargetInfo.detect( builder.allocator, .{}, ) catch unreachable; const target = target_info.target; const file_name = std.zig.binNameAlloc(builder.allocator, .{ .root_name = self.exercise.baseName(), .target = target, .output_mode = .Exe, .link_mode = .Static, .version = null, }) catch unreachable; return std.fs.path.join(builder.allocator, &[_][]const u8{ build_output_dir, file_name, }); } };