ip.go

package main

import (
	"fmt"
	"net/http"

	"golang.org/x/exp/slices"

	"github.com/gofiber/fiber/v2"
	"github.com/prophittcorey/tor"
)

func isTorExitNode(address string) bool {
	res, err := tor.IsExitNode(address)
	if err != nil {
		fmt.Println(err)
		return false
	}
	if res {
		return true
	}
	return false
}

func torMiddleware() fiber.Handler {
	return func(c *fiber.Ctx) error {
		if isTorExitNode(c.IP()) {
			return c.SendStatus(http.StatusForbidden)
		}

		return c.Next()
	}
}

func ipMiddleware() fiber.Handler {
	return func(c *fiber.Ctx) error {
		if slices.Contains(config.Limits.IpBlacklist, c.IP()) {
			return c.SendStatus(http.StatusForbidden)
		}

		return c.Next()
	}
}