ipbl/src/routers/funcs.go

package routers

import (
	"context"
	"fmt"
	"log"
	"net/http"
	"strconv"

	"git.paulbsd.com/paulbsd/ipbl/src/config"
	"git.paulbsd.com/paulbsd/ipbl/src/models"

	"github.com/labstack/echo/v4"
)

// RegisterRoutes runs the main echo HTTP server
func RegisterRoutes(e *echo.Echo, ctx *context.Context, cfg *config.Config) {
	e.GET("/", func(c echo.Context) error {
		return c.HTML(http.StatusOK, `<html>
		<body style="background-color: black">
		<p style="color:white">Welcome to ipbl software (<a href="https://git.paulbsd.com/paulbsd/ipbl">https://git.paulbsd.com/paulbsd/ipbl</a>)</p>
		</body>
		</html>`)
	})

	e.GET("/ip/:ip", func(c echo.Context) (err error) {
		ret, err := models.GetIP(ctx, cfg, c.Param("ip"))
		return Result(c, err, ret)
	})
	e.POST("/ip", func(c echo.Context) (err error) {
		var ip = new(models.IP)
		var msg = "No IP inserted"
		err = c.Bind(ip)
		if err != nil {
			return Result(c, fmt.Errorf("error when parsing body"), "")
		}
		num, err := ip.InsertIP(cfg)
		fmt.Println(err)
		if err != nil {
			return Result(c, err, "")
		}
		if num > 0 {
			msg = fmt.Sprintf("Inserted %d IP", num)
		}
		return Result(c, err, msg)
	})
	e.GET("/ips", func(c echo.Context) (err error) {
		limit := 50
		if c.QueryParam("limit") != "" {
			limit, _ = strconv.Atoi(c.QueryParam("limit"))
		}
		ret, err := models.GetIPs(ctx, cfg, limit)
		return Result(c, err, ret)
	})
	// retro-compat
	e.GET("/ips/lastday", func(c echo.Context) (err error) {
		interval := "1 day"
		ret, err := models.GetIPsLast(ctx, cfg, interval)
		return Result(c, err, ret)
	})
	e.GET("/ips/last", func(c echo.Context) (err error) {
		interval := "10 minutes"
		ret, err := models.GetIPsLast(ctx, cfg, interval)
		return Result(c, err, ret)
	})
	e.POST("/ips", func(c echo.Context) (err error) {
		var ips = []models.IP{}
		var msg string
		err = c.Bind(&ips)
		if err != nil {
			return Result(c, err, ips)
		}
		numinsert, numupdate, _, _ := models.InsertIPBulk(cfg, &ips)
		if numinsert > 0 {
			msg = fmt.Sprintf("Inserted %d IP", numinsert)
			log.Println(msg)
		}
		if numupdate > 0 {
			msg = fmt.Sprintf("Updated %d IP", numupdate)
			log.Println(msg)
		}
		return Result(c, err, msg)
	})
	e.GET("/ips/whitelist", func(c echo.Context) (err error) {
		whitelists, err := models.GetWhitelists(*cfg)
		return Result(c, err, whitelists)
	})
	e.POST("/ips/whitelist", func(c echo.Context) (err error) {
		var cidr models.Whitelist
		err = c.Bind(&cidr)
		if err == nil && cidr.Verify() {
			err = cidr.InsertOrUpdate(*cfg)
			return Result(c, err, cidr)
		}
		return Result(c, err, nil)
	})
	e.DELETE("/ips/whitelist/:ip", func(c echo.Context) (err error) {
		var ip = c.Param("ip")
		var cidr models.Whitelist
		err = cidr.Delete(*cfg, ip)
		if err != nil {
			return c.JSON(http.StatusOK, "Deleted old CIDR")
		}
		return
	})

	e.Logger.Fatal(
		e.Start(
			fmt.Sprintf(":%d",
				cfg.Switchs.Port)))

}
updated ipbl 2021-12-27 23:52:22 +01:00			`package routers`
initial commit for ipbl 2021-11-07 20:44:48 +01:00
			`import (`
