astreum 0.2.29__py3-none-any.whl → 0.2.30__py3-none-any.whl

astreum/node.py

@@ -463,7 +463,7 @@ class Node:
                 elif first.value == "def":
                     args = expr.elements[1:]
                     if len(args) != 2:
-                        return Expr.Error(message=f"'def' expects exactly 2 arguments, got {len(args)}", origin=expr)
+                        return Expr.Error("def expects key value", origin=expr)
                     if not isinstance(args[0], Expr.Symbol):
                         return Expr.Error(message="first argument to 'def' must be a symbol", origin=args[0])
                     result = self.machine_expr_eval(env_id=env_id, expr=args[1])
@@ -473,13 +473,191 @@ class Node:
                     self.machine_expr_put(env_id=env_id, name=args[0].value, expr=result)
                     return result
 
-                # # List
-                elif first.value == "list.each":
-                    internal_function = expr.elements[1]
+                ## List: ints -> (1 2)
+                # push: (list.push 3 ints) -> (1 2 3) / (list.push 0 0 ints) -> (0 1 2)
+                elif first.value == "list.push":
+                    args = expr.elements[1:]
+                    if len(args) == 2:
+                        val_expr, list_expr = args
+                        idx = None
+                    elif len(args) == 3:
+                        idx_expr, val_expr, list_expr = args
+                        idx = self.machine_expr_eval(env_id, idx_expr)
+                        if isinstance(idx, Expr.Error): return idx
+                        if not isinstance(idx, Expr.IntExpr):
+                            return Expr.Error("index must be int", origin=idx_expr)
+                        idx = idx.value
+                    else:
+                        return Expr.Error("list.push expects (value list) or (index value list)", origin=expr)
+
+                    lst = self.machine_expr_eval(env_id, list_expr)
+                    if isinstance(lst, Expr.Error): return lst
+                    if not isinstance(lst, Expr.ListExpr):
+                        return Expr.Error("last arg to list.push must be a list", origin=list_expr)
+
+                    val = self.machine_expr_eval(env_id, val_expr)
+                    if isinstance(val, Expr.Error): return val
 
-                
+                    elems = list(lst.elements)
+                    if idx is None:
+                        elems.append(val)
+                    else:
+                        if idx < 0 or idx > len(elems):
+                            return Expr.Error("index out of range", origin=idx_expr)
+                        elems.insert(idx, val)
+                    return Expr.ListExpr(elems)
+
+                # pop: (list.pop 1 ints) -> 2
+                elif first.value == "list.pop":
+                    if len(expr.elements) < 3:
+                        return Expr.Error("list.pop expects index list", origin=expr)
+
+                    idx_expr, list_expr = expr.elements[1], expr.elements[2]
+                    idx = self.machine_expr_eval(env_id, idx_expr)
+                    if isinstance(idx, Expr.Error): return idx
+                    if not isinstance(idx, Expr.IntExpr):
+                        return Expr.Error("index must be int", origin=idx_expr)
+                    idx = idx.value
+
+                    lst = self.machine_expr_eval(env_id, list_expr)
+                    if isinstance(lst, Expr.Error): return lst
+                    if not isinstance(lst, Expr.ListExpr):
+                        return Expr.Error("second arg to list.pop must be a list", origin=list_expr)
+
+                    elems = list(lst.elements)
+                    if idx < 0 or idx >= len(elems):
+                        return Expr.Error("index out of range", origin=idx_expr)
+                    del elems[idx]
+                    return Expr.ListExpr(elems)
+
+                # get: (list.get 1 ints) -> 2
+                elif first.value == "list.get":
+                    if len(expr.elements) < 3:
+                        return Expr.Error("list.get expects index list", origin=expr)
+
+                    idx_expr, list_expr = expr.elements[1], expr.elements[2]
+                    idx = self.machine_expr_eval(env_id, idx_expr)
+                    if isinstance(idx, Expr.Error): return idx
+                    if not isinstance(idx, Expr.IntExpr):
+                        return Expr.Error("index must be int", origin=idx_expr)
+                    idx = idx.value
+
+                    lst = self.machine_expr_eval(env_id, list_expr)
+                    if isinstance(lst, Expr.Error): return lst
+                    if not isinstance(lst, Expr.ListExpr):
+                        return Expr.Error("second arg to list.get must be a list", origin=list_expr)
+
+                    if idx < 0 or idx >= len(lst.elements):
+                        return Expr.Error("index out of range", origin=idx_expr)
+                    return lst.elements[idx]
+
+                # set: (list.set 1 3 ints) -> (1 3)
+                elif first.value == "list.set":
+                    if len(expr.elements) < 4:
+                        return Expr.Error("list.set expects index value list", origin=expr)
+                    idx_expr, val_expr, list_expr = expr.elements[1], expr.elements[2], expr.elements[3]
+                    idx = self.machine_expr_eval(env_id, idx_expr)
+                    if isinstance(idx, Expr.Error): return idx
+                    if not isinstance(idx, Expr.IntExpr):
+                        return Expr.Error("index must be int", origin=idx_expr)
+                    idx = idx.value
+
+                    val = self.machine_expr_eval(env_id, val_expr)
+                    if isinstance(val, Expr.Error): return val
+
+                    lst = self.machine_expr_eval(env_id, list_expr)
+                    if isinstance(lst, Expr.Error): return lst
+                    if not isinstance(lst, Expr.ListExpr):
+                        return Expr.Error("third arg to list.set must be a list", origin=list_expr)
 
