nyaa-pantsu/controllers/moderator/announcements.go

package moderatorController

import (
	"fmt"
	"html"
	"math"
	"net/http"
	"strconv"
	"time"

	"github.com/NyaaPantsu/nyaa/models"
	"github.com/NyaaPantsu/nyaa/models/notifications"
	"github.com/NyaaPantsu/nyaa/templates"
	"github.com/NyaaPantsu/nyaa/utils/log"
	msg "github.com/NyaaPantsu/nyaa/utils/messages"
	"github.com/NyaaPantsu/nyaa/utils/validator"
	"github.com/NyaaPantsu/nyaa/utils/validator/announcement"
	"github.com/gin-gonic/gin"
)

func listAnnouncements(c *gin.Context) {
	page := c.Param("page")
	pagenum := 1
	offset := 100
	var err error
	messages := msg.GetMessages(c)
	deleted := c.Request.URL.Query()["deleted"]
	if deleted != nil {
		messages.AddInfoTf("infos", "annoucement_deleted")
	}
	if page != "" {
		pagenum, err = strconv.Atoi(html.EscapeString(page))
		if !log.CheckError(err) {
			c.AbortWithError(http.StatusInternalServerError, err)
			return
		}
	}
	var conditions string
	var values []interface{}

	announcements, nbAnnouncements := notifications.FindAll(offset, (pagenum-1)*offset, conditions, values...)
	nav := templates.Navigation{nbAnnouncements, offset, pagenum, "mod/announcements/p"}
	templates.ModelList(c, "admin/announcements.jet.html", announcements, nav, templates.NewSearchForm(c))
}

func addAnnouncement(c *gin.Context) {
	announcement := &models.Notification{}
	messages := msg.GetMessages(c)

	id := c.Query("id")
	if id == "" && len(messages.GetInfos("ID_ANNOUNCEMENT")) > 0 {
		id = messages.GetInfos("ID_ANNOUNCEMENT")[0]
	}
	idInt, _ := strconv.Atoi(id)
	if idInt > 0 {
		var err error
		announcement, _ = notifications.FindByID(uint(idInt))
		if err != nil {
			c.AbortWithStatus(http.StatusNotFound)
		}
	}
	delay := int(math.Ceil(math.Max(1, float64(announcement.Expire.Sub(time.Now())/(24*time.Hour)))))
	form := &announcementValidator.CreateForm{
		ID:      announcement.ID,
		Message: announcement.Content,
		Delay:   delay,
	}
	c.Bind(form)
	if form.Delay == 0 {
		form.Delay = delay
	}
	templates.Form(c, "admin/announcement_form.jet.html", form)
}

func postAnnouncement(c *gin.Context) {
	messages := msg.GetMessages(c)
	announcement := &models.Notification{}
	id, _ := strconv.Atoi(c.Query("id"))
	if id > 0 {
		var err error
		announcement, err = notifications.FindByID(uint(id))
		if err != nil {
			c.AbortWithStatus(http.StatusNotFound)
		}
	}
	form := &announcementValidator.CreateForm{}
	// We bind the request to the form
	c.Bind(form)
	// We try to validate the form
	validator.ValidateForm(form, messages)
	// If validation has failed, errors are added in messages variable
	if !messages.HasErrors() {
		// No errors, check if we update or create
		if id > 0 { // announcement exists we update
			err := notifications.UpdateAnnouncement(announcement, form) // Making the update query
			if err != nil {
				// Error, we add it to the messages variable
				messages.AddErrorT("errors", "update_failed")
			} else {
				// Success, we add a notice to the messages variable
				messages.AddInfoT("infos", "update_success")
			}
		} else { // announcement doesn't exist, we create it
			var err error
			announcement, err := notifications.NotifyAll(form.Message, time.Now().AddDate(0, 0, form.Delay))
			if err != nil {
				// Error, we add it as a message
				messages.AddErrorT("errors", "create_failed")
			} else {
				// Success, we redirect to the edit form
				messages.AddInfoT("infos", "create_anouncement_success")
				id := fmt.Sprintf("%d", announcement.ID)
				messages.AddInfo("ID_ANNOUNCEMENT", id)
			}
		}
	}
	// If we are still here, we show the form
	addAnnouncement(c)
}

// deleteAnnouncement : Controller for deleting an announcement
func deleteAnnouncement(c *gin.Context) {
	id, _ := strconv.ParseInt(c.Query("id"), 10, 32)
	announcement, err := notifications.FindByID(uint(id))
	if err != nil {
		c.AbortWithStatus(http.StatusNotFound)
		return
	}
	err = announcement.Delete()
	if err != nil {
		c.AbortWithError(http.StatusInternalServerError, err)
		return
	}
	c.Redirect(http.StatusSeeOther, "/mod/announcement?deleted")
}