1{2 local std = self,3 local id = std.id,45 6 length:: $intrinsic(length),7 type:: $intrinsic(type),8 makeArray:: $intrinsic(makeArray),9 codepoint:: $intrinsic(codepoint),10 objectFieldsEx:: $intrinsic(objectFieldsEx),11 objectHasEx:: $intrinsic(objectHasEx),12 primitiveEquals:: $intrinsic(primitiveEquals),13 modulo:: $intrinsic(modulo),14 floor:: $intrinsic(floor),15 ceil:: $intrinsic(ceil),16 extVar:: $intrinsic(extVar),17 native:: $intrinsic(native),18 filter:: $intrinsic(filter),19 char:: $intrinsic(char),20 encodeUTF8:: $intrinsic(encodeUTF8),21 decodeUTF8:: $intrinsic(decodeUTF8),22 md5:: $intrinsic(md5),23 trace:: $intrinsic(trace),24 id:: $intrinsic(id),25 parseJson:: $intrinsic(parseJson),2627 log:: $intrinsic(log),28 pow:: $intrinsic(pow),29 sqrt:: $intrinsic(sqrt),3031 sin:: $intrinsic(sin),32 cos:: $intrinsic(cos),33 tan:: $intrinsic(tan),34 asin:: $intrinsic(asin),35 acos:: $intrinsic(acos),36 atan:: $intrinsic(atan),3738 exp:: $intrinsic(exp),39 mantissa:: $intrinsic(mantissa),40 exponent:: $intrinsic(exponent),4142 isString(v):: std.type(v) == 'string',43 isNumber(v):: std.type(v) == 'number',44 isBoolean(v):: std.type(v) == 'boolean',45 isObject(v):: std.type(v) == 'object',46 isArray(v):: std.type(v) == 'array',47 isFunction(v):: std.type(v) == 'function',4849 toString(a)::50 if std.type(a) == 'string' then a else '' + a,5152 substr:: $intrinsic(substr),5354 startsWith(a, b)::55 if std.length(a) < std.length(b) then56 false57 else58 std.substr(a, 0, std.length(b)) == b,5960 endsWith(a, b)::61 if std.length(a) < std.length(b) then62 false63 else64 std.substr(a, std.length(a) - std.length(b), std.length(b)) == b,6566 lstripChars(str, chars)::67 if std.length(str) > 0 && std.member(chars, str[0]) then68 std.lstripChars(str[1:], chars)69 else70 str,7172 rstripChars(str, chars)::73 local len = std.length(str);74 if len > 0 && std.member(chars, str[len - 1]) then75 std.rstripChars(str[:len - 1], chars)76 else77 str,7879 stripChars(str, chars)::80 std.lstripChars(std.rstripChars(str, chars), chars),8182 stringChars(str)::83 std.makeArray(std.length(str), function(i) str[i]),8485 local parse_nat(str, base) =86 assert base > 0 && base <= 16 : 'integer base %d invalid' % base;87 88 local zero_code = std.codepoint('0');89 local upper_a_code = std.codepoint('A');90 local lower_a_code = std.codepoint('a');91 local addDigit(aggregate, char) =92 local code = std.codepoint(char);93 local digit = if code >= lower_a_code then94 code - lower_a_code + 1095 else if code >= upper_a_code then96 code - upper_a_code + 1097 else98 code - zero_code;99 assert digit >= 0 && digit < base : '%s is not a base %d integer' % [str, base];100 base * aggregate + digit;101 std.foldl(addDigit, std.stringChars(str), 0),102103 parseInt(str)::104 assert std.isString(str) : 'Expected string, got ' + std.type(str);105 assert std.length(str) > 0 && str != '-' : 'Not an integer: "%s"' % [str];106 if str[0] == '-' then107 -parse_nat(str[1:], 10)108 else109 parse_nat(str, 10),110111 parseOctal(str)::112 assert std.isString(str) : 'Expected string, got ' + std.type(str);113 assert std.length(str) > 0 : 'Not an octal number: ""';114 parse_nat(str, 8),115116 parseHex(str)::117 assert std.isString(str) : 'Expected string, got ' + std.type(str);118 assert std.length(str) > 0 : 'Not hexadecimal: ""';119 parse_nat(str, 16),120121 split(str, c)::122 assert std.isString(str) : 'std.split first parameter should be a string, got ' + std.type(str);123 assert std.isString(c) : 'std.split second parameter should be a string, got ' + std.type(c);124 assert std.length(c) == 1 : 'std.split second parameter should have length 1, got ' + std.length(c);125 std.splitLimit(str, c, -1),126127 splitLimit:: $intrinsic(splitLimit),128129 strReplace:: $intrinsic(strReplace),130131 asciiUpper(str)::132 local cp = std.codepoint;133 local up_letter(c) = if cp(c) >= 97 && cp(c) < 123 then134 std.char(cp(c) - 32)135 else136 c;137 std.join('', std.map(up_letter, std.stringChars(str))),138139 asciiLower(str)::140 local cp = std.codepoint;141 local down_letter(c) = if cp(c) >= 65 && cp(c) < 91 then142 std.char(cp(c) + 32)143 else144 c;145 std.join('', std.map(down_letter, std.stringChars(str))),146147 range:: $intrinsic(range),148149 repeat(what, count)::150 local joiner =151 if std.isString(what) then ''152 else if std.isArray(what) then []153 else error 'std.repeat first argument must be an array or a string';154 std.join(joiner, std.makeArray(count, function(i) what)),155156 slice:: $intrinsic(slice),157158 member(arr, x)::159 if std.isArray(arr) then160 std.count(arr, x) > 0161 else if std.isString(arr) then162 std.length(std.findSubstr(x, arr)) > 0163 else error 'std.member first argument must be an array or a string',164165 count(arr, x):: std.length(std.filter(function(v) v == x, arr)),166167 mod:: $intrinsic(mod),168169 map:: $intrinsic(map),170171 mapWithIndex(func, arr)::172 if !std.isFunction(func) then173 error ('std.mapWithIndex first param must be function, got ' + std.type(func))174 else if !std.isArray(arr) && !std.isString(arr) then175 error ('std.mapWithIndex second param must be array, got ' + std.type(arr))176 else177 std.makeArray(std.length(arr), function(i) func(i, arr[i])),178179 mapWithKey(func, obj)::180 if !std.isFunction(func) then181 error ('std.mapWithKey first param must be function, got ' + std.type(func))182 else if !std.isObject(obj) then183 error ('std.mapWithKey second param must be object, got ' + std.type(obj))184 else185 { [k]: func(k, obj[k]) for k in std.objectFields(obj) },186187 flatMap:: $intrinsic(flatMap),188189 join:: $intrinsic(join),190191 lines(arr)::192 std.join('\n', arr + ['']),193194 deepJoin(arr)::195 if std.isString(arr) then196 arr197 else if std.isArray(arr) then198 std.join('', [std.deepJoin(x) for x in arr])199 else200 error 'Expected string or array, got %s' % std.type(arr),201202203 format:: $intrinsic(format),204205 foldr:: $intrinsic(foldr),206207 foldl:: $intrinsic(foldl),208209 filterMap(filter_func, map_func, arr)::210 if !std.isFunction(filter_func) then211 error ('std.filterMap first param must be function, got ' + std.type(filter_func))212 else if !std.isFunction(map_func) then213 error ('std.filterMap second param must be function, got ' + std.type(map_func))214 else if !std.isArray(arr) then215 error ('std.filterMap third param must be array, got ' + std.type(arr))216 else217 std.map(map_func, std.filter(filter_func, arr)),218219 assertEqual(a, b)::220 if a == b then221 true222 else223 error 'Assertion failed. ' + a + ' != ' + b,224225 abs(n)::226 if !std.isNumber(n) then227 error 'std.abs expected number, got ' + std.type(n)228 else229 if n > 0 then n else -n,230231 sign(n)::232 if !std.isNumber(n) then233 error 'std.sign expected number, got ' + std.type(n)234 else235 if n > 0 then236 1237 else if n < 0 then238 -1239 else 0,240241 max(a, b)::242 if !std.isNumber(a) then243 error 'std.max first param expected number, got ' + std.type(a)244 else if !std.isNumber(b) then245 error 'std.max second param expected number, got ' + std.type(b)246 else247 if a > b then a else b,248249 min(a, b)::250 if !std.isNumber(a) then251 error 'std.min first param expected number, got ' + std.type(a)252 else if !std.isNumber(b) then253 error 'std.min second param expected number, got ' + std.type(b)254 else255 if a < b then a else b,256257 clamp(x, minVal, maxVal)::258 if x < minVal then minVal259 else if x > maxVal then maxVal260 else x,261262 flattenArrays(arrs)::263 std.foldl(function(a, b) a + b, arrs, []),264265 manifestIni(ini)::266 local body_lines(body) =267 std.join([], [268 local value_or_values = body[k];269 if std.isArray(value_or_values) then270 ['%s = %s' % [k, value] for value in value_or_values]271 else272 ['%s = %s' % [k, value_or_values]]273274 for k in std.objectFields(body)275 ]);276277 local section_lines(sname, sbody) = ['[%s]' % [sname]] + body_lines(sbody),278 main_body = if std.objectHas(ini, 'main') then body_lines(ini.main) else [],279 all_sections = [280 section_lines(k, ini.sections[k])281 for k in std.objectFields(ini.sections)282 ];283 std.join('\n', main_body + std.flattenArrays(all_sections) + ['']),284285 manifestToml(value):: std.manifestTomlEx(value, ' '),286287 manifestTomlEx(value, indent)::288 local289 escapeStringToml = std.escapeStringJson,290 escapeKeyToml(key) =291 local bare_allowed = std.set(std.stringChars('ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789_-'));292 if std.setUnion(std.set(std.stringChars(key)), bare_allowed) == bare_allowed then key else escapeStringToml(key),293 isTableArray(v) = std.isArray(v) && std.length(v) > 0 && std.foldl(function(a, b) a && std.isObject(b), v, true),294 isSection(v) = std.isObject(v) || isTableArray(v),295 renderValue(v, indexedPath, inline, cindent) =296 if v == true then297 'true'298 else if v == false then299 'false'300 else if v == null then301 error 'Tried to manifest "null" at ' + indexedPath302 else if std.isNumber(v) then303 '' + v304 else if std.isString(v) then305 escapeStringToml(v)306 else if std.isFunction(v) then307 error 'Tried to manifest function at ' + indexedPath308 else if std.isArray(v) then309 if std.length(v) == 0 then310 '[]'311 else312 local range = std.range(0, std.length(v) - 1);313 local new_indent = if inline then '' else cindent + indent;314 local separator = if inline then ' ' else '\n';315 local lines = ['[' + separator]316 + std.join([',' + separator],317 [318 [new_indent + renderValue(v[i], indexedPath + [i], true, '')]319 for i in range320 ])321 + [separator + (if inline then '' else cindent) + ']'];322 std.join('', lines)323 else if std.isObject(v) then324 local lines = ['{ ']325 + std.join([', '],326 [327 [escapeKeyToml(k) + ' = ' + renderValue(v[k], indexedPath + [k], true, '')]328 for k in std.objectFields(v)329 ])330 + [' }'];331 std.join('', lines),332 renderTableInternal(v, path, indexedPath, cindent) =333 local kvp = std.flattenArrays([334 [cindent + escapeKeyToml(k) + ' = ' + renderValue(v[k], indexedPath + [k], false, cindent)]335 for k in std.objectFields(v)336 if !isSection(v[k])337 ]);338 local sections = [std.join('\n', kvp)] + [339 (340 if std.isObject(v[k]) then341 renderTable(v[k], path + [k], indexedPath + [k], cindent)342 else343 renderTableArray(v[k], path + [k], indexedPath + [k], cindent)344 )345 for k in std.objectFields(v)346 if isSection(v[k])347 ];348 std.join('\n\n', sections),349 renderTable(v, path, indexedPath, cindent) =350 cindent + '[' + std.join('.', std.map(escapeKeyToml, path)) + ']'351 + (if v == {} then '' else '\n')352 + renderTableInternal(v, path, indexedPath, cindent + indent),353 renderTableArray(v, path, indexedPath, cindent) =354 local range = std.range(0, std.length(v) - 1);355 local sections = [356 (cindent + '[[' + std.join('.', std.map(escapeKeyToml, path)) + ']]'357 + (if v[i] == {} then '' else '\n')358 + renderTableInternal(v[i], path, indexedPath + [i], cindent + indent))359 for i in range360 ];361 std.join('\n\n', sections);362 if std.isObject(value) then363 renderTableInternal(value, [], [], '')364 else365 error 'TOML body must be an object. Got ' + std.type(value),366367 escapeStringJson:: $intrinsic(escapeStringJson),368369 escapeStringPython(str)::370 std.escapeStringJson(str),371372 escapeStringBash(str_)::373 local str = std.toString(str_);374 local trans(ch) =375 if ch == "'" then376 "'\"'\"'"377 else378 ch;379 "'%s'" % std.join('', [trans(ch) for ch in std.stringChars(str)]),380381 escapeStringDollars(str_)::382 local str = std.toString(str_);383 local trans(ch) =384 if ch == '$' then385 '$$'386 else387 ch;388 std.foldl(function(a, b) a + trans(b), std.stringChars(str), ''),389390 manifestJson(value):: std.manifestJsonEx(value, ' '),391392 manifestJsonEx:: $intrinsic(manifestJsonEx),393394 manifestYamlDoc(value, indent_array_in_object=false)::395 local aux(v, path, cindent) =396 if v == true then397 'true'398 else if v == false then399 'false'400 else if v == null then401 'null'402 else if std.isNumber(v) then403 '' + v404 else if std.isString(v) then405 local len = std.length(v);406 if len == 0 then407 '""'408 else if v[len - 1] == '\n' then409 local split = std.split(v, '\n');410 std.join('\n' + cindent + ' ', ['|'] + split[0:std.length(split) - 1])411 else412 std.escapeStringJson(v)413 else if std.isFunction(v) then414 error 'Tried to manifest function at ' + path415 else if std.isArray(v) then416 if std.length(v) == 0 then417 '[]'418 else419 local params(value) =420 if std.isArray(value) && std.length(value) > 0 then {421 422 423 424 425 426 427 new_indent: cindent + ' ',428 space: '\n' + self.new_indent,429 } else if std.isObject(value) && std.length(value) > 0 then {430 new_indent: cindent + ' ',431 432 433 434 space: ' ',435 } else {436 437 new_indent: cindent,438 space: ' ',439 };440 local range = std.range(0, std.length(v) - 1);441 local parts = [442 '-' + param.space + aux(v[i], path + [i], param.new_indent)443 for i in range444 for param in [params(v[i])]445 ];446 std.join('\n' + cindent, parts)447 else if std.isObject(v) then448 if std.length(v) == 0 then449 '{}'450 else451 local params(value) =452 if std.isArray(value) && std.length(value) > 0 then {453 454 455 456 457 458 459 new_indent: if indent_array_in_object then cindent + ' ' else cindent,460 space: '\n' + self.new_indent,461 } else if std.isObject(value) && std.length(value) > 0 then {462 new_indent: cindent + ' ',463 space: '\n' + self.new_indent,464 } else {465 466 new_indent: cindent,467 space: ' ',468 };469 local lines = [470 std.escapeStringJson(k) + ':' + param.space + aux(v[k], path + [k], param.new_indent)471 for k in std.objectFields(v)472 for param in [params(v[k])]473 ];474 std.join('\n' + cindent, lines);475 aux(value, [], ''),476477 manifestYamlStream(value, indent_array_in_object=false, c_document_end=true)::478 if !std.isArray(value) then479 error 'manifestYamlStream only takes arrays, got ' + std.type(value)480 else481 '---\n' + std.join(482 '\n---\n', [std.manifestYamlDoc(e, indent_array_in_object) for e in value]483 ) + if c_document_end then '\n...\n' else '\n',484485486 manifestPython(v)::487 if std.isObject(v) then488 local fields = [489 '%s: %s' % [std.escapeStringPython(k), std.manifestPython(v[k])]490 for k in std.objectFields(v)491 ];492 '{%s}' % [std.join(', ', fields)]493 else if std.isArray(v) then494 '[%s]' % [std.join(', ', [std.manifestPython(v2) for v2 in v])]495 else if std.isString(v) then496 '%s' % [std.escapeStringPython(v)]497 else if std.isFunction(v) then498 error 'cannot manifest function'499 else if std.isNumber(v) then500 std.toString(v)501 else if v == true then502 'True'503 else if v == false then504 'False'505 else if v == null then506 'None',507508 manifestPythonVars(conf)::509 local vars = ['%s = %s' % [k, std.manifestPython(conf[k])] for k in std.objectFields(conf)];510 std.join('\n', vars + ['']),511512 manifestXmlJsonml(value)::513 if !std.isArray(value) then514 error 'Expected a JSONML value (an array), got %s' % std.type(value)515 else516 local aux(v) =517 if std.isString(v) then518 v519 else520 local tag = v[0];521 local has_attrs = std.length(v) > 1 && std.isObject(v[1]);522 local attrs = if has_attrs then v[1] else {};523 local children = if has_attrs then v[2:] else v[1:];524 local attrs_str =525 std.join('', [' %s="%s"' % [k, attrs[k]] for k in std.objectFields(attrs)]);526 std.deepJoin(['<', tag, attrs_str, '>', [aux(x) for x in children], '</', tag, '>']);527528 aux(value),529530 local base64_table = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/',531 local base64_inv = { [base64_table[i]]: i for i in std.range(0, 63) },532533 base64:: $intrinsic(base64),534535 base64DecodeBytes:: $intrinsic(base64DecodeBytes),536537 base64Decode:: $intrinsic(base64Decode),538539 reverse:: $intrinsic(reverse),540541 sortImpl:: $intrinsic(sortImpl),542543 sort(arr, keyF=id)::544 std.sortImpl(arr, keyF),545546 uniq(arr, keyF=id)::547 local f(a, b) =548 if std.length(a) == 0 then549 [b]550 else if keyF(a[std.length(a) - 1]) == keyF(b) then551 a552 else553 a + [b];554 std.foldl(f, arr, []),555556 set(arr, keyF=id)::557 std.uniq(std.sort(arr, keyF), keyF),558559 setMember(x, arr, keyF=id)::560 561 std.length(std.setInter([x], arr, keyF)) > 0,562563 setUnion(a, b, keyF=id)::564 565 local aux(a, b, i, j, acc) =566 if i >= std.length(a) then567 acc + b[j:]568 else if j >= std.length(b) then569 acc + a[i:]570 else571 local ak = keyF(a[i]);572 local bk = keyF(b[j]);573 if ak == bk then574 aux(a, b, i + 1, j + 1, acc + [a[i]]) tailstrict575 else if ak < bk then576 aux(a, b, i + 1, j, acc + [a[i]]) tailstrict577 else578 aux(a, b, i, j + 1, acc + [b[j]]) tailstrict;579 aux(a, b, 0, 0, []),580581 setInter(a, b, keyF=id)::582 local aux(a, b, i, j, acc) =583 if i >= std.length(a) || j >= std.length(b) then584 acc585 else586 if keyF(a[i]) == keyF(b[j]) then587 aux(a, b, i + 1, j + 1, acc + [a[i]]) tailstrict588 else if keyF(a[i]) < keyF(b[j]) then589 aux(a, b, i + 1, j, acc) tailstrict590 else591 aux(a, b, i, j + 1, acc) tailstrict;592 aux(a, b, 0, 0, []) tailstrict,593594 setDiff(a, b, keyF=id)::595 local aux(a, b, i, j, acc) =596 if i >= std.length(a) then597 acc598 else if j >= std.length(b) then599 acc + a[i:]600 else601 if keyF(a[i]) == keyF(b[j]) then602 aux(a, b, i + 1, j + 1, acc) tailstrict603 else if keyF(a[i]) < keyF(b[j]) then604 aux(a, b, i + 1, j, acc + [a[i]]) tailstrict605 else606 aux(a, b, i, j + 1, acc) tailstrict;607 aux(a, b, 0, 0, []) tailstrict,608609 mergePatch(target, patch)::610 if std.isObject(patch) then611 local target_object =612 if std.isObject(target) then target else {};613614 local target_fields =615 if std.isObject(target_object) then std.objectFields(target_object) else [];616617 local null_fields = [k for k in std.objectFields(patch) if patch[k] == null];618 local both_fields = std.setUnion(target_fields, std.objectFields(patch));619620 {621 [k]:622 if !std.objectHas(patch, k) then623 target_object[k]624 else if !std.objectHas(target_object, k) then625 std.mergePatch(null, patch[k]) tailstrict626 else627 std.mergePatch(target_object[k], patch[k]) tailstrict628 for k in std.setDiff(both_fields, null_fields)629 }630 else631 patch,632633 objectFields(o)::634 std.objectFieldsEx(o, false),635636 objectFieldsAll(o)::637 std.objectFieldsEx(o, true),638639 objectHas(o, f)::640 std.objectHasEx(o, f, false),641642 objectHasAll(o, f)::643 std.objectHasEx(o, f, true),644645 objectValues(o)::646 [o[k] for k in std.objectFields(o)],647648 objectValuesAll(o)::649 [o[k] for k in std.objectFieldsAll(o)],650651 equals:: $intrinsic(equals),652653 resolvePath(f, r)::654 local arr = std.split(f, '/');655 std.join('/', std.makeArray(std.length(arr) - 1, function(i) arr[i]) + [r]),656657 prune(a)::658 local isContent(b) =659 if b == null then660 false661 else if std.isArray(b) then662 std.length(b) > 0663 else if std.isObject(b) then664 std.length(b) > 0665 else666 true;667 if std.isArray(a) then668 [std.prune(x) for x in a if isContent($.prune(x))]669 else if std.isObject(a) then {670 [x]: $.prune(a[x])671 for x in std.objectFields(a)672 if isContent(std.prune(a[x]))673 } else674 a,675676 findSubstr(pat, str)::677 if !std.isString(pat) then678 error 'findSubstr first parameter should be a string, got ' + std.type(pat)679 else if !std.isString(str) then680 error 'findSubstr second parameter should be a string, got ' + std.type(str)681 else682 local pat_len = std.length(pat);683 local str_len = std.length(str);684 if pat_len == 0 || str_len == 0 || pat_len > str_len then685 []686 else687 std.filter(function(i) str[i:i + pat_len] == pat, std.range(0, str_len - pat_len)),688689 find(value, arr)::690 if !std.isArray(arr) then691 error 'find second parameter should be an array, got ' + std.type(arr)692 else693 std.filter(function(i) arr[i] == value, std.range(0, std.length(arr) - 1)),694}