package middleware

import (
	"net/http"

	"github.com/ekzyis/hermes/server/router"
	"github.com/labstack/echo/v4"
)

func LndGuard(hc router.Context) echo.MiddlewareFunc {
	return func(next echo.HandlerFunc) echo.HandlerFunc {
		return func(ec echo.Context) error {
			if hc.Lnd != nil {
				return next(ec)
			}
			return echo.NewHTTPError(http.StatusNotImplemented)
		}
	}
}

func DbGuard(hc router.Context) echo.MiddlewareFunc {
	return func(next echo.HandlerFunc) echo.HandlerFunc {
		return func(ec echo.Context) error {
			if hc.Db != nil {
				return next(ec)
			}
			return echo.NewHTTPError(http.StatusNotImplemented)
		}
	}
}