tvl-depot/users/Profpatsch/netencode/netencode.nix
Profpatsch b725e9b7e4 feat(users/Profpatsch/netencode): add dwim function to nix gen
Basically what you expect, strings to text, ints to 64-bit integers,
attrs and lists to nested records and lists.

Change-Id: I9d3d841f32ab32a152cd61522f02ebde4a9b11d3
Reviewed-on: https://cl.tvl.fyi/c/depot/+/2444
Reviewed-by: Profpatsch <mail@profpatsch.de>
Tested-by: BuildkiteCI
2021-01-27 13:41:18 +00:00

65 lines
1.1 KiB
Nix

{ lib }:
let
netstring = tag: suffix: s:
"${tag}${toString (builtins.stringLength s)}:${s}${suffix}";
unit = "u,";
n1 = b: if b then "n1:1," else "n1:0,";
n = i: n: netstring "n${toString i}" "," (toString n);
i = i: n: netstring "i${toString i}" "," (toString n);
n3 = n 3;
n6 = n 6;
n7 = n 7;
i3 = i 3;
i6 = i 6;
i7 = i 7;
text = netstring "t" ",";
binary = netstring "b" ",";
tag = key: val: netstring "<" "|" key + val;
concatStrings = builtins.concatStringsSep "";
record = lokv: netstring "{" "}"
(concatStrings (map ({key, val}: tag key val) lokv));
list = l: netstring "[" "]" (concatStrings l);
dwim = val:
let match = {
"bool" = n1;
"int" = i6;
"string" = text;
"set" = attrs: record (lib.mapAttrsToList
(k: v: {
key = k;
val = dwim v;
}) attrs);
"list" = l: list (map dwim l);
};
in match.${builtins.typeOf val} val;
in {
inherit
unit
n1
n3
n6
n7
i3
i6
i7
text
binary
tag
record
list
dwim
;
}