Merge pull request #82 from traP-jp/add-anonSodan_2

Add anon sodan 2

src/handler/sodanHandler.go

@@ -0,0 +1,96 @@
+package handler
+
+import (
+	"database/sql"
+	"errors"
+	"log"
+	"net/http"
+	"quickwiki-backend/model"
+	"strconv"
+
+	"github.com/labstack/echo"
+)
+
+func (h *Handler) PostMessageToTraQ(c echo.Context) error {
+	var message model.MessageToTraQ_POST
+	err := c.Bind(&message)
+	if err != nil {
+		return echo.NewHTTPError(http.StatusBadRequest, "bad request body")
+	}
+
+	channelId := "aff37b5f-0911-4255-81c3-b49985c8943f"
+	err = h.scraper.MessageToTraQ(message.Content, channelId)
+
+	if err != nil {
+		log.Println("post message err : ", err)
+		return echo.NewHTTPError(http.StatusInternalServerError, "Internal server error.")
+	}
+
+	return c.JSON(http.StatusOK, h.GetSodanHandler(c))
+}
+
+func (h *Handler) PatchMessageToTraQ(c echo.Context) error {
+	wikiId, err := strconv.Atoi(c.QueryParam("wikiId"))
+	if err != nil {
+		log.Printf("failed to convert wikiId to int: %v", err)
+		return c.JSON(http.StatusBadRequest, err)
+	}
+
+	var message model.MessageToTraQ_POST
+	err = c.Bind(&message)
+	if err != nil {
+		return echo.NewHTTPError(http.StatusBadRequest, "bad request body")
+	}
+
+	var messageContents model.SodanContent_fromDB
+	err = h.db.Select(&messageContents, "select * from messages where wiki_id = ? order by created_at desc limit 1", wikiId)
+	if err != nil {
+		if errors.Is(err, sql.ErrNoRows) {
+			return c.NoContent(http.StatusNotFound)
+		}
+		log.Printf("failed to get sodanContent: %s\n", err)
+		return c.NoContent(http.StatusInternalServerError)
+	}
+	messageID := messageContents.MessageID
+	err = h.scraper.MessageToTraQ(message.Content, messageID)
+
+	if err != nil {
+		log.Println("post message err : ", err)
+		return echo.NewHTTPError(http.StatusInternalServerError, "Internal server error.")
+	}
+
+	return c.JSON(http.StatusOK, h.GetSodanHandler(c))
+}
+
+func (h *Handler) PostRepliesToTraQ(c echo.Context) error {
+	wikiId, err := strconv.Atoi(c.QueryParam("wikiId"))
+	if err != nil {
+		log.Printf("failed to convert wikiId to int: %v", err)
+		return c.JSON(http.StatusBadRequest, err)
+	}
+
+	var message model.MessageToTraQ_POST
+	err = c.Bind(&message)
+	if err != nil {
+		return echo.NewHTTPError(http.StatusBadRequest, "bad request body")
+	}
+
+	var messageContents model.SodanContent_fromDB
+	err = h.db.Select(&messageContents, "select * from messages where wiki_id = ? order by created_at desc limit 1", wikiId)
+	if err != nil {
+		if errors.Is(err, sql.ErrNoRows) {
+			return c.NoContent(http.StatusNotFound)
+		}
+		log.Printf("failed to get sodanContent: %s\n", err)
+		return c.NoContent(http.StatusInternalServerError)
+	}
+	channelID := messageContents.ChannelID
+	err = h.scraper.MessageToTraQ(message.Content, channelID)
+
+	if err != nil {
+		log.Println("post message err : ", err)
+		return echo.NewHTTPError(http.StatusInternalServerError, "Internal server error.")
+	}
+
+	return c.JSON(http.StatusOK, h.GetSodanHandler(c))
+}

src/main.go

@@ -67,6 +67,10 @@ func main() {
 	e.POST("/wiki/user/favorite", h.PostUserFavoriteWikiHandler)
 	e.DELETE("/wiki/user/favorite", h.DeleteUserFavoriteWikiHandler)
 
+	e.POST("/anon-sodan", h.PostMessageToTraQ)
+	e.PATCH("/anon-sodan", h.PatchMessageToTraQ)
+	e.POST("/anon-sodan/replies", h.PostRepliesToTraQ)
+
 	e.GET("/setting/index", h.SetIndexingHandler)
 
 	e.Logger.Fatal(e.Start(":8080"))

src/model/model.go

@@ -187,3 +187,7 @@ type Folder_fromDB struct {
 	Name     string `db:"name"`
 	ParentID int    `db:"parent_id"`
 }
+
+type MessageToTraQ_POST struct {
+	Content string `json:"content"`
+}

src/scraper/messageBot.go

@@ -0,0 +1,30 @@
+package scraper
+
+import (
+	"context"
+
+	"github.com/traPtitech/go-traq"
+)
+
+func (s *Scraper) MessageToTraQ(message string, PostChanellId string) error {
+	PostChanellId = "01913f8b-8c05-76a2-b51f-bb83e9e93615" //DEV_MODE
+	_, _, err := s.bot.API().
+		MessageApi.
+		PostMessage(context.Background(), PostChanellId).
+		PostMessageRequest(traq.PostMessageRequest{
+			Content: message,
+		}).
+		Execute()
+	return err
+}
+
+func (s *Scraper) MessageEditOnTraQ(message string, editMessageId string) error {
+	editMessageId = "0191a3c2-6659-7c4e-a03f-6a37d080cd10" //DEV_MODE
+	_, err := s.bot.API().MessageApi.
+		EditMessage(context.Background(), editMessageId).
+		PostMessageRequest(traq.PostMessageRequest{
+			Content: message,
+		}).
+		Execute()
+	return err
+}

src/scraper/messages.go

@@ -85,7 +85,7 @@ func (s *Scraper) GetSodanMessages() {
 
 	s.GetSodanSubMessages("98ea48da-64e8-4f69-9d0d-80690b682670", 40, 52)
 	log.Println("sodan messages scraped")
-	s.GetSodanSubMessages("98ea48da-64e8-4f69-9d0d-80690b682670", 40, 52)
+	s.GetSodanSubMessages("98ea48da-64e8-4f69-9d0d-80690b682670", 48, 52)
 	log.Println("sodan sub messages 1 scraped")
 	s.GetSodanSubMessages("30c30aa5-c380-4324-b227-0ca85c34801c", 22, 32)
 	log.Println("sodan sub messages 2 scraped")

src/scraper/scraper.go

@@ -2,12 +2,13 @@ package scraper
 
 import (
 	"context"
+	"log"
+	"os"
+
 	"github.com/jmoiron/sqlx"
 	"github.com/traPtitech/go-traq"
 	traqwsbot "github.com/traPtitech/traq-ws-bot"
 	"github.com/traPtitech/traq-ws-bot/payload"
-	"log"
-	"os"
 )
 
 type Scraper struct {