Files
WulaFallenEmpireRW/Source/WulaFallenEmpire/EventSystem/AI/UI/Overlay_WulaLink.cs
2025-12-27 19:58:14 +08:00

635 lines
24 KiB
C#

using System;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;
using Verse;
using RimWorld;
using WulaFallenEmpire.EventSystem.AI;
namespace WulaFallenEmpire.EventSystem.AI.UI
{
public class Overlay_WulaLink : Window
{
// Core Connection
private AIIntelligenceCore _core;
private string _eventDefName;
private EventDef _def;
// UI State
private Vector2 _scrollPosition = Vector2.zero;
private string _inputText = "";
private bool _scrollToBottom = true;
// HUD / Minimized State
private bool _isMinimized = false;
private int _unreadCount = 0;
private Vector2 _expandedSize;
private Vector2 _minimizedSize = new Vector2(180f, 40f);
// Layout Constants
private const float HeaderHeight = 50f;
private const float FooterHeight = 50f;
private const float AvatarSize = 40f;
private const float BubblePadding = 10f;
private const float MessageSpacing = 15f;
private const float MaxBubbleWidthRatio = 0.75f;
public Overlay_WulaLink(EventDef def)
{
this._def = def;
this._eventDefName = def.defName;
// Window Properties (Floating, Non-Modal)
this.layer = WindowLayer.GameUI;
this.forcePause = false;
this.absorbInputAroundWindow = false;
this.closeOnClickedOutside = false;
this.closeOnAccept = false; // 防止 Enter 键误关闭
this.closeOnCancel = false; // 防止 Esc 键误关闭
this.doWindowBackground = false; // We draw our own
this.drawShadow = false; // 禁用阴影
this.draggable = true;
this.resizeable = true;
this.preventCameraMotion = false;
// Initial Size (Phone-like)
_expandedSize = new Vector2(380f, 600f);
}
public override Vector2 InitialSize => _isMinimized ? _minimizedSize : _expandedSize;
public void ToggleMinimize()
{
_isMinimized = !_isMinimized;
_unreadCount = 0; // Reset on toggle? Or only on expand? Let's say expand.
if (_isMinimized)
{
// 最小化时保持当前位置,只调整大小
windowRect.width = _minimizedSize.x;
windowRect.height = _minimizedSize.y;
// 确保不超出屏幕边界
windowRect.x = Mathf.Clamp(windowRect.x, 0, Verse.UI.screenWidth - _minimizedSize.x);
windowRect.y = Mathf.Clamp(windowRect.y, 0, Verse.UI.screenHeight - _minimizedSize.y);
}
else
{
// 展开时仅恢复大小,不强制居中
windowRect.width = _expandedSize.x;
windowRect.height = _expandedSize.y;
// 确保展开后不超出右侧边界(简单边界检查)
if (windowRect.xMax > Verse.UI.screenWidth)
{
windowRect.x = Verse.UI.screenWidth - windowRect.width - 20f;
}
}
}
public void Expand()
{
if (_isMinimized) ToggleMinimize();
Find.WindowStack.Notify_ManuallySetFocus(this);
}
public override void PreOpen()
{
base.PreOpen();
// Connect to Core
_core = Find.World.GetComponent<AIIntelligenceCore>();
if (_core != null)
{
_core.InitializeConversation(_eventDefName);
_core.OnMessageReceived += OnMessageReceived;
_core.OnThinkingStateChanged += OnThinkingStateChanged;
_core.OnExpressionChanged += OnExpressionChanged;
}
}
public override void PostClose()
{
base.PostClose();
if (_core != null)
{
_core.OnMessageReceived -= OnMessageReceived;
_core.OnThinkingStateChanged -= OnThinkingStateChanged;
_core.OnExpressionChanged -= OnExpressionChanged;
}
}
private void OnMessageReceived(string msg)
{
_scrollToBottom = true;
if (_isMinimized)
{
_unreadCount++;
// Spawn Notification Bubble
Find.WindowStack.Add(new Overlay_WulaLink_Notification(msg));
}
}
private void OnThinkingStateChanged(bool thinking)
{
// Trigger repaint or animation update if needed
}
protected override float Margin => 0f;
private void OnExpressionChanged(int id)
{
// Repaint happens next frame
}
public override void DoWindowContents(Rect inRect)
{
this.resizeable = !_isMinimized;
if (_isMinimized)
{
// 强制同步 windowRect 到设计尺寸 (Margin 为 0 时直接匹配)
if (windowRect.width != _minimizedSize.x || windowRect.height != _minimizedSize.y)
{
windowRect.width = _minimizedSize.x;
windowRect.height = _minimizedSize.y;
}
DrawMinimized(inRect);
return;
}
// Draw Main Background (Whole Window)
Widgets.DrawBoxSolid(inRect, WulaLinkStyles.BackgroundColor);
GUI.color = new Color(0.8f, 0.8f, 0.8f);
Widgets.DrawBox(inRect, 1); // Border
GUI.color = Color.white;
// Areas
Rect headerRect = new Rect(0, 0, inRect.width, HeaderHeight);
Rect footerRect = new Rect(0, inRect.height - FooterHeight, inRect.width, FooterHeight);
Rect contextRect = new Rect(0, inRect.height - FooterHeight - 30f, inRect.width, 30f); // Context Bar
Rect bodyRect = new Rect(0, HeaderHeight, inRect.width, inRect.height - HeaderHeight - FooterHeight - 30f);
// Draw Components
DrawHeader(headerRect);
DrawMessageList(bodyRect);
DrawContextBar(contextRect);
DrawFooter(footerRect);
}
private void DrawMinimized(Rect rect)
{
// AI 核心挂件背景
Widgets.DrawBoxSolid(rect, new Color(0.1f, 0.1f, 0.1f, 0.9f));
GUI.color = WulaLinkStyles.HeaderColor;
Widgets.DrawBox(rect, 2);
GUI.color = Color.white;
// 左侧:大型方形头像
float avaSize = rect.height - 16f;
Rect avatarRect = new Rect(8f, 8f, avaSize, avaSize);
int expId = _core?.ExpressionId ?? 1;
string portraitPath = _def.portraitPath ?? $"Wula/Events/Portraits/WULA_Legion_{expId}";
Texture2D portrait = ContentFinder<Texture2D>.Get(portraitPath, false);
if (portrait != null)
{
GUI.DrawTexture(avatarRect, portrait, ScaleMode.ScaleToFit);
}
else
{
Widgets.DrawBoxSolid(avatarRect, new Color(0.3f, 0.3f, 0.3f));
}
// 右侧:状态展示
float rightContentX = avatarRect.xMax + 12f;
float btnWidth = 30f;
// Status Info
string status = _core.IsThinking ? "Thinking..." : "Standby";
Color statusColor = _core.IsThinking ? Color.yellow : Color.green;
// 绘制状态文字
Rect textRect = new Rect(rightContentX, 0, rect.width - rightContentX - btnWidth - 5f, rect.height);
Text.Anchor = TextAnchor.MiddleLeft;
Text.Font = GameFont.Small;
GUI.color = statusColor;
Widgets.Label(textRect, status);
GUI.color = Color.white;
// 右侧:小巧的展开按钮
Rect expandBtnRect = new Rect(rect.width - 32f, (rect.height - 25f) / 2f, 25f, 25f);
if (DrawHeaderButton(expandBtnRect, "+"))
{
ToggleMinimize();
}
// 未读标签角标 (头像右上角)
if (_unreadCount > 0)
{
float badgeSize = 18f;
Rect badgeRect = new Rect(avatarRect.xMax - 10f, avatarRect.y - 5f, badgeSize, badgeSize);
GUI.DrawTexture(badgeRect, BaseContent.WhiteTex);
GUI.color = Color.red;
Widgets.DrawBoxSolid(badgeRect.ContractedBy(1f), Color.red);
GUI.color = Color.white;
Text.Font = GameFont.Tiny;
Text.Anchor = TextAnchor.MiddleCenter;
Widgets.Label(badgeRect, _unreadCount.ToString());
}
Text.Anchor = TextAnchor.UpperLeft;
}
private void DrawContextBar(Rect rect)
{
// Context Awareness
Widgets.DrawBoxSolid(rect, new Color(0.15f, 0.15f, 0.15f, 1f));
GUI.color = Color.grey;
Widgets.DrawLineHorizontal(rect.x, rect.y, rect.width);
string contextInfo = "Context: None";
if (Find.Selector.SingleSelectedThing != null)
{
contextInfo = $"Context: [{Find.Selector.SingleSelectedThing.LabelCap}]";
}
else if (Find.Selector.SelectedObjects.Count > 1)
{
contextInfo = $"Context: {Find.Selector.SelectedObjects.Count} objects selected";
}
Text.Anchor = TextAnchor.MiddleLeft;
Text.Font = GameFont.Tiny;
Widgets.Label(new Rect(rect.x + 10f, rect.y, rect.width - 20f, rect.height), contextInfo);
Text.Anchor = TextAnchor.UpperLeft;
GUI.color = Color.white;
}
private void DrawHeader(Rect rect)
{
// Header BG
Widgets.DrawBoxSolid(rect, WulaLinkStyles.HeaderColor);
// Title
Text.Font = GameFont.Medium;
Text.Anchor = TextAnchor.MiddleLeft;
GUI.color = Color.white;
Rect titleRect = rect;
titleRect.x += 10f;
Widgets.Label(titleRect, _def.characterName ?? "MomoTalk");
// Header Icons (Minimize/Close) - 自定义样式
Rect closeRect = new Rect(rect.width - 35f, 10f, 25f, 25f);
Rect minRect = new Rect(rect.width - 65f, 10f, 25f, 25f);
// 最小化按钮
if (DrawHeaderButton(minRect, "-"))
{
ToggleMinimize();
}
// 关闭按钮
if (DrawHeaderButton(closeRect, "X"))
{
Close();
}
GUI.color = Color.white;
Text.Anchor = TextAnchor.UpperLeft;
}
private bool DrawHeaderButton(Rect rect, string label)
{
bool isMouseOver = Mouse.IsOver(rect);
Color buttonColor = isMouseOver
? new Color(0.6f, 0.3f, 0.3f, 1f) // Hover: 深红色
: new Color(0.4f, 0.2f, 0.2f, 0.8f); // Normal: 暗红色
Color textColor = isMouseOver ? Color.white : new Color(0.9f, 0.9f, 0.9f);
var originalColor = GUI.color;
var originalAnchor = Text.Anchor;
var originalFont = Text.Font;
GUI.color = buttonColor;
Widgets.DrawBoxSolid(rect, buttonColor);
GUI.color = textColor;
Text.Font = GameFont.Small;
Text.Anchor = TextAnchor.MiddleCenter;
Widgets.Label(rect, label);
GUI.color = originalColor;
Text.Anchor = originalAnchor;
Text.Font = originalFont;
return Widgets.ButtonInvisible(rect);
}
private void DrawMessageList(Rect rect)
{
var history = _core?.GetHistorySnapshot();
if (history == null) return;
// Filter out tool messages, empty messages, and XML-only messages
var displayHistory = new List<(string role, string message, string displayText)>();
foreach (var msg in history)
{
// Skip tool/toolcall messages to avoid empty spacing
if (msg.role == "tool" || msg.role == "toolcall") continue;
if (msg.role == "system" && !Prefs.DevMode) continue;
// For assistant messages, strip XML and check if empty
string displayText = msg.message;
if (msg.role == "assistant")
{
displayText = StripXmlTags(msg.message)?.Trim() ?? "";
}
// Skip empty or whitespace-only messages
if (string.IsNullOrWhiteSpace(displayText)) continue;
displayHistory.Add((msg.role, msg.message, displayText));
}
// Setup ScrollView
float contentHeight = 0f;
float width = rect.width - 26f; // Scrollbar space
float reducedSpacing = 8f;
List<float> heights = new List<float>();
foreach (var msg in displayHistory)
{
string textToMeasure = msg.role == "assistant" ? msg.displayText : msg.message;
float h = CalcMessageHeight(textToMeasure, width);
heights.Add(h);
contentHeight += h + reducedSpacing;
}
if (_core.IsThinking)
{
contentHeight += 40f;
}
Rect viewRect = new Rect(0, 0, width, contentHeight);
if (_scrollToBottom)
{
_scrollPosition.y = contentHeight - rect.height;
_scrollToBottom = false;
}
Widgets.BeginScrollView(rect, ref _scrollPosition, viewRect);
float curY = 10f;
for (int i = 0; i < displayHistory.Count; i++)
{
var entry = displayHistory[i];
float h = heights[i];
Rect msgRect = new Rect(0, curY, width, h);
if (entry.role == "user")
{
DrawSenseiMessage(msgRect, entry.message);
}
else if (entry.role == "assistant")
{
DrawStudentMessage(msgRect, entry.displayText);
}
else if (entry.role == "tool" || entry.role == "toolcall")
{
DrawSystemMessage(msgRect, $"[Tool] {entry.message}");
}
else if (entry.role == "system")
{
DrawSystemMessage(msgRect, entry.message);
}
curY += h + reducedSpacing;
}
if (_core.IsThinking)
{
Rect thinkingRect = new Rect(0, curY, width, 30f);
DrawThinkingIndicator(thinkingRect);
}
Widgets.EndScrollView();
}
private static string StripXmlTags(string text)
{
if (string.IsNullOrEmpty(text)) return text;
// Remove XML tags with content: <tag>content</tag>
string stripped = System.Text.RegularExpressions.Regex.Replace(text, @"<([a-zA-Z0-9_]+)[^>]*>.*?</\1>", "", System.Text.RegularExpressions.RegexOptions.Singleline);
// Remove self-closing tags: <tag/>
stripped = System.Text.RegularExpressions.Regex.Replace(stripped, @"<([a-zA-Z0-9_]+)[^>]*/?>", "");
return stripped;
}
private void DrawFooter(Rect rect)
{
Widgets.DrawBoxSolid(rect, WulaLinkStyles.InputBarColor);
Widgets.DrawLineHorizontal(rect.x, rect.y, rect.width); // Top border
float padding = 8f;
float btnWidth = 40f;
float inputWidth = rect.width - btnWidth - (padding * 3);
Rect inputRect = new Rect(rect.x + padding, rect.y + padding, inputWidth, rect.height - (padding * 2));
Rect btnRect = new Rect(inputRect.xMax + padding, rect.y + padding, btnWidth, rect.height - (padding * 2));
// Input Field
string nextInput = Widgets.TextField(inputRect, _inputText);
if (nextInput != _inputText)
{
_inputText = nextInput;
}
// Send Button (Simulate Enter key or Click)
bool enterPressed = (Event.current.type == EventType.KeyDown && (Event.current.keyCode == KeyCode.Return || Event.current.keyCode == KeyCode.KeypadEnter) && GUI.GetNameOfFocusedControl() == "WulaInput");
bool sendClicked = DrawCustomButton(btnRect, ">", !string.IsNullOrWhiteSpace(_inputText));
if (sendClicked || enterPressed)
{
if (!string.IsNullOrWhiteSpace(_inputText))
{
bool wasFocused = GUI.GetNameOfFocusedControl() == "WulaInput";
_core.SendUserMessage(_inputText);
_inputText = "";
if (wasFocused) GUI.FocusControl("WulaInput");
}
}
// Handle focus for Enter key to work
if (Mouse.IsOver(inputRect) && Event.current.type == EventType.MouseDown)
{
GUI.FocusControl("WulaInput");
}
GUI.SetNextControlName("WulaInput");
}
// =================================================================================
// Message Rendering Helpers
// =================================================================================
private float CalcMessageHeight(string text, float containerWidth)
{
float maxBubbleWidth = containerWidth * MaxBubbleWidthRatio;
float effectiveWidth = maxBubbleWidth - AvatarSize - 20f;
Text.Font = WulaLinkStyles.MessageFont;
float textH = Text.CalcHeight(text, effectiveWidth - (BubblePadding * 2));
return Mathf.Max(textH + (BubblePadding * 2) + 8f, AvatarSize + 8f);
}
private void DrawSenseiMessage(Rect rect, string text)
{
// 用户消息 - 右对齐蓝色气泡
float maxBubbleWidth = rect.width * MaxBubbleWidthRatio;
Text.Font = WulaLinkStyles.MessageFont;
float textWidth = maxBubbleWidth - (BubblePadding * 2);
float textHeight = Text.CalcHeight(text, textWidth);
float bubbleHeight = textHeight + (BubblePadding * 2);
float bubbleWidth = Mathf.Min(Text.CalcSize(text).x + (BubblePadding * 2) + 10f, maxBubbleWidth);
Rect bubbleRect = new Rect(rect.xMax - bubbleWidth - 10f, rect.y, bubbleWidth, bubbleHeight);
// 绘制气泡背景 - 蓝色
Color bubbleColor = new Color(0.29f, 0.54f, 0.78f, 1f);
Widgets.DrawBoxSolid(bubbleRect, bubbleColor);
// 绘制文字
GUI.color = Color.white;
Text.Anchor = TextAnchor.MiddleLeft;
Widgets.Label(bubbleRect.ContractedBy(BubblePadding), text);
Text.Anchor = TextAnchor.UpperLeft;
}
private void DrawStudentMessage(Rect rect, string text)
{
// AI消息 - 左对齐带方形头像
float avatarX = 10f;
Rect avatarRect = new Rect(avatarX, rect.y, AvatarSize, AvatarSize);
// 绘制方形头像
int expId = _core?.ExpressionId ?? 1;
string portraitPath = _def.portraitPath ?? $"Wula/Events/Portraits/WULA_Legion_{expId}";
if (expId > 1 && _def.portraitPath == null)
{
portraitPath = $"Wula/Events/Portraits/WULA_Legion_{expId}";
}
Texture2D portrait = ContentFinder<Texture2D>.Get(portraitPath, false);
if (portrait != null)
{
GUI.DrawTexture(avatarRect, portrait, ScaleMode.ScaleToFit);
}
else
{
Widgets.DrawBoxSolid(avatarRect, Color.gray);
}
// 气泡
float maxBubbleWidth = rect.width * MaxBubbleWidthRatio;
float bubbleX = avatarRect.xMax + 10f;
Text.Font = WulaLinkStyles.MessageFont;
float textWidth = maxBubbleWidth - AvatarSize - 20f;
float textHeight = Text.CalcHeight(text, textWidth - (BubblePadding * 2));
float bubbleHeight = textHeight + (BubblePadding * 2);
float bubbleWidth = Mathf.Min(Text.CalcSize(text).x + (BubblePadding * 2) + 10f, maxBubbleWidth - AvatarSize - 20f);
Rect bubbleRect = new Rect(bubbleX, rect.y, bubbleWidth, bubbleHeight);
// 绘制气泡背景 - 灰色
Color bubbleColor = new Color(0.85f, 0.85f, 0.87f, 1f);
Widgets.DrawBoxSolid(bubbleRect, bubbleColor);
// 绘制边框
GUI.color = new Color(0.6f, 0.6f, 0.65f, 1f);
Widgets.DrawBox(bubbleRect, 1);
GUI.color = Color.white;
// 绘制文字
GUI.color = new Color(0.1f, 0.1f, 0.1f, 1f);
Text.Anchor = TextAnchor.MiddleLeft;
Widgets.Label(bubbleRect.ContractedBy(BubblePadding), text);
Text.Anchor = TextAnchor.UpperLeft;
GUI.color = Color.white;
}
private void DrawSystemMessage(Rect rect, string text)
{
// Centered gray text or Pink box
if (text.Contains("[Tool]")) return; // Skip logic log in main view if needed, but here we draw minimal
GUI.color = Color.gray;
Text.Font = GameFont.Tiny;
Text.Anchor = TextAnchor.MiddleCenter;
Widgets.Label(rect, text);
Text.Anchor = TextAnchor.UpperLeft;
GUI.color = Color.white;
}
private void DrawThinkingIndicator(Rect rect)
{
Text.Anchor = TextAnchor.MiddleLeft;
GUI.color = Color.gray;
Rect iconRect = new Rect(rect.x + 60f, rect.y, 24f, 24f);
Rect labelRect = new Rect(iconRect.xMax + 5f, rect.y, 200f, 24f);
// Draw a simple box as thinking indicator if TexUI is missing
Widgets.DrawBoxSolid(iconRect, Color.gray);
Widgets.Label(labelRect, "P.I.A is thinking...");
Text.Anchor = TextAnchor.UpperLeft;
GUI.color = Color.white;
}
private bool DrawCustomButton(Rect rect, string label, bool isEnabled)
{
bool isMouseOver = Mouse.IsOver(rect);
Color originalColor = GUI.color;
TextAnchor originalAnchor = Text.Anchor;
GameFont originalFont = Text.Font;
Color buttonColor;
Color textColor;
if (!isEnabled)
{
buttonColor = Dialog_CustomDisplay.CustomButtonDisabledColor;
textColor = Dialog_CustomDisplay.CustomButtonTextDisabledColor;
}
else if (isMouseOver)
{
buttonColor = Dialog_CustomDisplay.CustomButtonHoverColor;
textColor = Dialog_CustomDisplay.CustomButtonTextHoverColor;
}
else
{
buttonColor = Dialog_CustomDisplay.CustomButtonNormalColor;
textColor = Dialog_CustomDisplay.CustomButtonTextNormalColor;
}
GUI.color = buttonColor;
Widgets.DrawBoxSolid(rect, buttonColor);
Widgets.DrawBox(rect, 1);
GUI.color = textColor;
Text.Anchor = TextAnchor.MiddleCenter;
Text.Font = GameFont.Tiny;
Widgets.Label(rect.ContractedBy(4f), label);
GUI.color = originalColor;
Text.Anchor = originalAnchor;
Text.Font = originalFont;
if (!isEnabled)
{
return false;
}
return Widgets.ButtonInvisible(rect);
}
}
}