updated ipbl 2021-12-27 23:52:22 +01:00			`"context"`
initial commit for ipbl 2021-11-07 20:44:48 +01:00			`"fmt"`
updated ipbl 2021-12-12 19:02:35 +01:00			`"log"`
initial commit for ipbl 2021-11-07 20:44:48 +01:00			`"net/http"`
			`"strconv"`

			`"git.paulbsd.com/paulbsd/ipbl/src/config"`
updated ipbl 2021-12-27 23:52:22 +01:00			`"git.paulbsd.com/paulbsd/ipbl/src/models"`
initial commit for ipbl 2021-11-07 20:44:48 +01:00
			`"github.com/labstack/echo/v4"`
			`)`

updated ipbl 2021-12-27 23:52:22 +01:00			`// RegisterRoutes runs the main echo HTTP server`
			`func RegisterRoutes(e echo.Echo, ctx context.Context, cfg *config.Config) {`
initial commit for ipbl 2021-11-07 20:44:48 +01:00			`e.GET("/", func(c echo.Context) error {`
updated ipbl 2021-12-12 17:33:40 +01:00			return c.HTML(http.StatusOK, `<html>
			`<body style="background-color: black">`
added endpoints for cfg update 2022-02-07 23:08:22 +01:00			`<p style="color:white">Welcome to ipbl software (<a href="https://git.paulbsd.com/paulbsd/ipbl">https://git.paulbsd.com/paulbsd/ipbl</a>)</p>`
updated ipbl 2021-12-12 17:33:40 +01:00			`</body>`
			</html>`)
initial commit for ipbl 2021-11-07 20:44:48 +01:00			`})`
updated ipbl 2021-12-27 23:52:22 +01:00
initial commit for ipbl 2021-11-07 20:44:48 +01:00			`e.GET("/ip/:ip", func(c echo.Context) (err error) {`
updated ipbl 2021-12-30 12:03:26 +01:00			`ret, err := models.GetIP(ctx, cfg, c.Param("ip"))`
updated functions 2022-02-12 00:33:58 +01:00			`return Result(c, err, ret)`
initial commit for ipbl 2021-11-07 20:44:48 +01:00			`})`
updated ipbl 2021-12-11 12:59:43 +01:00			`e.POST("/ip", func(c echo.Context) (err error) {`
updated ipbl 2021-12-27 23:52:22 +01:00			`var ip = new(models.IP)`
updated msg in funcs.go after IP POST 2022-02-22 14:44:52 +01:00			`var msg = "No IP inserted"`
updated ipbl 2021-12-12 17:33:40 +01:00			`err = c.Bind(ip)`
			`if err != nil {`
updated functions 2022-02-12 00:33:58 +01:00			`return Result(c, fmt.Errorf("error when parsing body"), "")`
updated ipbl 2021-12-12 17:33:40 +01:00			`}`
			`num, err := ip.InsertIP(cfg)`
updated functions 2022-02-12 00:33:58 +01:00			`fmt.Println(err)`
updated ipbl 2021-12-12 17:33:40 +01:00			`if err != nil {`
updated functions 2022-02-12 00:33:58 +01:00			`return Result(c, err, "")`
updated ipbl 2021-12-12 17:33:40 +01:00			`}`
updated ipbl 2021-12-29 14:18:27 +01:00			`if num > 0 {`
			`msg = fmt.Sprintf("Inserted %d IP", num)`
			`}`
updated functions 2022-02-12 00:33:58 +01:00			`return Result(c, err, msg)`
updated ipbl 2021-12-11 12:59:43 +01:00			`})`
initial commit for ipbl 2021-11-07 20:44:48 +01:00			`e.GET("/ips", func(c echo.Context) (err error) {`
updated ipbl 2021-12-27 23:52:22 +01:00			`limit := 50`
initial commit for ipbl 2021-11-07 20:44:48 +01:00			`if c.QueryParam("limit") != "" {`
updated ipbl 2021-12-27 23:52:22 +01:00			`limit, _ = strconv.Atoi(c.QueryParam("limit"))`
initial commit for ipbl 2021-11-07 20:44:48 +01:00			`}`
updated ipbl 2021-12-30 12:03:26 +01:00			`ret, err := models.GetIPs(ctx, cfg, limit)`
updated functions 2022-02-12 00:33:58 +01:00			`return Result(c, err, ret)`
initial commit for ipbl 2021-11-07 20:44:48 +01:00			`})`
updated "last" endpoint 2022-02-09 10:18:28 +01:00			`// retro-compat`
updated ipbl with misc fixes 2022-01-30 18:38:03 +01:00			`e.GET("/ips/lastday", func(c echo.Context) (err error) {`
			`interval := "1 day"`
