68 lines
1.7 KiB
Go
68 lines
1.7 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"git.solsynth.dev/hypernet/nexus/pkg/internal/directory"
|
|
"git.solsynth.dev/hypernet/nexus/pkg/proto"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/rs/zerolog/log"
|
|
"google.golang.org/grpc/metadata"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func invokeCommand(c *fiber.Ctx) error {
|
|
command := c.Params("command")
|
|
method := strings.ToLower(c.Method())
|
|
|
|
handler := directory.GetCommandHandler(command, method)
|
|
if handler == nil {
|
|
return fiber.NewError(fiber.StatusNotFound, "command not found")
|
|
}
|
|
|
|
conn, err := handler.GetGrpcConn()
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusServiceUnavailable, "service unavailable")
|
|
}
|
|
|
|
log.Debug().Str("id", command).Str("method", method).Msg("Invoking command from HTTP Gateway...")
|
|
|
|
var meta []string
|
|
meta = append(meta, "client_id", "http-gateway")
|
|
meta = append(meta, "net.ip", c.IP())
|
|
meta = append(meta, "http.user_agent", c.Get(fiber.HeaderUserAgent))
|
|
for k, v := range c.GetReqHeaders() {
|
|
meta = append(
|
|
meta,
|
|
strings.ToLower(fmt.Sprintf("header.%s", strings.ReplaceAll(k, "-", "_"))),
|
|
strings.Join(v, "\n"),
|
|
)
|
|
}
|
|
|
|
for k, v := range c.Queries() {
|
|
meta = append(
|
|
meta,
|
|
strings.ToLower(fmt.Sprintf("query.%s", strings.ReplaceAll(k, "-", "_"))),
|
|
v,
|
|
)
|
|
}
|
|
|
|
ctx := metadata.AppendToOutgoingContext(c.Context(), meta...)
|
|
ctx, cancel := context.WithTimeout(ctx, time.Second*10)
|
|
defer cancel()
|
|
|
|
out, err := proto.NewCommandProviderClient(conn).SendCommand(ctx, &proto.CommandArgument{
|
|
Command: command,
|
|
Method: method,
|
|
Payload: c.Body(),
|
|
})
|
|
|
|
if err != nil {
|
|
return fiber.NewError(fiber.StatusInternalServerError, err.Error())
|
|
} else {
|
|
c.Set(fiber.HeaderContentType, out.ContentType)
|
|
return c.Status(int(out.Status)).Send(out.Payload)
|
|
}
|
|
}
|