| ... | ... | @@ -0,0 +1,221 @@ |
| 1 | const std = @import("../std.zig"); |
| 2 | const ConfigHeaderStep = @This(); |
| 3 | const Step = std.build.Step; |
| 4 | const Builder = std.build.Builder; |
| 5 | |
| 6 | pub const base_id: Step.Id = .config_header; |
| 7 | |
| 8 | pub const Style = enum { |
| 9 | /// The configure format supported by autotools. It uses `#undef foo` to |
| 10 | /// mark lines that can be substituted with different values. |
| 11 | autoconf, |
| 12 | /// The configure format supported by CMake. It uses `@@FOO@@` and |
| 13 | /// `#cmakedefine` for template substitution. |
| 14 | cmake, |
| 15 | }; |
| 16 | |
| 17 | pub const Value = union(enum) { |
| 18 | undef, |
| 19 | defined, |
| 20 | boolean: bool, |
| 21 | int: i64, |
| 22 | ident: []const u8, |
| 23 | string: []const u8, |
| 24 | }; |
| 25 | |
| 26 | step: Step, |
| 27 | builder: *Builder, |
| 28 | source: std.build.FileSource, |
| 29 | style: Style, |
| 30 | values: std.StringHashMap(Value), |
| 31 | max_bytes: usize = 2 * 1024 * 1024, |
| 32 | output_dir: []const u8, |
| 33 | output_basename: []const u8, |
| 34 | |
| 35 | pub fn create(builder: *Builder, source: std.build.FileSource, style: Style) *ConfigHeaderStep { |
| 36 | const self = builder.allocator.create(ConfigHeaderStep) catch @panic("OOM"); |
| 37 | const name = builder.fmt("configure header {s}", .{source.getDisplayName()}); |
| 38 | self.* = .{ |
| 39 | .builder = builder, |
| 40 | .step = Step.init(base_id, name, builder.allocator, make), |
| 41 | .source = source, |
| 42 | .style = style, |
| 43 | .values = std.StringHashMap(Value).init(builder.allocator), |
| 44 | .output_dir = undefined, |
| 45 | .output_basename = "config.h", |
| 46 | }; |
| 47 | switch (source) { |
| 48 | .path => |p| { |
| 49 | const basename = std.fs.path.basename(p); |
| 50 | if (std.mem.endsWith(u8, basename, ".h.in")) { |
| 51 | self.output_basename = basename[0 .. basename.len - 3]; |
| 52 | } |
| 53 | }, |
| 54 | else => {}, |
| 55 | } |
| 56 | return self; |
| 57 | } |
| 58 | |
| 59 | pub fn addValues(self: *ConfigHeaderStep, values: anytype) void { |
| 60 | return addValuesInner(self, values) catch @panic("OOM"); |
| 61 | } |
| 62 | |
| 63 | fn addValuesInner(self: *ConfigHeaderStep, values: anytype) !void { |
| 64 | inline for (@typeInfo(@TypeOf(values)).Struct.fields) |field| { |
| 65 | switch (@typeInfo(field.type)) { |
| 66 | .Null => { |
| 67 | try self.values.put(field.name, .undef); |
| 68 | }, |
| 69 | .Void => { |
| 70 | try self.values.put(field.name, .defined); |
| 71 | }, |
| 72 | .Bool => { |
| 73 | try self.values.put(field.name, .{ .boolean = @field(values, field.name) }); |
| 74 | }, |
| 75 | .ComptimeInt => { |
| 76 | try self.values.put(field.name, .{ .int = @field(values, field.name) }); |
| 77 | }, |
| 78 | .EnumLiteral => { |
| 79 | try self.values.put(field.name, .{ .ident = @tagName(@field(values, field.name)) }); |
| 80 | }, |
| 81 | .Pointer => |ptr| { |
| 82 | switch (@typeInfo(ptr.child)) { |
| 83 | .Array => |array| { |
| 84 | if (ptr.size == .One and array.child == u8) { |
| 85 | try self.values.put(field.name, .{ .string = @field(values, field.name) }); |
| 86 | continue; |
| 87 | } |
| 88 | }, |
| 89 | else => {}, |
| 90 | } |
| 91 | |
| 92 | @compileError("unsupported ConfigHeaderStep value type: " ++ |
| 93 | @typeName(field.type)); |
| 94 | }, |
| 95 | else => @compileError("unsupported ConfigHeaderStep value type: " ++ |
| 96 | @typeName(field.type)), |
| 97 | } |
| 98 | } |
| 99 | } |
| 100 | |
| 101 | fn make(step: *Step) !void { |
| 102 | const self = @fieldParentPtr(ConfigHeaderStep, "step", step); |
| 103 | const gpa = self.builder.allocator; |
| 104 | const src_path = self.source.getPath(self.builder); |
| 105 | const contents = try std.fs.cwd().readFileAlloc(gpa, src_path, self.max_bytes); |
| 106 | |
| 107 | // The cache is used here not really as a way to speed things up - because writing |
| 108 | // the data to a file would probably be very fast - but as a way to find a canonical |
| 109 | // location to put build artifacts. |
| 110 | |
| 111 | // If, for example, a hard-coded path was used as the location to put ConfigHeaderStep |
| 112 | // files, then two ConfigHeaderStep executing in parallel might clobber each other. |
| 113 | |
| 114 | // TODO port the cache system from the compiler to zig std lib. Until then |
| 115 | // we construct the path directly, and no "cache hit" detection happens; |
| 116 | // the files are always written. |
| 117 | // Note there is very similar code over in WriteFileStep |
| 118 | const Hasher = std.crypto.auth.siphash.SipHash128(1, 3); |
| 119 | // Random bytes to make ConfigHeaderStep unique. Refresh this with new |
| 120 | // random bytes when ConfigHeaderStep implementation is modified in a |
| 121 | // non-backwards-compatible way. |
| 122 | var hash = Hasher.init("X1pQzdDt91Zlh7Eh"); |
| 123 | hash.update(self.source.getDisplayName()); |
| 124 | hash.update(contents); |
| 125 | |
| 126 | var digest: [16]u8 = undefined; |
| 127 | hash.final(&digest); |
| 128 | var hash_basename: [digest.len * 2]u8 = undefined; |
| 129 | _ = std.fmt.bufPrint( |
| 130 | &hash_basename, |
| 131 | "{s}", |
| 132 | .{std.fmt.fmtSliceHexLower(&digest)}, |
| 133 | ) catch unreachable; |
| 134 | |
| 135 | self.output_dir = try std.fs.path.join(gpa, &[_][]const u8{ |
| 136 | self.builder.cache_root, "o", &hash_basename, |
| 137 | }); |
| 138 | |
| 139 | var dir = std.fs.cwd().makeOpenPath(self.output_dir, .{}) catch |err| { |
| 140 | std.debug.print("unable to make path {s}: {s}\n", .{ self.output_dir, @errorName(err) }); |
| 141 | return err; |
| 142 | }; |
| 143 | defer dir.close(); |
| 144 | |
| 145 | var values_copy = try self.values.clone(); |
| 146 | defer values_copy.deinit(); |
| 147 | |
| 148 | var output = std.ArrayList(u8).init(gpa); |
| 149 | defer output.deinit(); |
| 150 | try output.ensureTotalCapacity(contents.len); |
| 151 | |
| 152 | try output.appendSlice("/* This file was generated by ConfigHeaderStep using the Zig Build System. */\n"); |
| 153 | |
| 154 | var any_errors = false; |
| 155 | var line_index: u32 = 0; |
| 156 | var line_it = std.mem.split(u8, contents, "\n"); |
| 157 | while (line_it.next()) |line| : (line_index += 1) { |
| 158 | if (!std.mem.startsWith(u8, line, "#")) { |
| 159 | try output.appendSlice(line); |
| 160 | try output.appendSlice("\n"); |
| 161 | continue; |
| 162 | } |
| 163 | var it = std.mem.tokenize(u8, line[1..], " \t\r"); |
| 164 | const undef = it.next().?; |
| 165 | if (!std.mem.eql(u8, undef, "undef")) { |
| 166 | try output.appendSlice(line); |
| 167 | try output.appendSlice("\n"); |
| 168 | continue; |
| 169 | } |
| 170 | const name = it.rest(); |
| 171 | const kv = values_copy.fetchRemove(name) orelse { |
| 172 | std.debug.print("{s}:{d}: error: unspecified config header value: '{s}'\n", .{ |
| 173 | src_path, line_index + 1, name, |
| 174 | }); |
| 175 | any_errors = true; |
| 176 | continue; |
| 177 | }; |
| 178 | switch (kv.value) { |
| 179 | .undef => { |
| 180 | try output.appendSlice("/* #undef "); |
| 181 | try output.appendSlice(name); |
| 182 | try output.appendSlice(" */\n"); |
| 183 | }, |
| 184 | .defined => { |
| 185 | try output.appendSlice("#define "); |
| 186 | try output.appendSlice(name); |
| 187 | try output.appendSlice("\n"); |
| 188 | }, |
| 189 | .boolean => |b| { |
| 190 | try output.appendSlice("#define "); |
| 191 | try output.appendSlice(name); |
| 192 | try output.appendSlice(" "); |
| 193 | try output.appendSlice(if (b) "true\n" else "false\n"); |
| 194 | }, |
| 195 | .int => |i| { |
| 196 | try output.writer().print("#define {s} {d}\n", .{ name, i }); |
| 197 | }, |
| 198 | .ident => |ident| { |
| 199 | try output.writer().print("#define {s} {s}\n", .{ name, ident }); |
| 200 | }, |
| 201 | .string => |string| { |
| 202 | // TODO: use C-specific escaping instead of zig string literals |
| 203 | try output.writer().print("#define {s} \"{}\"\n", .{ name, std.zig.fmtEscapes(string) }); |
| 204 | }, |
| 205 | } |
| 206 | } |
| 207 | |
| 208 | { |
| 209 | var it = values_copy.iterator(); |
| 210 | while (it.next()) |entry| { |
| 211 | const name = entry.key_ptr.*; |
| 212 | std.debug.print("{s}: error: config header value unused: '{s}'\n", .{ src_path, name }); |
| 213 | } |
| 214 | } |
| 215 | |
| 216 | if (any_errors) { |
| 217 | return error.HeaderConfigFailed; |
| 218 | } |
| 219 | |
| 220 | try dir.writeFile(self.output_basename, output.items); |
| 221 | } |