updated "last" endpoint 2022-02-09 10:18:28 +01:00			`ret, err := models.GetIPsLast(ctx, cfg, interval)`
updated functions 2022-02-12 00:33:58 +01:00			`return Result(c, err, ret)`
updated "last" endpoint 2022-02-09 10:18:28 +01:00			`})`
			`e.GET("/ips/last", func(c echo.Context) (err error) {`
			`interval := "10 minutes"`
			`ret, err := models.GetIPsLast(ctx, cfg, interval)`
updated functions 2022-02-12 00:33:58 +01:00			`return Result(c, err, ret)`
updated ipbl with misc fixes 2022-01-30 18:38:03 +01:00			`})`
updated ipbl 2021-12-11 12:59:43 +01:00			`e.POST("/ips", func(c echo.Context) (err error) {`
updated ipbl 2021-12-27 23:52:22 +01:00			`var ips = []models.IP{}`
updated ipbl 2021-12-29 14:18:27 +01:00			`var msg string`
updated ipbl 2021-12-12 17:33:40 +01:00			`err = c.Bind(&ips)`
			`if err != nil {`
updated functions 2022-02-12 00:33:58 +01:00			`return Result(c, err, ips)`
updated ipbl 2021-12-12 17:33:40 +01:00			`}`
updated ipbl with misc fixes 2022-01-30 18:38:03 +01:00			`numinsert, numupdate, _, _ := models.InsertIPBulk(cfg, &ips)`
updated ipbl 2021-12-29 14:18:27 +01:00			`if numinsert > 0 {`
			`msg = fmt.Sprintf("Inserted %d IP", numinsert)`
updated ipbl 2021-12-29 14:24:26 +01:00			`log.Println(msg)`
updated ipbl 2021-12-29 14:18:27 +01:00			`}`
updated ipbl with misc fixes 2022-01-30 18:38:03 +01:00			`if numupdate > 0 {`
			`msg = fmt.Sprintf("Updated %d IP", numupdate)`
			`log.Println(msg)`
			`}`
updated functions 2022-02-12 00:33:58 +01:00			`return Result(c, err, msg)`
updated ipbl 2021-12-11 12:59:43 +01:00			`})`
updated ipbl 2021-12-27 23:52:22 +01:00			`e.GET("/ips/whitelist", func(c echo.Context) (err error) {`
updated functions 2022-02-12 00:33:58 +01:00			`whitelists, err := models.GetWhitelists(*cfg)`
			`return Result(c, err, whitelists)`
updated ipbl 2021-12-27 23:52:22 +01:00			`})`
updated functions 2022-02-12 00:33:58 +01:00			`e.POST("/ips/whitelist", func(c echo.Context) (err error) {`
added endpoints for cfg update 2022-02-07 23:08:22 +01:00			`var cidr models.Whitelist`
			`err = c.Bind(&cidr)`
			`if err == nil && cidr.Verify() {`
updated functions 2022-02-12 00:33:58 +01:00			`err = cidr.InsertOrUpdate(*cfg)`
			`return Result(c, err, cidr)`
added endpoints for cfg update 2022-02-07 23:08:22 +01:00			`}`
updated functions 2022-02-12 00:33:58 +01:00			`return Result(c, err, nil)`
added endpoints for cfg update 2022-02-07 23:08:22 +01:00			`})`
			`e.DELETE("/ips/whitelist/:ip", func(c echo.Context) (err error) {`
			`var ip = c.Param("ip")`
			`var cidr models.Whitelist`
			`err = cidr.Delete(*cfg, ip)`
			`if err != nil {`
			`return c.JSON(http.StatusOK, "Deleted old CIDR")`
			`}`
			`return`
			`})`
initial commit for ipbl 2021-11-07 20:44:48 +01:00
			`e.Logger.Fatal(`
			`e.Start(`
			`fmt.Sprintf(":%d",`
			`cfg.Switchs.Port)))`

			`}`