From 7f481f9ff656deb6e93e87486d5360581a9b4d90 Mon Sep 17 00:00:00 2001
From: Emery Hemingway <ehmry@posteo.net>
Date: Wed, 10 Jan 2024 10:16:42 +0200
Subject: [PATCH] WiP! Postgre actor

---
 config.prs                       |  6 +++
 shell.nix                        |  2 +-
 src/schema/config.nim            | 15 ++++++
 src/syndesizer.nim               |  8 +++
 src/syndesizer/postgre_actor.nim | 83 ++++++++++++++++++++++++++++++++
 5 files changed, 113 insertions(+), 1 deletion(-)
 create mode 100644 src/syndesizer/postgre_actor.nim

diff --git a/config.prs b/config.prs
index 3793d4b..4e1987d 100644
--- a/config.prs
+++ b/config.prs
@@ -22,6 +22,12 @@ JsonSocketTranslatorArguments = <json-socket-translator {
   socket: string
 }>.
 
+PostgreArguments = <postgre {
+  connection: [PostgreConnectionParameter ...]
+  dataspace: #!any
+}>.
+PostgreConnectionParameter = [@key string @val string].
+
 SqliteArguments = <sqlite {
   database: string
   dataspace: #!any
diff --git a/shell.nix b/shell.nix
index ea5fd1a..2333a0b 100644
--- a/shell.nix
+++ b/shell.nix
@@ -3,5 +3,5 @@
 pkgs.buildNimPackage {
   name = "dummy";
   propagatedNativeBuildInputs = [ pkgs.pkg-config ];
-  propagatedBuildInputs = [ pkgs.sqlite ];
+  propagatedBuildInputs = [ pkgs.postgresql pkgs.sqlite ];
 }
diff --git a/src/schema/config.nim b/src/schema/config.nim
index 7f1ac5e..c4291b0 100644
--- a/src/schema/config.nim
+++ b/src/schema/config.nim
@@ -54,6 +54,17 @@ type
   CacheArguments* {.preservesRecord: "cache".} = object
     `field0`*: CacheArgumentsField0
 
+  PostgreConnectionParameter* {.preservesTuple.} = object
+    `key`*: string
+    `val`*: string
+
+  PostgreArgumentsField0* {.preservesDictionary.} = object
+    `connection`*: seq[PostgreConnectionParameter]
+    `dataspace`* {.preservesEmbedded.}: EmbeddedRef
+
+  PostgreArguments* {.preservesRecord: "postgre".} = object
+    `field0`*: PostgreArgumentsField0
+
   Tcp* {.preservesRecord: "tcp".} = object
     `host`*: string
     `port`*: BiggestInt
@@ -65,6 +76,8 @@ proc `$`*(x: WebsocketArguments | JsonTranslatorArguments |
     FileSystemUsageArguments |
     SqliteArguments |
     CacheArguments |
+    PostgreConnectionParameter |
+    PostgreArguments |
     Tcp): string =
   `$`(toPreserves(x))
 
@@ -75,5 +88,7 @@ proc encode*(x: WebsocketArguments | JsonTranslatorArguments |
     FileSystemUsageArguments |
     SqliteArguments |
     CacheArguments |
+    PostgreConnectionParameter |
+    PostgreArguments |
     Tcp): seq[byte] =
   encode(toPreserves(x))
diff --git a/src/syndesizer.nim b/src/syndesizer.nim
index 2c83606..661c5e0 100644
--- a/src/syndesizer.nim
+++ b/src/syndesizer.nim
@@ -5,6 +5,9 @@
 
 import syndicate, syndicate/relays, syndicate/actors/timers
 
+const
+  withPostgre* {.booldefine.}: bool = true
+
 import ./syndesizer/[
   cache_actor,
   file_system_usage,
@@ -14,6 +17,9 @@ import ./syndesizer/[
   webhooks,
   websockets]
 
+when withPostgre:
+  import ./syndesizer/postgre_actor
+
 runActor("syndesizer") do (turn: var Turn; root: Cap):
   connectStdio(turn, root)
   discard spawnTimers(turn, root)
@@ -23,5 +29,7 @@ runActor("syndesizer") do (turn: var Turn; root: Cap):
   discard spawnJsonStdioTranslator(turn, root)
   discard spawnWebhookActor(turn, root)
   discard spawnWebsocketActor(turn, root)
+  when withPostgre:
+    discard spawnPostgreActor(turn, root)
   when withSqlite:
     discard spawnSqliteActor(turn, root)
diff --git a/src/syndesizer/postgre_actor.nim b/src/syndesizer/postgre_actor.nim
new file mode 100644
index 0000000..7bf5282
--- /dev/null
+++ b/src/syndesizer/postgre_actor.nim
@@ -0,0 +1,83 @@
+# SPDX-FileCopyrightText: ☭ Emery Hemingway
+# SPDX-License-Identifier: Unlicense
+
+
+import preserves, syndicate, syndicate/relays
+import ../schema/[config, sql]
+
+{.passL: "-lpq".}
+
+{.pragma: libpq, header: "libpq-fe.h", importc.}
+
+type
+  PGconn* {.libpq.} = ptr object
+  ConnStatusType* {.libpq.} = enum
+    CONNECTION_OK, CONNECTION_BAD, ##  Non-blocking mode only below here
+                                 ##
+                                 ##  The existence of these should never be relied upon - they should only
+                                 ##  be used for user feedback or similar purposes.
+                                 ##
+    CONNECTION_STARTED,       ##  Waiting for connection to be made.
+    CONNECTION_MADE,          ##  Connection OK; waiting to send.
+    CONNECTION_AWAITING_RESPONSE, ##  Waiting for a response from the
+                                 ##  postmaster.
+    CONNECTION_AUTH_OK,       ##  Received authentication; waiting for
+                       ##  backend startup.
+    CONNECTION_SETENV,        ##  This state is no longer used.
+    CONNECTION_SSL_STARTUP,   ##  Negotiating SSL.
+    CONNECTION_NEEDED,        ##  Internal state: connect() needed
+    CONNECTION_CHECK_WRITABLE, ##  Checking if session is read-write.
+    CONNECTION_CONSUME,       ##  Consuming any extra messages.
+    CONNECTION_GSS_STARTUP,   ##  Negotiating GSSAPI.
+    CONNECTION_CHECK_TARGET,  ##  Checking target server properties.
+    CONNECTION_CHECK_STANDBY  ##  Checking if server is in standby mode.
+
+
+proc PQconnectdbParams(
+    keywords: cstringArray; values: cstringArray; expand_dbname: cint): PGconn {.libpq.}
+
+proc PQerrorMessage(conn: PGconn): cstring {.libpq.}
+proc PQfinish(conn: PGconn) {.libpq.}
+proc PQstatus(conn: PGconn): ConnStatusType {.libpq.}
+
+# proc PQsocket(conn: PGconn): cint
+# proc PQconnectStartParams(
+#     keywords: cstringArray; values: cstringArray; expand_dbname: cint): PGconn
+# TODO: async
+
+
+type StringPairs = seq[tuple[key: string, val: string]]
+
+proc splitParams(params: StringPairs): (cstringArray, cstringArray) =
+  var strings = newSeq[string](params.len)
+  for i, _ in params: strings[i] = params[i][0]
+  result[0] = allocCStringArray(strings)
+  for i, _ in params: strings[i] = params[i][1]
+  result[1] = allocCStringArray(strings)
+
+proc spawnPostgreActor*(turn: var Turn; root: Cap): Actor {.discardable.} =
+  spawn("postgre", turn) do (turn: var Turn):
+    during(turn, root, ?:PostgreArguments) do (params: StringPairs, ds: Cap):
+      var
+        conn: PGconn
+        statusHandle: Handle
+      block:
+        var (keys, vals) = splitParams(params)
+        conn = PQconnectdbParams(keys, vals, 0)
+        deallocCStringArray(keys)
+        deallocCStringArray(vals)
+        if conn.isNil: raise newException(OutOfMemDefect, "failed to alloc PQ connection")
+      let
+        status = PQstatus(conn)
+        msg = $PQerrorMessage(conn)
+      statusHandle = publish(turn, ds,
+          initRecord("status", toSymbol($status), msg.toPreserves))
+      if status == CONNECTION_OK:
+        discard
+    do:
+      PQfinish(conn)
+
+when isMainModule:
+  runActor("main") do (turn: var Turn; root: Cap):
+    connectStdio(turn, root)
+    spawnPostgreActor(turn, root)