git.delta.rocks / jrsonnet / refs/commits / e23fa8691cef

1{2  local std = self,3  local id = std.id,45  # Those functions aren't normally located in stdlib6  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  md5:: $intrinsic(md5),22  trace:: $intrinsic(trace),23  id:: $intrinsic(id),24  parseJson:: $intrinsic(parseJson),2526  log:: $intrinsic(log),27  pow:: $intrinsic(pow),28  sqrt:: $intrinsic(sqrt),2930  sin:: $intrinsic(sin),31  cos:: $intrinsic(cos),32  tan:: $intrinsic(tan),33  asin:: $intrinsic(asin),34  acos:: $intrinsic(acos),35  atan:: $intrinsic(atan),3637  exp:: $intrinsic(exp),38  mantissa:: $intrinsic(mantissa),39  exponent:: $intrinsic(exponent),4041  isString(v):: std.type(v) == 'string',42  isNumber(v):: std.type(v) == 'number',43  isBoolean(v):: std.type(v) == 'boolean',44  isObject(v):: std.type(v) == 'object',45  isArray(v):: std.type(v) == 'array',46  isFunction(v):: std.type(v) == 'function',4748  toString(a)::49    if std.type(a) == 'string' then a else '' + a,5051  substr:: $intrinsic(substr),5253  startsWith(a, b)::54    if std.length(a) < std.length(b) then55      false56    else57      std.substr(a, 0, std.length(b)) == b,5859  endsWith(a, b)::60    if std.length(a) < std.length(b) then61      false62    else63      std.substr(a, std.length(a) - std.length(b), std.length(b)) == b,6465  lstripChars(str, chars)::66    if std.length(str) > 0 && std.member(chars, str[0]) then67      std.lstripChars(str[1:], chars)68    else69      str,7071  rstripChars(str, chars)::72    local len = std.length(str);73    if len > 0 && std.member(chars, str[len - 1]) then74      std.rstripChars(str[:len - 1], chars)75    else76      str,7778  stripChars(str, chars)::79    std.lstripChars(std.rstripChars(str, chars), chars),8081  stringChars(str)::82    std.makeArray(std.length(str), function(i) str[i]),8384  local parse_nat(str, base) =85    assert base > 0 && base <= 16 : 'integer base %d invalid' % base;86    // These codepoints are in ascending order:87    local zero_code = std.codepoint('0');88    local upper_a_code = std.codepoint('A');89    local lower_a_code = std.codepoint('a');90    local addDigit(aggregate, char) =91      local code = std.codepoint(char);92      local digit = if code >= lower_a_code then93        code - lower_a_code + 1094      else if code >= upper_a_code then95        code - upper_a_code + 1096      else97        code - zero_code;98      assert digit >= 0 && digit < base : '%s is not a base %d integer' % [str, base];99      base * aggregate + digit;100    std.foldl(addDigit, std.stringChars(str), 0),101102  parseInt(str)::103    assert std.isString(str) : 'Expected string, got ' + std.type(str);104    assert std.length(str) > 0 && str != '-' : 'Not an integer: "%s"' % [str];105    if str[0] == '-' then106      -parse_nat(str[1:], 10)107    else108      parse_nat(str, 10),109110  parseOctal(str)::111    assert std.isString(str) : 'Expected string, got ' + std.type(str);112    assert std.length(str) > 0 : 'Not an octal number: ""';113    parse_nat(str, 8),114115  parseHex(str)::116    assert std.isString(str) : 'Expected string, got ' + std.type(str);117    assert std.length(str) > 0 : 'Not hexadecimal: ""';118    parse_nat(str, 16),119120  split(str, c)::121    assert std.isString(str) : 'std.split first parameter should be a string, got ' + std.type(str);122    assert std.isString(c) : 'std.split second parameter should be a string, got ' + std.type(c);123    assert std.length(c) == 1 : 'std.split second parameter should have length 1, got ' + std.length(c);124    std.splitLimit(str, c, -1),125126  splitLimit:: $intrinsic(splitLimit),127128  strReplace:: $intrinsic(strReplace),129130  asciiUpper(str)::131    local cp = std.codepoint;132    local up_letter(c) = if cp(c) >= 97 && cp(c) < 123 then133      std.char(cp(c) - 32)134    else135      c;136    std.join('', std.map(up_letter, std.stringChars(str))),137138  asciiLower(str)::139    local cp = std.codepoint;140    local down_letter(c) = if cp(c) >= 65 && cp(c) < 91 then141      std.char(cp(c) + 32)142    else143      c;144    std.join('', std.map(down_letter, std.stringChars(str))),145146  range:: $intrinsic(range),147148  repeat(what, count)::149    local joiner =150      if std.isString(what) then ''151      else if std.isArray(what) then []152      else error 'std.repeat first argument must be an array or a string';153    std.join(joiner, std.makeArray(count, function(i) what)),154155  slice:: $intrinsic(slice),156157  member(arr, x)::158    if std.isArray(arr) then159      std.count(arr, x) > 0160    else if std.isString(arr) then161      std.length(std.findSubstr(x, arr)) > 0162    else error 'std.member first argument must be an array or a string',163164  count(arr, x):: std.length(std.filter(function(v) v == x, arr)),165166  mod:: $intrinsic(mod),167168  map:: $intrinsic(map),169170  mapWithIndex(func, arr)::171    if !std.isFunction(func) then172      error ('std.mapWithIndex first param must be function, got ' + std.type(func))173    else if !std.isArray(arr) && !std.isString(arr) then174      error ('std.mapWithIndex second param must be array, got ' + std.type(arr))175    else176      std.makeArray(std.length(arr), function(i) func(i, arr[i])),177178  mapWithKey(func, obj)::179    if !std.isFunction(func) then180      error ('std.mapWithKey first param must be function, got ' + std.type(func))181    else if !std.isObject(obj) then182      error ('std.mapWithKey second param must be object, got ' + std.type(obj))183    else184      { [k]: func(k, obj[k]) for k in std.objectFields(obj) },185186  flatMap:: $intrinsic(flatMap),187188  join:: $intrinsic(join),189190  lines(arr)::191    std.join('\n', arr + ['']),192193  deepJoin(arr)::194    if std.isString(arr) then195      arr196    else if std.isArray(arr) then197      std.join('', [std.deepJoin(x) for x in arr])198    else199      error 'Expected string or array, got %s' % std.type(arr),200201202  format:: $intrinsic(format),203204  foldr:: $intrinsic(foldr),205206  foldl:: $intrinsic(foldl),207208  filterMap(filter_func, map_func, arr)::209    if !std.isFunction(filter_func) then210      error ('std.filterMap first param must be function, got ' + std.type(filter_func))211    else if !std.isFunction(map_func) then212      error ('std.filterMap second param must be function, got ' + std.type(map_func))213    else if !std.isArray(arr) then214      error ('std.filterMap third param must be array, got ' + std.type(arr))215    else216      std.map(map_func, std.filter(filter_func, arr)),217218  assertEqual(a, b)::219    if a == b then220      true221    else222      error 'Assertion failed. ' + a + ' != ' + b,223224  abs(n)::225    if !std.isNumber(n) then226      error 'std.abs expected number, got ' + std.type(n)227    else228      if n > 0 then n else -n,229230  sign(n)::231    if !std.isNumber(n) then232      error 'std.sign expected number, got ' + std.type(n)233    else234      if n > 0 then235        1236      else if n < 0 then237        -1238      else 0,239240  max(a, b)::241    if !std.isNumber(a) then242      error 'std.max first param expected number, got ' + std.type(a)243    else if !std.isNumber(b) then244      error 'std.max second param expected number, got ' + std.type(b)245    else246      if a > b then a else b,247248  min(a, b)::249    if !std.isNumber(a) then250      error 'std.min first param expected number, got ' + std.type(a)251    else if !std.isNumber(b) then252      error 'std.min second param expected number, got ' + std.type(b)253    else254      if a < b then a else b,255256  clamp(x, minVal, maxVal)::257    if x < minVal then minVal258    else if x > maxVal then maxVal259    else x,260261  flattenArrays(arrs)::262    std.foldl(function(a, b) a + b, arrs, []),263264  manifestIni(ini)::265    local body_lines(body) =266      std.join([], [267        local value_or_values = body[k];268        if std.isArray(value_or_values) then269          ['%s = %s' % [k, value] for value in value_or_values]270        else271          ['%s = %s' % [k, value_or_values]]272273        for k in std.objectFields(body)274      ]);275276    local section_lines(sname, sbody) = ['[%s]' % [sname]] + body_lines(sbody),277          main_body = if std.objectHas(ini, 'main') then body_lines(ini.main) else [],278          all_sections = [279      section_lines(k, ini.sections[k])280      for k in std.objectFields(ini.sections)281    ];282    std.join('\n', main_body + std.flattenArrays(all_sections) + ['']),283284  manifestToml(value):: std.manifestTomlEx(value, '  '),285286  manifestTomlEx(value, indent)::287    local288      escapeStringToml = std.escapeStringJson,289      escapeKeyToml(key) =290        local bare_allowed = std.set(std.stringChars('ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789_-'));291        if std.setUnion(std.set(std.stringChars(key)), bare_allowed) == bare_allowed then key else escapeStringToml(key),292      isTableArray(v) = std.isArray(v) && std.length(v) > 0 && std.foldl(function(a, b) a && std.isObject(b), v, true),293      isSection(v) = std.isObject(v) || isTableArray(v),294      renderValue(v, indexedPath, inline, cindent) =295        if v == true then296          'true'297        else if v == false then298          'false'299        else if v == null then300          error 'Tried to manifest "null" at ' + indexedPath301        else if std.isNumber(v) then302          '' + v303        else if std.isString(v) then304          escapeStringToml(v)305        else if std.isFunction(v) then306          error 'Tried to manifest function at ' + indexedPath307        else if std.isArray(v) then308          if std.length(v) == 0 then309            '[]'310          else311            local range = std.range(0, std.length(v) - 1);312            local new_indent = if inline then '' else cindent + indent;313            local separator = if inline then ' ' else '\n';314            local lines = ['[' + separator]315                          + std.join([',' + separator],316                                     [317                                       [new_indent + renderValue(v[i], indexedPath + [i], true, '')]318                                       for i in range319                                     ])320                          + [separator + (if inline then '' else cindent) + ']'];321            std.join('', lines)322        else if std.isObject(v) then323          local lines = ['{ ']324                        + std.join([', '],325                                   [326                                     [escapeKeyToml(k) + ' = ' + renderValue(v[k], indexedPath + [k], true, '')]327                                     for k in std.objectFields(v)328                                   ])329                        + [' }'];330          std.join('', lines),331      renderTableInternal(v, path, indexedPath, cindent) =332        local kvp = std.flattenArrays([333          [cindent + escapeKeyToml(k) + ' = ' + renderValue(v[k], indexedPath + [k], false, cindent)]334          for k in std.objectFields(v)335          if !isSection(v[k])336        ]);337        local sections = [std.join('\n', kvp)] + [338          (339            if std.isObject(v[k]) then340              renderTable(v[k], path + [k], indexedPath + [k], cindent)341            else342              renderTableArray(v[k], path + [k], indexedPath + [k], cindent)343          )344          for k in std.objectFields(v)345          if isSection(v[k])346        ];347        std.join('\n\n', sections),348      renderTable(v, path, indexedPath, cindent) =349        cindent + '[' + std.join('.', std.map(escapeKeyToml, path)) + ']'350        + (if v == {} then '' else '\n')351        + renderTableInternal(v, path, indexedPath, cindent + indent),352      renderTableArray(v, path, indexedPath, cindent) =353        local range = std.range(0, std.length(v) - 1);354        local sections = [355          (cindent + '[[' + std.join('.', std.map(escapeKeyToml, path)) + ']]'356           + (if v[i] == {} then '' else '\n')357           + renderTableInternal(v[i], path, indexedPath + [i], cindent + indent))358          for i in range359        ];360        std.join('\n\n', sections);361    if std.isObject(value) then362      renderTableInternal(value, [], [], '')363    else364      error 'TOML body must be an object. Got ' + std.type(value),365366  escapeStringJson:: $intrinsic(escapeStringJson),367368  escapeStringPython(str)::369    std.escapeStringJson(str),370371  escapeStringBash(str_)::372    local str = std.toString(str_);373    local trans(ch) =374      if ch == "'" then375        "'\"'\"'"376      else377        ch;378    "'%s'" % std.join('', [trans(ch) for ch in std.stringChars(str)]),379380  escapeStringDollars(str_)::381    local str = std.toString(str_);382    local trans(ch) =383      if ch == '$' then384        '$$'385      else386        ch;387    std.foldl(function(a, b) a + trans(b), std.stringChars(str), ''),388389  manifestJson(value):: std.manifestJsonEx(value, '    '),390391  manifestJsonEx:: $intrinsic(manifestJsonEx),392393  manifestYamlDoc(value, indent_array_in_object=false)::394    local aux(v, path, cindent) =395      if v == true then396        'true'397      else if v == false then398        'false'399      else if v == null then400        'null'401      else if std.isNumber(v) then402        '' + v403      else if std.isString(v) then404        local len = std.length(v);405        if len == 0 then406          '""'407        else if v[len - 1] == '\n' then408          local split = std.split(v, '\n');409          std.join('\n' + cindent + '  ', ['|'] + split[0:std.length(split) - 1])410        else411          std.escapeStringJson(v)412      else if std.isFunction(v) then413        error 'Tried to manifest function at ' + path414      else if std.isArray(v) then415        if std.length(v) == 0 then416          '[]'417        else418          local params(value) =419            if std.isArray(value) && std.length(value) > 0 then {420              // While we could avoid the new line, it yields YAML that is421              // hard to read, e.g.:422              // - - - 1423              //     - 2424              //   - - 3425              //     - 4426              new_indent: cindent + '  ',427              space: '\n' + self.new_indent,428            } else if std.isObject(value) && std.length(value) > 0 then {429              new_indent: cindent + '  ',430              // In this case we can start on the same line as the - because the indentation431              // matches up then.  The converse is not true, because fields are not always432              // 1 character long.433              space: ' ',434            } else {435              // In this case, new_indent is only used in the case of multi-line strings.436              new_indent: cindent,437              space: ' ',438            };439          local range = std.range(0, std.length(v) - 1);440          local parts = [441            '-' + param.space + aux(v[i], path + [i], param.new_indent)442            for i in range443            for param in [params(v[i])]444          ];445          std.join('\n' + cindent, parts)446      else if std.isObject(v) then447        if std.length(v) == 0 then448          '{}'449        else450          local params(value) =451            if std.isArray(value) && std.length(value) > 0 then {452              // Not indenting allows e.g.453              // ports:454              // - 80455              // instead of456              // ports:457              //   - 80458              new_indent: if indent_array_in_object then cindent + '  ' else cindent,459              space: '\n' + self.new_indent,460            } else if std.isObject(value) && std.length(value) > 0 then {461              new_indent: cindent + '  ',462              space: '\n' + self.new_indent,463            } else {464              // In this case, new_indent is only used in the case of multi-line strings.465              new_indent: cindent,466              space: ' ',467            };468          local lines = [469            std.escapeStringJson(k) + ':' + param.space + aux(v[k], path + [k], param.new_indent)470            for k in std.objectFields(v)471            for param in [params(v[k])]472          ];473          std.join('\n' + cindent, lines);474    aux(value, [], ''),475476  manifestYamlStream(value, indent_array_in_object=false, c_document_end=true)::477    if !std.isArray(value) then478      error 'manifestYamlStream only takes arrays, got ' + std.type(value)479    else480      '---\n' + std.join(481        '\n---\n', [std.manifestYamlDoc(e, indent_array_in_object) for e in value]482      ) + if c_document_end then '\n...\n' else '\n',483484485  manifestPython(v)::486    if std.isObject(v) then487      local fields = [488        '%s: %s' % [std.escapeStringPython(k), std.manifestPython(v[k])]489        for k in std.objectFields(v)490      ];491      '{%s}' % [std.join(', ', fields)]492    else if std.isArray(v) then493      '[%s]' % [std.join(', ', [std.manifestPython(v2) for v2 in v])]494    else if std.isString(v) then495      '%s' % [std.escapeStringPython(v)]496    else if std.isFunction(v) then497      error 'cannot manifest function'498    else if std.isNumber(v) then499      std.toString(v)500    else if v == true then501      'True'502    else if v == false then503      'False'504    else if v == null then505      'None',506507  manifestPythonVars(conf)::508    local vars = ['%s = %s' % [k, std.manifestPython(conf[k])] for k in std.objectFields(conf)];509    std.join('\n', vars + ['']),510511  manifestXmlJsonml(value)::512    if !std.isArray(value) then513      error 'Expected a JSONML value (an array), got %s' % std.type(value)514    else515      local aux(v) =516        if std.isString(v) then517          v518        else519          local tag = v[0];520          local has_attrs = std.length(v) > 1 && std.isObject(v[1]);521          local attrs = if has_attrs then v[1] else {};522          local children = if has_attrs then v[2:] else v[1:];523          local attrs_str =524            std.join('', [' %s="%s"' % [k, attrs[k]] for k in std.objectFields(attrs)]);525          std.deepJoin(['<', tag, attrs_str, '>', [aux(x) for x in children], '</', tag, '>']);526527      aux(value),528529  local base64_table = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/',530  local base64_inv = { [base64_table[i]]: i for i in std.range(0, 63) },531532  base64:: $intrinsic(base64),533534  base64DecodeBytes:: $intrinsic(base64DecodeBytes),535536  base64Decode:: $intrinsic(base64Decode),537538  reverse:: $intrinsic(reverse),539540  sortImpl:: $intrinsic(sortImpl),541542  sort(arr, keyF=id)::543    std.sortImpl(arr, keyF),544545  uniq(arr, keyF=id)::546    local f(a, b) =547      if std.length(a) == 0 then548        [b]549      else if keyF(a[std.length(a) - 1]) == keyF(b) then550        a551      else552        a + [b];553    std.foldl(f, arr, []),554555  set(arr, keyF=id)::556    std.uniq(std.sort(arr, keyF), keyF),557558  setMember(x, arr, keyF=id)::559    // TODO(dcunnin): Binary chop for O(log n) complexity560    std.length(std.setInter([x], arr, keyF)) > 0,561562  setUnion(a, b, keyF=id)::563    // NOTE: order matters, values in `a` win564    local aux(a, b, i, j, acc) =565      if i >= std.length(a) then566        acc + b[j:]567      else if j >= std.length(b) then568        acc + a[i:]569      else570        local ak = keyF(a[i]);571        local bk = keyF(b[j]);572        if ak == bk then573          aux(a, b, i + 1, j + 1, acc + [a[i]]) tailstrict574        else if ak < bk then575          aux(a, b, i + 1, j, acc + [a[i]]) tailstrict576        else577          aux(a, b, i, j + 1, acc + [b[j]]) tailstrict;578    aux(a, b, 0, 0, []),579580  setInter(a, b, keyF=id)::581    local aux(a, b, i, j, acc) =582      if i >= std.length(a) || j >= std.length(b) then583        acc584      else585        if keyF(a[i]) == keyF(b[j]) then586          aux(a, b, i + 1, j + 1, acc + [a[i]]) tailstrict587        else if keyF(a[i]) < keyF(b[j]) then588          aux(a, b, i + 1, j, acc) tailstrict589        else590          aux(a, b, i, j + 1, acc) tailstrict;591    aux(a, b, 0, 0, []) tailstrict,592593  setDiff(a, b, keyF=id)::594    local aux(a, b, i, j, acc) =595      if i >= std.length(a) then596        acc597      else if j >= std.length(b) then598        acc + a[i:]599      else600        if keyF(a[i]) == keyF(b[j]) then601          aux(a, b, i + 1, j + 1, acc) tailstrict602        else if keyF(a[i]) < keyF(b[j]) then603          aux(a, b, i + 1, j, acc + [a[i]]) tailstrict604        else605          aux(a, b, i, j + 1, acc) tailstrict;606    aux(a, b, 0, 0, []) tailstrict,607608  mergePatch(target, patch)::609    if std.isObject(patch) then610      local target_object =611        if std.isObject(target) then target else {};612613      local target_fields =614        if std.isObject(target_object) then std.objectFields(target_object) else [];615616      local null_fields = [k for k in std.objectFields(patch) if patch[k] == null];617      local both_fields = std.setUnion(target_fields, std.objectFields(patch));618619      {620        [k]:621          if !std.objectHas(patch, k) then622            target_object[k]623          else if !std.objectHas(target_object, k) then624            std.mergePatch(null, patch[k]) tailstrict625          else626            std.mergePatch(target_object[k], patch[k]) tailstrict627        for k in std.setDiff(both_fields, null_fields)628      }629    else630      patch,631632  objectFields(o)::633    std.objectFieldsEx(o, false),634635  objectFieldsAll(o)::636    std.objectFieldsEx(o, true),637638  objectHas(o, f)::639    std.objectHasEx(o, f, false),640641  objectHasAll(o, f)::642    std.objectHasEx(o, f, true),643644  objectValues(o)::645    [o[k] for k in std.objectFields(o)],646647  objectValuesAll(o)::648    [o[k] for k in std.objectFieldsAll(o)],649650  equals:: $intrinsic(equals),651652  resolvePath(f, r)::653    local arr = std.split(f, '/');654    std.join('/', std.makeArray(std.length(arr) - 1, function(i) arr[i]) + [r]),655656  prune(a)::657    local isContent(b) =658      if b == null then659        false660      else if std.isArray(b) then661        std.length(b) > 0662      else if std.isObject(b) then663        std.length(b) > 0664      else665        true;666    if std.isArray(a) then667      [std.prune(x) for x in a if isContent($.prune(x))]668    else if std.isObject(a) then {669      [x]: $.prune(a[x])670      for x in std.objectFields(a)671      if isContent(std.prune(a[x]))672    } else673      a,674675  findSubstr(pat, str)::676    if !std.isString(pat) then677      error 'findSubstr first parameter should be a string, got ' + std.type(pat)678    else if !std.isString(str) then679      error 'findSubstr second parameter should be a string, got ' + std.type(str)680    else681      local pat_len = std.length(pat);682      local str_len = std.length(str);683      if pat_len == 0 || str_len == 0 || pat_len > str_len then684        []685      else686        std.filter(function(i) str[i:i + pat_len] == pat, std.range(0, str_len - pat_len)),687688  find(value, arr)::689    if !std.isArray(arr) then690      error 'find second parameter should be an array, got ' + std.type(arr)691    else692      std.filter(function(i) arr[i] == value, std.range(0, std.length(arr) - 1)),693}