-                # Integer arithmetic primitives
+                    elems = list(lst.elements)
+                    if idx < 0 or idx >= len(elems):
+                        return Expr.Error("index out of range", origin=idx_expr)
+                    elems[idx] = val
+                    return Expr.ListExpr(elems)
+
+                ### each: (list.each fn list) -> ()
+                elif first.value == "list.each":
+                    if len(expr.elements) < 3:
+                        return Expr.Error("list.each expects fn list", origin=expr)
+                    fn_expr, list_expr = expr.elements[1], expr.elements[2]
+                    lst = self.machine_expr_eval(env_id, list_expr)
+                    if isinstance(lst, Expr.Error):
+                        return lst
+                    if not isinstance(lst, Expr.ListExpr):
+                        return Expr.Error("second arg to list.each must be a list", origin=list_expr)
+
+                    for el in lst.elements:
+                        res = self.machine_expr_eval(env_id, Expr.ListExpr([fn_expr, el]))
+                        if isinstance(res, Expr.Error):
+                            return res
+                    return Expr.ListExpr([])
+
+                ### fold: (list.fold fn init list) / (list.fold + 0 ints) -> 3
+                elif first.value == "list.fold":
+                    fn_expr, init_expr, list_expr = expr.elements[1], expr.elements[2], expr.elements[3]
+                    acc = self.machine_expr_eval(env_id, init_expr)
+                    if isinstance(acc, Expr.Error):
+                        return acc
+
+                    lst = self.machine_expr_eval(env_id, list_expr)
+                    if isinstance(lst, Expr.Error):
+                        return lst
+                    if not isinstance(lst, Expr.ListExpr):
+                        return Expr.Error("third arg to list.fold must be a list", origin=list_expr)
+
+                    for el in lst.elements:
+                        call = Expr.ListExpr([fn_expr, acc, el])
+                        res = self.machine_expr_eval(env_id, call)
+                        if isinstance(res, Expr.Error):
+                            return res
+                        acc = res
+
+                    return acc
+
+                ### sort: (list.sort fn list) / (list.sort (fn (a b) (a < b)) ints) -> (2 1)
+                elif first.value == "list.sort":
+                    if len(expr.elements) < 3:
+                        return Expr.Error("list.sort fn list", origin=expr)
+                    fn_e, lst_e = expr.elements[1], expr.elements[2]
+
+                    lst = self.machine_expr_eval(env_id, lst_e)
+                    if isinstance(lst, Expr.Error): return lst
+                    if not isinstance(lst, Expr.ListExpr):
+                        return Expr.Error("second arg must be list", origin=lst_e)
+
+                    elems = list(lst.elements)
+                    for i in range(1, len(elems)):
+                        j = i
+                        while j > 0:
+                            cmp_res = self.machine_expr_eval(
+                                env_id,
+                                Expr.ListExpr([fn_e, elems[j-1], elems[j]])
+                            )
+                            if isinstance(cmp_res, Expr.Error): return cmp_res
+                            if not isinstance(cmp_res, Expr.BoolExpr):
+                                return Expr.Error("comparator must return bool", origin=fn_e)
+
+                            if cmp_res.value:
+                                elems[j-1], elems[j] = elems[j], elems[j-1]
+                                j -= 1
+                            else:
+                                break
+                    return Expr.ListExpr(elems)
+
+                ### len: (list.len list) -> Int / (list.len ints) -> Integer(2)
+                elif first.value == "list.len":
+                    if len(expr.elements) < 2:
+                        return Expr.Error("list.len list", origin=expr)
+                    lst_e = expr.elements[1]
+                    lst = self.machine_expr_eval(env_id, lst_e)
+                    if isinstance(lst, Expr.Error): return lst
+                    if not isinstance(lst, Expr.ListExpr):
+                        return Expr.Error("arg must be list", origin=lst_e)
+                    return Expr.Integer(len(lst.elements))
+
+                ## Integer
+                ### add
                 elif first.value == "+":
                     args = expr.elements[1:]
                     if not args:

