| ... | ... | @@ -1,295 +0,0 @@ |
| 1 | | const std = @import("std"); |
| 2 | | const string = []const u8; |
| 3 | | |
| 4 | | const c = @cImport({ |
| 5 | | @cInclude("yaml.h"); |
| 6 | | }); |
| 7 | | const u = @import("./index.zig"); |
| 8 | | |
| 9 | | // |
| 10 | | // |
| 11 | | |
| 12 | | pub const Stream = struct { |
| 13 | | docs: []const Document, |
| 14 | | }; |
| 15 | | |
| 16 | | pub const Document = struct { |
| 17 | | mapping: Mapping, |
| 18 | | }; |
| 19 | | |
| 20 | | pub const Item = union(enum) { |
| 21 | | event: Token, |
| 22 | | kv: Key, |
| 23 | | mapping: Mapping, |
| 24 | | sequence: Sequence, |
| 25 | | document: Document, |
| 26 | | string: string, |
| 27 | | stream: Stream, |
| 28 | | |
| 29 | | pub fn format(self: Item, comptime fmt: string, options: std.fmt.FormatOptions, writer: anytype) @TypeOf(writer).Error!void { |
| 30 | | _ = fmt; |
| 31 | | _ = options; |
| 32 | | |
| 33 | | try writer.writeAll("Item{"); |
| 34 | | switch (self) { |
| 35 | | .event => { |
| 36 | | // try std.fmt.format(writer, "{s}", .{@tagName(self.event.type)}); |
| 37 | | try std.fmt.format(writer, "event {d}", .{self.event}); |
| 38 | | }, |
| 39 | | .kv, .document, .stream => { |
| 40 | | unreachable; |
| 41 | | }, |
| 42 | | .mapping => { |
| 43 | | try std.fmt.format(writer, "{}", .{self.mapping}); |
| 44 | | }, |
| 45 | | .sequence => { |
| 46 | | try writer.writeAll("[ "); |
| 47 | | for (self.sequence) |it| { |
| 48 | | try std.fmt.format(writer, "{}, ", .{it}); |
| 49 | | } |
| 50 | | try writer.writeAll("]"); |
| 51 | | }, |
| 52 | | .string => { |
| 53 | | try std.fmt.format(writer, "{s}", .{self.string}); |
| 54 | | }, |
| 55 | | } |
| 56 | | try writer.writeAll("}"); |
| 57 | | } |
| 58 | | }; |
| 59 | | |
| 60 | | pub const Sequence = []const Item; |
| 61 | | |
| 62 | | pub const Key = struct { |
| 63 | | key: string, |
| 64 | | value: Value, |
| 65 | | }; |
| 66 | | |
| 67 | | pub const Value = union(enum) { |
| 68 | | string: string, |
| 69 | | mapping: Mapping, |
| 70 | | sequence: Sequence, |
| 71 | | |
| 72 | | pub fn format(self: Value, comptime fmt: string, options: std.fmt.FormatOptions, writer: anytype) @TypeOf(writer).Error!void { |
| 73 | | _ = fmt; |
| 74 | | _ = options; |
| 75 | | |
| 76 | | try writer.writeAll("Value{"); |
| 77 | | switch (self) { |
| 78 | | .string => { |
| 79 | | try std.fmt.format(writer, "{s}", .{self.string}); |
| 80 | | }, |
| 81 | | .mapping => { |
| 82 | | try std.fmt.format(writer, "{}", .{self.mapping}); |
| 83 | | }, |
| 84 | | .sequence => { |
| 85 | | try writer.writeAll("[ "); |
| 86 | | for (self.sequence) |it| { |
| 87 | | try std.fmt.format(writer, "{}, ", .{it}); |
| 88 | | } |
| 89 | | try writer.writeAll("]"); |
| 90 | | }, |
| 91 | | } |
| 92 | | try writer.writeAll("}"); |
| 93 | | } |
| 94 | | }; |
| 95 | | |
| 96 | | pub const Mapping = struct { |
| 97 | | items: []const Key, |
| 98 | | |
| 99 | | pub fn get(self: Mapping, k: string) ?Value { |
| 100 | | for (self.items) |item| { |
| 101 | | if (std.mem.eql(u8, item.key, k)) { |
| 102 | | return item.value; |
| 103 | | } |
| 104 | | } |
| 105 | | return null; |
| 106 | | } |
| 107 | | |
| 108 | | pub fn get_string(self: Mapping, k: string) string { |
| 109 | | return if (self.get(k)) |v| v.string else ""; |
| 110 | | } |
| 111 | | |
| 112 | | pub fn get_string_array(self: Mapping, alloc: std.mem.Allocator, k: string) ![]string { |
| 113 | | var list = std.ArrayList(string).init(alloc); |
| 114 | | defer list.deinit(); |
| 115 | | if (self.get(k)) |val| { |
| 116 | | if (val == .sequence) { |
| 117 | | for (val.sequence) |item| { |
| 118 | | if (item != .string) { |
| 119 | | continue; |
| 120 | | } |
| 121 | | try list.append(item.string); |
| 122 | | } |
| 123 | | } |
| 124 | | } |
| 125 | | return list.toOwnedSlice(); |
| 126 | | } |
| 127 | | |
| 128 | | pub fn format(self: Mapping, comptime fmt: string, options: std.fmt.FormatOptions, writer: anytype) @TypeOf(writer).Error!void { |
| 129 | | _ = fmt; |
| 130 | | _ = options; |
| 131 | | |
| 132 | | try writer.writeAll("{ "); |
| 133 | | for (self.items) |it| { |
| 134 | | try std.fmt.format(writer, "{s}: ", .{it.key}); |
| 135 | | try std.fmt.format(writer, "{}, ", .{it.value}); |
| 136 | | } |
| 137 | | try writer.writeAll("}"); |
| 138 | | } |
| 139 | | }; |
| 140 | | |
| 141 | | pub const Token = c.yaml_event_t; |
| 142 | | pub const TokenList = []const Token; |
| 143 | | |
| 144 | | // |
| 145 | | // |
| 146 | | |
| 147 | | pub fn parse(alloc: std.mem.Allocator, input: string) !Document { |
| 148 | | var parser: c.yaml_parser_t = undefined; |
| 149 | | _ = c.yaml_parser_initialize(&parser); |
| 150 | | defer c.yaml_parser_delete(&parser); |
| 151 | | |
| 152 | | const lines = try u.split(alloc, input, "\n"); |
| 153 | | |
| 154 | | _ = c.yaml_parser_set_input_string(&parser, input.ptr, input.len); |
| 155 | | |
| 156 | | var all_events = std.ArrayList(Token).init(alloc); |
| 157 | | var event: Token = undefined; |
| 158 | | while (true) { |
| 159 | | const p = c.yaml_parser_parse(&parser, &event); |
| 160 | | if (p == 0) { |
| 161 | | break; |
| 162 | | } |
| 163 | | |
| 164 | | const et = event.type; |
| 165 | | try all_events.append(event); |
| 166 | | c.yaml_event_delete(&event); |
| 167 | | |
| 168 | | if (et == c.YAML_STREAM_END_EVENT) { |
| 169 | | break; |
| 170 | | } |
| 171 | | } |
| 172 | | |
| 173 | | var p = Parser{ |
| 174 | | .alloc = alloc, |
| 175 | | .tokens = all_events.items, |
| 176 | | .lines = lines, |
| 177 | | .index = 0, |
| 178 | | }; |
| 179 | | const stream = try p.parse(); |
| 180 | | return stream.docs[0]; |
| 181 | | } |
| 182 | | |
| 183 | | pub const Parser = struct { |
| 184 | | alloc: std.mem.Allocator, |
| 185 | | tokens: TokenList, |
| 186 | | lines: []const string, |
| 187 | | index: usize, |
| 188 | | |
| 189 | | pub fn parse(self: *Parser) !Stream { |
| 190 | | const item = try parse_item(self, null); |
| 191 | | return item.stream; |
| 192 | | } |
| 193 | | |
| 194 | | fn next(self: *Parser) ?Token { |
| 195 | | if (self.index >= self.tokens.len) { |
| 196 | | return null; |
| 197 | | } |
| 198 | | defer self.index += 1; |
| 199 | | return self.tokens[self.index]; |
| 200 | | } |
| 201 | | }; |
| 202 | | |
| 203 | | pub const Error = |
| 204 | | std.mem.Allocator.Error || |
| 205 | | error{YamlUnexpectedToken}; |
| 206 | | |
| 207 | | fn parse_item(p: *Parser, start: ?Token) Error!Item { |
| 208 | | const tok = start orelse p.next(); |
| 209 | | return switch (tok.?.type) { |
| 210 | | c.YAML_STREAM_START_EVENT => Item{ .stream = try parse_stream(p) }, |
| 211 | | c.YAML_DOCUMENT_START_EVENT => Item{ .document = try parse_document(p) }, |
| 212 | | c.YAML_MAPPING_START_EVENT => Item{ .mapping = try parse_mapping(p) }, |
| 213 | | c.YAML_SEQUENCE_START_EVENT => Item{ .sequence = try parse_sequence(p) }, |
| 214 | | c.YAML_SCALAR_EVENT => Item{ .string = get_event_string(tok.?, p.lines) }, |
| 215 | | else => unreachable, |
| 216 | | }; |
| 217 | | } |
| 218 | | |
| 219 | | fn parse_stream(p: *Parser) Error!Stream { |
| 220 | | var res = std.ArrayList(Document).init(p.alloc); |
| 221 | | defer res.deinit(); |
| 222 | | |
| 223 | | while (true) { |
| 224 | | const tok = p.next(); |
| 225 | | if (tok.?.type == c.YAML_STREAM_END_EVENT) { |
| 226 | | return Stream{ .docs = res.toOwnedSlice() }; |
| 227 | | } |
| 228 | | if (tok.?.type != c.YAML_DOCUMENT_START_EVENT) { |
| 229 | | return error.YamlUnexpectedToken; |
| 230 | | } |
| 231 | | const item = try parse_item(p, tok); |
| 232 | | try res.append(item.document); |
| 233 | | } |
| 234 | | } |
| 235 | | |
| 236 | | fn parse_document(p: *Parser) Error!Document { |
| 237 | | const tok = p.next(); |
| 238 | | if (tok.?.type != c.YAML_MAPPING_START_EVENT) { |
| 239 | | return error.YamlUnexpectedToken; |
| 240 | | } |
| 241 | | const item = try parse_item(p, tok); |
| 242 | | |
| 243 | | if (p.next().?.type != c.YAML_DOCUMENT_END_EVENT) { |
| 244 | | return error.YamlUnexpectedToken; |
| 245 | | } |
| 246 | | return Document{ .mapping = item.mapping }; |
| 247 | | } |
| 248 | | |
| 249 | | fn parse_mapping(p: *Parser) Error!Mapping { |
| 250 | | var res = std.ArrayList(Key).init(p.alloc); |
| 251 | | defer res.deinit(); |
| 252 | | |
| 253 | | while (true) { |
| 254 | | const tok = p.next(); |
| 255 | | if (tok.?.type == c.YAML_MAPPING_END_EVENT) { |
| 256 | | return Mapping{ .items = res.toOwnedSlice() }; |
| 257 | | } |
| 258 | | if (tok.?.type != c.YAML_SCALAR_EVENT) { |
| 259 | | return error.YamlUnexpectedToken; |
| 260 | | } |
| 261 | | try res.append(Key{ |
| 262 | | .key = get_event_string(tok.?, p.lines), |
| 263 | | .value = try parse_value(p), |
| 264 | | }); |
| 265 | | } |
| 266 | | } |
| 267 | | |
| 268 | | fn parse_value(p: *Parser) Error!Value { |
| 269 | | const item = try parse_item(p, null); |
| 270 | | return switch (item) { |
| 271 | | .mapping => |x| Value{ .mapping = x }, |
| 272 | | .sequence => |x| Value{ .sequence = x }, |
| 273 | | .string => |x| Value{ .string = x }, |
| 274 | | else => unreachable, |
| 275 | | }; |
| 276 | | } |
| 277 | | |
| 278 | | fn parse_sequence(p: *Parser) Error!Sequence { |
| 279 | | var res = std.ArrayList(Item).init(p.alloc); |
| 280 | | defer res.deinit(); |
| 281 | | |
| 282 | | while (true) { |
| 283 | | const tok = p.next(); |
| 284 | | if (tok.?.type == c.YAML_SEQUENCE_END_EVENT) { |
| 285 | | return res.toOwnedSlice(); |
| 286 | | } |
| 287 | | try res.append(try parse_item(p, tok)); |
| 288 | | } |
| 289 | | } |
| 290 | | |
| 291 | | fn get_event_string(event: Token, lines: []const string) string { |
| 292 | | const sm = event.start_mark; |
| 293 | | const em = event.end_mark; |
| 294 | | return lines[sm.line][sm.column..em.column]; |
| 295 | | } |