Merge pull request #259 from beNative/tisi/add-visual/source-editor-toggle-for-new-html-documents

Ensure HTML language enables rich text editing

Author: beNative

App.tsx

@@ -2119,7 +2119,21 @@ export const MainApp: React.FC = () => {
     const handleLanguageChange = useCallback((newLanguage: string) => {
         if (activeNodeId && activeNode?.type === 'document') {
             addLog('INFO', `User action: Change language for document "${activeNode?.title}" to "${newLanguage}".`);
-            updateItem(activeNodeId, { language_hint: newLanguage });
+            const normalizedLanguage = newLanguage.toLowerCase();
+            let nextDocType: DocType | undefined;
+
+            if (normalizedLanguage === 'html') {
+                nextDocType = 'rich_text';
+            } else if (activeNode.doc_type === 'rich_text') {
+                nextDocType = 'prompt';
+            }
+
+            const updates: Partial<Omit<DocumentOrFolder, 'id' | 'content'>> = { language_hint: newLanguage };
+            if (nextDocType) {
+                updates.doc_type = nextDocType;
+            }
+
+            updateItem(activeNodeId, updates);
         }
     }, [activeNodeId, activeNode, updateItem, addLog]);
 

hooks/useNodes.ts

@@ -1,5 +1,5 @@
 import { useState, useEffect, useCallback } from 'react';
-import type { Node, ViewMode, ImportedNodeSummary, DraggedNodeTransfer, ClassificationSummary } from '../types';
+import type { Node, ViewMode, ImportedNodeSummary, DraggedNodeTransfer, ClassificationSummary, DocType } from '../types';
 import { repository } from '../services/repository';
 import { useLogger } from './useLogger';
 
@@ -32,7 +32,10 @@ export const useNodes = () => {
     return newNode;
   }, [addLog, refreshNodes]);
 
-  const updateNode = useCallback(async (nodeId: string, updates: Partial<Pick<Node, 'title' | 'parent_id'> & { language_hint?: string | null; default_view_mode?: ViewMode | null }>) => {
+  const updateNode = useCallback(async (
+    nodeId: string,
+    updates: Partial<Pick<Node, 'title' | 'parent_id'> & { language_hint?: string | null; default_view_mode?: ViewMode | null; doc_type?: DocType | null }>,
+  ) => {
     await repository.updateNode(nodeId, updates);
     addLog('DEBUG', `Node updated with ID: ${nodeId}. Refreshing tree.`);
     await refreshNodes(true);

hooks/usePrompts.ts

@@ -117,6 +117,7 @@ export const useDocuments = () => {
     if (updates.parentId !== undefined) nodeUpdates.parent_id = updates.parentId;
     if (updates.language_hint !== undefined) nodeUpdates.language_hint = updates.language_hint;
     if (updates.default_view_mode !== undefined) nodeUpdates.default_view_mode = updates.default_view_mode;
+    if (updates.doc_type !== undefined) nodeUpdates.doc_type = updates.doc_type as DocType;
 
     if (Object.keys(nodeUpdates).length > 0) {
         await updateNode(id, nodeUpdates);

services/repository.ts

@@ -965,7 +965,10 @@ export const repository = {
 
         return { node: newNode, summary: classification.summary };
     },
-    async updateNode(nodeId: string, updates: Partial<Pick<Node, 'title' | 'parent_id'> & { language_hint?: string | null; default_view_mode?: ViewMode | null }>) {
+    async updateNode(
+        nodeId: string,
+        updates: Partial<Pick<Node, 'title' | 'parent_id'> & { language_hint?: string | null; default_view_mode?: ViewMode | null; doc_type?: DocType | null }>,
+    ) {
         if (!isElectron) {
             const state = ensureBrowserState();
             const result = findNodeWithParent(nodeId, state.nodes);
@@ -1001,6 +1004,11 @@ export const repository = {
                 if (updates.default_view_mode !== undefined) {
                     node.document.default_view_mode = updates.default_view_mode;
                 }
+                if (updates.doc_type !== undefined) {
+                    node.document.doc_type = updates.doc_type ?? node.document.doc_type;
+                    node.document.doc_type_source = 'user';
+                    node.document.classification_updated_at = now;
+                }
             }
 
             node.updated_at = now;
@@ -1038,6 +1046,14 @@ export const repository = {
             );
             await window.electronAPI!.dbRun(`UPDATE nodes SET updated_at = ? WHERE node_id = ?`, [now, nodeId]);
         }
+
+        if (updates.doc_type !== undefined) {
+            await window.electronAPI!.dbRun(
+                `UPDATE documents SET doc_type = ?, doc_type_source = 'user', classification_updated_at = ? WHERE node_id = ?`,
+                [updates.doc_type, now, nodeId]
+            );
+            await window.electronAPI!.dbRun(`UPDATE nodes SET updated_at = ? WHERE node_id = ?`, [now, nodeId]);
+        }
     },
 
     async updateDocumentContent(nodeId: string, newContent: string, documentId?: number) {