// *****************************************************************************
// Copyright (C) 2024 EclipseSource GmbH.
//
// This program and the accompanying materials are made available under the
// terms of the Eclipse Public License v. 2.0 which is available at
// http://www.eclipse.org/legal/epl-2.0.
//
// This Source Code may also be made available under the following Secondary
// Licenses when the conditions for such availability set forth in the Eclipse
// Public License v. 2.0 are satisfied: GNU General Public License, version 2
// with the GNU Classpath Exception which is available at
// https://www.gnu.org/software/classpath/license.html.
//
// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
// *****************************************************************************

import { Agent, AgentService, AISettingsService, AIVariableContribution, bindToolProvider, CustomAgentPromptVariant } from '@theia/ai-core/lib/common';
import { bindRootContributionProvider, CommandContribution, PreferenceContribution } from '@theia/core';
import { FrontendApplicationContribution, LabelProviderContribution } from '@theia/core/lib/browser';
import { ContainerModule } from '@theia/core/shared/inversify';
import {
    ChatAgent,
    ChatAgentService,
    ChatAgentServiceImpl,
    ChatRequestParser,
    ChatRequestParserImpl,
    ChatService,
    ToolCallChatResponseContentFactory,
    ServerToolCallResponseContentFactory,
    PinChatAgent,
    ChatServiceFactory,
    ChatAgentServiceFactory
} from '../common';
import { ChatAgentsVariableContribution } from '../common/chat-agents-variable-contribution';
import { CustomChatAgent } from '../common/custom-chat-agent';
import { DefaultResponseContentFactory, DefaultResponseContentMatcherProvider, ResponseContentMatcherProvider } from '../common/response-content-matcher';
import { aiChatPreferences } from '../common/ai-chat-preferences';
import { ChangeSetElementArgs, ChangeSetFileElement, ChangeSetFileElementFactory } from './change-set-file-element';
import { AICustomAgentsFrontendApplicationContribution } from './custom-agent-frontend-application-contribution';
import { FrontendChatServiceImpl } from './frontend-chat-service';
import { CustomAgentFactory } from './custom-agent-factory';
import { ChatToolRequestService } from '../common/chat-tool-request-service';
import { FrontendChatToolRequestService } from './chat-tool-request-service';
import { ChangeSetFileService } from './change-set-file-service';
import { FileReadTracker } from '../common/file-read-tracker';
import { FileReadTrackerImpl } from './file-read-tracker-impl';
import { ContextVariableLabelProvider } from './context-variable-label-provider';
import { ContextFileVariableLabelProvider } from './context-file-variable-label-provider';
import { FileChatVariableContribution } from './file-chat-variable-contribution';
import { ContextSummaryVariableContribution } from '../common/context-summary-variable';
import { ContextDetailsVariableContribution } from '../common/context-details-variable';
import { ChangeSetVariableContribution } from './change-set-variable';
import { ChatSessionNamingAgent, ChatSessionNamingService } from '../common/chat-session-naming-service';
import { ChangeSetDecorator, ChangeSetDecoratorService } from './change-set-decorator-service';
import { ChatSessionSummaryAgent } from '../common/chat-session-summary-agent';
import { TaskContextVariableContribution } from './task-context-variable-contribution';
import { TaskContextVariableLabelProvider } from './task-context-variable-label-provider';
import { TaskContextService, TaskContextStorageService } from './task-context-service';
import { InMemoryTaskContextStorage } from './task-context-storage-service';
import { AIChatFrontendContribution } from './ai-chat-frontend-contribution';
import { ImageContextVariableContribution } from './image-context-variable-contribution';
import { DefaultPendingImageRegistry, PendingImageRegistry } from './pending-image-registry';
import { AgentDelegationTool } from './agent-delegation-tool';
import { ToolConfirmationManager } from './chat-tool-preference-bindings';
import { bindChatToolPreferences } from '../common/chat-tool-preferences';
import { PendingToolConfirmationTracker } from './pending-tool-confirmation-tracker';
import { ChatSessionStore } from '../common/chat-session-store';
import { ChatSessionStoreImpl } from './chat-session-store-impl';
import {
    ChatContentDeserializerContribution,
    ChatContentDeserializerRegistry,
    ChatContentDeserializerRegistryImpl,
    DefaultChatContentDeserializerContribution
} from '../common/chat-content-deserializer';
import {
    ChangeSetElementDeserializerContribution,
    ChangeSetElementDeserializerRegistry,
    ChangeSetElementDeserializerRegistryImpl
} from '../common/change-set-element-deserializer';
import { ChangeSetFileElementDeserializerContribution } from './change-set-file-element-deserializer';

