astro-tractstack 2.0.11 → 2.0.13

package/templates/src/components/compositor/Compositor.tsx

@@ -387,12 +387,6 @@ export const Compositor = (props: CompositorProps) => {
       const ctx = getCtx(props);
       const range = $selection;
 
-      if ($selection.pendingAction === 'style') {
-        if (VERBOSE) console.log(LOG_PREFIX + 'useEffect acting on: style');
-        await ctx.wrapRangeInSpan(range as SelectionStoreState, 'span');
-        resetSelectionStore();
-      }
-
       if ($selection.pendingAction === 'link') {
         if (VERBOSE) console.log(LOG_PREFIX + 'useEffect acting on: link');
         const newAnchorNodeId = await ctx.wrapRangeInAnchor(
@@ -402,6 +396,20 @@ export const Compositor = (props: CompositorProps) => {
           ctx.handleInsertSignal('a', newAnchorNodeId);
         }
         resetSelectionStore();
+      } else if ($selection.pendingAction === 'style') {
+        if (VERBOSE) console.log(LOG_PREFIX + 'useEffect acting on: style');
+        const newSpanNodeId = await ctx.wrapRangeInSpan(
+          range as SelectionStoreState,
+          'span'
+        );
+        if (newSpanNodeId) {
+          settingsPanelStore.set({
+            action: 'style-element',
+            nodeId: newSpanNodeId,
+            expanded: true,
+          });
+        }
+        resetSelectionStore();
       }
       ctx.notifyNode('root');
     };

package/templates/src/components/compositor/Node.tsx

@@ -45,6 +45,8 @@ import type {
   BaseNode,
   FlatNode,
 } from '@/types/compositorTypes';
+import { handleClickEventDefault } from '@/utils/compositor/handleClickEvent';
+import { selectionStore } from '@/stores/selection';
 import type { NodeProps, SelectionOrigin } from '@/types/nodeProps';
 
 const VERBOSE = false;
@@ -153,6 +155,7 @@ const getElement = (
               nodeId={props.nodeId}
               ctx={getCtx(props)}
               isTemplate={isTemplate}
+              config={props.config!}
             />
           ) : (
             <>
@@ -213,6 +216,7 @@ const getElement = (
                       nodeId={node.id}
                       first={true}
                       ctx={getCtx(props)}
+                      config={props.config!}
                       isContextPane={true}
                     />
                   </PanelVisibilityWrapper>
@@ -240,7 +244,12 @@ const getElement = (
               panelType="add"
               ctx={getCtx(props)}
             >
-              <AddPanePanel nodeId={node.id} first={true} ctx={getCtx(props)} />
+              <AddPanePanel
+                nodeId={node.id}
+                first={true}
+                ctx={getCtx(props)}
+                config={props.config!}
+              />
             </PanelVisibilityWrapper>
           )}
           <div className="py-0.5">
@@ -264,7 +273,12 @@ const getElement = (
             panelType="add"
             ctx={getCtx(props)}
           >
-            <AddPanePanel nodeId={node.id} first={false} ctx={getCtx(props)} />
+            <AddPanePanel
+              nodeId={node.id}
+              first={false}
+              ctx={getCtx(props)}
+              config={props.config!}
+            />
           </PanelVisibilityWrapper>
         </>
       );
@@ -291,6 +305,31 @@ const getElement = (
           viewportKeyStore.get().value
         );
 
+        const handleElementClick = (e: MouseEvent<HTMLElement>) => {
+          // 1. ALWAYS stop the event from bubbling up to the Pane.
+          e.stopPropagation();
+
+          // 2. Check the selection store. The handleMouseUp in Compositor.tsx
+          // has already run by the time this 'click' event fires.
+          //
+          // - If it was a drag, Compositor.tsx set 'isActive: true' (line 267).
+          // - If it was a click, Compositor.tsx called 'resetSelectionStore'
+          //   (line 242), so 'isActive: false'.
+          //
+          const { isActive } = selectionStore.get();
+
+          if (isActive) {
+            // A drag just finished. The user's intent was to select text.
+            // Do NOT open the panel.
+            return;
+          }
+
+          // 3. 'isActive' was false. This was a genuine click.
+          // We can safely open the settings panel.
+          // 'node' is already in scope from the getElement function.
+          handleClickEventDefault(node as FlatNode, true);
+        };
+
         const handleMouseDown = (e: MouseEvent<HTMLElement>) => {
           if (VERBOSE)
             console.log('[Node.tsx] handleMouseDown FIRED', { event: e });
@@ -346,6 +385,7 @@ const getElement = (
           {
             className: className,
             onMouseDown: handleMouseDown,
+            onClick: handleElementClick,
             'data-node-id': node.id,
             style: { userSelect: 'none' },
           },

package/templates/src/components/compositor/nodes/Pane.tsx

@@ -294,11 +294,6 @@ const Pane = memo(
   (prevProps: NodeProps, nextProps: NodeProps) => {
     const isEqual =
       prevProps.nodeId === nextProps.nodeId && prevProps.ctx === nextProps.ctx;
-    if (!isEqual) {
-      console.log(
-        ` !! Pane rerender triggered by props change: ${prevProps.nodeId} -> ${nextProps.nodeId}`
-      );
-    }
     return isEqual;
   }
 );

package/templates/src/components/compositor/nodes/tagElements/NodeBasicTag.tsx

@@ -379,8 +379,9 @@ export const NodeBasicTag = (props: NodeTagProps) => {
           .filter(
             (childNode): childNode is FlatNode =>
               'tagName' in childNode &&
-              ['a', 'button'].includes(childNode.tagName as string)
+              ['a', 'button', 'span'].includes(childNode.tagName as string)
           ) as FlatNode[];
+        if (VERBOSE) console.log('originalNodes to save:', originalNodes);
 
         const parsedNodes = processRichTextToNodes(
           currentContent,

package/templates/src/components/edit/pane/AddPanePanel.tsx

@@ -5,8 +5,9 @@ import AddPaneNewPanel from './AddPanePanel_new';
 import AddPaneBreakPanel from './AddPanePanel_break';
 import AddPaneReUsePanel from './AddPanePanel_reuse';
 import AddPaneCodeHookPanel from './AddPanePanel_codehook';
-import { NodesContext, ROOT_NODE_NAME, getCtx } from '@/stores/nodes'; // Import getCtx
+import { NodesContext, ROOT_NODE_NAME, getCtx } from '@/stores/nodes';
 import { PaneAddMode } from '@/types/compositorTypes';
+import type { BrandConfig } from '@/types/tractstack';
 
 interface AddPanePanelProps {
   nodeId: string;
@@ -14,6 +15,7 @@ interface AddPanePanelProps {
   ctx?: NodesContext;
   isStoryFragment?: boolean;
   isContextPane?: boolean;
+  config?: BrandConfig;
 }
 
 const AddPanePanel = ({
@@ -22,6 +24,7 @@ const AddPanePanel = ({
   ctx,
   isStoryFragment = false,
   isContextPane = false,
+  config,
 }: AddPanePanelProps) => {
   const [reset, setReset] = useState(false);
   const lookup = first ? `${nodeId}-0` : nodeId;
@@ -62,6 +65,7 @@ const AddPanePanel = ({
           ctx={nodesCtx}
           isStoryFragment={isStoryFragment}
           isContextPane={isContextPane}
+          config={config!}
         />
       ) : mode === PaneAddMode.BREAK && !isContextPane ? (
         <AddPaneBreakPanel

package/templates/src/components/edit/pane/AddPanePanel_new.tsx

@@ -24,13 +24,11 @@ import {
   hasAssemblyAIStore,
 } from '@/stores/storykeep';
 import { templateCategories } from '@/utils/compositor/templateMarkdownStyles';
-import { AddPanePanel_newAICopy } from './AddPanePanel_newAICopy';
-import { AddPaneNewCopyMode, type CopyMode } from './AddPanePanel_newCopyMode';
+import { AiPaneGenerator } from './AiPaneGenerator';
 import { AddPaneNewCustomCopy } from './AddPanePanel_newCustomCopy';
-import { getTitleSlug } from '@/utils/aai/getTitleSlug';
-import { fullContentMapStore } from '@/stores/storykeep';
-import { themes, type Theme } from '@/types/tractstack';
-import { PaneAddMode } from '@/types/compositorTypes';
+import { themes, type Theme, type BrandConfig } from '@/types/tractstack';
+import { PaneAddMode, type TemplatePane } from '@/types/compositorTypes';
+import { useStore } from '@nanostores/react';
 
 interface AddPaneNewPanelProps {
   nodeId: string;
@@ -39,6 +37,7 @@ interface AddPaneNewPanelProps {
   ctx?: NodesContext;
   isStoryFragment?: boolean;
   isContextPane?: boolean;
+  config?: BrandConfig;
 }
 
 interface PreviewPane {
@@ -58,47 +57,47 @@ interface TemplateCategory {
 
 const ITEMS_PER_PAGE = 8;
 
+type Mode = 'template' | 'custom' | 'ai';
+
 const AddPaneNewPanel = ({
   nodeId,
   first,
-  setMode,
+  setMode: setParentMode, // Renamed prop to avoid conflict with local state
   ctx,
   isStoryFragment = false,
   isContextPane = false,
+  config,
 }: AddPaneNewPanelProps) => {
-  const brand = brandColourStore.get();
-  const hasAssemblyAI = hasAssemblyAIStore.get();
-  const [copyMode, setCopyMode] = useState<CopyMode>('design');
+  const brand = useStore(brandColourStore);
+  const hasAssemblyAI = useStore(hasAssemblyAIStore);
+  const [mode, setMode] = useState<Mode>('template'); // Local mode state
   const [customMarkdown, setCustomMarkdown] = useState<string>(`...`);
   const [previews, setPreviews] = useState<PreviewPane[]>([]);
   const [currentPage, setCurrentPage] = useState(0);
   const [renderedPages, setRenderedPages] = useState<Set<number>>(new Set());
   const [selectedTheme, setSelectedTheme] = useState<Theme>(
-    preferredThemeStore.get()
+    useStore(preferredThemeStore)
   );
   const [useOddVariant, setUseOddVariant] = useState(false);
   const [selectedCategory, setSelectedCategory] = useState<TemplateCategory>(
     templateCategories[isContextPane ? 1 : first ? 4 : 0]
   );
   const [isInserting, setIsInserting] = useState(false);
-  const [aiContentGenerated, setAiContentGenerated] = useState(false);
   const [fragmentsToGenerate, setFragmentsToGenerate] = useState<
     PanePreviewRequest[]
   >([]);
-  const shouldShowDesigns = copyMode !== 'ai' || aiContentGenerated;
 
   const categoryCollection = useMemo(() => {
-    const categories =
-      copyMode === `ai` || isContextPane
-        ? [templateCategories[1]]
-        : templateCategories;
+    const categories = isContextPane
+      ? [templateCategories[1]]
+      : templateCategories;
 
     return createListCollection({
       items: categories,
       itemToValue: (item) => item.id,
       itemToString: (item) => item.title,
     });
-  }, [copyMode, isContextPane]);
+  }, [isContextPane]);
 
   const themesCollection = useMemo(() => {
     return createListCollection({
@@ -109,38 +108,25 @@ const AddPaneNewPanel = ({
   }, []);
 
   const filteredTemplates = useMemo(() => {
-    if (copyMode === `ai` || isContextPane)
-      return templateCategories[1].getTemplates(
-        selectedTheme,
-        brand,
-        useOddVariant
-      );
-    if (isContextPane)
-      return templateCategories[1].getTemplates(
-        selectedTheme,
-        brand,
-        useOddVariant
-      );
     return selectedCategory.getTemplates(selectedTheme, brand, useOddVariant);
-  }, [selectedTheme, useOddVariant, selectedCategory, copyMode, isContextPane]);
+  }, [selectedTheme, useOddVariant, selectedCategory, brand]);
 
   useEffect(() => {
-    if (copyMode !== 'ai') setAiContentGenerated(false);
-    if (copyMode !== 'ai' || isContextPane)
-      setSelectedCategory(templateCategories[first ? 4 : 0]);
-  }, [copyMode, first, isContextPane]);
-
-  const handleAiContentGenerated = (content: string) => {
-    setCustomMarkdown(content);
-    setAiContentGenerated(true);
-  };
+    if (isContextPane) {
+      setSelectedCategory(templateCategories[1]);
+    } else if (first) {
+      setSelectedCategory(templateCategories[4]);
+    } else {
+      setSelectedCategory(templateCategories[0]);
+    }
+  }, [isContextPane, first]);
 
   useEffect(() => {
     const newPreviews = filteredTemplates.map((template, index: number) => {
-      const ctx = new NodesContext();
-      ctx.addNode(createEmptyStorykeep('tmp'));
+      const previewCtx = new NodesContext();
+      previewCtx.addNode(createEmptyStorykeep('tmp'));
       const thisTemplate =
-        copyMode === 'custom' || (copyMode === 'ai' && aiContentGenerated)
+        mode === 'custom'
           ? {
               ...template,
               markdown: template.markdown && {
@@ -149,13 +135,13 @@ const AddPaneNewPanel = ({
               },
             }
           : template;
-      ctx.addTemplatePane('tmp', thisTemplate);
-      return { ctx, template: thisTemplate, index };
+      previewCtx.addTemplatePane('tmp', thisTemplate);
+      return { ctx: previewCtx, template: thisTemplate, index };
     });
     setPreviews(newPreviews);
     setCurrentPage(0);
     setRenderedPages(new Set());
-  }, [filteredTemplates, customMarkdown, copyMode, aiContentGenerated]);
+  }, [filteredTemplates, customMarkdown, mode]);
 
   const totalPages = Math.ceil(previews.length / ITEMS_PER_PAGE);
 
@@ -223,81 +209,47 @@ const AddPaneNewPanel = ({
     });
   };
 
-  const handleTemplateInsert = async (
-    template: any,
-    nodeId: string,
-    first: boolean
-  ) => {
-    if (isInserting) return;
+  const handleApplyTemplate = async (template: any) => {
+    if (isInserting || !ctx) return;
     setIsInserting(true);
 
     try {
-      if (ctx) {
-        const hasMarkdownContent =
-          template?.markdown?.markdownBody &&
-          template.markdown.markdownBody.trim() !== '...' &&
-          template.markdown.markdownBody.trim().length > 0;
-
-        const insertTemplate = [`blank`, `custom`].includes(copyMode)
+      const insertTemplate =
+        mode === 'custom'
           ? {
               ...cloneDeep(template),
               markdown: template.markdown && {
                 ...template.markdown,
-                markdownBody: copyMode === `blank` ? `...` : customMarkdown,
+                markdownBody: customMarkdown,
               },
             }
           : cloneDeep(template);
 
-        const markdownContent = [`blank`].includes(copyMode)
-          ? null
-          : copyMode === `custom`
-            ? customMarkdown
-            : insertTemplate?.markdown?.markdownBody;
-
-        insertTemplate.title = '';
-        insertTemplate.slug = '';
-
-        if (
-          copyMode === `ai` &&
-          hasAssemblyAI &&
-          markdownContent &&
-          hasMarkdownContent
-        ) {
-          const existingSlugs = fullContentMapStore
-            .get()
-            .filter((item) => ['Pane', 'StoryFragment'].includes(item.type))
-            .map((item) => item.slug);
-
-          const titleSlugResult = await getTitleSlug(
-            markdownContent,
-            existingSlugs
-          );
-
-          if (titleSlugResult) {
-            insertTemplate.title = titleSlugResult.title;
-            insertTemplate.slug = titleSlugResult.slug;
-          }
-        }
-
-        const ownerId =
-          isStoryFragment || isContextPane
-            ? nodeId
-            : ctx.getClosestNodeTypeFromId(nodeId, 'StoryFragment');
-
-        if (isContextPane) {
-          insertTemplate.isContextPane = true;
-          ctx.addContextTemplatePane(ownerId, insertTemplate);
-        } else {
-          const newPaneId = ctx.addTemplatePane(
+      insertTemplate.title = '';
+      insertTemplate.slug = '';
+
+      const ownerId =
+        isStoryFragment || isContextPane
+          ? nodeId
+          : ctx.getClosestNodeTypeFromId(nodeId, 'StoryFragment');
+
+      if (isContextPane) {
+        insertTemplate.isContextPane = true;
+        await ctx.applyAtomicUpdate(async (tmpCtx) => {
+          tmpCtx.addContextTemplatePane(ownerId, insertTemplate);
+        });
+      } else {
+        await ctx.applyAtomicUpdate(async (tmpCtx) => {
+          tmpCtx.addTemplatePane(
             ownerId,
             insertTemplate,
             nodeId,
             first ? 'before' : 'after'
           );
-          if (newPaneId) ctx.notifyNode(`root`);
-        }
-        setMode(PaneAddMode.DEFAULT, false);
+        });
+        ctx.notifyNode(`root`);
       }
+      setParentMode(PaneAddMode.DEFAULT, false);
     } catch (error) {
       console.error('Error inserting template:', error);
     } finally {
@@ -305,6 +257,39 @@ const AddPaneNewPanel = ({
     }
   };
 
+  const handleApplyGeneratedPane = async (pane: TemplatePane) => {
+    if (isInserting || !ctx) return;
+    setIsInserting(true);
+    try {
+      const ownerId =
+        isStoryFragment || isContextPane
+          ? nodeId
+          : ctx.getClosestNodeTypeFromId(nodeId, 'StoryFragment');
+
+      if (isContextPane) {
+        pane.isContextPane = true;
+        await ctx.applyAtomicUpdate(async (tmpCtx) => {
+          tmpCtx.addContextTemplatePane(ownerId, pane);
+        });
+      } else {
+        await ctx.applyAtomicUpdate(async (tmpCtx) => {
+          tmpCtx.addTemplatePane(
+            ownerId,
+            pane,
+            nodeId,
+            first ? 'before' : 'after'
+          );
+        });
+        ctx.notifyNode(`root`);
+      }
+      setParentMode(PaneAddMode.DEFAULT, false);
+    } catch (error) {
+      console.error('Error applying generated pane:', error);
+    } finally {
+      setIsInserting(false);
+    }
+  };
+
   const handleThemeChange = (details: { value: string[] }) => {
     const newTheme = details.value[0] as Theme;
     if (newTheme) {
@@ -338,8 +323,9 @@ const AddPaneNewPanel = ({
       <style>{customStyles}</style>
       <div className="group flex w-full gap-1 rounded-md bg-white p-1.5">
         <button
-          onClick={() => setMode(PaneAddMode.DEFAULT, first)}
+          onClick={() => setParentMode(PaneAddMode.DEFAULT, first)}
           className="w-fit rounded bg-gray-100 px-3 py-1 text-sm text-gray-700 transition-colors hover:bg-gray-200 focus:bg-gray-200"
+          type="button"
         >
           ← Go Back
         </button>
@@ -348,11 +334,45 @@ const AddPaneNewPanel = ({
           <div className="font-action flex-none rounded px-2 py-2.5 text-sm font-bold text-cyan-700 shadow-sm">
             + Design New Pane
           </div>
+          <div className="flex items-center space-x-2 rounded-lg bg-gray-100 p-1">
+            <button
+              onClick={() => setMode('template')}
+              className={`rounded-md px-3 py-1 text-sm font-medium transition-colors ${
+                mode === 'template'
+                  ? 'bg-white text-cyan-700 shadow'
+                  : 'text-gray-600 hover:text-gray-800'
+              }`}
+              type="button"
+            >
+              Use Template
+            </button>
+            <button
+              onClick={() => setMode('custom')}
+              className={`rounded-md px-3 py-1 text-sm font-medium transition-colors ${
+                mode === 'custom'
+                  ? 'bg-white text-cyan-700 shadow'
+                  : 'text-gray-600 hover:text-gray-800'
+              }`}
+              type="button"
+            >
+              Paste Markdown
+            </button>
+            {hasAssemblyAI && (
+              <button
+                onClick={() => setMode('ai')}
+                className={`rounded-md px-3 py-1 text-sm font-medium transition-colors ${
+                  mode === 'ai'
+                    ? 'rounded-md border border-transparent bg-cyan-600 px-3 py-1.5 text-sm font-bold text-white shadow-sm hover:bg-cyan-700 focus:outline-none focus:ring-2 focus:ring-cyan-500 focus:ring-offset-2'
+                    : 'text-gray-600 hover:text-gray-800'
+                }`}
+                type="button"
+              >
+                ✨ Generate with AI
+              </button>
+            )}
+          </div>
 
-          {!(copyMode === 'ai' && aiContentGenerated) && (
-            <AddPaneNewCopyMode selected={copyMode} onChange={setCopyMode} />
-          )}
-          {copyMode === 'custom' && (
+          {mode === 'custom' && (
             <div className="mt-4 w-full">
               <AddPaneNewCustomCopy
                 value={customMarkdown}
@@ -360,18 +380,20 @@ const AddPaneNewPanel = ({
               />
             </div>
           )}
-          {copyMode === 'ai' && !aiContentGenerated && (
+          {mode === 'ai' && (
             <div className="mt-4 w-full">
-              <AddPanePanel_newAICopy
-                onChange={handleAiContentGenerated}
-                isContextPane={isContextPane}
+              <AiPaneGenerator
+                ownerId={nodeId}
+                onComplete={handleApplyGeneratedPane}
+                onCancel={() => setMode('template')}
+                config={config!}
               />
             </div>
           )}
         </div>
       </div>
 
-      {shouldShowDesigns && (
+      {mode !== 'ai' && (
         <>
           <h3 className="font-action px-3.5 pb-1.5 pt-4 text-xl font-bold text-black">
             1. Template design settings
@@ -521,8 +543,7 @@ const AddPaneNewPanel = ({
                   onClick={
                     isInserting
                       ? undefined
-                      : () =>
-                          handleTemplateInsert(preview.template, nodeId, first)
+                      : () => handleApplyTemplate(preview.template)
                   }
                   className={`bg-mywhite group relative w-full rounded-sm shadow-inner ${
                     isInserting
@@ -588,6 +609,7 @@ const AddPaneNewPanel = ({
               onClick={() => handlePageChange(currentPage - 1)}
               disabled={currentPage === 0}
               className="rounded bg-gray-100 px-3 py-1 text-sm text-gray-700 transition-colors hover:bg-gray-200 focus:bg-gray-200 disabled:cursor-not-allowed disabled:opacity-50"
+              type="button"
             >
               Previous
             </button>
@@ -601,6 +623,7 @@ const AddPaneNewPanel = ({
                       ? 'bg-cyan-700 text-white'
                       : 'bg-gray-100 text-gray-700 hover:bg-gray-200'
                   }`}
+                  type="button"
                 >
                   {index + 1}
                 </button>
@@ -610,6 +633,7 @@ const AddPaneNewPanel = ({
               onClick={() => handlePageChange(currentPage + 1)}
               disabled={currentPage === totalPages - 1}
               className="rounded bg-gray-100 px-3 py-1 text-sm text-gray-700 transition-colors hover:bg-gray-200 focus:bg-gray-200 disabled:cursor-not-allowed disabled:opacity-50"
+              type="button"
             >
               Next
             </button>