main.go

package main

import (
	"os"
	"fmt"
	"log"
	"context"
	"github.com/gofiber/fiber/v2"
	"github.com/gofiber/contrib/otelfiber"
	telemetry "github.com/fireacademy/telemetry"
)


func getPort() string {
	port := os.Getenv("LEAFLET_LISTEN_PORT")
	if port == "" {
		panic("LEAFLET_LISTEN_PORT not set")
	}

	return port
}

func Index(c *fiber.Ctx) error {
	return c.SendString("Leaflet server is running.")
}

func ReadinessCheck(c *fiber.Ctx) error {
	ready := IsReady()

	if ready {
		return c.SendString("OK")
	}

	return c.Status(400).SendString("NOT READY BRO")
}

func ProxyToRPCEndpoint(c *fiber.Ctx) error {
	endpoint := c.Params("endpoint")
	
	body := string(c.Body())
	if len(body) < 2 {
		body = "{}"
	}

	ctx := c.UserContext()
	resp, err := DoRPCRequest(ctx, "POST", endpoint, body)
	if err != nil {
		telemetry.LogError(ctx, err, "error while calling RPC")
		return c.Status(500).JSON(fiber.Map{
			"success": false,
			"message": "error while calling RPC",
		})
	}

	c.Set("Content-Type", "application/json")
	return c.Status(resp.StatusCode).SendStream(resp.Body)
}

func main() {
	cleanup := telemetry.Initialize()
	defer cleanup(context.Background())

	SetupRPCClient()

	app := fiber.New()
	app.Use(otelfiber.Middleware())

	app.Get("/", Index)
	app.Get("/ready", ReadinessCheck)
	app.Get("/rpc/:endpoint", ProxyToRPCEndpoint)
	app.Post("/rpc/:endpoint", ProxyToRPCEndpoint)

	port := getPort()
	log.Fatalln(app.Listen(fmt.Sprintf(":%v", port)))
}