@gguf/coder 0.3.1 → 0.3.2

package/source/hooks/useAppState.tsx

@@ -1,311 +0,0 @@
-import { loadPreferences } from '@/config/preferences';
-import { defaultTheme } from '@/config/themes';
-import { CustomCommandExecutor } from '@/custom-commands/executor';
-import { CustomCommandLoader } from '@/custom-commands/loader';
-import { createTokenizer } from '@/tokenization/index.js';
-import { ToolManager } from '@/tools/tool-manager';
-import type { CheckpointListItem } from '@/types/checkpoint';
-import type { CustomCommand } from '@/types/commands';
-import {
-	DevelopmentMode,
-	LLMClient,
-	LSPConnectionStatus,
-	MCPConnectionStatus,
-	Message,
-	ToolCall,
-} from '@/types/core';
-import type { ToolResult, UpdateInfo } from '@/types/index';
-import type { Tokenizer } from '@/types/tokenization.js';
-import type { ThemePreset } from '@/types/ui';
-import { BoundedMap } from '@/utils/bounded-map';
-import { useCallback, useEffect, useMemo, useState } from 'react';
-import React from 'react';
-
-export interface ConversationContext {
-	/**
-	 * All messages up to (but not including) tool execution.
-	 * Includes user message, auto-executed messages, and assistant message with tool_calls.
-	 */
-	messagesBeforeToolExecution: Message[];
-	/**
-	 * The assistant message that triggered tool execution.
-	 * Included in messagesBeforeToolExecution for reference.
-	 */
-	assistantMsg: Message;
-	/**
-	 * System message for the next turn after tool execution.
-	 */
-	systemMessage: Message;
-}
-
-export function useAppState() {
-	// Initialize theme from preferences
-	const preferences = loadPreferences();
-	const initialTheme = preferences.selectedTheme || defaultTheme;
-
-	const [client, setClient] = useState<LLMClient | null>(null);
-	const [messages, setMessages] = useState<Message[]>([]);
-	const [displayMessages, setDisplayMessages] = useState<Message[]>([]);
-	const [messageTokenCache, setMessageTokenCache] = useState<
-		BoundedMap<string, number>
-	>(
-		new BoundedMap({
-			maxSize: 1000,
-			// No TTL - cache is session-based and cleared on app restart
-		}),
-	);
-	const [currentModel, setCurrentModel] = useState<string>('');
-	const [currentProvider, setCurrentProvider] =
-		useState<string>('openai-compatible');
-	const [currentTheme, setCurrentTheme] = useState<ThemePreset>(initialTheme);
-	const [toolManager, setToolManager] = useState<ToolManager | null>(null);
-	const [customCommandLoader, setCustomCommandLoader] =
-		useState<CustomCommandLoader | null>(null);
-	const [customCommandExecutor, setCustomCommandExecutor] =
-		useState<CustomCommandExecutor | null>(null);
-	const [customCommandCache, setCustomCommandCache] = useState<
-		Map<string, CustomCommand>
-	>(new Map());
-	const [startChat, setStartChat] = useState<boolean>(false);
-	const [mcpInitialized, setMcpInitialized] = useState<boolean>(false);
-	const [updateInfo, setUpdateInfo] = useState<UpdateInfo | null>(null);
-
-	// Connection status states
-	const [mcpServersStatus, setMcpServersStatus] = useState<
-		MCPConnectionStatus[]
-	>([]);
-	const [lspServersStatus, setLspServersStatus] = useState<
-		LSPConnectionStatus[]
-	>([]);
-
-	// Initialization status states
-	const [preferencesLoaded, setPreferencesLoaded] = useState<boolean>(false);
-	const [customCommandsCount, setCustomCommandsCount] = useState<number>(0);
-
-	// Cancelling indicator state
-	const [isCancelling, setIsCancelling] = useState<boolean>(false);
-	const [isConversationComplete, setIsConversationComplete] =
-		useState<boolean>(false);
-
-	// Cancellation state
-	const [abortController, setAbortController] =
-		useState<AbortController | null>(null);
-
-	// Mode states
-	const [isModelSelectionMode, setIsModelSelectionMode] =
-		useState<boolean>(false);
-	const [isProviderSelectionMode, setIsProviderSelectionMode] =
-		useState<boolean>(false);
-	const [isThemeSelectionMode, setIsThemeSelectionMode] =
-		useState<boolean>(false);
-	const [isModelDatabaseMode, setIsModelDatabaseMode] =
-		useState<boolean>(false);
-	const [isConfigWizardMode, setIsConfigWizardMode] = useState<boolean>(false);
-	const [isCheckpointLoadMode, setIsCheckpointLoadMode] =
-		useState<boolean>(false);
-	const [checkpointLoadData, setCheckpointLoadData] = useState<{
-		checkpoints: CheckpointListItem[];
-		currentMessageCount: number;
-	} | null>(null);
-	const [isToolConfirmationMode, setIsToolConfirmationMode] =
-		useState<boolean>(false);
-	const [isToolExecuting, setIsToolExecuting] = useState<boolean>(false);
-	const [isBashExecuting, setIsBashExecuting] = useState<boolean>(false);
-	const [currentBashCommand, setCurrentBashCommand] = useState<string>('');
-
-	// Development mode state
-	const [developmentMode, setDevelopmentMode] =
-		useState<DevelopmentMode>('auto-accept');
-
-	// Tool confirmation state
-	const [pendingToolCalls, setPendingToolCalls] = useState<ToolCall[]>([]);
-	const [currentToolIndex, setCurrentToolIndex] = useState<number>(0);
-	const [completedToolResults, setCompletedToolResults] = useState<
-		ToolResult[]
-	>([]);
-	const [currentConversationContext, setCurrentConversationContext] =
-		useState<ConversationContext | null>(null);
-
-	// Chat queue for components
-	const [chatComponents, setChatComponents] = useState<React.ReactNode[]>([]);
-	const [componentKeyCounter, setComponentKeyCounter] = useState(0);
-
-	// Helper function to add components to the chat queue with stable keys
-	const addToChatQueue = useCallback(
-		(component: React.ReactNode) => {
-			const newCounter = componentKeyCounter + 1;
-			setComponentKeyCounter(newCounter);
-
-			let componentWithKey = component;
-			if (React.isValidElement(component) && !component.key) {
-				componentWithKey = React.cloneElement(component, {
-					key: `chat-component-${newCounter}`,
-				});
-			}
-
-			setChatComponents(prevComponents => [
-				...prevComponents,
-				componentWithKey,
-			]);
-		},
-		[componentKeyCounter],
-	);
-
-	// Create tokenizer based on current provider and model
-	const tokenizer = useMemo<Tokenizer>(() => {
-		if (currentProvider && currentModel) {
-			return createTokenizer(currentProvider, currentModel);
-		}
-
-		// Fallback to simple char/4 heuristic if provider/model not set
-		return createTokenizer('', '');
-	}, [currentProvider, currentModel]);
-
-	// Cleanup tokenizer resources when it changes
-	useEffect(() => {
-		return () => {
-			if (tokenizer.free) {
-				tokenizer.free();
-			}
-		};
-	}, [tokenizer]);
-
-	// Helper function for token calculation with caching
-	const getMessageTokens = useCallback(
-		(message: Message) => {
-			const cacheKey = (message.content || '') + message.role + currentModel;
-
-			const cachedTokens = messageTokenCache.get(cacheKey);
-			if (cachedTokens !== undefined) {
-				return cachedTokens;
-			}
-
-			const tokens = tokenizer.countTokens(message);
-			// Defer cache update to avoid "Cannot update a component while rendering" error
-			// This can happen when components call getMessageTokens during their render
-			queueMicrotask(() => {
-				setMessageTokenCache(prev => {
-					const newCache = new BoundedMap<string, number>({
-						maxSize: 1000,
-					});
-					// Copy existing entries
-					for (const [k, v] of prev.entries()) {
-						newCache.set(k, v);
-					}
-					// Add new entry
-					newCache.set(cacheKey, tokens);
-					return newCache;
-				});
-			});
-			return tokens;
-		},
-		[messageTokenCache, tokenizer, currentModel],
-	);
-
-	// Message updater - no limits, display all messages
-	const updateMessages = useCallback((newMessages: Message[]) => {
-		setMessages(newMessages);
-		setDisplayMessages(newMessages);
-	}, []);
-
-	// Reset tool confirmation state
-	const resetToolConfirmationState = () => {
-		setIsToolConfirmationMode(false);
-		setIsToolExecuting(false);
-		setPendingToolCalls([]);
-		setCurrentToolIndex(0);
-		setCompletedToolResults([]);
-		setCurrentConversationContext(null);
-	};
-
-	return {
-		// State
-		client,
-		messages,
-		displayMessages,
-		messageTokenCache,
-		currentModel,
-		currentProvider,
-		currentTheme,
-		toolManager,
-		customCommandLoader,
-		customCommandExecutor,
-		customCommandCache,
-		startChat,
-		mcpInitialized,
-		updateInfo,
-		mcpServersStatus,
-		lspServersStatus,
-		preferencesLoaded,
-		customCommandsCount,
-		isCancelling,
-		isConversationComplete,
-		abortController,
-		isModelSelectionMode,
-		isProviderSelectionMode,
-		isThemeSelectionMode,
-		isModelDatabaseMode,
-		isConfigWizardMode,
-		isCheckpointLoadMode,
-		checkpointLoadData,
-		isToolConfirmationMode,
-		isToolExecuting,
-		isBashExecuting,
-		currentBashCommand,
-		developmentMode,
-		pendingToolCalls,
-		currentToolIndex,
-		completedToolResults,
-		currentConversationContext,
-		chatComponents,
-		componentKeyCounter,
-		tokenizer,
-
-		// Setters
-		setClient,
-		setMessages,
-		setDisplayMessages,
-		setMessageTokenCache,
-		setCurrentModel,
-		setCurrentProvider,
-		setCurrentTheme,
-		setToolManager,
-		setCustomCommandLoader,
-		setCustomCommandExecutor,
-		setCustomCommandCache,
-		setStartChat,
-		setMcpInitialized,
-		setUpdateInfo,
-		setMcpServersStatus,
-		setLspServersStatus,
-		setPreferencesLoaded,
-		setCustomCommandsCount,
-		setIsCancelling,
-		setIsConversationComplete,
-		setAbortController,
-		setIsModelSelectionMode,
-		setIsProviderSelectionMode,
-		setIsThemeSelectionMode,
-		setIsModelDatabaseMode,
-		setIsConfigWizardMode,
-		setIsCheckpointLoadMode,
-		setCheckpointLoadData,
-		setIsToolConfirmationMode,
-		setIsToolExecuting,
-		setIsBashExecuting,
-		setCurrentBashCommand,
-		setDevelopmentMode,
-		setPendingToolCalls,
-		setCurrentToolIndex,
-		setCompletedToolResults,
-		setCurrentConversationContext,
-		setChatComponents,
-		setComponentKeyCounter,
-
-		// Utilities
-		addToChatQueue,
-		getMessageTokens,
-		updateMessages,
-		resetToolConfirmationState,
-	};
-}

