waveterm/pkg/web/webvdomproto.go

// Copyright 2024, Command Line Inc.
// SPDX-License-Identifier: Apache-2.0

package web

import (
	"fmt"
	"io"
	"log"
	"net/http"
	"strings"

	"github.com/wavetermdev/waveterm/pkg/wshrpc"
	"github.com/wavetermdev/waveterm/pkg/wshrpc/wshclient"
	"github.com/wavetermdev/waveterm/pkg/wshrpc/wshserver"
	"github.com/wavetermdev/waveterm/pkg/wshutil"
)

// Add the new handler function
func handleVDom(w http.ResponseWriter, r *http.Request) {
	// Extract UUID and path from URL
	pathParts := strings.Split(strings.TrimPrefix(r.URL.Path, "/vdom/"), "/")
	if len(pathParts) < 1 {
		http.Error(w, "Invalid VDOM URL format", http.StatusBadRequest)
		return
	}

	uuid := pathParts[0]
	// Simple UUID validation
	if len(uuid) != 36 {
		http.Error(w, "Invalid UUID format", http.StatusBadRequest)
		return
	}

	// Reconstruct the remaining path
	path := "/" + strings.Join(pathParts[1:], "/")
	if r.URL.RawQuery != "" {
		path += "?" + r.URL.RawQuery
	}

	// Read request body if present
	var body []byte
	var err error
	if r.Body != nil {
		body, err = io.ReadAll(r.Body)
		if err != nil {
			http.Error(w, fmt.Sprintf("Error reading request body: %v", err), http.StatusInternalServerError)
			return
		}
		defer r.Body.Close()
	}

	// Convert headers to map
	headers := make(map[string]string)
	for key, values := range r.Header {
		if len(values) > 0 {
			headers[key] = values[0]
		}
	}

	// Prepare RPC request data
	data := wshrpc.VDomUrlRequestData{
		Method:  r.Method,
		URL:     path,
		Headers: headers,
		Body:    body,
	}

	// Get RPC client
	client := wshserver.GetMainRpcClient()

	// Make RPC call with route to specific process
	route := wshutil.MakeProcRouteId(uuid)
	respCh := wshclient.VDomUrlRequestCommand(client, data, &wshrpc.RpcOpts{
		Route: route,
	})

	// Handle first response to set headers
	firstResp := true
	for respUnion := range respCh {
		if respUnion.Error != nil {
			http.Error(w, fmt.Sprintf("RPC error: %v", respUnion.Error), http.StatusInternalServerError)
			return
		}

		resp := respUnion.Response
		if firstResp {
			firstResp = false
			// Set status code and headers from first response
			if resp.StatusCode > 0 {
				w.WriteHeader(resp.StatusCode)
			} else {
				w.WriteHeader(http.StatusOK)
			}
			// Copy headers
			for key, value := range resp.Headers {
				w.Header().Set(key, value)
			}
		}

		// Write body chunk if present
		if len(resp.Body) > 0 {
			_, err = w.Write(resp.Body)
			if err != nil {
				log.Printf("Error writing response: %v", err)
				return
			}
		}
	}
}