feat: add http server

2024-09-13 21:52:15 +08:00
parent e4e4861d42
commit b8499ca97d
7 changed files with 222 additions and 2 deletions
--- a/internal/cmd_serve.go
+++ b/internal/cmd_serve.go
@@ -0,0 +1,187 @@
+package internal
+
+import (
+	"fmt"
+	"strconv"
+
+	"exporter/database/telegram_resource/public/model"
+	"exporter/database/telegram_resource/public/table"
+
+	. "github.com/go-jet/jet/v2/postgres"
+	"github.com/gofiber/fiber/v2"
+	"github.com/gofiber/fiber/v2/middleware/recover"
+	"github.com/spf13/cobra"
+)
+
+func ServeCmd() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "serve",
+		Short: "http server",
+		RunE:  serveCmd,
+	}
+
+	return cmd
+}
+
+func serveCmd(cmd *cobra.Command, args []string) error {
+	var err error
+	var port int64 = 3000
+	if len(args) > 0 {
+		port, err = strconv.ParseInt(args[0], 10, 64)
+		if err != nil {
+			return err
+		}
+	}
+
+	app := fiber.New()
+
+	// Initialize default config
+	app.Use(recover.New())
+
+	app.Get("/channels", func(c *fiber.Ctx) error {
+		var channels []model.Channels
+
+		tbl := table.Channels
+		if err := tbl.SELECT(tbl.AllColumns).QueryContext(c.Context(), db, &channels); err != nil {
+			return err
+		}
+
+		return c.JSON(channels)
+	})
+
+	// 获取频道最新一条数据
+	app.Get("/channels/:id", func(c *fiber.Ctx) error {
+		channelID, err := c.ParamsInt("id")
+		if err != nil {
+			return err
+		}
+
+		var channel model.Channels
+
+		tbl := table.Channels
+		err = tbl.
+			SELECT(tbl.AllColumns).
+			WHERE(tbl.ID.EQ(Int64(int64(channelID)))).
+			QueryContext(c.Context(), db, &channel)
+		if err != nil {
+			return err
+		}
+
+		return c.JSON(channel)
+	})
+
+	app.Get("/channels/:id/message", func(c *fiber.Ctx) error {
+		channelID, err := c.ParamsInt("id")
+		if err != nil {
+			return err
+		}
+
+		var msg model.ChannelMessages
+
+		tbl := table.ChannelMessages
+		err = tbl.
+			SELECT(tbl.AllColumns).
+			WHERE(tbl.ChannelID.EQ(Int64(int64(channelID)))).
+			ORDER_BY(tbl.ID.DESC()).
+			LIMIT(1).
+			QueryContext(c.Context(), db, &msg)
+		if err != nil {
+			return err
+		}
+
+		return c.JSON(msg)
+	})
+
+	app.Get("/channels/:id/message/like", func(c *fiber.Ctx) error {
+		channelID, err := c.ParamsInt("id")
+		if err != nil {
+			return err
+		}
+
+		var msg model.ChannelMessages
+
+		tbl := table.ChannelMessages
+		err = tbl.
+			SELECT(tbl.AllColumns).
+			WHERE(
+				tbl.ChannelID.EQ(Int64(int64(channelID))).AND(
+					tbl.Like.EQ(Bool(true)),
+				),
+			).
+			ORDER_BY(tbl.ID.DESC()).
+			LIMIT(1).
+			QueryContext(c.Context(), db, &msg)
+		if err != nil {
+			return err
+		}
+
+		return c.JSON(msg)
+	})
+
+	// toggle msg likes
+	app.Patch("/channels/:channelID/message/:id/like", func(c *fiber.Ctx) error {
+		channelID, err := c.ParamsInt("channelID")
+		if err != nil {
+			return err
+		}
+
+		msgID, err := c.ParamsInt("id")
+		if err != nil {
+			return err
+		}
+
+		var msg model.ChannelMessages
+
+		tbl := table.ChannelMessages
+		cond := tbl.ChannelID.EQ(Int64(int64(channelID))).AND(tbl.ID.EQ(Int64(int64(msgID))))
+
+		err = tbl.
+			SELECT(tbl.AllColumns).
+			WHERE(cond).
+			ORDER_BY(tbl.ID.DESC()).
+			LIMIT(1).
+			QueryContext(c.Context(), db, &msg)
+		if err != nil {
+			return err
+		}
+
+		_, err = tbl.
+			UPDATE().
+			SET(tbl.Like.SET(Bool(!msg.Like))).
+			WHERE(cond).
+			ExecContext(c.Context(), db)
+		if err != nil {
+			return err
+		}
+
+		return nil
+	})
+
+	// delete message
+	app.Delete("/channels/:channelID/message/:id", func(c *fiber.Ctx) error {
+		channelID, err := c.ParamsInt("channelID")
+		if err != nil {
+			return err
+		}
+
+		msgID, err := c.ParamsInt("id")
+		if err != nil {
+			return err
+		}
+
+		tbl := table.ChannelMessages
+		cond := tbl.ChannelID.EQ(Int64(int64(channelID))).AND(tbl.ID.EQ(Int64(int64(msgID))))
+
+		_, err = tbl.
+			DELETE().
+			WHERE(cond).
+			ExecContext(c.Context(), db)
+		if err != nil {
+			return err
+		}
+
+		return nil
+	})
+
+	return app.Listen(fmt.Sprintf(":%d", port))
+}