From 087a1095c834749dc708cd8c1c6ab101235573fa Mon Sep 17 00:00:00 2001
From: Emir Karabeg <emirkarabeg@berkeley.edu>
Date: Thu, 30 Jan 2025 18:54:23 -0800
Subject: [PATCH] Tag works on long input

---
 .../sub-block/components/long-input.tsx       | 95 +++++++++++++++++--
 .../components/sub-block/sub-block.tsx        |  1 +
 2 files changed, 89 insertions(+), 7 deletions(-)

diff --git a/app/w/components/workflow-block/components/sub-block/components/long-input.tsx b/app/w/components/workflow-block/components/sub-block/components/long-input.tsx
index 60ed7c7857..385b256a5a 100644
--- a/app/w/components/workflow-block/components/sub-block/components/long-input.tsx
+++ b/app/w/components/workflow-block/components/sub-block/components/long-input.tsx
@@ -1,26 +1,107 @@
 import { Textarea } from '@/components/ui/textarea'
 import { useSubBlockValue } from '../hooks/use-sub-block-value'
+import { cn } from '@/lib/utils'
+import { useState, useRef, useEffect } from 'react'
 
 interface LongInputProps {
   placeholder?: string
   blockId: string
   subBlockId: string
+  isConnecting: boolean
 }
 
 export function LongInput({
   placeholder,
   blockId,
   subBlockId,
+  isConnecting,
 }: LongInputProps) {
   const [value, setValue] = useSubBlockValue(blockId, subBlockId)
+  const textareaRef = useRef<HTMLTextAreaElement>(null)
+  const overlayRef = useRef<HTMLDivElement>(null)
+
+  // Sync scroll position between textarea and overlay
+  const handleScroll = (e: React.UIEvent<HTMLTextAreaElement>) => {
+    if (overlayRef.current) {
+      overlayRef.current.scrollTop = e.currentTarget.scrollTop
+      overlayRef.current.scrollLeft = e.currentTarget.scrollLeft
+    }
+  }
+
+  // Drag and Drop handlers
+  const handleDrop = (e: React.DragEvent<HTMLTextAreaElement>) => {
+    e.preventDefault()
+    try {
+      const data = JSON.parse(e.dataTransfer.getData('application/json'))
+
+      const isValidConnectionBlock =
+        data.type === 'connectionBlock' &&
+        data.connectionData.sourceBlockId === blockId
+
+      if (!isValidConnectionBlock) return
+
+      const currentValue = value?.toString() ?? ''
+      const connectionName = data.connectionData.name
+        .replace(/\s+/g, '')
+        .toLowerCase()
+      const outputSuffix =
+        data.connectionData.outputType === 'any'
+          ? 'res'
+          : data.connectionData.outputType
+
+      const newValue = `${currentValue}<${connectionName}.${outputSuffix}>`
+      setValue(newValue)
+    } catch (error) {
+      console.error('Failed to parse drop data:', error)
+    }
+  }
+
+  const handleDragOver = (e: React.DragEvent<HTMLTextAreaElement>) => {
+    e.preventDefault()
+  }
+
+  const formatDisplayText = (text: string) => {
+    if (!text) return null
+
+    // Split the text by tag pattern <something.something>
+    const parts = text.split(/(<[^>]+>)/g)
+
+    return parts.map((part, index) => {
+      // Check if the part matches tag pattern
+      if (part.match(/^<[^>]+>$/)) {
+        return (
+          <span key={index} className="text-blue-500">
+            {part}
+          </span>
+        )
+      }
+      return <span key={index}>{part}</span>
+    })
+  }
 
   return (
-    <Textarea
-      className="w-full resize-none placeholder:text-muted-foreground/50 allow-scroll"
-      rows={4}
-      placeholder={placeholder ?? ''}
-      value={value?.toString() ?? ''}
-      onChange={(e) => setValue(e.target.value)}
-    />
+    <div className="relative w-full">
+      <Textarea
+        ref={textareaRef}
+        className={cn(
+          'w-full resize-none placeholder:text-muted-foreground/50 allow-scroll text-transparent caret-foreground',
+          isConnecting &&
+            'focus-visible:ring-blue-500 ring-2 ring-blue-500 ring-offset-2'
+        )}
+        rows={4}
+        placeholder={placeholder ?? ''}
+        value={value?.toString() ?? ''}
+        onChange={(e) => setValue(e.target.value)}
+        onDrop={handleDrop}
+        onDragOver={handleDragOver}
+        onScroll={handleScroll}
+      />
+      <div
+        ref={overlayRef}
+        className="absolute inset-0 pointer-events-none px-3 py-2 overflow-auto whitespace-pre-wrap scrollbar-none text-sm bg-transparent"
+      >
+        {formatDisplayText(value?.toString() ?? '')}
+      </div>
+    </div>
   )
 }
diff --git a/app/w/components/workflow-block/components/sub-block/sub-block.tsx b/app/w/components/workflow-block/components/sub-block/sub-block.tsx
index 27d4249c4b..c72329c9bb 100644
--- a/app/w/components/workflow-block/components/sub-block/sub-block.tsx
+++ b/app/w/components/workflow-block/components/sub-block/sub-block.tsx
@@ -37,6 +37,7 @@ export function SubBlock({ blockId, config, isConnecting }: SubBlockProps) {
             blockId={blockId}
             subBlockId={config.id}
             placeholder={config.placeholder}
+            isConnecting={isConnecting}
           />
         )
       case 'dropdown':