Add code to submit image generation instructions to the server

[zombocom-ai] / index.js

diff --git a/index.js b/index.js
index 80526a3f6c879e14fb4a4055fe28504372f0d489..249537fd3414cd2541b16513901ab6eaeb68359a 100644 (file)
--- a/index.js
+++ b/index.js
@@ -1,3 +1,5 @@
+const fs = require('fs');
+
 const express = require('express');
 const app = express();
 const http = require('http');
@@ -6,22 +8,56 @@ const { Server } = require("socket.io");
 const io = new Server(server);
 const port = 2122;
 
-comments = [];
+const state_file = 'zombocom-state.json'
+
+var comments;
+
+// Load comments at server startup
+fs.readFile(state_file, (err, data) => {
+    if (err)
+        comments = [];
+    else
+        comments = JSON.parse(data);
+});
+
+// Save comments when server is shutting down
+function cleanup() {
+    fs.writeFileSync('zombocom-state.json', JSON.stringify(comments), (error) => {
+        if (error)
+            throw error;
+    })
+}
+
+// And connect to that on either clean exit...
+process.on('exit', cleanup);
+
+// ... or on a SIGINT (control-C)
+process.on('SIGINT', () => {
+    cleanup();
+    process.exit();
+});
 
 app.get('/index.html', (req, res) => {
     res.sendFile(__dirname + '/index.html');
 });
 
 io.on('connection', (socket) => {
+
     // Replay old comments to a newly-joining client
     comments.forEach((comment) => {
         socket.emit('comment', comment)
     });
+
     // When any client comments, send that to all clients (including sender)
     socket.on('comment', (comment) => {
         io.emit('comment', comment);
         comments.push(comment);
     });
+
+    // Generate an image when requested
+    socket.on('generate', (request) => {
+        console.log(`Generating image with code=${request['code']} and prompt=${request['prompt']}`);
+    });
 });
 
 server.listen(port, () => {