UnrealBlueprintMCP

// Copyright Epic Games, Inc. All Rights Reserved. // UnrealBlueprintMCP Plugin - Production Implementation #include "UnrealBlueprintMCP.h" #include "MCPStatusWidget.h" #include "MCPSettings.h" #include "MCPClient.h" // Unreal Engine includes #include "Framework/MultiBox/MultiBoxBuilder.h" #include "Framework/MultiBox/MultiBoxExtender.h" #include "ToolMenus.h" #include "Widgets/Docking/SDockTab.h" #include "WorkspaceMenuStructure.h" #include "WorkspaceMenuStructureModule.h" #include "Styling/AppStyle.h" // Logging DEFINE_LOG_CATEGORY_STATIC(LogUnrealBlueprintMCP, Log, All); #define LOCTEXT_NAMESPACE "FUnrealBlueprintMCPModule" // Tab identifier for the MCP status window static const FName MCPStatusTabName("MCPStatusTab"); void FUnrealBlueprintMCPModule::StartupModule() { UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("UnrealBlueprintMCP module starting up...")); // Register editor menu extensions RegisterMenuExtensions(); // Register tab spawner for MCP status window FGlobalTabmanager::Get()->RegisterNomadTabSpawner(MCPStatusTabName, FOnSpawnTab::CreateRaw(this, &FUnrealBlueprintMCPModule::SpawnMCPStatusTab)) .SetDisplayName(LOCTEXT("MCPStatusTabTitle", "MCP Status")) .SetTooltipText(LOCTEXT("MCPStatusTabTooltip", "Monitor MCP server connection status and settings")) .SetGroup(WorkspaceMenu::GetMenuStructure().GetDeveloperToolsDebugCategory()) .SetIcon(FSlateIcon(FAppStyle::GetAppStyleSetName(), "LevelEditor.GameSettings")); // Initialize MCP client and attempt auto-connection if enabled InitializeMCPClient(); UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("UnrealBlueprintMCP module started successfully")); } void FUnrealBlueprintMCPModule::ShutdownModule() { UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("UnrealBlueprintMCP module shutting down...")); // Disconnect MCP client if connected ShutdownMCPClient(); // Unregister menu extensions UnregisterMenuExtensions(); // Unregister tab spawner FGlobalTabmanager::Get()->UnregisterNomadTabSpawner(MCPStatusTabName); UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("UnrealBlueprintMCP module shut down successfully")); } void FUnrealBlueprintMCPModule::RegisterMenuExtensions() { // Register toolbar extension UToolMenus* ToolMenus = UToolMenus::Get(); if (ToolMenus) { UToolMenu* ToolbarMenu = ToolMenus->ExtendMenu("LevelEditor.LevelEditorToolBar.PlayToolBar"); if (ToolbarMenu) { FToolMenuSection& Section = ToolbarMenu->FindOrAddSection("MCP"); Section.Label = LOCTEXT("MCPSectionLabel", "MCP"); // Add MCP status button to toolbar Section.AddEntry(FToolMenuEntry::InitToolBarButton( "MCPStatus", FUIAction(FExecuteAction::CreateRaw(this, &FUnrealBlueprintMCPModule::OnOpenMCPStatusAction)), LOCTEXT("MCPStatusLabel", "MCP Status"), LOCTEXT("MCPStatusTooltip", "Open MCP Server Status Window"), FSlateIcon(FAppStyle::GetAppStyleSetName(), "LevelEditor.GameSettings") )); UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("Registered MCP toolbar extension")); } } } void FUnrealBlueprintMCPModule::UnregisterMenuExtensions() { // Cleanup is handled automatically when the module shuts down UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("Unregistered MCP menu extensions")); } void FUnrealBlueprintMCPModule::OnOpenMCPStatusAction() { // Open the MCP status tab FGlobalTabmanager::Get()->TryInvokeTab(MCPStatusTabName); } TSharedRef<SDockTab> FUnrealBlueprintMCPModule::SpawnMCPStatusTab(const FSpawnTabArgs& Args) { // Create the MCP status widget TSharedRef<SMCPStatusWidget> StatusWidget = SNew(SMCPStatusWidget); // Create and return the dock tab containing the status widget return SNew(SDockTab) .TabRole(ETabRole::NomadTab) [ StatusWidget ]; } void FUnrealBlueprintMCPModule::InitializeMCPClient() { UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("Initializing MCP client...")); // Get MCP settings UMCPSettings* MCPSettings = UMCPSettings::Get(); if (!MCPSettings) { UE_LOG(LogUnrealBlueprintMCP, Error, TEXT("Failed to get MCP settings")); return; } // Get MCP client instance UMCPClient* MCPClient = UMCPClient::Get(); if (!MCPClient) { UE_LOG(LogUnrealBlueprintMCP, Error, TEXT("Failed to get MCP client instance")); return; } // Initialize the client if (!MCPClient->Initialize(MCPSettings)) { UE_LOG(LogUnrealBlueprintMCP, Error, TEXT("Failed to initialize MCP client")); return; } // Attempt auto-connection if enabled if (MCPSettings->bAutoConnectOnStartup) { UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("Auto-connect enabled, attempting connection...")); if (MCPClient->Connect()) { UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("Auto-connection attempt initiated")); } else { UE_LOG(LogUnrealBlueprintMCP, Warning, TEXT("Auto-connection attempt failed")); } } else { UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("Auto-connect disabled in settings")); } } void FUnrealBlueprintMCPModule::ShutdownMCPClient() { UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("Shutting down MCP client...")); // 안전한 종료 - Get() 대신 직접 Shutdown() 호출 UMCPClient::Shutdown(); UE_LOG(LogUnrealBlueprintMCP, Log, TEXT("MCP client shutdown complete")); } #undef LOCTEXT_NAMESPACE // Required module implementation IMPLEMENT_MODULE(FUnrealBlueprintMCPModule, UnrealBlueprintMCP)