From cc7157e9aba9bbe5d14b82167eec143e66ae0a41 Mon Sep 17 00:00:00 2001
From: Grant Linville <grant@acorn.io>
Date: Wed, 4 Dec 2024 16:31:23 -0500
Subject: [PATCH] enhance: add functions for daemon tools to do mTLS

Signed-off-by: Grant Linville <grant@acorn.io>
---
 src/gptscript.ts | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/src/gptscript.ts b/src/gptscript.ts
index d241cb0..bea6175 100644
--- a/src/gptscript.ts
+++ b/src/gptscript.ts
@@ -3,6 +3,7 @@ import path from "path"
 import child_process from "child_process"
 import {fileURLToPath} from "url"
 import {gunzipSync} from "zlib"
+import https from "https"
 
 export interface GlobalOpts {
     URL?: string
@@ -1283,3 +1284,48 @@ export interface DatasetElement {
     contents?: string
     binaryContents?: ArrayBuffer
 }
+
+// Functions for use in daemon tools:
+
+export function createServer(listener: http.RequestListener<typeof http.IncomingMessage, typeof http.ServerResponse>): https.Server {
+    const certB64 = process.env.CERT
+    const privateKeyB64 = process.env.PRIVATE_KEY
+    const gptscriptCertB64 = process.env.GPTSCRIPT_CERT
+
+    if (!certB64) {
+        console.log('Missing CERT env var')
+        process.exit(1)
+    } else if (!privateKeyB64) {
+        console.log('Missing PRIVATE_KEY env var')
+        process.exit(1)
+    } else if (!gptscriptCertB64) {
+        console.log('Missing GPTSCRIPT_CERT env var')
+        process.exit(1)
+    }
+
+    const cert = Buffer.from(certB64, 'base64').toString('utf-8')
+    const privateKey = Buffer.from(privateKeyB64, 'base64').toString('utf-8')
+    const gptscriptCert = Buffer.from(gptscriptCertB64, 'base64').toString('utf-8')
+
+    const options = {
+        key: privateKey,
+        cert: cert,
+        ca: gptscriptCert,
+        requestCert: true,
+        rejectUnauthorized: true,
+    }
+
+    return https.createServer(options, listener)
+}
+
+export function startServer(server: https.Server) {
+    const port = process.env.PORT
+    if (!port) {
+        console.log('Missing PORT env var')
+        process.exit(1)
+    }
+
+    server.listen(port, () => {
+        console.log(`Server listening on port ${port}`)
+    })
+}