feat: improve graph structure display

Author: Planeshifter

lib/repl.js

@@ -305,6 +305,78 @@ async function showTextOutput( format, replState, prefix='', suffix='' ) {
  *     when autoshow is not displayed, this just returns baseOutput.
  *
  */
+/**
+ * Formats graph output with nice styling and colors
+ */
+function formatGraphOutput(graphText) {
+    if (!graphText) return '';
+
+    const lines = graphText.split('\n');
+    const styledLines = [];
+
+    for (let line of lines) {
+        if (line.includes('--') || line.includes('->')) {
+            // This is a node connection line
+            const parts = line.split(/(\s+--\s+|\s+->\s+)/);
+            if (parts.length >= 3) {
+                const nodeName = parts[0];
+                const connector = parts[1];
+                const connections = parts[2];
+
+                // Style the node name and connections
+                const styledNode = chalk.cyan.bold(nodeName);
+                const styledConnector = chalk.gray(connector);
+                const styledConnections = connections.split(', ')
+                    .map(conn => chalk.green(conn.trim()))
+                    .join(chalk.gray(', '));
+
+                styledLines.push(`  ${styledNode} ${styledConnector} ${styledConnections}`);
+            } else {
+                styledLines.push(`  ${line}`);
+            }
+        } else if (line.includes(':') && (line.includes('color') || line.includes('shape') || line.includes('size'))) {
+            // This is a node property line (like "A: color "red"")
+            const parts = line.split(':');
+            if (parts.length >= 2) {
+                const nodeName = parts[0].trim();
+                const properties = parts[1].trim();
+
+                const styledNode = chalk.cyan.bold(nodeName);
+                const styledProperties = chalk.yellow(properties);
+
+                styledLines.push(`  ${styledNode}: ${styledProperties}`);
+            } else {
+                styledLines.push(`  ${line}`);
+            }
+        } else if (line.includes('This is')) {
+            // This is the graph classification line
+            styledLines.push(`\n${chalk.italic.gray(line)}`);
+        } else if (line.trim() === '') {
+            // Empty line - add some spacing
+            styledLines.push('');
+        } else if (line.trim()) {
+            // Other non-empty lines
+            styledLines.push(`  ${line}`);
+        }
+    }
+
+    // Create a nice box around the graph
+    const graphTitle = chalk.bold.blue('📊 Current Graph Structure');
+    const boxedGraph = boxen(
+        styledLines.join('\n'),
+        {
+            padding: 1,
+            margin: 1,
+            borderStyle: 'round',
+            borderColor: 'blue',
+            title: graphTitle,
+            titleAlignment: 'center'
+        }
+    );
+
+    return boxedGraph;
+}
+
 async function autoshowText( replState, baseOutput='', { prefix='', beginHistoryIndex, endHistoryIndex } = {} ) {
     const textFormats = [ 'tex', 'latex', 'json' ];
     const summaryFormats = [ 'auto', 'graph', 'nodes', 'edges', 'decorations', 'constraints', 'data' ];
@@ -319,9 +391,17 @@ async function autoshowText( replState, baseOutput='', { prefix='', beginHistory
             aggregatedDiff = merge( aggregatedDiff, JSON.parse( replState.history[i].diff ) );
         }
         const summarizer = autoSummarizer( aggregatedDiff );
-        autoshow = init + summarizer(replState.graph) + '\n';
+        const rawOutput = summarizer(replState.graph);
+
+        // Always include the full graph structure for context
+        const graphStructure = summarize('graph', replState.graph);
+        const combinedOutput = rawOutput.includes('--') || rawOutput.includes('->') ?
+            rawOutput : `${graphStructure}\n\n${rawOutput}`;
+
+        autoshow = init + formatGraphOutput(combinedOutput) + '\n';
     } else if ( replState.config.autoshow && summaryFormats.includes(replState.config.autoshow) ) {
-        autoshow = init + summarize( replState.config.autoshow, replState.graph) + '\n';
+        const rawOutput = summarize( replState.config.autoshow, replState.graph);
+        autoshow = init + formatGraphOutput(rawOutput) + '\n';
     } else if ( replState.config.autoshow && textFormats.includes(replState.config.autoshow) ) {
         autoshow = init + showTextOutput( replState.config.autoshow, replState ) + '\n';
     }