287 lines
12 KiB
C#
287 lines
12 KiB
C#
using System;
|
||
using System.Collections.Concurrent;
|
||
using System.Collections.Generic;
|
||
using System.Collections.ObjectModel;
|
||
using System.ComponentModel;
|
||
using System.IO;
|
||
using System.Net.Http;
|
||
using System.Net.Http.Headers;
|
||
using System.Windows;
|
||
using System.Windows.Controls;
|
||
using Microsoft.Extensions.DependencyInjection;
|
||
using Microsoft.Extensions.Hosting;
|
||
using Microsoft.Extensions.Logging;
|
||
using System.Text.Json;
|
||
using System.Threading;
|
||
using System.Windows.Documents;
|
||
using System.Windows.Media;
|
||
using ArcGIS.Desktop.Core;
|
||
using ArcGIS.Desktop.Core.Geoprocessing;
|
||
using LinkToolAddin.client;
|
||
using LinkToolAddin.common;
|
||
using LinkToolAddin.host;
|
||
using LinkToolAddin.host.llm;
|
||
using LinkToolAddin.host.llm.entity;
|
||
using LinkToolAddin.message;
|
||
using LinkToolAddin.resource;
|
||
using LinkToolAddin.server;
|
||
using log4net;
|
||
using log4net.Appender;
|
||
using log4net.Config;
|
||
using log4net.Layout;
|
||
using ModelContextProtocol.Client;
|
||
using ModelContextProtocol.Protocol.Types;
|
||
using ModelContextProtocol.Server;
|
||
using Newtonsoft.Json;
|
||
|
||
namespace LinkToolAddin.ui.dockpane
|
||
{
|
||
public class ItemModel
|
||
{
|
||
public string Content { get; set; }
|
||
}
|
||
/// <summary>
|
||
/// Interaction logic for DialogDockpaneView.xaml
|
||
/// </summary>
|
||
public partial class DialogDockpaneView : UserControl
|
||
{
|
||
private static ILog log = LogManager.GetLogger(typeof(DialogDockpaneView));
|
||
|
||
private List<string> idList = new List<string>();
|
||
private ConcurrentDictionary<string,MessageListItem> messageDict = new ConcurrentDictionary<string,MessageListItem>();
|
||
private ConcurrentDictionary<string,Border> borderItemsDict = new ConcurrentDictionary<string,Border>();
|
||
|
||
public DialogDockpaneView()
|
||
{
|
||
InitLogger();
|
||
InitializeComponent();
|
||
DataContext = this;
|
||
}
|
||
|
||
private async void TestServer_OnClick(object sender, RoutedEventArgs e)
|
||
{
|
||
log.Info("TestServer Clicked");
|
||
}
|
||
|
||
protected void InitLogger()
|
||
{
|
||
// 1. 创建控制台输出器（Appender）
|
||
var consoleAppender = new ConsoleAppender
|
||
{
|
||
Layout = new PatternLayout("%date [%thread] %-5level %logger - %message%newline"),
|
||
Threshold = log4net.Core.Level.Info // 仅输出 Info 及以上级别
|
||
};
|
||
consoleAppender.ActivateOptions(); // 激活配置
|
||
|
||
// 2. 创建文件滚动输出器（按大小滚动）
|
||
var fileAppender = new RollingFileAppender
|
||
{
|
||
File = Path.Combine("Logs", "D:\\linktool_app.log"), // 日志文件路径
|
||
AppendToFile = true, // 追加模式
|
||
RollingStyle = RollingFileAppender.RollingMode.Size, // 按文件大小滚动
|
||
MaxSizeRollBackups = 10, // 保留 10 个历史文件
|
||
MaximumFileSize = "1MB", // 单个文件最大 1MB
|
||
StaticLogFileName = true, // 固定文件名（否则自动追加序号）
|
||
Layout = new PatternLayout("%date [%thread] %-5level %logger - %message%newline"),
|
||
Threshold = log4net.Core.Level.Info // 仅输出 Info 及以上级别
|
||
};
|
||
fileAppender.ActivateOptions(); // 激活配置
|
||
|
||
// 3. 直接通过 BasicConfigurator 注册 Appender
|
||
BasicConfigurator.Configure(consoleAppender, fileAppender);
|
||
|
||
log = LogManager.GetLogger(typeof(DialogDockpaneView));
|
||
|
||
// 测试日志输出
|
||
log.Debug("Debug 日志（控制台可见）");
|
||
log.Info("Info 日志（控制台和文件可见）");
|
||
log.Error("Error 日志（严重问题）");
|
||
}
|
||
|
||
private void SendButton_OnClick(object sender, RoutedEventArgs e)
|
||
{
|
||
string question = QuestionTextbox.Text;
|
||
string defaultGdbPath = Project.Current.DefaultGeodatabasePath;
|
||
string gdbPath = @"";
|
||
long timestamp = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
|
||
MessageListItem userMsg = new ChatMessageItem()
|
||
{
|
||
content = question,
|
||
role = "user",
|
||
type = MessageType.CHAT_MESSAGE,
|
||
id = timestamp.ToString()
|
||
};
|
||
Border userMsgBoder = GetUserChatBorder(userMsg);
|
||
idList.Add(timestamp.ToString());
|
||
messageDict[timestamp.ToString()] = userMsg;
|
||
QuestionTextbox.Text = "";
|
||
borderItemsDict[timestamp.ToString()] = userMsgBoder;
|
||
ChatHistoryStackPanel.Children.Add(userMsgBoder);
|
||
Gateway.SendMessageStream(question,"qwen-max",defaultGdbPath,NewMessage_Recall);
|
||
}
|
||
|
||
public void NewMessage_Recall(MessageListItem msg)
|
||
{
|
||
string msgId = msg.id;
|
||
log.Info(msg.content);
|
||
if (!idList.Contains(msgId))
|
||
{
|
||
//不存在该消息，需添加到ListView中
|
||
if (msg.content == "")
|
||
{
|
||
return;
|
||
}
|
||
idList.Add(msgId);
|
||
messageDict[msgId] = msg;
|
||
if (msg.role == "user")
|
||
{
|
||
if (msg.type == MessageType.TOOL_MESSAGE)
|
||
{
|
||
Border border = GetToolChatBorder(msg);
|
||
borderItemsDict[msgId] = border;
|
||
ChatHistoryStackPanel.Children.Add(border);
|
||
}else if (msg.type == MessageType.CHAT_MESSAGE)
|
||
{
|
||
Border border = GetUserChatBorder(msg);
|
||
borderItemsDict[msgId] = border;
|
||
ChatHistoryStackPanel.Children.Add(border);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
Border border = GetAiChatBorder(msg);
|
||
borderItemsDict[msgId] = border;
|
||
ChatHistoryStackPanel.Children.Add(border);
|
||
}
|
||
}
|
||
else
|
||
{
|
||
//已有该消息，只需要修改内容
|
||
messageDict[msgId] = msg;
|
||
if (msg.role == "user")
|
||
{
|
||
if (msg.type == MessageType.TOOL_MESSAGE)
|
||
{
|
||
Border borderItem = borderItemsDict[msgId];
|
||
Grid grid = borderItem.Child as Grid;
|
||
TextBlock textBlock = grid.Children[1] as TextBlock;
|
||
textBlock.Text = (msg as ToolMessageItem).toolName;
|
||
}else if (msg.type == MessageType.CHAT_MESSAGE)
|
||
{
|
||
Border borderItem = borderItemsDict[msgId];
|
||
Grid grid = borderItem.Child as Grid;
|
||
TextBox textBox = grid.Children[1] as TextBox;
|
||
textBox.Text = msg.content;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
Border borderItem = borderItemsDict[msgId];
|
||
Grid grid = borderItem.Child as Grid;
|
||
TextBox textBox = grid.Children[1] as TextBox;
|
||
textBox.Text = msg.content;
|
||
}
|
||
}
|
||
}
|
||
|
||
private Border GetAiChatBorder(MessageListItem msg)
|
||
{
|
||
Border border = new Border();
|
||
border.Margin = new Thickness(8, 12, 8, 12);
|
||
border.BorderThickness = new Thickness(0);
|
||
// border.Background = Brushes.DarkSeaGreen;
|
||
Grid grid = new Grid();
|
||
Image icon = new Image()
|
||
{
|
||
Source = LocalResource.ReadImageByResource("LinkToolAddin.resource.img.linktool.png"),
|
||
Stretch = Stretch.Fill,
|
||
HorizontalAlignment = HorizontalAlignment.Center, VerticalAlignment = VerticalAlignment.Top
|
||
};
|
||
grid.ColumnDefinitions.Add(new ColumnDefinition(){Width = new GridLength(24, GridUnitType.Pixel)});
|
||
grid.ColumnDefinitions.Add(new ColumnDefinition(){Width = new GridLength(1, GridUnitType.Star)});
|
||
TextBox textBox = new TextBox();
|
||
grid.Children.Add(icon);
|
||
grid.Children.Add(textBox);
|
||
Grid.SetColumn(icon, 0);
|
||
Grid.SetColumn(textBox, 1);
|
||
textBox.Background = Brushes.Transparent;
|
||
textBox.Text = msg.content;
|
||
textBox.TextWrapping = TextWrapping.Wrap;
|
||
border.Child = grid;
|
||
return border;
|
||
}
|
||
|
||
private Border GetUserChatBorder(MessageListItem msg)
|
||
{
|
||
Border border = new Border();
|
||
border.Margin = new Thickness(8, 12, 8, 12);
|
||
border.BorderThickness = new Thickness(0);
|
||
// border.Background = Brushes.DarkSeaGreen;
|
||
Grid grid = new Grid();
|
||
Image icon = new Image()
|
||
{
|
||
Source = LocalResource.ReadImageByResource("LinkToolAddin.resource.img.user.png"),
|
||
Stretch = Stretch.Fill,
|
||
HorizontalAlignment = HorizontalAlignment.Center, VerticalAlignment = VerticalAlignment.Top
|
||
};
|
||
grid.ColumnDefinitions.Add(new ColumnDefinition(){Width = new GridLength(1, GridUnitType.Star)});
|
||
grid.ColumnDefinitions.Add(new ColumnDefinition(){Width = new GridLength(24, GridUnitType.Pixel)});
|
||
TextBox textBox = new TextBox();
|
||
grid.Children.Add(icon);
|
||
grid.Children.Add(textBox);
|
||
Grid.SetColumn(icon, 1);
|
||
Grid.SetColumn(textBox, 0);
|
||
textBox.Background = Brushes.Transparent;
|
||
textBox.Text = msg.content;
|
||
textBox.TextWrapping = TextWrapping.Wrap;
|
||
border.Child = grid;
|
||
return border;
|
||
}
|
||
|
||
private Border GetToolChatBorder(MessageListItem msg)
|
||
{
|
||
Border border = new Border();
|
||
border.Margin = new Thickness(24);
|
||
border.Padding = new Thickness(8);
|
||
border.BorderThickness = new Thickness(1);
|
||
border.BorderBrush = Brushes.Gray;
|
||
// border.Background = Brushes.DarkSeaGreen;
|
||
Grid grid = new Grid();
|
||
Image icon = new Image()
|
||
{
|
||
Source = LocalResource.ReadImageByResource("LinkToolAddin.resource.img.tool.png"),
|
||
Stretch = Stretch.Fill,
|
||
HorizontalAlignment = HorizontalAlignment.Center, VerticalAlignment = VerticalAlignment.Center
|
||
};
|
||
icon.Margin = new Thickness(0, 0, 8, 0);
|
||
grid.ColumnDefinitions.Add(new ColumnDefinition(){Width = new GridLength(24, GridUnitType.Pixel)});
|
||
grid.ColumnDefinitions.Add(new ColumnDefinition(){Width = new GridLength(1, GridUnitType.Star)});
|
||
TextBlock textBlock = new TextBlock();
|
||
textBlock.Text = (msg as ToolMessageItem).toolName;
|
||
textBlock.TextWrapping = TextWrapping.Wrap;
|
||
grid.Children.Add(icon);
|
||
grid.Children.Add(textBlock);
|
||
Grid.SetColumn(icon, 0);
|
||
Grid.SetColumn(textBlock, 1);
|
||
border.Child = grid;
|
||
return border;
|
||
}
|
||
|
||
private void TestButton_OnClick(object sender, RoutedEventArgs e)
|
||
{
|
||
long timestamp = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
|
||
MessageListItem toolMessageItem = new ToolMessageItem
|
||
{
|
||
toolName = "toolName",
|
||
toolParams = new Dictionary<string,object>(),
|
||
type = MessageType.TOOL_MESSAGE,
|
||
status = "success",
|
||
content = "JsonConvert.SerializeObject(toolResponse)",
|
||
id = (timestamp + 1).ToString(),
|
||
role = "user"
|
||
};
|
||
NewMessage_Recall(toolMessageItem);
|
||
}
|
||
}
|
||
}
|