From 10b658adfe54867575ca603293f2b707fff36847 Mon Sep 17 00:00:00 2001
From: Kaitlyn Michael <kaitlyn.michael@redis.com>
Date: Wed, 22 Oct 2025 09:27:25 -0500
Subject: [PATCH] enhanced copy button functionality

---
 layouts/partials/scripts.html | 78 +++++++++++++++++++++++++++++++-
 static/js/agent-builder.js    | 84 ++++++++++++++++++++++++++++++++---
 static/js/codetabs.js         | 81 +++++++++++++++++++++++++++++++--
 3 files changed, 232 insertions(+), 11 deletions(-)

diff --git a/layouts/partials/scripts.html b/layouts/partials/scripts.html
index 919be752b..57b56aa5c 100644
--- a/layouts/partials/scripts.html
+++ b/layouts/partials/scripts.html
@@ -1,5 +1,76 @@
 <!-- Code for copy to clipboard button -->
 <script>
+    // Smart copy processing function to handle shell prompts
+    function smartCopyProcessing(text) {
+        const lines = text.split('\n');
+        const processedLines = [];
+        let inMultiLineCommand = false;
+        let currentCommand = '';
+
+        // Regex patterns for different shell prompts
+        const promptPatterns = [
+            /^\s*\$\s+(.+)$/,                    // $ command
+            /^\s*#\s+(.+)$/,                     // # command (root)
+            /^\s*[\w.-]+>\s+(.+)$/,              // redis-cli>, rladmin>, etc.
+            /^\s*[\d.:]+>\s+(.+)$/,              // 127.0.0.1:6379> command
+        ];
+
+        for (let line of lines) {
+            let isPromptLine = false;
+            let commandPart = '';
+
+            // Check if this line matches any prompt pattern
+            for (let pattern of promptPatterns) {
+                const match = line.match(pattern);
+                if (match) {
+                    isPromptLine = true;
+                    commandPart = match[1];
+                    break;
+                }
+            }
+
+            if (isPromptLine) {
+                // Handle multi-line commands with backslash continuation
+                if (commandPart.endsWith('\\')) {
+                    inMultiLineCommand = true;
+                    currentCommand = commandPart.slice(0, -1).trim() + ' ';
+                } else {
+                    if (inMultiLineCommand) {
+                        // End of multi-line command
+                        currentCommand += commandPart;
+                        processedLines.push(currentCommand.trim());
+                        inMultiLineCommand = false;
+                        currentCommand = '';
+                    } else {
+                        // Single line command
+                        processedLines.push(commandPart);
+                    }
+                }
+            } else if (inMultiLineCommand) {
+                // Continuation line of a multi-line command
+                const trimmedLine = line.trim();
+                if (trimmedLine.endsWith('\\')) {
+                    currentCommand += trimmedLine.slice(0, -1).trim() + ' ';
+                } else {
+                    currentCommand += trimmedLine;
+                    processedLines.push(currentCommand.trim());
+                    inMultiLineCommand = false;
+                    currentCommand = '';
+                }
+            } else {
+                // Regular line (output, comments, etc.) - copy as-is
+                processedLines.push(line);
+            }
+        }
+
+        // Handle case where multi-line command was not completed
+        if (inMultiLineCommand && currentCommand) {
+            processedLines.push(currentCommand.trim());
+        }
+
+        return processedLines.join('\n');
+    }
+
     document.addEventListener("DOMContentLoaded", function () {
 
         // Add copy buttons to code blocks (but skip those inside codetabs)
@@ -48,9 +119,12 @@
                 // Handle copy logic
                 button.addEventListener('click', () => {
                     const lines = block.querySelectorAll('span.cl');
-                    const text = Array.from(lines).map(line => line.textContent).join('');
+                    const rawText = Array.from(lines).map(line => line.textContent).join('');
+
+                    // Smart copy: detect shell prompts and extract commands
+                    const processedText = smartCopyProcessing(rawText);
 
-                    navigator.clipboard.writeText(text).then(() => {
+                    navigator.clipboard.writeText(processedText).then(() => {
                         tooltip.style.display = 'block';
                         setTimeout(() => {
                             tooltip.style.display = 'none';
diff --git a/static/js/agent-builder.js b/static/js/agent-builder.js
index 0acef6603..4d0a4d46d 100644
--- a/static/js/agent-builder.js
+++ b/static/js/agent-builder.js
@@ -823,27 +823,101 @@ public class ${formData.agentName.replace(/\s+/g, '')}
         return `${cleanName}_agent${extensions[formData.programmingLanguage]}`;
     }
 
+    // Smart copy processing function to handle shell prompts
+    function smartCopyProcessing(text) {
+        const lines = text.split('\n');
+        const processedLines = [];
+        let inMultiLineCommand = false;
+        let currentCommand = '';
+
+        // Regex patterns for different shell prompts
+        const promptPatterns = [
+            /^\s*\$\s+(.+)$/,                    // $ command
+            /^\s*#\s+(.+)$/,                     // # command (root)
+            /^\s*[\w.-]+>\s+(.+)$/,              // redis-cli>, rladmin>, etc.
+            /^\s*[\d.:]+>\s+(.+)$/,              // 127.0.0.1:6379> command
+        ];
+
+        for (let line of lines) {
+            let isPromptLine = false;
+            let commandPart = '';
+
+            // Check if this line matches any prompt pattern
+            for (let pattern of promptPatterns) {
+                const match = line.match(pattern);
+                if (match) {
+                    isPromptLine = true;
+                    commandPart = match[1];
+                    break;
+                }
+            }
+
+            if (isPromptLine) {
+                // Handle multi-line commands with backslash continuation
+                if (commandPart.endsWith('\\')) {
+                    inMultiLineCommand = true;
+                    currentCommand = commandPart.slice(0, -1).trim() + ' ';
+                } else {
+                    if (inMultiLineCommand) {
+                        // End of multi-line command
+                        currentCommand += commandPart;
+                        processedLines.push(currentCommand.trim());
+                        inMultiLineCommand = false;
+                        currentCommand = '';
+                    } else {
+                        // Single line command
+                        processedLines.push(commandPart);
+                    }
+                }
+            } else if (inMultiLineCommand) {
+                // Continuation line of a multi-line command
+                const trimmedLine = line.trim();
+                if (trimmedLine.endsWith('\\')) {
+                    currentCommand += trimmedLine.slice(0, -1).trim() + ' ';
+                } else {
+                    currentCommand += trimmedLine;
+                    processedLines.push(currentCommand.trim());
+                    inMultiLineCommand = false;
+                    currentCommand = '';
+                }
+            } else {
+                // Regular line (output, comments, etc.) - copy as-is
+                processedLines.push(line);
+            }
+        }
+
+        // Handle case where multi-line command was not completed
+        if (inMultiLineCommand && currentCommand) {
+            processedLines.push(currentCommand.trim());
+        }
+
+        return processedLines.join('\n');
+    }
+
     function copyCode() {
         // Get the raw code from the dataset (not the highlighted version)
-        const code = elements.codeSection.dataset.code;
+        const rawCode = elements.codeSection.dataset.code;
         const copyBtn = document.getElementById('copy-code-btn');
 
-        if (!code) {
+        if (!rawCode) {
             alert('No code available to copy');
             return;
         }
 
+        // Process the code with smart copy functionality
+        const processedCode = smartCopyProcessing(rawCode);
+
         // Try to copy to clipboard
         if (navigator.clipboard && navigator.clipboard.writeText) {
-            navigator.clipboard.writeText(code).then(() => {
+            navigator.clipboard.writeText(processedCode).then(() => {
                 showCopyFeedback(copyBtn, true);
             }).catch(err => {
                 console.error('Failed to copy to clipboard:', err);
-                fallbackCopyToClipboard(code, copyBtn);
+                fallbackCopyToClipboard(processedCode, copyBtn);
             });
         } else {
             // Fallback for older browsers
-            fallbackCopyToClipboard(code, copyBtn);
+            fallbackCopyToClipboard(processedCode, copyBtn);
         }
     }
 
diff --git a/static/js/codetabs.js b/static/js/codetabs.js
index e2dbfc6f9..d4e5108e5 100644
--- a/static/js/codetabs.js
+++ b/static/js/codetabs.js
@@ -1,9 +1,81 @@
 // TODO: URI-able tabs
 
+// Smart copy processing function to handle shell prompts
+function smartCopyProcessing(text) {
+    const lines = text.split('\n');
+    const processedLines = [];
+    let inMultiLineCommand = false;
+    let currentCommand = '';
+
+    // Regex patterns for different shell prompts
+    const promptPatterns = [
+        /^\s*\$\s+(.+)$/,                    // $ command
+        /^\s*#\s+(.+)$/,                     // # command (root)
+        /^\s*[\w.-]+>\s+(.+)$/,              // redis-cli>, rladmin>, etc.
+        /^\s*[\d.:]+>\s+(.+)$/,              // 127.0.0.1:6379> command
+    ];
+
+    for (let line of lines) {
+        let isPromptLine = false;
+        let commandPart = '';
+
+        // Check if this line matches any prompt pattern
+        for (let pattern of promptPatterns) {
+            const match = line.match(pattern);
+            if (match) {
+                isPromptLine = true;
+                commandPart = match[1];
+                break;
+            }
+        }
+
+        if (isPromptLine) {
+            // Handle multi-line commands with backslash continuation
+            if (commandPart.endsWith('\\')) {
+                inMultiLineCommand = true;
+                currentCommand = commandPart.slice(0, -1).trim() + ' ';
+            } else {
+                if (inMultiLineCommand) {
+                    // End of multi-line command
+                    currentCommand += commandPart;
+                    processedLines.push(currentCommand.trim());
+                    inMultiLineCommand = false;
+                    currentCommand = '';
+                } else {
+                    // Single line command
+                    processedLines.push(commandPart);
+                }
+            }
+        } else if (inMultiLineCommand) {
+            // Continuation line of a multi-line command
+            const trimmedLine = line.trim();
+            if (trimmedLine.endsWith('\\')) {
+                currentCommand += trimmedLine.slice(0, -1).trim() + ' ';
+            } else {
+                currentCommand += trimmedLine;
+                processedLines.push(currentCommand.trim());
+                inMultiLineCommand = false;
+                currentCommand = '';
+            }
+        } else {
+            // Regular line (output, comments, etc.) - copy as-is
+            processedLines.push(line);
+        }
+    }
+
+    // Handle case where multi-line command was not completed
+    if (inMultiLineCommand && currentCommand) {
+        processedLines.push(currentCommand.trim());
+    }
+
+    return processedLines.join('\n');
+}
+
 function copyCodeToClipboard(panelId) {
   // Get the last <code>, path depends on highlighter options
-  const code = [...document.querySelectorAll(`${panelId} code`)].pop().textContent;
-  navigator.clipboard.writeText(code);
+  const rawCode = [...document.querySelectorAll(`${panelId} code`)].pop().textContent;
+  const processedCode = smartCopyProcessing(rawCode);
+  navigator.clipboard.writeText(processedCode);
 
   // Toggle tooltip
   const tooltip = document.querySelector(`${panelId} .tooltiptext`);
@@ -22,8 +94,9 @@ function copyCodeToClipboardForCodetabs(button) {
   if (!visiblePanel) return;
 
   // Get the code from the visible panel
-  const code = [...visiblePanel.querySelectorAll('code')].pop().textContent;
-  navigator.clipboard.writeText(code);
+  const rawCode = [...visiblePanel.querySelectorAll('code')].pop().textContent;
+  const processedCode = smartCopyProcessing(rawCode);
+  navigator.clipboard.writeText(processedCode);
 
   // Toggle tooltip
   const tooltip = button.querySelector('.tooltiptext');