generate full graph

Author: electron271

.gitignore

@@ -1,5 +1,6 @@
 docker-compose.yml
 data/graph.gexf
+data/graph-full.gexf
 postgres-data/
 .env
 node_modules/

docker-compose.yml.example

@@ -21,6 +21,8 @@ services:
     restart: unless-stopped
     ports:
       - 8080:8080 # this is a really common port and should be changed most likely to something else
+    depends_on:
+      - app
     volumes:
       - ./nginx.conf:/etc/nginx/nginx.conf
       - ./public/:/usr/share/nginx/html/public
@@ -34,6 +36,5 @@ services:
       - ./data/:/app/data
     depends_on:
       - db
-      - nginx
     environment:
       - DATABASE_URL=postgresql://postgres:slightlysecurepassword123@db:5432/postgres

nginx.conf

@@ -13,9 +13,14 @@ http {
       proxy_pass http://app:8000;
     }
 
-    # Serve static files in /data
+    # Serve static files in /data and block /data/graph-full.gexf as 403
     location /data {
       alias /usr/share/nginx/html/data;
     }
+
+    location /data/graph-full.gexf {
+      deny all;
+      return 403;
+    }
   }
 }

src/bot.js

@@ -101,6 +101,39 @@ async function generateGEXF() {
     }
 }
 
+async function generateFullGEXF() { // generate full graph with all users
+    await console.log('Generating FULL graph...');
+
+    // Create a new graph
+    const graph = new Graph();
+
+    // Fetch all mentions and user lookups
+    const mentions = await prisma.mention.findMany();
+    const userLookups = await prisma.userLookup.findMany();
+
+    // add nodes to the graph
+    userLookups.forEach((user) => {
+        graph.addNode(user.id, { label: user.username });
+    });
+
+    // add edges to the graph
+    mentions.forEach((mention) => {
+        if (graph.hasEdge(mention.user1Id, mention.user2Id)) {
+            graph.updateEdgeAttribute(mention.user1Id, mention.user2Id, 'weight', (w) => w + mention.count);
+        } else {
+            graph.addEdge(mention.user1Id, mention.user2Id, { weight: mention.count });
+        }
+    });
+
+    // Export the graph
+    try {
+        fs.writeFileSync('data/graph-full.gexf', gexf.write(graph));
+        console.log('Graph successfully exported to data/graph-full.gexf');
+    } catch (err) {
+        console.error('Error exporting graph:', err);
+    }
+}
+
 // On client ready
 client.once(Events.ClientReady, (readyClient) => {
     console.log(`Ready! Logged in as ${readyClient.user.tag}`);
@@ -249,6 +282,7 @@ client.on(Events.MessageCreate, async (message) => {
     if (mentionsCounter >= MENTIONS_THRESHOLD) {
         mentionsCounter = 0; // Reset counter
         await generateGEXF();
+        await generateFullGEXF();
     }
 });