mirror of
https://github.com/hanzoai/runtime.git
synced 2026-10-30 18:23:13 +00:00
zeekay
ea80dc3638
- Add main Dockerfile for production builds - Add Dockerfile.runtime for secure code execution - Add Dockerfile.runtime.fixed with improved configuration - Include .dockerignore for build optimization - Add integration documentation and publishing guides - Update API and CLI components for better runtime support - Enhance sandbox and container management features
79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
// Copyright 2025 Hanzo Industries Inc.
|
|
// SPDX-License-Identifier: AGPL-3.0
|
|
|
|
package controllers
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/hanzoai/common-go/pkg/errors"
|
|
"github.com/hanzoai/common-go/pkg/proxy"
|
|
"github.com/gin-gonic/gin"
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func ProxyCommandLogsStream(ctx *gin.Context) {
|
|
targetURL, extraHeaders, err := getProxyTarget(ctx)
|
|
if err != nil {
|
|
// Error already sent to the context
|
|
return
|
|
}
|
|
|
|
if ctx.Query("follow") != "true" {
|
|
proxy.NewProxyRequestHandler(func(ctx *gin.Context) (*url.URL, map[string]string, error) {
|
|
return targetURL, extraHeaders, nil
|
|
})(ctx)
|
|
return
|
|
}
|
|
|
|
fullTargetURL := strings.Replace(targetURL.String(), "http://", "ws://", 1)
|
|
|
|
ws, _, err := websocket.DefaultDialer.DialContext(context.Background(), fullTargetURL+"?follow=true", nil)
|
|
if err != nil {
|
|
ctx.Error(errors.NewBadRequestError(fmt.Errorf("failed to create outgoing request: %w", err)))
|
|
return
|
|
}
|
|
|
|
ctx.Header("Content-Type", "application/octet-stream")
|
|
|
|
ws.SetCloseHandler(func(code int, text string) error {
|
|
if code == websocket.CloseNormalClosure {
|
|
return nil
|
|
}
|
|
ctx.AbortWithStatus(code)
|
|
return nil
|
|
})
|
|
|
|
defer ws.Close()
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
default:
|
|
_, msg, err := ws.ReadMessage()
|
|
if err != nil {
|
|
if websocket.IsCloseError(err, websocket.CloseNormalClosure) {
|
|
return
|
|
}
|
|
log.Errorf("Error reading message: %v", err)
|
|
ws.Close()
|
|
return
|
|
}
|
|
|
|
_, err = ctx.Writer.Write(msg)
|
|
if err != nil {
|
|
log.Errorf("Error writing message: %v", err)
|
|
ws.Close()
|
|
return
|
|
}
|
|
ctx.Writer.Flush()
|
|
}
|
|
}
|
|
}
|