1{2 __intrinsic_namespace__:: 'std',34 local std = self,5 local id = std.id,67 isString(v):: std.type(v) == 'string',8 isNumber(v):: std.type(v) == 'number',9 isBoolean(v):: std.type(v) == 'boolean',10 isObject(v):: std.type(v) == 'object',11 isArray(v):: std.type(v) == 'array',12 isFunction(v):: std.type(v) == 'function',1314 toString(a)::15 if std.type(a) == 'string' then a else '' + a,1617 substr(str, from, len)::18 assert std.isString(str) : 'substr first parameter should be a string, got ' + std.type(str);19 assert std.isNumber(from) : 'substr second parameter should be a string, got ' + std.type(from);20 assert std.isNumber(len) : 'substr third parameter should be a string, got ' + std.type(len);21 assert len >= 0 : 'substr third parameter should be greater than zero, got ' + len;22 std.join('', std.makeArray(std.max(0, std.min(len, std.length(str) - from)), function(i) str[i + from])),2324 startsWith(a, b)::25 if std.length(a) < std.length(b) then26 false27 else28 std.substr(a, 0, std.length(b)) == b,2930 endsWith(a, b)::31 if std.length(a) < std.length(b) then32 false33 else34 std.substr(a, std.length(a) - std.length(b), std.length(b)) == b,3536 lstripChars(str, chars)::37 if std.length(str) > 0 && std.member(chars, str[0]) then38 std.lstripChars(str[1:], chars)39 else40 str,4142 rstripChars(str, chars)::43 local len = std.length(str);44 if len > 0 && std.member(chars, str[len - 1]) then45 std.rstripChars(str[:len - 1], chars)46 else47 str,4849 stripChars(str, chars)::50 std.lstripChars(std.rstripChars(str, chars), chars),5152 stringChars(str)::53 std.makeArray(std.length(str), function(i) str[i]),5455 local parse_nat(str, base) =56 assert base > 0 && base <= 16 : 'integer base %d invalid' % base;57 58 local zero_code = std.codepoint('0');59 local upper_a_code = std.codepoint('A');60 local lower_a_code = std.codepoint('a');61 local addDigit(aggregate, char) =62 local code = std.codepoint(char);63 local digit = if code >= lower_a_code then64 code - lower_a_code + 1065 else if code >= upper_a_code then66 code - upper_a_code + 1067 else68 code - zero_code;69 assert digit >= 0 && digit < base : '%s is not a base %d integer' % [str, base];70 base * aggregate + digit;71 std.foldl(addDigit, std.stringChars(str), 0),7273 parseInt(str)::74 assert std.isString(str) : 'Expected string, got ' + std.type(str);75 assert std.length(str) > 0 && str != '-' : 'Not an integer: "%s"' % [str];76 if str[0] == '-' then77 -parse_nat(str[1:], 10)78 else79 parse_nat(str, 10),8081 parseOctal(str)::82 assert std.isString(str) : 'Expected string, got ' + std.type(str);83 assert std.length(str) > 0 : 'Not an octal number: ""';84 parse_nat(str, 8),8586 parseHex(str)::87 assert std.isString(str) : 'Expected string, got ' + std.type(str);88 assert std.length(str) > 0 : 'Not hexadecimal: ""';89 parse_nat(str, 16),9091 split(str, c)::92 assert std.isString(str) : 'std.split first parameter should be a string, got ' + std.type(str);93 assert std.isString(c) : 'std.split second parameter should be a string, got ' + std.type(c);94 assert std.length(c) == 1 : 'std.split second parameter should have length 1, got ' + std.length(c);95 std.splitLimit(str, c, -1),9697 splitLimit(str, c, maxsplits)::98 assert std.isString(str) : 'std.splitLimit first parameter should be a string, got ' + std.type(str);99 assert std.isString(c) : 'std.splitLimit second parameter should be a string, got ' + std.type(c);100 assert std.length(c) == 1 : 'std.splitLimit second parameter should have length 1, got ' + std.length(c);101 assert std.isNumber(maxsplits) : 'std.splitLimit third parameter should be a number, got ' + std.type(maxsplits);102 local aux(str, delim, i, arr, v) =103 local c = str[i];104 local i2 = i + 1;105 if i >= std.length(str) then106 arr + [v]107 else if c == delim && (maxsplits == -1 || std.length(arr) < maxsplits) then108 aux(str, delim, i2, arr + [v], '') tailstrict109 else110 aux(str, delim, i2, arr, v + c) tailstrict;111 aux(str, c, 0, [], ''),112113 strReplace(str, from, to)::114 assert std.isString(str);115 assert std.isString(from);116 assert std.isString(to);117 assert from != '' : "'from' string must not be zero length.";118119 120 local str_len = std.length(str);121 local from_len = std.length(from);122123 124 local found_at(i) = str[i:i + from_len] == from;125126 127 128 local replace_after(start_index, curr_index, acc) =129 if curr_index > str_len then130 acc + str[start_index:curr_index]131 else if found_at(curr_index) then132 local new_index = curr_index + std.length(from);133 replace_after(new_index, new_index, acc + str[start_index:curr_index] + to) tailstrict134 else135 replace_after(start_index, curr_index + 1, acc) tailstrict;136137 138 139 if from_len == 1 then140 std.join(to, std.split(str, from))141 else142 replace_after(0, 0, ''),143144 asciiUpper(str)::145 local cp = std.codepoint;146 local up_letter(c) = if cp(c) >= 97 && cp(c) < 123 then147 std.char(cp(c) - 32)148 else149 c;150 std.join('', std.map(up_letter, std.stringChars(str))),151152 asciiLower(str)::153 local cp = std.codepoint;154 local down_letter(c) = if cp(c) >= 65 && cp(c) < 91 then155 std.char(cp(c) + 32)156 else157 c;158 std.join('', std.map(down_letter, std.stringChars(str))),159160 range(from, to)::161 std.makeArray(to - from + 1, function(i) i + from),162163 repeat(what, count)::164 local joiner =165 if std.isString(what) then ''166 else if std.isArray(what) then []167 else error 'std.repeat first argument must be an array or a string';168 std.join(joiner, std.makeArray(count, function(i) what)),169170 slice(indexable, index, end, step)::171 local invar =172 173 {174 indexable: indexable,175 index:176 if index == null then 0177 else index,178 end:179 if end == null then std.length(indexable)180 else end,181 step:182 if step == null then 1183 else step,184 length: std.length(indexable),185 type: std.type(indexable),186 };187 assert invar.index >= 0 && invar.end >= 0 && invar.step >= 0 : 'got [%s:%s:%s] but negative index, end, and steps are not supported' % [invar.index, invar.end, invar.step];188 assert step != 0 : 'got %s but step must be greater than 0' % step;189 assert std.isString(indexable) || std.isArray(indexable) : 'std.slice accepts a string or an array, but got: %s' % std.type(indexable);190 local build(slice, cur) =191 if cur >= invar.end || cur >= invar.length then192 slice193 else194 build(195 if invar.type == 'string' then196 slice + invar.indexable[cur]197 else198 slice + [invar.indexable[cur]],199 cur + invar.step200 ) tailstrict;201 build(if invar.type == 'string' then '' else [], invar.index),202203 member(arr, x)::204 if std.isArray(arr) then205 std.count(arr, x) > 0206 else if std.isString(arr) then207 std.length(std.findSubstr(x, arr)) > 0208 else error 'std.member first argument must be an array or a string',209210 count(arr, x):: std.length(std.filter(function(v) v == x, arr)),211212 mod(a, b)::213 if std.isNumber(a) && std.isNumber(b) then214 std.modulo(a, b)215 else if std.isString(a) then216 std.format(a, b)217 else218 error 'Operator % cannot be used on types ' + std.type(a) + ' and ' + std.type(b) + '.',219220 map(func, arr)::221 if !std.isFunction(func) then222 error ('std.map first param must be function, got ' + std.type(func))223 else if !std.isArray(arr) && !std.isString(arr) then224 error ('std.map second param must be array / string, got ' + std.type(arr))225 else226 std.makeArray(std.length(arr), function(i) func(arr[i])),227228 mapWithIndex(func, arr)::229 if !std.isFunction(func) then230 error ('std.mapWithIndex first param must be function, got ' + std.type(func))231 else if !std.isArray(arr) && !std.isString(arr) then232 error ('std.mapWithIndex second param must be array, got ' + std.type(arr))233 else234 std.makeArray(std.length(arr), function(i) func(i, arr[i])),235236 mapWithKey(func, obj)::237 if !std.isFunction(func) then238 error ('std.mapWithKey first param must be function, got ' + std.type(func))239 else if !std.isObject(obj) then240 error ('std.mapWithKey second param must be object, got ' + std.type(obj))241 else242 { [k]: func(k, obj[k]) for k in std.objectFields(obj) },243244 flatMap(func, arr)::245 if !std.isFunction(func) then246 error ('std.flatMap first param must be function, got ' + std.type(func))247 else if std.isArray(arr) then248 std.flattenArrays(std.makeArray(std.length(arr), function(i) func(arr[i])))249 else if std.isString(arr) then250 std.join('', std.makeArray(std.length(arr), function(i) func(arr[i])))251 else error ('std.flatMap second param must be array / string, got ' + std.type(arr)),252253 join(sep, arr)::254 local aux(arr, i, first, running) =255 if i >= std.length(arr) then256 running257 else if arr[i] == null then258 aux(arr, i + 1, first, running) tailstrict259 else if std.type(arr[i]) != std.type(sep) then260 error 'expected %s but arr[%d] was %s ' % [std.type(sep), i, std.type(arr[i])]261 else if first then262 aux(arr, i + 1, false, running + arr[i]) tailstrict263 else264 aux(arr, i + 1, false, running + sep + arr[i]) tailstrict;265 if !std.isArray(arr) then266 error 'join second parameter should be array, got ' + std.type(arr)267 else if std.isString(sep) then268 aux(arr, 0, true, '')269 else if std.isArray(sep) then270 aux(arr, 0, true, [])271 else272 error 'join first parameter should be string or array, got ' + std.type(sep),273274 lines(arr)::275 std.join('\n', arr + ['']),276277 deepJoin(arr)::278 if std.isString(arr) then279 arr280 else if std.isArray(arr) then281 std.join('', [std.deepJoin(x) for x in arr])282 else283 error 'Expected string or array, got %s' % std.type(arr),284285286 format(str, vals)::287288 289 290 291292 local try_parse_mapping_key(str, i) =293 assert i < std.length(str) : 'Truncated format code.';294 local c = str[i];295 if c == '(' then296 local consume(str, j, v) =297 if j >= std.length(str) then298 error 'Truncated format code.'299 else300 local c = str[j];301 if c != ')' then302 consume(str, j + 1, v + c)303 else304 { i: j + 1, v: v };305 consume(str, i + 1, '')306 else307 { i: i, v: null };308309 local try_parse_cflags(str, i) =310 local consume(str, j, v) =311 assert j < std.length(str) : 'Truncated format code.';312 local c = str[j];313 if c == '#' then314 consume(str, j + 1, v { alt: true })315 else if c == '0' then316 consume(str, j + 1, v { zero: true })317 else if c == '-' then318 consume(str, j + 1, v { left: true })319 else if c == ' ' then320 consume(str, j + 1, v { blank: true })321 else if c == '+' then322 consume(str, j + 1, v { sign: true })323 else324 { i: j, v: v };325 consume(str, i, { alt: false, zero: false, left: false, blank: false, sign: false });326327 local try_parse_field_width(str, i) =328 if i < std.length(str) && str[i] == '*' then329 { i: i + 1, v: '*' }330 else331 local consume(str, j, v) =332 assert j < std.length(str) : 'Truncated format code.';333 local c = str[j];334 if c == '0' then335 consume(str, j + 1, v * 10 + 0)336 else if c == '1' then337 consume(str, j + 1, v * 10 + 1)338 else if c == '2' then339 consume(str, j + 1, v * 10 + 2)340 else if c == '3' then341 consume(str, j + 1, v * 10 + 3)342 else if c == '4' then343 consume(str, j + 1, v * 10 + 4)344 else if c == '5' then345 consume(str, j + 1, v * 10 + 5)346 else if c == '6' then347 consume(str, j + 1, v * 10 + 6)348 else if c == '7' then349 consume(str, j + 1, v * 10 + 7)350 else if c == '8' then351 consume(str, j + 1, v * 10 + 8)352 else if c == '9' then353 consume(str, j + 1, v * 10 + 9)354 else355 { i: j, v: v };356 consume(str, i, 0);357358 local try_parse_precision(str, i) =359 assert i < std.length(str) : 'Truncated format code.';360 local c = str[i];361 if c == '.' then362 try_parse_field_width(str, i + 1)363 else364 { i: i, v: null };365366 367 local try_parse_length_modifier(str, i) =368 assert i < std.length(str) : 'Truncated format code.';369 local c = str[i];370 if c == 'h' || c == 'l' || c == 'L' then371 i + 1372 else373 i;374375 local parse_conv_type(str, i) =376 assert i < std.length(str) : 'Truncated format code.';377 local c = str[i];378 if c == 'd' || c == 'i' || c == 'u' then379 { i: i + 1, v: 'd', caps: false }380 else if c == 'o' then381 { i: i + 1, v: 'o', caps: false }382 else if c == 'x' then383 { i: i + 1, v: 'x', caps: false }384 else if c == 'X' then385 { i: i + 1, v: 'x', caps: true }386 else if c == 'e' then387 { i: i + 1, v: 'e', caps: false }388 else if c == 'E' then389 { i: i + 1, v: 'e', caps: true }390 else if c == 'f' then391 { i: i + 1, v: 'f', caps: false }392 else if c == 'F' then393 { i: i + 1, v: 'f', caps: true }394 else if c == 'g' then395 { i: i + 1, v: 'g', caps: false }396 else if c == 'G' then397 { i: i + 1, v: 'g', caps: true }398 else if c == 'c' then399 { i: i + 1, v: 'c', caps: false }400 else if c == 's' then401 { i: i + 1, v: 's', caps: false }402 else if c == '%' then403 { i: i + 1, v: '%', caps: false }404 else405 error 'Unrecognised conversion type: ' + c;406407408 409 local parse_code(str, i) =410 assert i < std.length(str) : 'Truncated format code.';411 local mkey = try_parse_mapping_key(str, i);412 local cflags = try_parse_cflags(str, mkey.i);413 local fw = try_parse_field_width(str, cflags.i);414 local prec = try_parse_precision(str, fw.i);415 local len_mod = try_parse_length_modifier(str, prec.i);416 local ctype = parse_conv_type(str, len_mod);417 {418 i: ctype.i,419 code: {420 mkey: mkey.v,421 cflags: cflags.v,422 fw: fw.v,423 prec: prec.v,424 ctype: ctype.v,425 caps: ctype.caps,426 },427 };428429 430 local parse_codes(str, i, out, cur) =431 if i >= std.length(str) then432 out + [cur]433 else434 local c = str[i];435 if c == '%' then436 local r = parse_code(str, i + 1);437 parse_codes(str, r.i, out + [cur, r.code], '') tailstrict438 else439 parse_codes(str, i + 1, out, cur + c) tailstrict;440441 local codes = parse_codes(str, 0, [], '');442443444 445 446 447448 449 local padding(w, s) =450 local aux(w, v) =451 if w <= 0 then452 v453 else454 aux(w - 1, v + s);455 aux(w, '');456457 458 local pad_left(str, w, s) =459 padding(w - std.length(str), s) + str;460461 462 local pad_right(str, w, s) =463 str + padding(w - std.length(str), s);464465 466 local render_int(n__, min_chars, min_digits, blank, sign, radix, zero_prefix) =467 local n_ = std.abs(n__);468 local aux(n) =469 if n == 0 then470 zero_prefix471 else472 aux(std.floor(n / radix)) + (n % radix);473 local dec = if std.floor(n_) == 0 then '0' else aux(std.floor(n_));474 local neg = n__ < 0;475 local zp = min_chars - (if neg || blank || sign then 1 else 0);476 local zp2 = std.max(zp, min_digits);477 local dec2 = pad_left(dec, zp2, '0');478 (if neg then '-' else if sign then '+' else if blank then ' ' else '') + dec2;479480 481 local render_hex(n__, min_chars, min_digits, blank, sign, add_zerox, capitals) =482 local numerals = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]483 + if capitals then ['A', 'B', 'C', 'D', 'E', 'F']484 else ['a', 'b', 'c', 'd', 'e', 'f'];485 local n_ = std.abs(n__);486 local aux(n) =487 if n == 0 then488 ''489 else490 aux(std.floor(n / 16)) + numerals[n % 16];491 local hex = if std.floor(n_) == 0 then '0' else aux(std.floor(n_));492 local neg = n__ < 0;493 local zp = min_chars - (if neg || blank || sign then 1 else 0)494 - (if add_zerox then 2 else 0);495 local zp2 = std.max(zp, min_digits);496 local hex2 = (if add_zerox then (if capitals then '0X' else '0x') else '')497 + pad_left(hex, zp2, '0');498 (if neg then '-' else if sign then '+' else if blank then ' ' else '') + hex2;499500 local strip_trailing_zero(str) =501 local aux(str, i) =502 if i < 0 then503 ''504 else505 if str[i] == '0' then506 aux(str, i - 1)507 else508 std.substr(str, 0, i + 1);509 aux(str, std.length(str) - 1);510511 512 local render_float_dec(n__, zero_pad, blank, sign, ensure_pt, trailing, prec) =513 local n_ = std.abs(n__);514 local whole = std.floor(n_);515 local dot_size = if prec == 0 && !ensure_pt then 0 else 1;516 local zp = zero_pad - prec - dot_size;517 local str = render_int(std.sign(n__) * whole, zp, 0, blank, sign, 10, '');518 if prec == 0 then519 str + if ensure_pt then '.' else ''520 else521 local frac = std.floor((n_ - whole) * std.pow(10, prec) + 0.5);522 if trailing || frac > 0 then523 local frac_str = render_int(frac, prec, 0, false, false, 10, '');524 str + '.' + if !trailing then strip_trailing_zero(frac_str) else frac_str525 else526 str;527528 529 local render_float_sci(n__, zero_pad, blank, sign, ensure_pt, trailing, caps, prec) =530 local exponent = if n__ == 0 then 0 else std.floor(std.log(std.abs(n__)) / std.log(10));531 local suff = (if caps then 'E' else 'e')532 + render_int(exponent, 3, 0, false, true, 10, '');533 local mantissa = if exponent == -324 then534 535 536 n__ * 10 / std.pow(10, exponent + 1)537 else538 n__ / std.pow(10, exponent);539 local zp2 = zero_pad - std.length(suff);540 render_float_dec(mantissa, zp2, blank, sign, ensure_pt, trailing, prec) + suff;541542 543 local format_code(val, code, fw, prec_or_null, i) =544 local cflags = code.cflags;545 local fpprec = if prec_or_null != null then prec_or_null else 6;546 local iprec = if prec_or_null != null then prec_or_null else 0;547 local zp = if cflags.zero && !cflags.left then fw else 0;548 if code.ctype == 's' then549 std.toString(val)550 else if code.ctype == 'd' then551 if std.type(val) != 'number' then552 error 'Format required number at '553 + i + ', got ' + std.type(val)554 else555 render_int(val, zp, iprec, cflags.blank, cflags.sign, 10, '')556 else if code.ctype == 'o' then557 if std.type(val) != 'number' then558 error 'Format required number at '559 + i + ', got ' + std.type(val)560 else561 local zero_prefix = if cflags.alt then '0' else '';562 render_int(val, zp, iprec, cflags.blank, cflags.sign, 8, zero_prefix)563 else if code.ctype == 'x' then564 if std.type(val) != 'number' then565 error 'Format required number at '566 + i + ', got ' + std.type(val)567 else568 render_hex(val,569 zp,570 iprec,571 cflags.blank,572 cflags.sign,573 cflags.alt,574 code.caps)575 else if code.ctype == 'f' then576 if std.type(val) != 'number' then577 error 'Format required number at '578 + i + ', got ' + std.type(val)579 else580 render_float_dec(val,581 zp,582 cflags.blank,583 cflags.sign,584 cflags.alt,585 true,586 fpprec)587 else if code.ctype == 'e' then588 if std.type(val) != 'number' then589 error 'Format required number at '590 + i + ', got ' + std.type(val)591 else592 render_float_sci(val,593 zp,594 cflags.blank,595 cflags.sign,596 cflags.alt,597 true,598 code.caps,599 fpprec)600 else if code.ctype == 'g' then601 if std.type(val) != 'number' then602 error 'Format required number at '603 + i + ', got ' + std.type(val)604 else605 local exponent = std.floor(std.log(std.abs(val)) / std.log(10));606 if exponent < -4 || exponent >= fpprec then607 render_float_sci(val,608 zp,609 cflags.blank,610 cflags.sign,611 cflags.alt,612 cflags.alt,613 code.caps,614 fpprec - 1)615 else616 local digits_before_pt = std.max(1, exponent + 1);617 render_float_dec(val,618 zp,619 cflags.blank,620 cflags.sign,621 cflags.alt,622 cflags.alt,623 fpprec - digits_before_pt)624 else if code.ctype == 'c' then625 if std.type(val) == 'number' then626 std.char(val)627 else if std.type(val) == 'string' then628 if std.length(val) == 1 then629 val630 else631 error '%c expected 1-sized string got: ' + std.length(val)632 else633 error '%c expected number / string, got: ' + std.type(val)634 else635 error 'Unknown code: ' + code.ctype;636637 638 local format_codes_arr(codes, arr, i, j, v) =639 if i >= std.length(codes) then640 if j < std.length(arr) then641 error ('Too many values to format: ' + std.length(arr) + ', expected ' + j)642 else643 v644 else645 local code = codes[i];646 if std.type(code) == 'string' then647 format_codes_arr(codes, arr, i + 1, j, v + code) tailstrict648 else649 local tmp = if code.fw == '*' then {650 j: j + 1,651 fw: if j >= std.length(arr) then652 error ('Not enough values to format: ' + std.length(arr) + ', expected at least ' + j)653 else654 arr[j],655 } else {656 j: j,657 fw: code.fw,658 };659 local tmp2 = if code.prec == '*' then {660 j: tmp.j + 1,661 prec: if tmp.j >= std.length(arr) then662 error ('Not enough values to format: ' + std.length(arr) + ', expected at least ' + tmp.j)663 else664 arr[tmp.j],665 } else {666 j: tmp.j,667 prec: code.prec,668 };669 local j2 = tmp2.j;670 local val =671 if j2 < std.length(arr) then672 arr[j2]673 else674 error ('Not enough values to format: ' + std.length(arr) + ', expected more than ' + j2);675 local s =676 if code.ctype == '%' then677 '%'678 else679 format_code(val, code, tmp.fw, tmp2.prec, j2);680 local s_padded =681 if code.cflags.left then682 pad_right(s, tmp.fw, ' ')683 else684 pad_left(s, tmp.fw, ' ');685 local j3 =686 if code.ctype == '%' then687 j2688 else689 j2 + 1;690 format_codes_arr(codes, arr, i + 1, j3, v + s_padded) tailstrict;691692 693 local format_codes_obj(codes, obj, i, v) =694 if i >= std.length(codes) then695 v696 else697 local code = codes[i];698 if std.type(code) == 'string' then699 format_codes_obj(codes, obj, i + 1, v + code) tailstrict700 else701 local f =702 if code.mkey == null then703 error 'Mapping keys required.'704 else705 code.mkey;706 local fw =707 if code.fw == '*' then708 error 'Cannot use * field width with object.'709 else710 code.fw;711 local prec =712 if code.prec == '*' then713 error 'Cannot use * precision with object.'714 else715 code.prec;716 local val =717 if std.objectHasAll(obj, f) then718 obj[f]719 else720 error 'No such field: ' + f;721 local s =722 if code.ctype == '%' then723 '%'724 else725 format_code(val, code, fw, prec, f);726 local s_padded =727 if code.cflags.left then728 pad_right(s, fw, ' ')729 else730 pad_left(s, fw, ' ');731 format_codes_obj(codes, obj, i + 1, v + s_padded) tailstrict;732733 if std.isArray(vals) then734 format_codes_arr(codes, vals, 0, 0, '')735 else if std.isObject(vals) then736 format_codes_obj(codes, vals, 0, '')737 else738 format_codes_arr(codes, [vals], 0, 0, ''),739740 foldr(func, arr, init)::741 local aux(func, arr, running, idx) =742 if idx < 0 then743 running744 else745 aux(func, arr, func(arr[idx], running), idx - 1) tailstrict;746 aux(func, arr, init, std.length(arr) - 1),747748 foldl(func, arr, init)::749 local aux(func, arr, running, idx) =750 if idx >= std.length(arr) then751 running752 else753 aux(func, arr, func(running, arr[idx]), idx + 1) tailstrict;754 aux(func, arr, init, 0),755756757 filterMap(filter_func, map_func, arr)::758 if !std.isFunction(filter_func) then759 error ('std.filterMap first param must be function, got ' + std.type(filter_func))760 else if !std.isFunction(map_func) then761 error ('std.filterMap second param must be function, got ' + std.type(map_func))762 else if !std.isArray(arr) then763 error ('std.filterMap third param must be array, got ' + std.type(arr))764 else765 std.map(map_func, std.filter(filter_func, arr)),766767 assertEqual(a, b)::768 if a == b then769 true770 else771 error 'Assertion failed. ' + a + ' != ' + b,772773 abs(n)::774 if !std.isNumber(n) then775 error 'std.abs expected number, got ' + std.type(n)776 else777 if n > 0 then n else -n,778779 sign(n)::780 if !std.isNumber(n) then781 error 'std.sign expected number, got ' + std.type(n)782 else783 if n > 0 then784 1785 else if n < 0 then786 -1787 else 0,788789 max(a, b)::790 if !std.isNumber(a) then791 error 'std.max first param expected number, got ' + std.type(a)792 else if !std.isNumber(b) then793 error 'std.max second param expected number, got ' + std.type(b)794 else795 if a > b then a else b,796797 min(a, b)::798 if !std.isNumber(a) then799 error 'std.min first param expected number, got ' + std.type(a)800 else if !std.isNumber(b) then801 error 'std.min second param expected number, got ' + std.type(b)802 else803 if a < b then a else b,804805 clamp(x, minVal, maxVal)::806 if x < minVal then minVal807 else if x > maxVal then maxVal808 else x,809810 flattenArrays(arrs)::811 std.foldl(function(a, b) a + b, arrs, []),812813 manifestIni(ini)::814 local body_lines(body) =815 std.join([], [816 local value_or_values = body[k];817 if std.isArray(value_or_values) then818 ['%s = %s' % [k, value] for value in value_or_values]819 else820 ['%s = %s' % [k, value_or_values]]821822 for k in std.objectFields(body)823 ]);824825 local section_lines(sname, sbody) = ['[%s]' % [sname]] + body_lines(sbody),826 main_body = if std.objectHas(ini, 'main') then body_lines(ini.main) else [],827 all_sections = [828 section_lines(k, ini.sections[k])829 for k in std.objectFields(ini.sections)830 ];831 std.join('\n', main_body + std.flattenArrays(all_sections) + ['']),832833 manifestToml(value):: std.manifestTomlEx(value, ' '),834835 manifestTomlEx(value, indent)::836 local837 escapeStringToml = std.escapeStringJson,838 escapeKeyToml(key) =839 local bare_allowed = std.set(std.stringChars('ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789_-'));840 if std.setUnion(std.set(std.stringChars(key)), bare_allowed) == bare_allowed then key else escapeStringToml(key),841 isTableArray(v) = std.isArray(v) && std.length(v) > 0 && std.foldl(function(a, b) a && std.isObject(b), v, true),842 isSection(v) = std.isObject(v) || isTableArray(v),843 renderValue(v, indexedPath, inline, cindent) =844 if v == true then845 'true'846 else if v == false then847 'false'848 else if v == null then849 error 'Tried to manifest "null" at ' + indexedPath850 else if std.isNumber(v) then851 '' + v852 else if std.isString(v) then853 escapeStringToml(v)854 else if std.isFunction(v) then855 error 'Tried to manifest function at ' + indexedPath856 else if std.isArray(v) then857 if std.length(v) == 0 then858 '[]'859 else860 local range = std.range(0, std.length(v) - 1);861 local new_indent = if inline then '' else cindent + indent;862 local separator = if inline then ' ' else '\n';863 local lines = ['[' + separator]864 + std.join([',' + separator],865 [866 [new_indent + renderValue(v[i], indexedPath + [i], true, '')]867 for i in range868 ])869 + [separator + (if inline then '' else cindent) + ']'];870 std.join('', lines)871 else if std.isObject(v) then872 local lines = ['{ ']873 + std.join([', '],874 [875 [escapeKeyToml(k) + ' = ' + renderValue(v[k], indexedPath + [k], true, '')]876 for k in std.objectFields(v)877 ])878 + [' }'];879 std.join('', lines),880 renderTableInternal(v, path, indexedPath, cindent) =881 local kvp = std.flattenArrays([882 [cindent + escapeKeyToml(k) + ' = ' + renderValue(v[k], indexedPath + [k], false, cindent)]883 for k in std.objectFields(v)884 if !isSection(v[k])885 ]);886 local sections = [std.join('\n', kvp)] + [887 (888 if std.isObject(v[k]) then889 renderTable(v[k], path + [k], indexedPath + [k], cindent)890 else891 renderTableArray(v[k], path + [k], indexedPath + [k], cindent)892 )893 for k in std.objectFields(v)894 if isSection(v[k])895 ];896 std.join('\n\n', sections),897 renderTable(v, path, indexedPath, cindent) =898 cindent + '[' + std.join('.', std.map(escapeKeyToml, path)) + ']'899 + (if v == {} then '' else '\n')900 + renderTableInternal(v, path, indexedPath, cindent + indent),901 renderTableArray(v, path, indexedPath, cindent) =902 local range = std.range(0, std.length(v) - 1);903 local sections = [904 (cindent + '[[' + std.join('.', std.map(escapeKeyToml, path)) + ']]'905 + (if v[i] == {} then '' else '\n')906 + renderTableInternal(v[i], path, indexedPath + [i], cindent + indent))907 for i in range908 ];909 std.join('\n\n', sections);910 if std.isObject(value) then911 renderTableInternal(value, [], [], '')912 else913 error 'TOML body must be an object. Got ' + std.type(value),914915 escapeStringJson(str_)::916 local str = std.toString(str_);917 local trans(ch) =918 if ch == '"' then919 '\\"'920 else if ch == '\\' then921 '\\\\'922 else if ch == '\b' then923 '\\b'924 else if ch == '\f' then925 '\\f'926 else if ch == '\n' then927 '\\n'928 else if ch == '\r' then929 '\\r'930 else if ch == '\t' then931 '\\t'932 else933 local cp = std.codepoint(ch);934 if cp < 32 || (cp >= 127 && cp <= 159) then935 '\\u%04x' % [cp]936 else937 ch;938 '"%s"' % std.join('', [trans(ch) for ch in std.stringChars(str)]),939940 escapeStringPython(str)::941 std.escapeStringJson(str),942943 escapeStringBash(str_)::944 local str = std.toString(str_);945 local trans(ch) =946 if ch == "'" then947 "'\"'\"'"948 else949 ch;950 "'%s'" % std.join('', [trans(ch) for ch in std.stringChars(str)]),951952 escapeStringDollars(str_)::953 local str = std.toString(str_);954 local trans(ch) =955 if ch == '$' then956 '$$'957 else958 ch;959 std.foldl(function(a, b) a + trans(b), std.stringChars(str), ''),960961 manifestJson(value):: std.manifestJsonEx(value, ' '),962963 manifestJsonEx(value, indent)::964 local aux(v, path, cindent) =965 if v == true then966 'true'967 else if v == false then968 'false'969 else if v == null then970 'null'971 else if std.isNumber(v) then972 '' + v973 else if std.isString(v) then974 std.escapeStringJson(v)975 else if std.isFunction(v) then976 error 'Tried to manifest function at ' + path977 else if std.isArray(v) then978 local range = std.range(0, std.length(v) - 1);979 local new_indent = cindent + indent;980 local lines = ['[\n']981 + std.join([',\n'],982 [983 [new_indent + aux(v[i], path + [i], new_indent)]984 for i in range985 ])986 + ['\n' + cindent + ']'];987 std.join('', lines)988 else if std.isObject(v) then989 local lines = ['{\n']990 + std.join([',\n'],991 [992 [cindent + indent + std.escapeStringJson(k) + ': '993 + aux(v[k], path + [k], cindent + indent)]994 for k in std.objectFields(v)995 ])996 + ['\n' + cindent + '}'];997 std.join('', lines);998 aux(value, [], ''),9991000 manifestYamlDoc(value, indent_array_in_object=false)::1001 local aux(v, path, cindent) =1002 if v == true then1003 'true'1004 else if v == false then1005 'false'1006 else if v == null then1007 'null'1008 else if std.isNumber(v) then1009 '' + v1010 else if std.isString(v) then1011 local len = std.length(v);1012 if len == 0 then1013 '""'1014 else if v[len - 1] == '\n' then1015 local split = std.split(v, '\n');1016 std.join('\n' + cindent + ' ', ['|'] + split[0:std.length(split) - 1])1017 else1018 std.escapeStringJson(v)1019 else if std.isFunction(v) then1020 error 'Tried to manifest function at ' + path1021 else if std.isArray(v) then1022 if std.length(v) == 0 then1023 '[]'1024 else1025 local params(value) =1026 if std.isArray(value) && std.length(value) > 0 then {1027 1028 1029 1030 1031 1032 1033 new_indent: cindent + ' ',1034 space: '\n' + self.new_indent,1035 } else if std.isObject(value) && std.length(value) > 0 then {1036 new_indent: cindent + ' ',1037 1038 1039 1040 space: ' ',1041 } else {1042 1043 new_indent: cindent,1044 space: ' ',1045 };1046 local range = std.range(0, std.length(v) - 1);1047 local parts = [1048 '-' + param.space + aux(v[i], path + [i], param.new_indent)1049 for i in range1050 for param in [params(v[i])]1051 ];1052 std.join('\n' + cindent, parts)1053 else if std.isObject(v) then1054 if std.length(v) == 0 then1055 '{}'1056 else1057 local params(value) =1058 if std.isArray(value) && std.length(value) > 0 then {1059 1060 1061 1062 1063 1064 1065 new_indent: if indent_array_in_object then cindent + ' ' else cindent,1066 space: '\n' + self.new_indent,1067 } else if std.isObject(value) && std.length(value) > 0 then {1068 new_indent: cindent + ' ',1069 space: '\n' + self.new_indent,1070 } else {1071 1072 new_indent: cindent,1073 space: ' ',1074 };1075 local lines = [1076 std.escapeStringJson(k) + ':' + param.space + aux(v[k], path + [k], param.new_indent)1077 for k in std.objectFields(v)1078 for param in [params(v[k])]1079 ];1080 std.join('\n' + cindent, lines);1081 aux(value, [], ''),10821083 manifestYamlStream(value, indent_array_in_object=false, c_document_end=true)::1084 if !std.isArray(value) then1085 error 'manifestYamlStream only takes arrays, got ' + std.type(value)1086 else1087 '---\n' + std.join(1088 '\n---\n', [std.manifestYamlDoc(e, indent_array_in_object) for e in value]1089 ) + if c_document_end then '\n...\n' else '\n',109010911092 manifestPython(v)::1093 if std.isObject(v) then1094 local fields = [1095 '%s: %s' % [std.escapeStringPython(k), std.manifestPython(v[k])]1096 for k in std.objectFields(v)1097 ];1098 '{%s}' % [std.join(', ', fields)]1099 else if std.isArray(v) then1100 '[%s]' % [std.join(', ', [std.manifestPython(v2) for v2 in v])]1101 else if std.isString(v) then1102 '%s' % [std.escapeStringPython(v)]1103 else if std.isFunction(v) then1104 error 'cannot manifest function'1105 else if std.isNumber(v) then1106 std.toString(v)1107 else if v == true then1108 'True'1109 else if v == false then1110 'False'1111 else if v == null then1112 'None',11131114 manifestPythonVars(conf)::1115 local vars = ['%s = %s' % [k, std.manifestPython(conf[k])] for k in std.objectFields(conf)];1116 std.join('\n', vars + ['']),11171118 manifestXmlJsonml(value)::1119 if !std.isArray(value) then1120 error 'Expected a JSONML value (an array), got %s' % std.type(value)1121 else1122 local aux(v) =1123 if std.isString(v) then1124 v1125 else1126 local tag = v[0];1127 local has_attrs = std.length(v) > 1 && std.isObject(v[1]);1128 local attrs = if has_attrs then v[1] else {};1129 local children = if has_attrs then v[2:] else v[1:];1130 local attrs_str =1131 std.join('', [' %s="%s"' % [k, attrs[k]] for k in std.objectFields(attrs)]);1132 std.deepJoin(['<', tag, attrs_str, '>', [aux(x) for x in children], '</', tag, '>']);11331134 aux(value),11351136 local base64_table = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/',1137 local base64_inv = { [base64_table[i]]: i for i in std.range(0, 63) },11381139 base64(input)::1140 local bytes =1141 if std.isString(input) then1142 std.map(function(c) std.codepoint(c), input)1143 else1144 input;11451146 local aux(arr, i, r) =1147 if i >= std.length(arr) then1148 r1149 else if i + 1 >= std.length(arr) then1150 local str =1151 1152 base64_table[(arr[i] & 252) >> 2] +1153 1154 base64_table[(arr[i] & 3) << 4] +1155 '==';1156 aux(arr, i + 3, r + str) tailstrict1157 else if i + 2 >= std.length(arr) then1158 local str =1159 1160 base64_table[(arr[i] & 252) >> 2] +1161 1162 base64_table[(arr[i] & 3) << 4 | (arr[i + 1] & 240) >> 4] +1163 1164 base64_table[(arr[i + 1] & 15) << 2] +1165 '=';1166 aux(arr, i + 3, r + str) tailstrict1167 else1168 local str =1169 1170 base64_table[(arr[i] & 252) >> 2] +1171 1172 base64_table[(arr[i] & 3) << 4 | (arr[i + 1] & 240) >> 4] +1173 1174 base64_table[(arr[i + 1] & 15) << 2 | (arr[i + 2] & 192) >> 6] +1175 1176 base64_table[(arr[i + 2] & 63)];1177 aux(arr, i + 3, r + str) tailstrict;11781179 local sanity = std.foldl(function(r, a) r && (a < 256), bytes, true);1180 if !sanity then1181 error 'Can only base64 encode strings / arrays of single bytes.'1182 else1183 aux(bytes, 0, ''),118411851186 base64DecodeBytes(str)::1187 if std.length(str) % 4 != 0 then1188 error 'Not a base64 encoded string "%s"' % str1189 else1190 local aux(str, i, r) =1191 if i >= std.length(str) then1192 r1193 else1194 1195 local n1 = [base64_inv[str[i]] << 2 | (base64_inv[str[i + 1]] >> 4)];1196 1197 local n2 =1198 if str[i + 2] == '=' then []1199 else [(base64_inv[str[i + 1]] & 15) << 4 | (base64_inv[str[i + 2]] >> 2)];1200 1201 local n3 =1202 if str[i + 3] == '=' then []1203 else [(base64_inv[str[i + 2]] & 3) << 6 | base64_inv[str[i + 3]]];1204 aux(str, i + 4, r + n1 + n2 + n3) tailstrict;1205 aux(str, 0, []),12061207 base64Decode(str)::1208 local bytes = std.base64DecodeBytes(str);1209 std.join('', std.map(function(b) std.char(b), bytes)),12101211 reverse(arr)::1212 local l = std.length(arr);1213 std.makeArray(l, function(i) arr[l - i - 1]),12141215 1216 sortImpl(arr, keyF)::1217 local quickSort(arr, keyF=id) =1218 local l = std.length(arr);1219 if std.length(arr) <= 1 then1220 arr1221 else1222 local pos = 0;1223 local pivot = keyF(arr[pos]);1224 local rest = std.makeArray(l - 1, function(i) if i < pos then arr[i] else arr[i + 1]);1225 local left = std.filter(function(x) keyF(x) < pivot, rest);1226 local right = std.filter(function(x) keyF(x) >= pivot, rest);1227 quickSort(left, keyF) + [arr[pos]] + quickSort(right, keyF);12281229 local merge(a, b) =1230 local la = std.length(a), lb = std.length(b);1231 local aux(i, j, prefix) =1232 if i == la then1233 prefix + b[j:]1234 else if j == lb then1235 prefix + a[i:]1236 else1237 if keyF(a[i]) <= keyF(b[j]) then1238 aux(i + 1, j, prefix + [a[i]]) tailstrict1239 else1240 aux(i, j + 1, prefix + [b[j]]) tailstrict;1241 aux(0, 0, []);12421243 local l = std.length(arr);1244 if std.length(arr) <= 30 then1245 quickSort(arr, keyF=keyF)1246 else1247 local mid = std.floor(l / 2);1248 local left = arr[:mid], right = arr[mid:];1249 merge(std.sort(left, keyF=keyF), std.sort(right, keyF=keyF)),12501251 sort(arr, keyF=id)::1252 std.sortImpl(arr, keyF),12531254 uniq(arr, keyF=id)::1255 local f(a, b) =1256 if std.length(a) == 0 then1257 [b]1258 else if keyF(a[std.length(a) - 1]) == keyF(b) then1259 a1260 else1261 a + [b];1262 std.foldl(f, arr, []),12631264 set(arr, keyF=id)::1265 std.uniq(std.sort(arr, keyF), keyF),12661267 setMember(x, arr, keyF=id)::1268 1269 std.length(std.setInter([x], arr, keyF)) > 0,12701271 setUnion(a, b, keyF=id)::1272 1273 local aux(a, b, i, j, acc) =1274 if i >= std.length(a) then1275 acc + b[j:]1276 else if j >= std.length(b) then1277 acc + a[i:]1278 else1279 local ak = keyF(a[i]);1280 local bk = keyF(b[j]);1281 if ak == bk then1282 aux(a, b, i + 1, j + 1, acc + [a[i]]) tailstrict1283 else if ak < bk then1284 aux(a, b, i + 1, j, acc + [a[i]]) tailstrict1285 else1286 aux(a, b, i, j + 1, acc + [b[j]]) tailstrict;1287 aux(a, b, 0, 0, []),12881289 setInter(a, b, keyF=id)::1290 local aux(a, b, i, j, acc) =1291 if i >= std.length(a) || j >= std.length(b) then1292 acc1293 else1294 if keyF(a[i]) == keyF(b[j]) then1295 aux(a, b, i + 1, j + 1, acc + [a[i]]) tailstrict1296 else if keyF(a[i]) < keyF(b[j]) then1297 aux(a, b, i + 1, j, acc) tailstrict1298 else1299 aux(a, b, i, j + 1, acc) tailstrict;1300 aux(a, b, 0, 0, []) tailstrict,13011302 setDiff(a, b, keyF=id)::1303 local aux(a, b, i, j, acc) =1304 if i >= std.length(a) then1305 acc1306 else if j >= std.length(b) then1307 acc + a[i:]1308 else1309 if keyF(a[i]) == keyF(b[j]) then1310 aux(a, b, i + 1, j + 1, acc) tailstrict1311 else if keyF(a[i]) < keyF(b[j]) then1312 aux(a, b, i + 1, j, acc + [a[i]]) tailstrict1313 else1314 aux(a, b, i, j + 1, acc) tailstrict;1315 aux(a, b, 0, 0, []) tailstrict,13161317 mergePatch(target, patch)::1318 if std.isObject(patch) then1319 local target_object =1320 if std.isObject(target) then target else {};13211322 local target_fields =1323 if std.isObject(target_object) then std.objectFields(target_object) else [];13241325 local null_fields = [k for k in std.objectFields(patch) if patch[k] == null];1326 local both_fields = std.setUnion(target_fields, std.objectFields(patch));13271328 {1329 [k]:1330 if !std.objectHas(patch, k) then1331 target_object[k]1332 else if !std.objectHas(target_object, k) then1333 std.mergePatch(null, patch[k]) tailstrict1334 else1335 std.mergePatch(target_object[k], patch[k]) tailstrict1336 for k in std.setDiff(both_fields, null_fields)1337 }1338 else1339 patch,13401341 objectFields(o)::1342 std.objectFieldsEx(o, false),13431344 objectFieldsAll(o)::1345 std.objectFieldsEx(o, true),13461347 objectHas(o, f)::1348 std.objectHasEx(o, f, false),13491350 objectHasAll(o, f)::1351 std.objectHasEx(o, f, true),13521353 objectValues(o)::1354 [o[k] for k in std.objectFields(o)],13551356 objectValuesAll(o)::1357 [o[k] for k in std.objectFieldsAll(o)],13581359 equals(a, b)::1360 local ta = std.type(a);1361 local tb = std.type(b);1362 if !std.primitiveEquals(ta, tb) then1363 false1364 else1365 if std.primitiveEquals(ta, 'array') then1366 local la = std.length(a);1367 if !std.primitiveEquals(la, std.length(b)) then1368 false1369 else1370 local aux(a, b, i) =1371 if i >= la then1372 true1373 else if a[i] != b[i] then1374 false1375 else1376 aux(a, b, i + 1) tailstrict;1377 aux(a, b, 0)1378 else if std.primitiveEquals(ta, 'object') then1379 local fields = std.objectFields(a);1380 local lfields = std.length(fields);1381 if fields != std.objectFields(b) then1382 false1383 else1384 local aux(a, b, i) =1385 if i >= lfields then1386 true1387 else if local f = fields[i]; a[f] != b[f] then1388 false1389 else1390 aux(a, b, i + 1) tailstrict;1391 aux(a, b, 0)1392 else1393 std.primitiveEquals(a, b),139413951396 resolvePath(f, r)::1397 local arr = std.split(f, '/');1398 std.join('/', std.makeArray(std.length(arr) - 1, function(i) arr[i]) + [r]),13991400 prune(a)::1401 local isContent(b) =1402 if b == null then1403 false1404 else if std.isArray(b) then1405 std.length(b) > 01406 else if std.isObject(b) then1407 std.length(b) > 01408 else1409 true;1410 if std.isArray(a) then1411 [std.prune(x) for x in a if isContent($.prune(x))]1412 else if std.isObject(a) then {1413 [x]: $.prune(a[x])1414 for x in std.objectFields(a)1415 if isContent(std.prune(a[x]))1416 } else1417 a,14181419 findSubstr(pat, str)::1420 if !std.isString(pat) then1421 error 'findSubstr first parameter should be a string, got ' + std.type(pat)1422 else if !std.isString(str) then1423 error 'findSubstr second parameter should be a string, got ' + std.type(str)1424 else1425 local pat_len = std.length(pat);1426 local str_len = std.length(str);1427 if pat_len == 0 || str_len == 0 || pat_len > str_len then1428 []1429 else1430 std.filter(function(i) str[i:i + pat_len] == pat, std.range(0, str_len - pat_len)),14311432 find(value, arr)::1433 if !std.isArray(arr) then1434 error 'find second parameter should be an array, got ' + std.type(arr)1435 else1436 std.filter(function(i) arr[i] == value, std.range(0, std.length(arr) - 1)),1437}