ABI Strings

pyteal/ast/abi/__init__.py

@@ -1,3 +1,4 @@
+from .string import String, Address, StringTypeSpec, AddressTypeSpec
 from .type import TypeSpec, BaseType, ComputedValue
 from .bool import BoolTypeSpec, Bool
 from .uint import (
@@ -32,6 +33,10 @@
 from .method_return import MethodReturn
 
 __all__ = [
+    "String",
+    "StringTypeSpec",
+    "Address",
+    "AddressTypeSpec",
     "TypeSpec",
     "BaseType",
     "ComputedValue",

pyteal/ast/abi/string.py

@@ -0,0 +1,55 @@
+from .array_static import StaticArray, StaticArrayTypeSpec
+from .array_dynamic import DynamicArray, DynamicArrayTypeSpec
+from .uint import ByteTypeSpec
+
+address_length = 32
+
+
+class AddressTypeSpec(StaticArrayTypeSpec):
+    def __init__(self) -> None:
+        super().__init__(ByteTypeSpec, address_length)
+
+    def new_instance(self) -> "Address":
+        return Address()
+
+    def __str__(self) -> str:
+        return "address"
+
+
+AddressTypeSpec.__module__ = "pyteal"
+
+
+class Address(StaticArray):
+    def __init__(self) -> None:
+        super().__init__(AddressTypeSpec(), address_length)
+
+    def type_spec(self) -> AddressTypeSpec:
+        return AddressTypeSpec()
+
+
+Address.__module__ = "pyteal"
+
+
+class StringTypeSpec(DynamicArrayTypeSpec):
+    def __init__(self) -> None:
+        super().__init__(ByteTypeSpec)
+
+    def new_instance(self) -> "String":
+        return String()
+
+    def __str__(self) -> str:
+        return "string"
+
+
+StringTypeSpec.__module__ = "pyteal"
+
+
+class String(DynamicArray):
+    def __init__(self) -> None:
+        super().__init__(StringTypeSpec())
+
+    def type_spec(self) -> StringTypeSpec:
+        return StringTypeSpec()
+
+
+String.__module__ = "pyteal"

pyteal/ast/abi/util.py

@@ -107,6 +107,7 @@ def type_spec_from_annotation(annotation: Any) -> TypeSpec:
         Tuple4,
         Tuple5,
     )
+    from .string import String, Address, StringTypeSpec, AddressTypeSpec
 
     origin = get_origin(annotation)
     if origin is None:
@@ -144,6 +145,16 @@ def type_spec_from_annotation(annotation: Any) -> TypeSpec:
             raise TypeError("Uint64 expects 0 type arguments. Got: {}".format(args))
         return Uint64TypeSpec()
 
+    if origin is String:
+        if len(args) != 0:
+            raise TypeError("String expects 0 arguments. Got: {}".format(args))
+        return StringTypeSpec()
+
+    if origin is Address:
+        if len(args) != 0:
+            raise TypeError("Address expects 0 arguments. Got: {}".format(args))
+        return AddressTypeSpec()
+
     if origin is DynamicArray:
         if len(args) != 1:
             raise TypeError(