package/source/hooks/useDirectoryTrust.tsx

@@ -1,98 +0,0 @@
-import path from 'path';
-import {loadPreferences, savePreferences} from '@/config/preferences';
-import {logError, logInfo} from '@/utils/message-queue';
-import {useCallback, useEffect, useState} from 'react';
-
-interface UseDirectoryTrustReturn {
-	isTrusted: boolean;
-	handleConfirmTrust: () => void;
-	isTrustLoading: boolean;
-	isTrustedError: string | null;
-}
-
-/**
- * Custom hook for managing directory trust functionality.
- * Handles checking if a directory is trusted and adding it to trusted directories.
- *
- * @param directory - The directory path to check trust for (defaults to current working directory)
- * @returns Object containing trust state and handler functions
- */
-export function useDirectoryTrust(
-	directory: string = process.cwd(),
-): UseDirectoryTrustReturn {
-	const [isTrusted, setIsTrusted] = useState(false);
-	const [isLoading, setIsLoading] = useState(true);
-	const [error, setError] = useState<string | null>(null);
-
-	// Check if directory is trusted on mount and when directory changes
-	useEffect(() => {
-		const checkTrustStatus = () => {
-			try {
-				setIsLoading(true);
-				setError(null);
-
-				const preferences = loadPreferences();
-				const trustedDirectories = preferences.trustedDirectories || [];
-
-				// Normalize paths for comparison (resolve any relative path components)
-				const normalizedDirectory = path.resolve(directory); // nosemgrep
-				const isTrustedDir = trustedDirectories.some(
-					trustedDir => path.resolve(trustedDir) === normalizedDirectory, // nosemgrep
-				);
-
-				setIsTrusted(isTrustedDir);
-			} catch (err) {
-				const errorMessage =
-					err instanceof Error ? err.message : 'Unknown error occurred';
-				setError(`Failed to check directory trust status: ${errorMessage}`);
-
-				logError(`${errorMessage}`);
-			} finally {
-				setIsLoading(false);
-			}
-		};
-
-		checkTrustStatus();
-	}, [directory]);
-
-	// Handler to confirm trust for the current directory
-	const handleConfirmTrust = useCallback(() => {
-		try {
-			setError(null);
-
-			const preferences = loadPreferences();
-			const trustedDirectories = preferences.trustedDirectories || [];
-
-			// Normalize the directory path before storing and checking
-			const normalizedDirectory = path.resolve(directory); // nosemgrep
-
-			// Only add if not already trusted (check using normalized paths)
-			if (
-				!trustedDirectories.some(
-					trustedDir => path.resolve(trustedDir) === normalizedDirectory, // nosemgrep
-				)
-			) {
-				trustedDirectories.push(normalizedDirectory);
-				preferences.trustedDirectories = trustedDirectories;
-				savePreferences(preferences);
-
-				logInfo(`Directory added to trusted list: ${normalizedDirectory}`);
-			}
-
-			setIsTrusted(true);
-		} catch (err) {
-			const errorMessage =
-				err instanceof Error ? err.message : 'Unknown error occurred';
-			setError(`Failed to save directory trust: ${errorMessage}`);
-
-			logError(`${errorMessage}`);
-		}
-	}, [directory]);
-
-	return {
-		isTrusted,
-		handleConfirmTrust,
-		isTrustLoading: isLoading,
-		isTrustedError: error,
-	};
-}