101 lines
2.2 KiB
Go
101 lines
2.2 KiB
Go
package handlers
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"devone.aplikasi.web.id/gitea/mario/go-ohif-proxy/internal/proxy"
|
|
"github.com/go-chi/chi/v5"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// DicomHandler handles DICOM Web requests
|
|
type DicomHandler struct {
|
|
client *proxy.Client
|
|
logger *zap.Logger
|
|
}
|
|
|
|
// NewDicomHandler creates a new DICOM handler
|
|
func NewDicomHandler(client *proxy.Client, logger *zap.Logger) *DicomHandler {
|
|
return &DicomHandler{
|
|
client: client,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
// ForwardRequest forwards the request to Google Healthcare API
|
|
func (h *DicomHandler) ForwardRequest(w http.ResponseWriter, r *http.Request) {
|
|
// Get the path after /dicomWeb
|
|
urlPath := chi.URLParam(r, "*")
|
|
|
|
// If the URL parameter is empty, try to extract it from the URL path
|
|
if urlPath == "" {
|
|
// Remove /dicomWeb prefix from the URL
|
|
prefix := "/dicomWeb"
|
|
if strings.HasPrefix(r.URL.Path, prefix) {
|
|
urlPath = r.URL.Path[len(prefix):]
|
|
}
|
|
}
|
|
|
|
h.logger.Debug("Forwarding request",
|
|
zap.String("path", urlPath),
|
|
zap.String("method", r.Method),
|
|
zap.String("url", r.URL.String()),
|
|
)
|
|
|
|
// Read request body if present
|
|
var bodyBytes []byte
|
|
if r.Body != nil && r.ContentLength > 0 {
|
|
var err error
|
|
bodyBytes, err = io.ReadAll(r.Body)
|
|
if err != nil {
|
|
h.logger.Error("Failed to read request body", zap.Error(err))
|
|
http.Error(w, "Failed to read request body", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
|
|
// Copy query parameters
|
|
queryParams := r.URL.Query().Encode()
|
|
if queryParams != "" {
|
|
if !strings.HasPrefix(urlPath, "/") {
|
|
urlPath = "/" + urlPath
|
|
}
|
|
urlPath = urlPath + "?" + queryParams
|
|
}
|
|
|
|
// Get request headers
|
|
headers := make(map[string]string)
|
|
for k, v := range r.Header {
|
|
if len(v) > 0 {
|
|
headers[k] = v[0]
|
|
}
|
|
}
|
|
|
|
// Forward the request to Healthcare API
|
|
response, err := h.client.ForwardRequest(
|
|
r.Context(),
|
|
r.Method,
|
|
urlPath,
|
|
headers,
|
|
bodyBytes,
|
|
)
|
|
|
|
if err != nil {
|
|
h.logger.Error("Request forwarding failed", zap.Error(err))
|
|
http.Error(w, fmt.Sprintf("Request failed: %v", err), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
// Set response headers
|
|
for k, v := range response.Headers {
|
|
w.Header().Set(k, v)
|
|
}
|
|
|
|
// Set status and write response body
|
|
w.WriteHeader(response.StatusCode)
|
|
w.Write(response.Body)
|
|
}
|