{astreum-0.2.29.dist-info → astreum-0.2.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: astreum
-Version: 0.2.29
+Version: 0.2.30
 Summary: Python library to interact with the Astreum blockchain and its Lispeum virtual machine.
 Author-email: "Roy R. O. Okello" <roy@stelar.xyz>
 Project-URL: Homepage, https://github.com/astreum/lib

{astreum-0.2.29.dist-info → astreum-0.2.30.dist-info}/RECORD

@@ -1,6 +1,6 @@
 astreum/__init__.py,sha256=y2Ok3EY_FstcmlVASr80lGR_0w-dH-SXDCCQFmL6uwA,28
 astreum/format.py,sha256=X4tG5GGPweNCE54bHYkLFiuLTbmpy5upO_s1Cef-MGA,2711
-astreum/node.py,sha256=OhiApRqQcPc6-CWbk2NeKdOiXQapy0mbl3uLK_xjmYU,28971
+astreum/node.py,sha256=hBux-hMj2vLtXsWaEiMd9y8PgKCh3Sp1-tmaKs0B5ZU,38163
 astreum/crypto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 astreum/crypto/ed25519.py,sha256=FRnvlN0kZlxn4j-sJKl-C9tqiz_0z4LZyXLj3KIj1TQ,1760
 astreum/crypto/quadratic_form.py,sha256=pJgbORey2NTWbQNhdyvrjy_6yjORudQ67jBz2ScHptg,4037
@@ -26,8 +26,8 @@ astreum/relay/setup.py,sha256=ynvGaJdlDtw_f5LLiow2Wo7IRzUjvgk8eSr1Sv4_zTg,2090
 astreum/storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 astreum/storage/object.py,sha256=knFlvw_tpcC4twSu1DGNpHX31wlANN8E5dgEqIfU--Q,2041
 astreum/storage/setup.py,sha256=1-9ztEFI_BvRDvAA0lAn4mFya8iq65THTArlj--M3Hg,626
-astreum-0.2.29.dist-info/licenses/LICENSE,sha256=gYBvRDP-cPLmTyJhvZ346QkrYW_eleke4Z2Yyyu43eQ,1089
-astreum-0.2.29.dist-info/METADATA,sha256=JSIc8N98KW4Wmm89uMMFxyJKiEFxIU-zGRjIXf9JL24,5478
-astreum-0.2.29.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-astreum-0.2.29.dist-info/top_level.txt,sha256=1EG1GmkOk3NPmUA98FZNdKouhRyget-KiFiMk0i2Uz0,8
-astreum-0.2.29.dist-info/RECORD,,
+astreum-0.2.30.dist-info/licenses/LICENSE,sha256=gYBvRDP-cPLmTyJhvZ346QkrYW_eleke4Z2Yyyu43eQ,1089
+astreum-0.2.30.dist-info/METADATA,sha256=LqByNjd5zu9TMcCuIw6j-YuRAEYqCQHMP1prBdjFZaM,5478
+astreum-0.2.30.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+astreum-0.2.30.dist-info/top_level.txt,sha256=1EG1GmkOk3NPmUA98FZNdKouhRyget-KiFiMk0i2Uz0,8
+astreum-0.2.30.dist-info/RECORD,,