修改提交
This commit is contained in:
Bob.Song
@@ -0,0 +1,470 @@
|
||||
using UnityEditor;
|
||||
using UnityEngine;
|
||||
using UnityTcp.Editor.Helpers;
|
||||
|
||||
namespace UnityTcp.Editor.Windows
|
||||
{
|
||||
public class TcpBridgeControlWindow : EditorWindow
|
||||
{
|
||||
private enum BridgeState { NotStarted, Starting, Started }
|
||||
|
||||
private double lastStatusUpdate;
|
||||
private const double StatusUpdateInterval = 0.5;
|
||||
private BridgeState currentState = BridgeState.NotStarted;
|
||||
private bool isStarting = false;
|
||||
private double startingTime = 0;
|
||||
private const double StartingTimeout = 3.0;
|
||||
|
||||
// Window title icon
|
||||
private Texture2D titleIcon;
|
||||
|
||||
// Client icons
|
||||
private Texture2D codelyIcon;
|
||||
private Texture2D vscodeIcon;
|
||||
private Texture2D visualStudioIcon;
|
||||
private Texture2D jetbrainsIcon;
|
||||
private Texture2D cliIcon;
|
||||
private Texture2D unityEditorIcon;
|
||||
private Texture2D tuanjieIcon;
|
||||
|
||||
// Fonts
|
||||
private Font firaMono;
|
||||
private Font inter;
|
||||
|
||||
// Button textures
|
||||
private Texture2D connectTex;
|
||||
private Texture2D connectHoverTex;
|
||||
private Texture2D connectPressedTex;
|
||||
private Texture2D connectingTex;
|
||||
private Texture2D disconnectTex;
|
||||
private Texture2D disconnectHoverTex;
|
||||
private Texture2D disconnectPressedTex;
|
||||
|
||||
// Connection tracking
|
||||
private bool connectedToCLI;
|
||||
private bool connectedToVSCode;
|
||||
private bool connectedToVisualStudio;
|
||||
private bool connectedToJetBrains;
|
||||
private bool connectedToUnityEditor;
|
||||
private bool connectedToTuanjie;
|
||||
|
||||
private static readonly Color GreenColor = new Color(0x01 / 255f, 0xA7 / 255f, 0x7F / 255f);
|
||||
private static readonly Color GrayColor = new Color(0.5f, 0.5f, 0.5f);
|
||||
private static readonly Color SubtextColor = new Color(0x9C / 255f, 0xA3 / 255f, 0xAF / 255f);
|
||||
|
||||
// ── Scale ────────────────────────────────────────────────────────────────
|
||||
private const float ScaleFactor = 2 / 1.5f;
|
||||
|
||||
// ── Base sizes (ScaleFactor = 1) ─────────────────────────────────────────
|
||||
private const float BaseWindowWidth = 430f;
|
||||
private const float BaseLeftMargin = 25f;
|
||||
private const float BaseTopPadding = 40f;
|
||||
private const float BaseBottomPadding = 40f;
|
||||
private const float BaseCodelyIconSize = 40f;
|
||||
private const float BaseSpaceAfterIcon = 25f;
|
||||
private const int BaseTitleFontSize = 20;
|
||||
private const float BaseSpaceAfterTitle = 15f;
|
||||
private const int BaseBodyFontSize = 17;
|
||||
private const float BaseSpaceAfterBody = 30f;
|
||||
private const int BaseStatusFontSize = 12;
|
||||
private const float BaseSpaceBeforeStatus = 30f;
|
||||
private const float BaseMinWindowHeight = 80f;
|
||||
private const int BaseConnectedFontSize = 16;
|
||||
private const float BaseConnectedItemSpacing = 6f;
|
||||
private const float BaseClientIconSize = 24f;
|
||||
private const float BaseClientIconLabelGap = 6f;
|
||||
private const int BaseClientLabelFontSize = 13;
|
||||
private const float BaseClientLabelHeight = 18f;
|
||||
private const float BaseClientItemSpacing = 12f;
|
||||
private const float BaseConnectBtnW = 159f;
|
||||
private const float BaseConnectBtnH = 40f;
|
||||
|
||||
// ── Scaled values ─────────────────────────────────────────────────────────
|
||||
private static float WindowWidth => BaseWindowWidth * ScaleFactor;
|
||||
private static float LeftMargin => BaseLeftMargin * ScaleFactor;
|
||||
private static float TopPadding => BaseTopPadding * ScaleFactor;
|
||||
private static float BottomPadding => BaseBottomPadding * ScaleFactor;
|
||||
private static float CodelyIconSize => BaseCodelyIconSize * ScaleFactor;
|
||||
private static float SpaceAfterIcon => BaseSpaceAfterIcon * ScaleFactor;
|
||||
private static int TitleFontSize => Mathf.RoundToInt(BaseTitleFontSize * ScaleFactor);
|
||||
private static float SpaceAfterTitle => BaseSpaceAfterTitle * ScaleFactor;
|
||||
private static int BodyFontSize => Mathf.RoundToInt(BaseBodyFontSize * ScaleFactor);
|
||||
private static float SpaceAfterBody => BaseSpaceAfterBody * ScaleFactor;
|
||||
private static int StatusFontSize => Mathf.RoundToInt(BaseStatusFontSize * ScaleFactor);
|
||||
private static float SpaceBeforeStatus => BaseSpaceBeforeStatus * ScaleFactor;
|
||||
private static float MinWindowHeight => BaseMinWindowHeight * ScaleFactor;
|
||||
private static int ConnectedFontSize => Mathf.RoundToInt(BaseConnectedFontSize * ScaleFactor);
|
||||
private static float ConnectedItemSpacing => BaseConnectedItemSpacing * ScaleFactor;
|
||||
private static float ClientIconSize => BaseClientIconSize * ScaleFactor;
|
||||
private static float ClientIconLabelGap => BaseClientIconLabelGap * ScaleFactor;
|
||||
private static int ClientLabelFontSize => Mathf.RoundToInt(BaseClientLabelFontSize * ScaleFactor);
|
||||
private static float ClientLabelHeight => BaseClientLabelHeight * ScaleFactor;
|
||||
private static float ClientItemSpacing => BaseClientItemSpacing * ScaleFactor;
|
||||
private static float ConnectBtnW => BaseConnectBtnW * ScaleFactor;
|
||||
private static float ConnectBtnH => BaseConnectBtnH * ScaleFactor;
|
||||
|
||||
[MenuItem("Window/Codely Bridge", priority = 1001)]
|
||||
public static void ShowWindow()
|
||||
{
|
||||
var window = GetWindow<TcpBridgeControlWindow>("Codely Bridge");
|
||||
window.minSize = new Vector2(WindowWidth, MinWindowHeight);
|
||||
window.maxSize = new Vector2(WindowWidth, 2000f);
|
||||
window.Show();
|
||||
}
|
||||
|
||||
private void OnEnable()
|
||||
{
|
||||
EditorPrefs.SetBool("UnityTcp.DebugLogs", false);
|
||||
LoadIcons();
|
||||
UpdateStatus();
|
||||
UnityTcpBridge.OnClientPlatformsChanged += OnClientPlatformsChanged;
|
||||
}
|
||||
|
||||
private void OnDisable()
|
||||
{
|
||||
UnityTcpBridge.OnClientPlatformsChanged -= OnClientPlatformsChanged;
|
||||
}
|
||||
|
||||
private void OnClientPlatformsChanged() => UpdateConnectionStatus();
|
||||
|
||||
private static Texture2D Load(string path) =>
|
||||
AssetDatabase.LoadAssetAtPath<Texture2D>(path);
|
||||
|
||||
private void LoadIcons()
|
||||
{
|
||||
const string b = "Packages/cn.tuanjie.codely.bridge/Editor/Icons/";
|
||||
titleIcon = Load(b + "title_icon.png");
|
||||
codelyIcon = Load(b + "codely.png");
|
||||
vscodeIcon = Load(b + "vscode.png");
|
||||
visualStudioIcon = Load(b + "visualstudio.png");
|
||||
jetbrainsIcon = Load(b + "jetbrains.png");
|
||||
cliIcon = Load(b + "cli.png");
|
||||
unityEditorIcon = Load(b + "unity_editor.png");
|
||||
tuanjieIcon = Load(b + "tuanjie.png");
|
||||
connectTex = Load(b + "connect.png");
|
||||
connectHoverTex = Load(b + "connect_hover.png");
|
||||
connectPressedTex = Load(b + "connect_pressed.png");
|
||||
connectingTex = Load(b + "connecting.png");
|
||||
disconnectTex = Load(b + "disconnect.png");
|
||||
disconnectHoverTex = Load(b + "disconnect_hover.png");
|
||||
disconnectPressedTex = Load(b + "disconnect_pressed.png");
|
||||
|
||||
firaMono = AssetDatabase.LoadAssetAtPath<Font>("Packages/cn.tuanjie.codely.bridge/Editor/fonts/FiraMono-Regular.ttf");
|
||||
inter = AssetDatabase.LoadAssetAtPath<Font>("Packages/cn.tuanjie.codely.bridge/Editor/fonts/Inter.ttc");
|
||||
|
||||
if (titleIcon != null)
|
||||
titleContent = new GUIContent(" Codely Bridge", titleIcon);
|
||||
}
|
||||
|
||||
private void OnGUI()
|
||||
{
|
||||
if (EditorApplication.timeSinceStartup - lastStatusUpdate > StatusUpdateInterval)
|
||||
{
|
||||
UpdateStatus();
|
||||
lastStatusUpdate = EditorApplication.timeSinceStartup;
|
||||
}
|
||||
|
||||
DrawUI();
|
||||
}
|
||||
|
||||
private static readonly Color BgColor = new Color(0.18f, 0.18f, 0.18f);
|
||||
|
||||
private void DrawUI()
|
||||
{
|
||||
EditorGUI.DrawRect(new Rect(0, 0, position.width, position.height), BgColor);
|
||||
|
||||
bool started = currentState == BridgeState.Started;
|
||||
|
||||
// ── Top padding ──────────────────────────────────────────────────────
|
||||
GUILayout.Space(TopPadding);
|
||||
|
||||
// ── Codely icon ──────────────────────────────────────────────────────
|
||||
GUILayout.BeginHorizontal();
|
||||
GUILayout.Space(LeftMargin);
|
||||
if (codelyIcon != null)
|
||||
GUILayout.Box(codelyIcon, GUIStyle.none, GUILayout.Width(CodelyIconSize), GUILayout.Height(CodelyIconSize));
|
||||
GUILayout.EndHorizontal();
|
||||
|
||||
GUILayout.Space(SpaceAfterIcon);
|
||||
|
||||
// ── Title ────────────────────────────────────────────────────────────
|
||||
GUILayout.BeginHorizontal();
|
||||
GUILayout.Space(LeftMargin);
|
||||
var titleStyle = new GUIStyle(EditorStyles.boldLabel) { fontSize = TitleFontSize, wordWrap = false, font = inter };
|
||||
if (started) {
|
||||
titleStyle.normal.textColor = GreenColor;
|
||||
titleStyle.hover.textColor = GreenColor;
|
||||
}
|
||||
else {
|
||||
titleStyle.normal.textColor = Color.white;
|
||||
titleStyle.hover.textColor = Color.white;
|
||||
}
|
||||
GUILayout.Label(
|
||||
started ? "Codely bridge is ready \u2713" : "Connect to Codely Agent",
|
||||
titleStyle);
|
||||
GUILayout.EndHorizontal();
|
||||
|
||||
GUILayout.Space(SpaceAfterTitle);
|
||||
|
||||
// ── Body (Form 1: client list | Form 2: text + status) ───────────────
|
||||
bool hasClients = started && (connectedToCLI || connectedToVSCode || connectedToVisualStudio
|
||||
|| connectedToJetBrains || connectedToUnityEditor || connectedToTuanjie);
|
||||
|
||||
GUILayout.BeginHorizontal();
|
||||
GUILayout.Space(LeftMargin);
|
||||
GUILayout.BeginVertical();
|
||||
if (hasClients)
|
||||
DrawClientListBody();
|
||||
else
|
||||
DrawNoClientBody();
|
||||
GUILayout.EndVertical();
|
||||
GUILayout.EndHorizontal();
|
||||
|
||||
GUILayout.Space(SpaceAfterBody);
|
||||
|
||||
// ── Action button ────────────────────────────────────────────────────
|
||||
Texture2D btnNormal = null, btnPressed = null, btnHover = null;
|
||||
bool btnEnabled = true;
|
||||
System.Action btnAction = null;
|
||||
|
||||
switch (currentState)
|
||||
{
|
||||
case BridgeState.NotStarted:
|
||||
btnNormal = connectTex;
|
||||
btnPressed = connectPressedTex;
|
||||
btnHover = connectHoverTex;
|
||||
btnAction = StartBridge;
|
||||
break;
|
||||
case BridgeState.Starting:
|
||||
btnNormal = connectingTex;
|
||||
btnPressed = connectingTex;
|
||||
btnHover = connectingTex;
|
||||
btnEnabled = false;
|
||||
break;
|
||||
case BridgeState.Started:
|
||||
btnNormal = disconnectTex;
|
||||
btnPressed = disconnectPressedTex;
|
||||
btnHover = disconnectHoverTex;
|
||||
btnAction = StopBridge;
|
||||
break;
|
||||
}
|
||||
|
||||
GUILayout.BeginHorizontal();
|
||||
GUILayout.Space(LeftMargin);
|
||||
|
||||
if (btnNormal != null)
|
||||
{
|
||||
float btnW = ConnectBtnW;
|
||||
float btnH = ConnectBtnH;
|
||||
|
||||
var style = new GUIStyle(GUIStyle.none);
|
||||
style.normal.background = btnNormal;
|
||||
style.active.background = btnPressed ?? btnNormal;
|
||||
style.hover.background = btnHover ?? btnNormal;
|
||||
|
||||
if (!btnEnabled) GUI.enabled = false;
|
||||
if (GUILayout.Button(GUIContent.none, style, GUILayout.Width(btnW), GUILayout.Height(btnH))
|
||||
&& btnAction != null)
|
||||
btnAction();
|
||||
if (!btnEnabled) GUI.enabled = true;
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
string fallback = currentState == BridgeState.NotStarted ? "Connect \u2192"
|
||||
: currentState == BridgeState.Starting ? "Connecting..."
|
||||
: "Disconnect";
|
||||
if (!btnEnabled) GUI.enabled = false;
|
||||
if (GUILayout.Button(fallback, GUILayout.Width(ConnectBtnW), GUILayout.Height(ConnectBtnH))
|
||||
&& btnAction != null)
|
||||
btnAction();
|
||||
if (!btnEnabled) GUI.enabled = true;
|
||||
}
|
||||
|
||||
GUILayout.EndHorizontal();
|
||||
|
||||
// ── Bottom padding ───────────────────────────────────────────────────
|
||||
GUILayout.Space(BottomPadding);
|
||||
|
||||
// ── Auto-resize height to fit content ────────────────────────────────
|
||||
if (Event.current.type == EventType.Repaint)
|
||||
{
|
||||
float targetH = Mathf.Max(MinWindowHeight, GUILayoutUtility.GetLastRect().yMax);
|
||||
if (Mathf.Abs(position.height - targetH) > 1f)
|
||||
{
|
||||
minSize = new Vector2(WindowWidth, targetH);
|
||||
maxSize = new Vector2(WindowWidth, targetH);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Form 1 – bridge started and at least one client is connected
|
||||
private void DrawClientListBody()
|
||||
{
|
||||
var labelStyle = new GUIStyle(EditorStyles.label) { fontSize = ConnectedFontSize, font = inter };
|
||||
labelStyle.normal.textColor = SubtextColor;
|
||||
labelStyle.hover.textColor = SubtextColor;
|
||||
|
||||
GUILayout.BeginHorizontal();
|
||||
|
||||
// Left column: "Connected to :" label, top-aligned
|
||||
float labelW = labelStyle.CalcSize(new GUIContent("Connected to :")).x + 4f;
|
||||
GUILayout.Label("Connected to :", labelStyle, GUILayout.Width(labelW));
|
||||
|
||||
// Right column: all clients stacked vertically
|
||||
GUILayout.BeginVertical();
|
||||
if (connectedToVSCode) DrawClient(vscodeIcon, "VS Code");
|
||||
if (connectedToVisualStudio) DrawClient(visualStudioIcon, "Visual Studio");
|
||||
if (connectedToJetBrains) DrawClient(jetbrainsIcon, "JetBrains");
|
||||
if (connectedToCLI) DrawClient(cliIcon, "CLI");
|
||||
if (connectedToUnityEditor) DrawClient(unityEditorIcon, "Unity Editor");
|
||||
if (connectedToTuanjie) DrawClient(tuanjieIcon, "Tuanjie");
|
||||
|
||||
GUILayout.EndVertical();
|
||||
|
||||
GUILayout.EndHorizontal();
|
||||
}
|
||||
|
||||
// Form 2 – not started, starting, or started but no clients yet
|
||||
private void DrawNoClientBody()
|
||||
{
|
||||
string bodyText = currentState == BridgeState.Started
|
||||
? "Waiting for clients..."
|
||||
: "Start Codely Bridge to connect to Codely Agents.";
|
||||
|
||||
var bodyStyle = new GUIStyle(EditorStyles.label) { fontSize = BodyFontSize, wordWrap = true, font = inter };
|
||||
bodyStyle.normal.textColor = SubtextColor;
|
||||
bodyStyle.hover.textColor = SubtextColor;
|
||||
GUILayout.Label(bodyText, bodyStyle);
|
||||
|
||||
GUILayout.Space(SpaceBeforeStatus);
|
||||
|
||||
string dots = new string('.', (int)(EditorApplication.timeSinceStartup * 2) % 3 + 1);
|
||||
string statusText = currentState == BridgeState.Starting ? $"\u2022 Status: Connecting{dots}"
|
||||
: currentState == BridgeState.Started ? "\u2022 Status: Waiting for clients..."
|
||||
: "\u2022 Status: Disconnected";
|
||||
|
||||
GUILayout.Label(
|
||||
statusText,
|
||||
new GUIStyle(EditorStyles.label)
|
||||
{
|
||||
fontSize = StatusFontSize,
|
||||
font = firaMono,
|
||||
normal = { textColor = GrayColor },
|
||||
hover = { textColor = GrayColor }
|
||||
});
|
||||
}
|
||||
|
||||
private void DrawClient(Texture2D icon, string label)
|
||||
{
|
||||
GUILayout.BeginHorizontal();
|
||||
if (icon != null)
|
||||
GUILayout.Box(icon, GUIStyle.none, GUILayout.Width(ClientIconSize), GUILayout.Height(ClientIconSize));
|
||||
else
|
||||
GUILayout.Space(ClientIconSize);
|
||||
GUILayout.Space(ClientIconLabelGap);
|
||||
GUILayout.Label(
|
||||
label,
|
||||
new GUIStyle(EditorStyles.label) { fontSize = ClientLabelFontSize, font = inter, alignment = TextAnchor.MiddleLeft, normal = { textColor = Color.white }, hover = { textColor = Color.white } },
|
||||
GUILayout.Height(ClientLabelHeight));
|
||||
GUILayout.EndHorizontal();
|
||||
GUILayout.Space(ClientItemSpacing);
|
||||
}
|
||||
|
||||
private void StartBridge()
|
||||
{
|
||||
try
|
||||
{
|
||||
UnityTcpBridge.Start();
|
||||
isStarting = true;
|
||||
startingTime = EditorApplication.timeSinceStartup;
|
||||
currentState = BridgeState.Starting;
|
||||
TcpLog.Info("Codely Bridge started via Control Window");
|
||||
}
|
||||
catch (System.Exception ex)
|
||||
{
|
||||
TcpLog.Error($"Failed to start Codely Bridge: {ex.Message}");
|
||||
currentState = BridgeState.NotStarted;
|
||||
isStarting = false;
|
||||
}
|
||||
}
|
||||
|
||||
private void StopBridge()
|
||||
{
|
||||
try
|
||||
{
|
||||
UnityTcpBridge.Stop(true);
|
||||
TcpLog.Info("Codely Bridge manually stopped via Control Window");
|
||||
currentState = BridgeState.NotStarted;
|
||||
isStarting = false;
|
||||
ResetConnections();
|
||||
}
|
||||
catch (System.Exception ex)
|
||||
{
|
||||
TcpLog.Error($"Failed to stop Codely Bridge: {ex.Message}");
|
||||
}
|
||||
}
|
||||
|
||||
private void ResetConnections()
|
||||
{
|
||||
connectedToCLI = connectedToVSCode = connectedToVisualStudio =
|
||||
connectedToJetBrains = connectedToUnityEditor = connectedToTuanjie = false;
|
||||
}
|
||||
|
||||
private void UpdateStatus()
|
||||
{
|
||||
bool isRunning = UnityTcpBridge.IsRunning;
|
||||
|
||||
if (isStarting && isRunning)
|
||||
{
|
||||
currentState = BridgeState.Started;
|
||||
isStarting = false;
|
||||
}
|
||||
else if (isStarting && !isRunning)
|
||||
{
|
||||
if (EditorApplication.timeSinceStartup - startingTime > StartingTimeout)
|
||||
{
|
||||
currentState = BridgeState.NotStarted;
|
||||
isStarting = false;
|
||||
}
|
||||
}
|
||||
else if (!isStarting && isRunning && currentState != BridgeState.Started)
|
||||
{
|
||||
currentState = BridgeState.Started;
|
||||
}
|
||||
else if (!isRunning && currentState == BridgeState.Started)
|
||||
{
|
||||
currentState = BridgeState.NotStarted;
|
||||
ResetConnections();
|
||||
}
|
||||
|
||||
if (currentState == BridgeState.Started)
|
||||
UpdateConnectionStatus();
|
||||
}
|
||||
|
||||
private void UpdateConnectionStatus()
|
||||
{
|
||||
var clients = UnityTcpBridge.GetConnectedClients();
|
||||
ResetConnections();
|
||||
|
||||
foreach (var platform in clients.Values)
|
||||
{
|
||||
var p = platform.ToLower();
|
||||
if (p.Contains("cli"))
|
||||
connectedToCLI = true;
|
||||
else if (p.Contains("vscode") || p.Contains("vs code"))
|
||||
connectedToVSCode = true;
|
||||
else if (p.Contains("visualstudio") || p.Contains("visual studio"))
|
||||
connectedToVisualStudio = true;
|
||||
else if (p.Contains("jetbrains") || p.Contains("intellij") || p.Contains("rider"))
|
||||
connectedToJetBrains = true;
|
||||
else if (p.Contains("tuanjie") || p.Contains("codely") || p.Contains("agent"))
|
||||
connectedToTuanjie = true;
|
||||
else if (p.Contains("unity"))
|
||||
connectedToUnityEditor = true;
|
||||
}
|
||||
}
|
||||
|
||||
private void OnInspectorUpdate() => Repaint();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user