poc

.gitignore

@@ -1,9 +1,19 @@
+# Testing
+db.sqlite3
+
+# Nix
+result
+# Python
 .direnv
 __pycache__
 .ruff_cache
 .mypy_cache
 .coverage
 .coverage.*
-result
-db.sqlite3
 htmlcov
+
+# Zig
+*.so
+*.so.o
+zig-cache
+zig-out

Makefile

@@ -24,4 +24,7 @@ lint:
 	ruff check --fix .
 	prettier --write .
 
-.PHONY: check test typecheck lintcheck lint
+clean:
+	git clean -xdf
+
+.PHONY: check test typecheck lintcheck lint clean

flake.nix

@@ -26,8 +26,9 @@
         };
         doCheck = false;
       };
-      prod-deps = with python.pkgs; [
+      prod-py-deps = with python.pkgs; [
         appdirs
+        cffi
         click
         jinja2
         llfuse
@@ -38,22 +39,15 @@
         uuid6-python
         watchdog
       ];
-      dev-deps = with python.pkgs; [
+      dev-py-deps = with python.pkgs; [
         mypy
         pytest
         pytest-timeout
         pytest-cov
         pytest-xdist
         snapshottest
       ];
-      zig-pkgs = with pkgs; [
-        zig
-        zls
-      ];
-      dev-cli = pkgs.writeShellScriptBin "rose" ''
-        cd $ROSE_ROOT
-        python -m rose_cli "$@"
-      '';
+      version = nixpkgs.lib.strings.removeSuffix "\n" (builtins.readFile ./rose/.version);
     in
     {
       devShells.default = pkgs.mkShell {
@@ -70,25 +64,37 @@
         buildInputs = [
           (pkgs.buildEnv {
             name = "rose-devshell";
-            paths = with pkgs; ([
-              (python.withPackages (_: prod-deps ++ dev-deps))
+            paths = with pkgs; [
+              (python.withPackages (_: prod-py-deps ++ dev-py-deps))
               ruff
-              dev-cli
               nodePackages.pyright
               nodePackages.prettier
-            ] ++ zig-pkgs);
+              zig
+              zls
+            ];
           })
         ];
       };
       packages = rec {
-        rose = python.pkgs.buildPythonPackage {
+        rose-zig = pkgs.stdenv.mkDerivation {
+          pname = "rose";
+          version = version;
+          src = ./rose_zig;
+          nativeBuildInputs = [ pkgs.zig.hook ];
+        };
+        # TODO: Split up into multiple packages.
+        rose-py = python.pkgs.buildPythonPackage {
           pname = "rose";
-          version = nixpkgs.lib.strings.removeSuffix "\n" (builtins.readFile ./rose/.version);
+          version = version;
           src = ./.;
-          propagatedBuildInputs = prod-deps;
+          propagatedBuildInputs = prod-py-deps ++ [ rose-zig ];
+          nativeBuildInputs = [ pkgs.makeWrapper ];
+          postInstall = ''
+            wrapProgram $out/bin/rose \
+              --prefix LD_LIBRARY_PATH : "${pkgs.lib.makeLibraryPath [ rose-zig ]}"
+          '';
           doCheck = false;
         };
-        default = rose;
       };
     });
 }

pyproject.toml

@@ -86,6 +86,9 @@ ignore_missing_imports = true
 [[tool.mypy.overrides]]
 module = "snapshottest"
 ignore_missing_imports = true
+[[tool.mypy.overrides]]
+module = "cffi"
+ignore_missing_imports = true
 
 [tool.pytest.ini_options]
 addopts = [

rose/ffi.py

@@ -0,0 +1,16 @@
+# type: ignore
+from cffi import FFI
+
+ffi = FFI()
+
+lib = ffi.dlopen("rose")
+
+ffi.cdef("""
+    void free_str(void *str);
+
+    char *getRelease();
+""")
+
+
+def get_release() -> str:
+    return ffi.string(ffi.gc(lib.getRelease(), lib.free_str)).decode("utf-8")

rose_cli/__main__.py

@@ -7,6 +7,10 @@
 
 def main() -> None:
     from rose import RoseExpectedError
+    from rose.ffi import get_release
+
+    print(get_release())
+    exit(0)
 
     try:
         cli()

rose_zig/build.zig

@@ -0,0 +1,14 @@
+const std = @import("std");
+
+pub fn build(b: *std.Build) void {
+    const target = b.standardTargetOptions(.{});
+    const optimize = b.standardOptimizeOption(.{});
+
+    const libroseffi = b.addSharedLibrary(.{
+        .name = "rose",
+        .root_source_file = .{ .path = "src/ffi.zig" },
+        .target = target,
+        .optimize = optimize,
+    });
+    b.installArtifact(libroseffi);
+}

rose_zig/src/ffi.zig

@@ -0,0 +1,17 @@
+const std = @import("std");
+const rose = @import("rose.zig");
+
+var gpa = std.heap.GeneralPurposeAllocator(.{}){};
+const allocator = gpa.allocator();
+
+export fn getRelease() [*:0]const u8 {
+    const message = rose.getRelease(allocator) catch |err| switch (err) {
+        error.OutOfMemory => @panic("Out of memory"),
+    };
+    return message.ptr;
+}
+
+export fn free_str(str: [*:0]const u8) void {
+    const len = std.mem.len(str);
+    allocator.free(str[0 .. len + 1]);
+}

rose_zig/src/rose.zig

@@ -0,0 +1,5 @@
+const std = @import("std");
+
+pub fn getRelease(allocator: std.mem.Allocator) ![:0]const u8 {
+    return try allocator.dupeZ(u8, "hello");
+}