package routes import ( "encoding/json" "log" "net/http" "strings" "github.com/go-chi/chi/v5" "github.com/google/uuid" "github.com/jtom38/newsbot/collector/database" ) // GetDiscordWebHooks // @Summary Returns the top 100 entries from the queue to be processed. // @Produce application/json // @Tags Config, Discord, Webhook // @Router /discord/webhooks [get] func (s *Server) GetDiscordWebHooks(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "application/json") res, err := s.Db.ListDiscordWebhooks(*s.ctx, 100) if err != nil { w.Write([]byte(err.Error())) panic(err) } bres, err := json.Marshal(res) if err != nil { w.Write([]byte(err.Error())) panic(err) } w.Write(bres) } // GetDiscordWebHook // @Summary Returns the top 100 entries from the queue to be processed. // @Produce application/json // @Param id path string true "id" // @Tags Config, Discord, Webhook // @Router /discord/webhooks/{id} [get] func (s *Server) GetDiscordWebHooksById(w http.ResponseWriter, r *http.Request) { w.Header().Set("Content-Type", "application/json") query := r.URL.Query() _id := query["id"][0] if _id == "" { http.Error(w, "id is missing", http.StatusBadRequest) return } uuid, err := uuid.Parse(_id) if err != nil { http.Error(w, "unable to parse id value", http.StatusBadRequest) return } res, err := s.Db.GetDiscordWebHooksByID(*s.ctx, uuid) if err != nil { http.Error(w, "no record found", http.StatusBadRequest) return } bres, err := json.Marshal(res) if err != nil { http.Error(w, "unable to convert to json", http.StatusBadRequest) panic(err) } w.Write(bres) } // NewDiscordWebHook // @Summary Creates a new record for a discord web hook to post data to. // @Param url query string true "url" // @Param server query string true "Server name" // @Param channel query string true "Channel name" // @Tags Config, Discord, Webhook // @Router /discord/webhooks/new [post] func (s *Server) NewDiscordWebHook(w http.ResponseWriter, r *http.Request) { query := r.URL.Query() _url := query["url"][0] _server := query["server"][0] _channel := query["channel"][0] if _url == "" { http.Error(w, "url is missing a value", http.StatusBadRequest) return } if !strings.Contains(_url, "discord.com/api/webhooks") { http.Error(w, "invalid url", http.StatusBadRequest) return } if _server == ""{ http.Error(w, "server is missing", http.StatusBadRequest) } if _channel == "" { http.Error(w, "channel is missing", http.StatusBadRequest) } params := database.CreateDiscordWebHookParams{ ID: uuid.New(), Url: _url, Server: _server, Channel: _channel, Enabled: true, } s.Db.CreateDiscordWebHook(*s.ctx, params) bJson, err := json.Marshal(¶ms) if err != nil { log.Panicln(err) } w.Header().Set("Content-Type", "application/json") w.Write(bJson) } // DisableDiscordWebHooks // @Summary Disables a Webhook from being used. // @Param id path string true "id" // @Tags Config, Discord, Webhook // @Router /discord/webhooks/{id}/disable [post] func (s *Server) disableDiscordWebHook(w http.ResponseWriter, r *http.Request) { id := chi.URLParam(r, "ID") uuid, err := uuid.Parse(id) if err != nil { log.Panicln(err) } // Check to make sure we can find the record _, err = s.Db.GetDiscordWebHooksByID(*s.ctx, uuid) if err != nil { http.Error(w, err.Error(), http.StatusBadRequest) } err = s.Db.DisableDiscordWebHook(*s.ctx, uuid) if err != nil { log.Panic(err) } } // EnableDiscordWebHook // @Summary Enables a source to continue processing. // @Param id path string true "id" // @Tags Config, Discord, Webhook // @Router /discord/webhooks/{id}/enable [post] func (s *Server) enableDiscordWebHook(w http.ResponseWriter, r *http.Request) { id := chi.URLParam(r, "ID") uuid, err := uuid.Parse(id) if err != nil { log.Panicln(err) } // Check to make sure we can find the record _, err = s.Db.GetDiscordWebHooksByID(*s.ctx, uuid) if err != nil { http.Error(w, err.Error(), http.StatusBadRequest) } err = s.Db.EnableDiscordWebHook(*s.ctx, uuid) if err != nil { log.Panic(err) } } // DeleteDiscordWebHook // @Summary Deletes a record by ID. // @Param id path string true "id" // @Tags Config, Discord, Webhook // @Router /discord/webhooks/{id} [delete] func (s *Server) deleteDiscordWebHook(w http.ResponseWriter, r *http.Request) { //var item model.Sources = model.Sources{} id := chi.URLParam(r, "ID") uuid, err := uuid.Parse(id) if err != nil { log.Panicln(err) } // Check to make sure we can find the record _, err = s.Db.GetDiscordQueueByID(*s.ctx, uuid) if err != nil { http.Error(w, err.Error(), http.StatusBadRequest) } // Delete the record err = s.Db.DeleteDiscordWebHooks(*s.ctx, uuid) if err != nil { log.Panic(err) } }