1 files changed, 131 insertions, 0 deletions
diff --git a/ffi/lang/python/sequoia/core.py b/ffi/lang/python/sequoia/core.py
new file mode 100644
index 00000000..9810c27b
--- /dev/null
+++ b/ffi/lang/python/sequoia/core.py
@@ -0,0 +1,131 @@
+import io
+from enum import Enum
+
+from _sequoia import ffi, lib
+from .error import Error
+from .glue import SQObject
+
+class NetworkPolicy(Enum):
+    Offline = lib.SQ_NETWORK_POLICY_OFFLINE
+    Anonymized = lib.SQ_NETWORK_POLICY_ANONYMIZED
+    Encrypted = lib.SQ_NETWORK_POLICY_ENCRYPTED
+    Insecure = lib.SQ_NETWORK_POLICY_INSECURE
+
+class IPCPolicy(Enum):
+    External = lib.SQ_IPC_POLICY_EXTERNAL
+    Internal = lib.SQ_IPC_POLICY_INTERNAL
+    Robust = lib.SQ_IPC_POLICY_ROBUST
+
+class Context(SQObject):
+    _del = lib.sq_context_free
+    def __init__(self, domain,
+                 home=None,
+                 network_policy=NetworkPolicy.Encrypted,
+                 ipc_policy=IPCPolicy.Robust,
+                 ephemeral=False):
+        cfg = lib.sq_context_configure(domain.encode())
+        if home:
+            lib.sq_config_home(cfg, home.encode())
+        lib.sq_config_network_policy(cfg, network_policy.value)
+        lib.sq_config_ipc_policy(cfg, ipc_policy.value)
+        if ephemeral:
+            lib.sq_config_ephemeral(cfg)
+        err = ffi.new("sq_error_t[1]")
+        o = lib.sq_config_build(cfg, err)
+        if o == ffi.NULL:
+            raise Error._from(err[0])
+        super(Context, self).__init__(o)
+
+class AbstractReader(SQObject, io.RawIOBase):
+    _del = lib.sq_reader_free
+
+    def readable(self):
+        return True
+    def writable(self):
+        return False
+
+    def readinto(self, buf):
+        bytes_read = lib.sq_reader_read(
+            self.context().ref(), self.ref(),
+            ffi.cast("uint8_t *", ffi.from_buffer(buf)), len(buf))
+        if bytes_read < 0:
+            raise Error._last(self.context())
+        return bytes_read
+
+    def close(self):
+        self._delete()
+
+    # Implement the context manager protocol.
+    def __enter__(self):
+        return self
+    def __exit__(self, *args):
+        self.close()
+        return False
+
+class Reader(AbstractReader):
+    @classmethod
+    def open(cls, ctx, filename):
+        return Reader(
+            lib.sq_reader_from_file(
+                ctx.ref(),
+                filename.encode()),
+            context=ctx)
+
+    @classmethod
+    def from_fd(cls, ctx, fd):
+        return Reader(lib.sq_reader_from_fd(fd),
+                      context=ctx)
+
+    @classmethod
+    def from_bytes(cls, ctx, buf):
+        return Reader(
+            lib.sq_reader_from_bytes(
+                ffi.cast("uint8_t *", ffi.from_buffer(buf)), len(buf)),
+            context=ctx)
+
+class AbstractWriter(SQObject, io.RawIOBase):
+    _del = lib.sq_writer_free
+
+    def readable(self):
+        return False
+    def writable(self):
+        return True
+
+    def write(self, buf):
+        bytes_written = lib.sq_writer_write(
+            self.context().ref(), self.ref(),
+            ffi.cast("const uint8_t *", ffi.from_buffer(buf)), len(buf))
+        if bytes_written < 0:
+            raise Error._last(self.context())
+        return bytes_written
+
+    def close(self):
+        self._delete()
+
+    # Implement the context manager protocol.
+    def __enter__(self):
+        return self
+    def __exit__(self, *args):
+        self.close()
+        return False
+
+class Writer(AbstractWriter):
+    @classmethod
+    def open(cls, ctx, filename):
+        return Writer(
+            lib.sq_writer_from_file(
+                ctx.ref(),
+                filename.encode()),
+            context=ctx)
+
+    @classmethod
+    def from_fd(cls, ctx, fd):
+        return Writer(lib.sq_writer_from_fd(fd),
+                      context=ctx)
+
+    @classmethod
+    def from_bytes(cls, ctx, buf):
+        return Writer(
+            lib.sq_writer_from_bytes(
+                ffi.cast("uint8_t *", ffi.from_buffer(buf)), len(buf)),
+            context=ctx)