git.delta.rocks / jrsonnet / refs/commits / 2c370b494610

1{2  local std = self,3  local id = std.id,45  # Magic legacy field6  thisFile:: $intrinsicThisFile,78  # Those functions aren't normally located in stdlib9  length:: $intrinsic(length),10  type:: $intrinsic(type),11  makeArray:: $intrinsic(makeArray),12  codepoint:: $intrinsic(codepoint),13  objectFieldsEx:: $intrinsic(objectFieldsEx),14  objectHasEx:: $intrinsic(objectHasEx),15  primitiveEquals:: $intrinsic(primitiveEquals),16  modulo:: $intrinsic(modulo),17  floor:: $intrinsic(floor),18  ceil:: $intrinsic(ceil),19  extVar:: $intrinsic(extVar),20  native:: $intrinsic(native),21  filter:: $intrinsic(filter),22  char:: $intrinsic(char),23  encodeUTF8:: $intrinsic(encodeUTF8),24  decodeUTF8:: $intrinsic(decodeUTF8),25  md5:: $intrinsic(md5),26  trace:: $intrinsic(trace),27  id:: $intrinsic(id),28  parseJson:: $intrinsic(parseJson),29  parseYaml:: $intrinsic(parseYaml),3031  log:: $intrinsic(log),32  pow:: $intrinsic(pow),33  sqrt:: $intrinsic(sqrt),3435  sin:: $intrinsic(sin),36  cos:: $intrinsic(cos),37  tan:: $intrinsic(tan),38  asin:: $intrinsic(asin),39  acos:: $intrinsic(acos),40  atan:: $intrinsic(atan),4142  exp:: $intrinsic(exp),43  mantissa:: $intrinsic(mantissa),44  exponent:: $intrinsic(exponent),4546  any:: $intrinsic(any),47  all:: $intrinsic(all),4849  isString(v):: std.type(v) == 'string',50  isNumber(v):: std.type(v) == 'number',51  isBoolean(v):: std.type(v) == 'boolean',52  isObject(v):: std.type(v) == 'object',53  isArray(v):: std.type(v) == 'array',54  isFunction(v):: std.type(v) == 'function',5556  toString(a)::57    if std.type(a) == 'string' then a else '' + a,5859  substr:: $intrinsic(substr),6061  startsWith(a, b)::62    if std.length(a) < std.length(b) then63      false64    else65      std.substr(a, 0, std.length(b)) == b,6667  endsWith(a, b)::68    if std.length(a) < std.length(b) then69      false70    else71      std.substr(a, std.length(a) - std.length(b), std.length(b)) == b,7273  lstripChars(str, chars)::74    if std.length(str) > 0 && std.member(chars, str[0]) then75      std.lstripChars(str[1:], chars)76    else77      str,7879  rstripChars(str, chars)::80    local len = std.length(str);81    if len > 0 && std.member(chars, str[len - 1]) then82      std.rstripChars(str[:len - 1], chars)83    else84      str,8586  stripChars(str, chars)::87    std.lstripChars(std.rstripChars(str, chars), chars),8889  stringChars(str)::90    std.makeArray(std.length(str), function(i) str[i]),9192  local parse_nat(str, base) =93    assert base > 0 && base <= 16 : 'integer base %d invalid' % base;94    // These codepoints are in ascending order:95    local zero_code = std.codepoint('0');96    local upper_a_code = std.codepoint('A');97    local lower_a_code = std.codepoint('a');98    local addDigit(aggregate, char) =99      local code = std.codepoint(char);100      local digit = if code >= lower_a_code then101        code - lower_a_code + 10102      else if code >= upper_a_code then103        code - upper_a_code + 10104      else105        code - zero_code;106      assert digit >= 0 && digit < base : '%s is not a base %d integer' % [str, base];107      base * aggregate + digit;108    std.foldl(addDigit, std.stringChars(str), 0),109110  parseInt(str)::111    assert std.isString(str) : 'Expected string, got ' + std.type(str);112    assert std.length(str) > 0 && str != '-' : 'Not an integer: "%s"' % [str];113    if str[0] == '-' then114      -parse_nat(str[1:], 10)115    else116      parse_nat(str, 10),117118  parseOctal(str)::119    assert std.isString(str) : 'Expected string, got ' + std.type(str);120    assert std.length(str) > 0 : 'Not an octal number: ""';121    parse_nat(str, 8),122123  parseHex(str)::124    assert std.isString(str) : 'Expected string, got ' + std.type(str);125    assert std.length(str) > 0 : 'Not hexadecimal: ""';126    parse_nat(str, 16),127128  split(str, c):: std.splitLimit(str, c, -1),129130  splitLimit:: $intrinsic(splitLimit),131132  strReplace:: $intrinsic(strReplace),133134  asciiUpper:: $intrinsic(asciiUpper),135136  asciiLower:: $intrinsic(asciiLower),137138  range:: $intrinsic(range),139140  repeat(what, count)::141    local joiner =142      if std.isString(what) then ''143      else if std.isArray(what) then []144      else error 'std.repeat first argument must be an array or a string';145    std.join(joiner, std.makeArray(count, function(i) what)),146147  slice:: $intrinsic(slice),148149  member:: $intrinsic(member),150151  count:: $intrinsic(count),152153  mod:: $intrinsic(mod),154155  map:: $intrinsic(map),156157  mapWithIndex(func, arr)::158    if !std.isFunction(func) then159      error ('std.mapWithIndex first param must be function, got ' + std.type(func))160    else if !std.isArray(arr) && !std.isString(arr) then161      error ('std.mapWithIndex second param must be array, got ' + std.type(arr))162    else163      std.makeArray(std.length(arr), function(i) func(i, arr[i])),164165  mapWithKey(func, obj)::166    if !std.isFunction(func) then167      error ('std.mapWithKey first param must be function, got ' + std.type(func))168    else if !std.isObject(obj) then169      error ('std.mapWithKey second param must be object, got ' + std.type(obj))170    else171      { [k]: func(k, obj[k]) for k in std.objectFields(obj) },172173  flatMap:: $intrinsic(flatMap),174175  join:: $intrinsic(join),176177  lines(arr)::178    std.join('\n', arr + ['']),179180  deepJoin(arr)::181    if std.isString(arr) then182      arr183    else if std.isArray(arr) then184      std.join('', [std.deepJoin(x) for x in arr])185    else186      error 'Expected string or array, got %s' % std.type(arr),187188189  format:: $intrinsic(format),190191  foldr:: $intrinsic(foldr),192193  foldl:: $intrinsic(foldl),194195  filterMap(filter_func, map_func, arr)::196    if !std.isFunction(filter_func) then197      error ('std.filterMap first param must be function, got ' + std.type(filter_func))198    else if !std.isFunction(map_func) then199      error ('std.filterMap second param must be function, got ' + std.type(map_func))200    else if !std.isArray(arr) then201      error ('std.filterMap third param must be array, got ' + std.type(arr))202    else203      std.map(map_func, std.filter(filter_func, arr)),204205  assertEqual(a, b)::206    if a == b then207      true208    else209      error 'Assertion failed. ' + a + ' != ' + b,210211  abs(n)::212    if !std.isNumber(n) then213      error 'std.abs expected number, got ' + std.type(n)214    else215      if n > 0 then n else -n,216217  sign(n)::218    if !std.isNumber(n) then219      error 'std.sign expected number, got ' + std.type(n)220    else221      if n > 0 then222        1223      else if n < 0 then224        -1225      else 0,226227  max(a, b)::228    if !std.isNumber(a) then229      error 'std.max first param expected number, got ' + std.type(a)230    else if !std.isNumber(b) then231      error 'std.max second param expected number, got ' + std.type(b)232    else233      if a > b then a else b,234235  min(a, b)::236    if !std.isNumber(a) then237      error 'std.min first param expected number, got ' + std.type(a)238    else if !std.isNumber(b) then239      error 'std.min second param expected number, got ' + std.type(b)240    else241      if a < b then a else b,242243  clamp(x, minVal, maxVal)::244    if x < minVal then minVal245    else if x > maxVal then maxVal246    else x,247248  flattenArrays(arrs)::249    std.foldl(function(a, b) a + b, arrs, []),250251  manifestIni(ini)::252    local body_lines(body) =253      std.join([], [254        local value_or_values = body[k];255        if std.isArray(value_or_values) then256          ['%s = %s' % [k, value] for value in value_or_values]257        else258          ['%s = %s' % [k, value_or_values]]259260        for k in std.objectFields(body)261      ]);262263    local section_lines(sname, sbody) = ['[%s]' % [sname]] + body_lines(sbody),264          main_body = if std.objectHas(ini, 'main') then body_lines(ini.main) else [],265          all_sections = [266      section_lines(k, ini.sections[k])267      for k in std.objectFields(ini.sections)268    ];269    std.join('\n', main_body + std.flattenArrays(all_sections) + ['']),270271  manifestToml(value):: std.manifestTomlEx(value, '  '),272273  manifestTomlEx(value, indent)::274    local275      escapeStringToml = std.escapeStringJson,276      escapeKeyToml(key) =277        local bare_allowed = std.set(std.stringChars('ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789_-'));278        if std.setUnion(std.set(std.stringChars(key)), bare_allowed) == bare_allowed then key else escapeStringToml(key),279      isTableArray(v) = std.isArray(v) && std.length(v) > 0 && std.foldl(function(a, b) a && std.isObject(b), v, true),280      isSection(v) = std.isObject(v) || isTableArray(v),281      renderValue(v, indexedPath, inline, cindent) =282        if v == true then283          'true'284        else if v == false then285          'false'286        else if v == null then287          error 'Tried to manifest "null" at ' + indexedPath288        else if std.isNumber(v) then289          '' + v290        else if std.isString(v) then291          escapeStringToml(v)292        else if std.isFunction(v) then293          error 'Tried to manifest function at ' + indexedPath294        else if std.isArray(v) then295          if std.length(v) == 0 then296            '[]'297          else298            local range = std.range(0, std.length(v) - 1);299            local new_indent = if inline then '' else cindent + indent;300            local separator = if inline then ' ' else '\n';301            local lines = ['[' + separator]302                          + std.join([',' + separator],303                                     [304                                       [new_indent + renderValue(v[i], indexedPath + [i], true, '')]305                                       for i in range306                                     ])307                          + [separator + (if inline then '' else cindent) + ']'];308            std.join('', lines)309        else if std.isObject(v) then310          local lines = ['{ ']311                        + std.join([', '],312                                   [313                                     [escapeKeyToml(k) + ' = ' + renderValue(v[k], indexedPath + [k], true, '')]314                                     for k in std.objectFields(v)315                                   ])316                        + [' }'];317          std.join('', lines),318      renderTableInternal(v, path, indexedPath, cindent) =319        local kvp = std.flattenArrays([320          [cindent + escapeKeyToml(k) + ' = ' + renderValue(v[k], indexedPath + [k], false, cindent)]321          for k in std.objectFields(v)322          if !isSection(v[k])323        ]);324        local sections = [std.join('\n', kvp)] + [325          (326            if std.isObject(v[k]) then327              renderTable(v[k], path + [k], indexedPath + [k], cindent)328            else329              renderTableArray(v[k], path + [k], indexedPath + [k], cindent)330          )331          for k in std.objectFields(v)332          if isSection(v[k])333        ];334        std.join('\n\n', sections),335      renderTable(v, path, indexedPath, cindent) =336        cindent + '[' + std.join('.', std.map(escapeKeyToml, path)) + ']'337        + (if v == {} then '' else '\n')338        + renderTableInternal(v, path, indexedPath, cindent + indent),339      renderTableArray(v, path, indexedPath, cindent) =340        local range = std.range(0, std.length(v) - 1);341        local sections = [342          (cindent + '[[' + std.join('.', std.map(escapeKeyToml, path)) + ']]'343           + (if v[i] == {} then '' else '\n')344           + renderTableInternal(v[i], path, indexedPath + [i], cindent + indent))345          for i in range346        ];347        std.join('\n\n', sections);348    if std.isObject(value) then349      renderTableInternal(value, [], [], '')350    else351      error 'TOML body must be an object. Got ' + std.type(value),352353  escapeStringJson:: $intrinsic(escapeStringJson),354355  escapeStringPython(str)::356    std.escapeStringJson(str),357358  escapeStringBash(str_)::359    local str = std.toString(str_);360    local trans(ch) =361      if ch == "'" then362        "'\"'\"'"363      else364        ch;365    "'%s'" % std.join('', [trans(ch) for ch in std.stringChars(str)]),366367  escapeStringDollars(str_)::368    local str = std.toString(str_);369    local trans(ch) =370      if ch == '$' then371        '$$'372      else373        ch;374    std.foldl(function(a, b) a + trans(b), std.stringChars(str), ''),375376  manifestJson(value):: std.manifestJsonEx(value, '    ') tailstrict,377378  manifestJsonMinified(value):: std.manifestJsonEx(value, '', '', ':'),379380  manifestJsonEx:: $intrinsic(manifestJsonEx),381382  manifestYamlDoc:: $intrinsic(manifestYamlDoc),383384  manifestYamlStream(value, indent_array_in_object=false, c_document_end=true)::385    if !std.isArray(value) then386      error 'manifestYamlStream only takes arrays, got ' + std.type(value)387    else388      '---\n' + std.join(389        '\n---\n', [std.manifestYamlDoc(e, indent_array_in_object) for e in value]390      ) + if c_document_end then '\n...\n' else '\n',391392393  manifestPython(v)::394    if std.isObject(v) then395      local fields = [396        '%s: %s' % [std.escapeStringPython(k), std.manifestPython(v[k])]397        for k in std.objectFields(v)398      ];399      '{%s}' % [std.join(', ', fields)]400    else if std.isArray(v) then401      '[%s]' % [std.join(', ', [std.manifestPython(v2) for v2 in v])]402    else if std.isString(v) then403      '%s' % [std.escapeStringPython(v)]404    else if std.isFunction(v) then405      error 'cannot manifest function'406    else if std.isNumber(v) then407      std.toString(v)408    else if v == true then409      'True'410    else if v == false then411      'False'412    else if v == null then413      'None',414415  manifestPythonVars(conf)::416    local vars = ['%s = %s' % [k, std.manifestPython(conf[k])] for k in std.objectFields(conf)];417    std.join('\n', vars + ['']),418419  manifestXmlJsonml(value)::420    if !std.isArray(value) then421      error 'Expected a JSONML value (an array), got %s' % std.type(value)422    else423      local aux(v) =424        if std.isString(v) then425          v426        else427          local tag = v[0];428          local has_attrs = std.length(v) > 1 && std.isObject(v[1]);429          local attrs = if has_attrs then v[1] else {};430          local children = if has_attrs then v[2:] else v[1:];431          local attrs_str =432            std.join('', [' %s="%s"' % [k, attrs[k]] for k in std.objectFields(attrs)]);433          std.deepJoin(['<', tag, attrs_str, '>', [aux(x) for x in children], '</', tag, '>']);434435      aux(value),436437  local base64_table = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/',438  local base64_inv = { [base64_table[i]]: i for i in std.range(0, 63) },439440  base64:: $intrinsic(base64),441442  base64DecodeBytes:: $intrinsic(base64DecodeBytes),443444  base64Decode:: $intrinsic(base64Decode),445446  reverse:: $intrinsic(reverse),447448  sort:: $intrinsic(sort),449450  uniq(arr, keyF=id)::451    local f(a, b) =452      if std.length(a) == 0 then453        [b]454      else if keyF(a[std.length(a) - 1]) == keyF(b) then455        a456      else457        a + [b];458    std.foldl(f, arr, []),459460  set(arr, keyF=id)::461    std.uniq(std.sort(arr, keyF), keyF),462463  setMember(x, arr, keyF=id)::464    // TODO(dcunnin): Binary chop for O(log n) complexity465    std.length(std.setInter([x], arr, keyF)) > 0,466467  setUnion(a, b, keyF=id)::468    // NOTE: order matters, values in `a` win469    local aux(a, b, i, j, acc) =470      if i >= std.length(a) then471        acc + b[j:]472      else if j >= std.length(b) then473        acc + a[i:]474      else475        local ak = keyF(a[i]);476        local bk = keyF(b[j]);477        if ak == bk then478          aux(a, b, i + 1, j + 1, acc + [a[i]]) tailstrict479        else if ak < bk then480          aux(a, b, i + 1, j, acc + [a[i]]) tailstrict481        else482          aux(a, b, i, j + 1, acc + [b[j]]) tailstrict;483    aux(a, b, 0, 0, []),484485  setInter(a, b, keyF=id)::486    local aux(a, b, i, j, acc) =487      if i >= std.length(a) || j >= std.length(b) then488        acc489      else490        if keyF(a[i]) == keyF(b[j]) then491          aux(a, b, i + 1, j + 1, acc + [a[i]]) tailstrict492        else if keyF(a[i]) < keyF(b[j]) then493          aux(a, b, i + 1, j, acc) tailstrict494        else495          aux(a, b, i, j + 1, acc) tailstrict;496    aux(a, b, 0, 0, []) tailstrict,497498  setDiff(a, b, keyF=id)::499    local aux(a, b, i, j, acc) =500      if i >= std.length(a) then501        acc502      else if j >= std.length(b) then503        acc + a[i:]504      else505        if keyF(a[i]) == keyF(b[j]) then506          aux(a, b, i + 1, j + 1, acc) tailstrict507        else if keyF(a[i]) < keyF(b[j]) then508          aux(a, b, i + 1, j, acc + [a[i]]) tailstrict509        else510          aux(a, b, i, j + 1, acc) tailstrict;511    aux(a, b, 0, 0, []) tailstrict,512513  mergePatch(target, patch)::514    if std.isObject(patch) then515      local target_object =516        if std.isObject(target) then target else {};517518      local target_fields =519        if std.isObject(target_object) then std.objectFields(target_object) else [];520521      local null_fields = [k for k in std.objectFields(patch) if patch[k] == null];522      local both_fields = std.setUnion(target_fields, std.objectFields(patch));523524      {525        [k]:526          if !std.objectHas(patch, k) then527            target_object[k]528          else if !std.objectHas(target_object, k) then529            std.mergePatch(null, patch[k]) tailstrict530          else531            std.mergePatch(target_object[k], patch[k]) tailstrict532        for k in std.setDiff(both_fields, null_fields)533      }534    else535      patch,536537  get(o, f, default = null, inc_hidden = true)::538    if std.objectHasEx(o, f, inc_hidden) then o[f] else default,539540  objectFields(o)::541    std.objectFieldsEx(o, false),542543  objectFieldsAll(o)::544    std.objectFieldsEx(o, true),545546  objectHas(o, f)::547    std.objectHasEx(o, f, false),548549  objectHasAll(o, f)::550    std.objectHasEx(o, f, true),551552  objectValues(o)::553    [o[k] for k in std.objectFields(o)],554555  objectValuesAll(o)::556    [o[k] for k in std.objectFieldsAll(o)],557558  equals:: $intrinsic(equals),559560  resolvePath(f, r)::561    local arr = std.split(f, '/');562    std.join('/', std.makeArray(std.length(arr) - 1, function(i) arr[i]) + [r]),563564  prune(a)::565    local isContent(b) =566      if b == null then567        false568      else if std.isArray(b) then569        std.length(b) > 0570      else if std.isObject(b) then571        std.length(b) > 0572      else573        true;574    if std.isArray(a) then575      [std.prune(x) for x in a if isContent($.prune(x))]576    else if std.isObject(a) then {577      [x]: $.prune(a[x])578      for x in std.objectFields(a)579      if isContent(std.prune(a[x]))580    } else581      a,582583  findSubstr(pat, str)::584    if !std.isString(pat) then585      error 'findSubstr first parameter should be a string, got ' + std.type(pat)586    else if !std.isString(str) then587      error 'findSubstr second parameter should be a string, got ' + std.type(str)588    else589      local pat_len = std.length(pat);590      local str_len = std.length(str);591      if pat_len == 0 || str_len == 0 || pat_len > str_len then592        []593      else594        std.filter(function(i) str[i:i + pat_len] == pat, std.range(0, str_len - pat_len)),595596  find(value, arr)::597    if !std.isArray(arr) then598      error 'find second parameter should be an array, got ' + std.type(arr)599    else600      std.filter(function(i) arr[i] == value, std.range(0, std.length(arr) - 1)),601}