feat: optional token echo in sse endpoint

2026-03-15 13:08:00 +08:00
parent c20b33adb0
commit f5086a9342
2 changed files with 9 additions and 3 deletions
--- a/main.go
+++ b/main.go
@@ -323,11 +323,16 @@ func (s *TaoServer) SSEHandler(w http.ResponseWriter, r *http.Request) {
  client = generateClientID()
 }

- if queryToken != "" {
+ endpointToken := queryToken
+ if endpointToken == "" && getEnvBool("TAO_ENDPOINT_TOKEN_ECHO", false) {
+  endpointToken = token
+ }
+
+ if endpointToken != "" {
  if strings.Contains(endpoint, "?") {
-   endpoint = endpoint + "&token=" + queryToken
+   endpoint = endpoint + "&token=" + endpointToken
  } else {
-   endpoint = endpoint + "?token=" + queryToken
+   endpoint = endpoint + "?token=" + endpointToken
  }
 }
 if client != "" {