| ... | @@ -1,47 +0,0 @@ |
| 1 | const std = @import("std"); |
| 2 | |
| 3 | /// This script replaces a matching license header from .zig source files in a directory tree |
| 4 | /// with the `new_header` below. |
| 5 | const new_header = ""; |
| 6 | |
| 7 | pub fn main() !void { |
| 8 | var progress = std.Progress{}; |
| 9 | const root_node = progress.start("", 0); |
| 10 | defer root_node.end(); |
| 11 | |
| 12 | var arena_allocator = std.heap.ArenaAllocator.init(std.heap.page_allocator); |
| 13 | const arena = arena_allocator.allocator(); |
| 14 | |
| 15 | const args = try std.process.argsAlloc(arena); |
| 16 | const path_to_walk = args[1]; |
| 17 | const dir = try std.fs.cwd().openDir(path_to_walk, .{ .iterate = true }); |
| 18 | |
| 19 | var walker = try dir.walk(arena); |
| 20 | defer walker.deinit(); |
| 21 | |
| 22 | var buffer: [500]u8 = undefined; |
| 23 | const expected_header = buffer[0..try std.io.getStdIn().readAll(&buffer)]; |
| 24 | |
| 25 | while (try walker.next()) |entry| { |
| 26 | if (!std.mem.endsWith(u8, entry.basename, ".zig")) |
| 27 | continue; |
| 28 | |
| 29 | var node = root_node.start(entry.basename, 0); |
| 30 | node.activate(); |
| 31 | defer node.end(); |
| 32 | |
| 33 | const source = try dir.readFileAlloc(arena, entry.path, 20 * 1024 * 1024); |
| 34 | if (!std.mem.startsWith(u8, source, expected_header)) { |
| 35 | std.debug.print("no match: {s}\n", .{entry.path}); |
| 36 | continue; |
| 37 | } |
| 38 | |
| 39 | const truncated_source = source[expected_header.len..]; |
| 40 | |
| 41 | const new_source = try arena.alloc(u8, truncated_source.len + new_header.len); |
| 42 | @memcpy(new_source[0..new_source.len], new_header); |
| 43 | @memcpy(new_source[new_header.len .. new_header.len + truncated_source.len], truncated_source); |
| 44 | |
| 45 | try dir.writeFile(entry.path, new_source); |
| 46 | } |
| 47 | } |
Andrew Kelley <andrew@ziglang.org> 2024-01-07 17:02:17-07:00