Nexus/pkg/internal/http/api/command.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)
	}
}