listmonk/cmd/lists.go

189 lines
4.7 KiB
Go
Raw Permalink Normal View History

2018-10-25 15:51:47 +02:00
package main
import (
"fmt"
"net/http"
"strconv"
2020-03-07 16:07:48 +01:00
"github.com/gofrs/uuid"
2018-10-25 15:51:47 +02:00
"github.com/knadh/listmonk/models"
"github.com/lib/pq"
"github.com/labstack/echo"
)
2019-05-14 13:11:05 +02:00
type listsWrap struct {
Results []models.List `json:"results"`
Total int `json:"total"`
PerPage int `json:"per_page"`
Page int `json:"page"`
}
var (
listQuerySortFields = []string{"name", "type", "subscriber_count", "created_at", "updated_at"}
)
2018-10-25 15:51:47 +02:00
// handleGetLists handles retrieval of lists.
func handleGetLists(c echo.Context) error {
var (
app = c.Get("app").(*App)
2019-05-14 13:11:05 +02:00
out listsWrap
2018-10-25 15:51:47 +02:00
2020-08-01 15:27:33 +02:00
pg = getPagination(c.QueryParams(), 20, 50)
orderBy = c.FormValue("order_by")
order = c.FormValue("order")
2018-10-25 15:51:47 +02:00
listID, _ = strconv.Atoi(c.Param("id"))
single = false
)
// Fetch one list.
if listID > 0 {
single = true
}
// Sort params.
if !strSliceContains(orderBy, listQuerySortFields) {
orderBy = "created_at"
}
if order != sortAsc && order != sortDesc {
order = sortAsc
}
if err := db.Select(&out.Results, fmt.Sprintf(app.queries.QueryLists, orderBy, order), listID, pg.Offset, pg.Limit); err != nil {
app.log.Printf("error fetching lists: %v", err)
2018-10-25 15:51:47 +02:00
return echo.NewHTTPError(http.StatusInternalServerError,
app.i18n.Ts("globals.messages.errorFetching",
"name", "{globals.terms.lists}", "error", pqErrMsg(err)))
}
if single && len(out.Results) == 0 {
2020-12-19 11:55:52 +01:00
return echo.NewHTTPError(http.StatusBadRequest,
app.i18n.Ts("globals.messages.notFound", "name", "{globals.terms.list}"))
}
if len(out.Results) == 0 {
2018-10-25 15:51:47 +02:00
return c.JSON(http.StatusOK, okResp{[]struct{}{}})
}
// Replace null tags.
2019-05-14 13:11:05 +02:00
for i, v := range out.Results {
2018-10-25 15:51:47 +02:00
if v.Tags == nil {
2019-05-14 13:11:05 +02:00
out.Results[i].Tags = make(pq.StringArray, 0)
2018-10-25 15:51:47 +02:00
}
}
if single {
2019-05-14 13:11:05 +02:00
return c.JSON(http.StatusOK, okResp{out.Results[0]})
2018-10-25 15:51:47 +02:00
}
2019-05-14 13:11:05 +02:00
// Meta.
out.Total = out.Results[0].Total
out.Page = pg.Page
out.PerPage = pg.PerPage
2018-10-25 15:51:47 +02:00
return c.JSON(http.StatusOK, okResp{out})
}
// handleCreateList handles list creation.
func handleCreateList(c echo.Context) error {
var (
app = c.Get("app").(*App)
o = models.List{}
)
if err := c.Bind(&o); err != nil {
return err
}
// Validate.
2020-03-08 08:33:38 +01:00
if !strHasLen(o.Name, 1, stdInputMaxLen) {
2020-12-19 11:55:52 +01:00
return echo.NewHTTPError(http.StatusBadRequest, app.i18n.T("lists.invalidName"))
2018-10-25 15:51:47 +02:00
}
2020-03-07 16:07:48 +01:00
uu, err := uuid.NewV4()
if err != nil {
app.log.Printf("error generating UUID: %v", err)
2020-12-19 11:55:52 +01:00
return echo.NewHTTPError(http.StatusInternalServerError,
app.i18n.Ts("globals.messages.errorUUID", "error", err.Error()))
2020-03-07 16:07:48 +01:00
}
2018-10-25 15:51:47 +02:00
// Insert and read ID.
var newID int
2020-03-07 16:07:48 +01:00
o.UUID = uu.String()
if err := app.queries.CreateList.Get(&newID,
2018-10-25 15:51:47 +02:00
o.UUID,
o.Name,
o.Type,
o.Optin,
2018-10-25 15:51:47 +02:00
pq.StringArray(normalizeTags(o.Tags))); err != nil {
app.log.Printf("error creating list: %v", err)
2018-10-25 15:51:47 +02:00
return echo.NewHTTPError(http.StatusInternalServerError,
app.i18n.Ts("globals.messages.errorCreating",
2020-12-19 11:55:52 +01:00
"name", "{globals.terms.list}", "error", pqErrMsg(err)))
2018-10-25 15:51:47 +02:00
}
// Hand over to the GET handler to return the last insertion.
return handleGetLists(copyEchoCtx(c, map[string]string{
"id": fmt.Sprintf("%d", newID),
}))
2018-10-25 15:51:47 +02:00
}
// handleUpdateList handles list modification.
func handleUpdateList(c echo.Context) error {
var (
app = c.Get("app").(*App)
id, _ = strconv.Atoi(c.Param("id"))
)
if id < 1 {
2020-12-19 11:55:52 +01:00
return echo.NewHTTPError(http.StatusBadRequest, app.i18n.T("globals.messages.invalidID"))
2018-10-25 15:51:47 +02:00
}
// Incoming params.
var o models.List
if err := c.Bind(&o); err != nil {
return err
}
res, err := app.queries.UpdateList.Exec(id,
2020-03-07 18:30:55 +01:00
o.Name, o.Type, o.Optin, pq.StringArray(normalizeTags(o.Tags)))
2018-10-25 15:51:47 +02:00
if err != nil {
app.log.Printf("error updating list: %v", err)
2020-12-19 11:55:52 +01:00
return echo.NewHTTPError(http.StatusInternalServerError,
app.i18n.Ts("globals.messages.errorUpdating",
2020-12-19 11:55:52 +01:00
"name", "{globals.terms.list}", "error", pqErrMsg(err)))
2018-10-25 15:51:47 +02:00
}
if n, _ := res.RowsAffected(); n == 0 {
2020-12-19 11:55:52 +01:00
return echo.NewHTTPError(http.StatusBadRequest,
app.i18n.Ts("globals.messages.notFound", "name", "{globals.terms.list}"))
2018-10-25 15:51:47 +02:00
}
return handleGetLists(c)
}
// handleDeleteLists handles deletion deletion,
// either a single one (ID in the URI), or a list.
func handleDeleteLists(c echo.Context) error {
var (
app = c.Get("app").(*App)
id, _ = strconv.ParseInt(c.Param("id"), 10, 64)
ids pq.Int64Array
)
if id < 1 && len(ids) == 0 {
2020-12-19 11:55:52 +01:00
return echo.NewHTTPError(http.StatusBadRequest, app.i18n.T("globals.messages.invalidID"))
2018-10-25 15:51:47 +02:00
}
if id > 0 {
ids = append(ids, id)
}
if _, err := app.queries.DeleteLists.Exec(ids); err != nil {
app.log.Printf("error deleting lists: %v", err)
2018-10-25 15:51:47 +02:00
return echo.NewHTTPError(http.StatusInternalServerError,
app.i18n.Ts("globals.messages.errorDeleting",
2020-12-19 11:55:52 +01:00
"name", "{globals.terms.list}", "error", pqErrMsg(err)))
2018-10-25 15:51:47 +02:00
}
return c.JSON(http.StatusOK, okResp{true})
}