/**
 * @license
 * Copyright 2025 Vybestack LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import { useCallback } from 'react';
import { getErrorMessage } from '@vybestack/llxprt-code-core';
import { debugLogger } from '@vybestack/llxprt-code-telemetry';
import { type HistoryItem, MessageType } from '../../../types.js';
import type { LoadedSettings } from '../../../../config/settings.js';
import { loadHierarchicalLlxprtMemory } from '../../../../config/environmentLoader.js';
import type {
  AppStateRuntime,
  ExtensionRuntime,
  FileWorkspaceState,
  MemoryState,
  SessionIdentity,
} from '../../../cliUiRuntime.js';

type MemoryRefreshRuntime = MemoryState &
  AppStateRuntime &
  FileWorkspaceState &
  ExtensionRuntime &
  SessionIdentity;

interface UseMemoryRefreshActionParams {
  config: MemoryRefreshRuntime;
  settings: LoadedSettings;
  addItem: (item: Omit<HistoryItem, 'id'>, timestamp: number) => number;
  setLlxprtMdFileCount: (fileCount: number) => void;
}

export function useMemoryRefreshAction({
  config,
  settings,
  addItem,
  setLlxprtMdFileCount,
}: UseMemoryRefreshActionParams): () => Promise<void> {
  return useCallback(async () => {
    addItem(
      {
        type: MessageType.INFO,
        text: 'Refreshing hierarchical memory (LLXPRT.md or other context files)...',
      },
      Date.now(),
    );

    try {
      let memoryContent = '';
      let fileCount = 0;

      if (config.isJitContextEnabled()) {
        const result = await config.refreshMemory();
        memoryContent = result.memoryContent;
        fileCount = result.fileCount;
      } else {
        const result = await loadHierarchicalLlxprtMemory(
          config.getWorkingDir(),
          settings.merged.loadMemoryFromIncludeDirectories === true
            ? config.getWorkspaceContext().getDirectories()
            : [],
          config.getDebugMode(),
          config.getFileService(),
          settings.merged,
          config.getExtensions(),
          config.getFolderTrust(),
          settings.merged.ui.memoryImportFormat ?? 'tree',
          config.getFileFilteringOptions(),
        );

        memoryContent = result.memoryContent;
        fileCount = result.fileCount;

        config.setUserMemory(memoryContent);
        config.setLlxprtMdFileCount(fileCount);
        config.setLlxprtMdFilePaths(result.filePaths);
      }

      setLlxprtMdFileCount(fileCount);

      addItem(
        {
          type: MessageType.INFO,
          text: `Memory refreshed successfully. ${memoryContent.length > 0 ? `Loaded ${memoryContent.length} characters from ${fileCount} file(s).` : 'No memory content found.'}`,
        },
        Date.now(),
      );

      if (config.getDebugMode()) {
        debugLogger.log(
          `[DEBUG] Refreshed memory content in config (${memoryContent.length} chars from ${fileCount} file(s))`,
        );
      }
    } catch (error) {
      const errorMessage = getErrorMessage(error);
      addItem(
        {
          type: MessageType.ERROR,
          text: `Error refreshing memory: ${errorMessage}`,
        },
        Date.now(),
      );
      debugLogger.error('Error refreshing memory:', error);
    }
    // settings.merged reference changes when any setting changes, so this dep is correct.
  }, [config, addItem, settings.merged, setLlxprtMdFileCount]);
}