additional option for snake case

This will usually be enabled in conjunction with camelCase, to the degree
that maybe this is a bit confused?
This commit is contained in:
Emil Lerch 2021-05-13 10:56:14 -07:00
parent 1d10ec3e17
commit da5e58872f
Signed by: lobo
GPG Key ID: A7B62D657EF764F8

View File

@ -1452,6 +1452,7 @@ pub const ParseOptions = struct {
} = .Error, } = .Error,
allow_camel_case_conversion: bool = false, allow_camel_case_conversion: bool = false,
allow_snake_case_conversion: bool = false,
allow_unknown_fields: bool = false, allow_unknown_fields: bool = false,
}; };
@ -1470,6 +1471,52 @@ fn camelCaseComp(field: []const u8, key: []const u8, options: ParseOptions) !boo
} }
return std.mem.eql(u8, field, key); return std.mem.eql(u8, field, key);
} }
fn snakeCaseComp(field: []const u8, key: []const u8, options: ParseOptions) !bool {
// snake case is much more intricate. Input:
// Field: user_id
// Key: UserId
// We can duplicate the field and remove all _ characters safely
// Then take the key and lowercase all the uppercase.
// Then compare
var found: u32 = 0;
for (field) |ch| {
if (ch == '_') {
found = found + 1;
break;
}
}
if (found == 0)
return std.mem.eql(u8, field, key);
// We have a snake case field. Let's do this
const allocator = options.allocator orelse return error.AllocatorRequired;
const comp_field = try allocator.alloc(u8, field.len - found);
defer allocator.free(comp_field);
var inx: u32 = 0;
for (field) |ch| {
if (ch != '_') {
comp_field[inx] = ch;
inx = inx + 1;
}
}
// field = 'user_id', comp_field = 'userid'
// We now transform the key by lowercasing. We will only deal with Latin
var utf8_source_key = (std.unicode.Utf8View.init(key) catch unreachable).iterator();
const normalized_key = try allocator.dupeZ(u8, key);
defer allocator.free(normalized_key);
inx = 0;
while (utf8_source_key.nextCodepoint()) |codepoint| {
if (codepoint > 255) return error.InvalidLiteral;
if (codepoint >= 'A' and codepoint <= 'Z') {
// First codepoint is uppercase Latin char, which is all we're handling atm
normalized_key[inx] = normalized_key[inx] + ('a' - 'A');
// We will assume the target field is in camelCase
}
inx = inx + 1;
}
return std.mem.eql(u8, comp_field, normalized_key);
}
fn parseInternal(comptime T: type, token: Token, tokens: *TokenStream, options: ParseOptions) !T { fn parseInternal(comptime T: type, token: Token, tokens: *TokenStream, options: ParseOptions) !T {
switch (@typeInfo(T)) { switch (@typeInfo(T)) {
@ -1573,7 +1620,7 @@ fn parseInternal(comptime T: type, token: Token, tokens: *TokenStream, options:
var found = false; var found = false;
inline for (structInfo.fields) |field, i| { inline for (structInfo.fields) |field, i| {
// TODO: using switches here segfault the compiler (#2727?) // TODO: using switches here segfault the compiler (#2727?)
if ((stringToken.escapes == .None and mem.eql(u8, field.name, key_source_slice)) or (stringToken.escapes == .Some and (field.name.len == stringToken.decodedLength() and encodesTo(field.name, key_source_slice))) or (stringToken.escapes == .None and options.allow_camel_case_conversion and try camelCaseComp(field.name, key_source_slice, options))) { if ((stringToken.escapes == .None and mem.eql(u8, field.name, key_source_slice)) or (stringToken.escapes == .Some and (field.name.len == stringToken.decodedLength() and encodesTo(field.name, key_source_slice))) or (stringToken.escapes == .None and options.allow_camel_case_conversion and try camelCaseComp(field.name, key_source_slice, options)) or (stringToken.escapes == .None and options.allow_snake_case_conversion and try snakeCaseComp(field.name, key_source_slice, options))) {
// if (switch (stringToken.escapes) { // if (switch (stringToken.escapes) {
// .None => mem.eql(u8, field.name, key_source_slice), // .None => mem.eql(u8, field.name, key_source_slice),
// .Some => (field.name.len == stringToken.decodedLength() and encodesTo(field.name, key_source_slice)), // .Some => (field.name.len == stringToken.decodedLength() and encodesTo(field.name, key_source_slice)),