elvish_config/lib/moi/util/map.elv

fn empty {|container| == (count $container) 0 }

fn values {|container| keys $container | each {|v| put $container[$v] } }
fn pvalues {|container| keys $container | peach {|v| put $container[$v] } }

fn value-of {|container k &default=$nil|
  if (has-key $container $k) {
    put $container[$k]
  } else {
    put $default
  }
}

fn unzip {|container|
  var lkeys lvalues = [] []
  keys $container | each {|k|
    set lkeys   = [(all $lkeys) $k]
    set lvalues = [(all $lvalues) $container[$k]]
  }
  put $lkeys $lvalues
}

fn zip {|lkeys lvalues|
  var ck cv = (count $lkeys) (count $lvalues)
  var c = $ck
  if (> $ck $cv) {
    set c = $cv
  }
  var result = [&]
  range $c | peach {|i|
    put [&k=$lkeys[$i] &i=$i]
  } | peach {|e|
    assoc $e v $lvalues[$e[i]]
  } | each {|e|
    set result[$e[k]] = $e[v]
  }
  put $result
}

fn to-map {|@argv|
  use ./list
  var container = (list:to-list $@argv)
  zip [(range (count $container))] $container
}

fn to-set {|@argv|
  use ./list
  var m = [&]
  each {|k| set m[$k] = $nil } (list:to-list $@argv)
  put $m
}

fn mdissoc {|container @argv|
  each {|k|
    set container = (dissoc $container $k)
  } $argv
  put $container
}

fn massoc {|&replace=$true container @argv|
  each {|e|
    var k v = $@e
    if (or $replace (not (has-key $container $k))) {
      set container[$k] = $v
    }
  } $argv
  put $container
}

fn add {|container key @values|
  use ./list
  var values = (list:to-list $values)
  if (has-key $container $key) {
    set container[$key] = [(all $container[$key]) $@values]
  } else {
    set container[$key] = $values
  }
  put $container
}

fn join {|container container2|
  keys $container2 | each {|k|
    set container[$k] = $container2[$k]
  }
  put $container
}

fn sorted-keys {|&reverse=$false &less-than=$nil container|
  keys $container | order &reverse=$reverse &less-than=$less-than
}

fn sorted-values {|&reverse=$false &less-than=$nil container|
  values $container | order &reverse=$reverse &less-than=$less-than
}