export default new ContainerModule(bind => {
    bindRootContributionProvider(bind, ChatAgent);

    bind(ChatContentDeserializerRegistryImpl).toSelf().inSingletonScope();
    bind(ChatContentDeserializerRegistry).toService(ChatContentDeserializerRegistryImpl);
    bindRootContributionProvider(bind, ChatContentDeserializerContribution);
    bind(ChatContentDeserializerContribution).to(DefaultChatContentDeserializerContribution).inSingletonScope();

    bind(ChangeSetElementDeserializerRegistryImpl).toSelf().inSingletonScope();
    bind(ChangeSetElementDeserializerRegistry).toService(ChangeSetElementDeserializerRegistryImpl);
    bindRootContributionProvider(bind, ChangeSetElementDeserializerContribution);
    bind(ChangeSetElementDeserializerContribution).to(ChangeSetFileElementDeserializerContribution).inSingletonScope();

    bind(ChatSessionStoreImpl).toSelf().inSingletonScope();
    bind(ChatSessionStore).toService(ChatSessionStoreImpl);

    bind(FrontendChatToolRequestService).toSelf().inSingletonScope();
    bind(ChatToolRequestService).toService(FrontendChatToolRequestService);

    bind(ChatAgentServiceImpl).toSelf().inSingletonScope();
    bind(ChatAgentService).toService(ChatAgentServiceImpl);
    bind(PinChatAgent).toConstantValue(true);

    bind(ChatSessionNamingService).toSelf().inSingletonScope();
    bind(ChatSessionNamingAgent).toSelf().inSingletonScope();
    bind(Agent).toService(ChatSessionNamingAgent);

    bindRootContributionProvider(bind, ResponseContentMatcherProvider);
    bind(DefaultResponseContentMatcherProvider).toSelf().inSingletonScope();
    bind(ResponseContentMatcherProvider).toService(DefaultResponseContentMatcherProvider);
    bind(DefaultResponseContentFactory).toSelf().inSingletonScope();

    bind(AIVariableContribution).to(ChatAgentsVariableContribution).inSingletonScope();

    bind(ChatRequestParserImpl).toSelf().inSingletonScope();
    bind(ChatRequestParser).toService(ChatRequestParserImpl);

    bind(FrontendChatServiceImpl).toSelf().inSingletonScope();
    bind(ChatService).toService(FrontendChatServiceImpl);

    bind(ChatServiceFactory).toDynamicValue(ctx => () =>
        ctx.container.get<ChatService>(ChatService)
    );
    bind(ChatAgentServiceFactory).toDynamicValue(ctx => () =>
        ctx.container.get<ChatAgentService>(ChatAgentService)
    );

    bind(PreferenceContribution).toConstantValue({ schema: aiChatPreferences });

    // Tool confirmation preferences
    bindChatToolPreferences(bind);
    bind(ToolConfirmationManager).toSelf().inSingletonScope();
    bind(PendingToolConfirmationTracker).toSelf().inSingletonScope();

    bind(CustomChatAgent).toSelf();
    bind(CustomAgentFactory).toFactory<CustomChatAgent, [string, string, string, string, string, boolean | undefined, CustomAgentPromptVariant[] | undefined]>(
        ctx => (id, name, description, prompt, defaultLLM, showInChat, promptVariants) => {
            const agent = ctx.container.get<CustomChatAgent>(CustomChatAgent);
            agent.id = id;
            agent.name = name;
            agent.description = description;
            agent.prompt = prompt;
            agent.promptVariants = promptVariants;
            agent.languageModelRequirements = [{
                purpose: 'chat',
                identifier: defaultLLM,
            }];
            ctx.container.get<ChatAgentService>(ChatAgentService).registerChatAgent(agent);
            ctx.container.get<AgentService>(AgentService).registerAgent(agent);

            // Initialize showInChat preference from YAML if not already set
            if (showInChat === false) {
                const settingsService = ctx.container.get<AISettingsService>(AISettingsService);
                settingsService.getAgentSettings(id).then(settings => {
                    if (settings?.showInChat === undefined) {
                        settingsService.updateAgentSettings(id, { showInChat: false });
                    }
                });
            }

            return agent;
        });
    bind(AICustomAgentsFrontendApplicationContribution).toSelf().inSingletonScope();
    bind(FrontendApplicationContribution).toService(AICustomAgentsFrontendApplicationContribution);
    bind(CommandContribution).toService(AICustomAgentsFrontendApplicationContribution);

    bind(ContextVariableLabelProvider).toSelf().inSingletonScope();
    bind(LabelProviderContribution).toService(ContextVariableLabelProvider);
    bind(ContextFileVariableLabelProvider).toSelf().inSingletonScope();
    bind(LabelProviderContribution).toService(ContextFileVariableLabelProvider);

    bind(FileReadTrackerImpl).toSelf().inSingletonScope();
    bind(FileReadTracker).toService(FileReadTrackerImpl);

    bind(ChangeSetFileService).toSelf().inSingletonScope();
    bind(ChangeSetFileElementFactory).toFactory(ctx => (args: ChangeSetElementArgs) => {
        const container = ctx.container.createChild();
        container.bind(ChangeSetElementArgs).toConstantValue(args);
        container.bind(ChangeSetFileElement).toSelf().inSingletonScope();
        return container.get(ChangeSetFileElement);
    });

    bind(ChangeSetDecoratorService).toSelf().inSingletonScope();
    bind(FrontendApplicationContribution).toService(ChangeSetDecoratorService);
    bindRootContributionProvider(bind, ChangeSetDecorator);
    bind(ToolCallChatResponseContentFactory).toSelf().inSingletonScope();
    bind(ServerToolCallResponseContentFactory).toSelf().inSingletonScope();
    bind(AIVariableContribution).to(FileChatVariableContribution).inSingletonScope();
    bind(AIVariableContribution).to(ContextSummaryVariableContribution).inSingletonScope();
    bind(AIVariableContribution).to(ContextDetailsVariableContribution).inSingletonScope();
    bind(AIVariableContribution).to(ChangeSetVariableContribution).inSingletonScope();

    bind(ChatSessionSummaryAgent).toSelf().inSingletonScope();
    bind(Agent).toService(ChatSessionSummaryAgent);

    bind(TaskContextVariableContribution).toSelf().inSingletonScope();
    bind(AIVariableContribution).toService(TaskContextVariableContribution);
    bind(TaskContextVariableLabelProvider).toSelf().inSingletonScope();
    bind(LabelProviderContribution).toService(TaskContextVariableLabelProvider);

    bind(ImageContextVariableContribution).toSelf().inSingletonScope();
    bind(AIVariableContribution).toService(ImageContextVariableContribution);
    bind(LabelProviderContribution).toService(ImageContextVariableContribution);

    bind(DefaultPendingImageRegistry).toSelf().inSingletonScope();
    bind(PendingImageRegistry).toService(DefaultPendingImageRegistry);

    bind(TaskContextService).toSelf().inSingletonScope();
    bind(InMemoryTaskContextStorage).toSelf().inSingletonScope();
    bind(TaskContextStorageService).toService(InMemoryTaskContextStorage);
    bind(AIChatFrontendContribution).toSelf().inSingletonScope();
    bind(CommandContribution).toService(AIChatFrontendContribution);

    bindToolProvider(AgentDelegationTool, bind);
});