Add Unity Pipeline package support

2026-07-31 17:34:04 +08:00
parent d1a526094e
commit 786b7ffff9
590 changed files with 51995 additions and 2 deletions
@@ -0,0 +1,101 @@
+using System;
+using Unity.Pipeline.Models;
+using Unity.Pipeline.Security;
+using UnityEngine;
+
+namespace Unity.Pipeline.Editor
+{
+    public class EditorPipelineServer : BasePipelineServer
+    {
+        private InstanceDescriptor m_InstanceDescriptor;
+
+        /// <summary>When true, every command transaction is written to the project transaction log.</summary>
+        public bool LogRequestsResponses { get; set; }
+
+        public override DateTime StartedAt => m_InstanceDescriptor == null ? new DateTime() : m_InstanceDescriptor.StartedAt;
+
+        protected override void ServerStarted()
+        {
+            var isAutomated = false;
+            foreach (var arg in System.Environment.GetCommandLineArgs())
+            {
+                if (arg == "-automated")
+                {
+                    isAutomated = true;
+                    break;
+                }
+            }
+            // Only warn in an interactive Editor: batchmode (CI, upm-pvp, UTR) can't show modal
+            // popups, so the warning is pure noise there.
+            if (!isAutomated && !Application.isBatchMode)
+            {
+                Debug.LogWarning("Editor is not in automated mode. Modal Pop up might break continuous command workflow. Start the editor with -automated");
+            }
+        }
+
+        protected override void OnTransactionProcessed(string requestJson, string responseJson)
+        {
+            if (!LogRequestsResponses)
+                return;
+
+            PipelineTransactionLog.Append(requestJson, responseJson);
+        }
+
+        // Runtime-only commands (eval, reload_file_override, …) are part of the Player surface; don't
+        // advertise them when a client is connected to the Editor.
+        protected override bool IncludeRuntimeOnlyCommands => false;
+        protected override void CreateInstanceDescriptor()
+        {
+            // Create and write instance descriptor for CLI discovery
+            m_InstanceDescriptor = InstanceDescriptor.CreateCurrent(Port);
+            InstanceDescriptor.WriteToProjectRoot(m_InstanceDescriptor);
+        }
+
+        protected override void DeleteInstanceDescriptor()
+        {
+            // Clean up instance descriptor file
+            if (m_InstanceDescriptor != null)
+            {
+                InstanceDescriptor.RemoveFromProjectRoot(m_InstanceDescriptor.ProjectPath);
+            }
+            m_InstanceDescriptor = null;
+        }
+
+        protected override void UpdateHeartBeat()
+        {
+            // Update heartbeat in instance descriptor
+            if (m_InstanceDescriptor != null)
+            {
+                // Keep the port file's token current for discovery only — GetToken() validates the
+                // live token, so this just catches up to a rotation on the next heartbeat.
+                m_InstanceDescriptor.EvalToken = SecurityTokenManager.GetOrCreateToken();
+                m_InstanceDescriptor.LastHeartbeat = DateTime.UtcNow;
+                try
+                {
+                    InstanceDescriptor.WriteToProjectRoot(m_InstanceDescriptor);
+                }
+                catch (Exception ex)
+                {
+                    Debug.LogWarning($"Failed to update instance descriptor: {ex.Message}");
+                }
+            }
+        }
+
+        protected override object GetServerStatus()
+        {
+            UpdateHeartBeat();
+            return new
+            {
+                status = m_InstanceDescriptor == null ? "error" : "ready",
+                lastHeartbeat = m_InstanceDescriptor?.LastHeartbeat
+            };
+        }
+
+        protected override string GetToken()
+        {
+            // Validate the live token so a rotation/revocation takes effect on the next request (not
+            // the next heartbeat). The descriptor's EvalToken is discovery-only.
+            return SecurityTokenManager.GetOrCreateToken();
+        }
+    }
+}