miniflux_v2/server/ui/controller/category.go

// Copyright 2017 Frédéric Guillot. All rights reserved.
// Use of this source code is governed by the Apache 2.0
// license that can be found in the LICENSE file.

package controller

import (
	"errors"
	"log"

	"github.com/miniflux/miniflux2/model"
	"github.com/miniflux/miniflux2/server/core"
	"github.com/miniflux/miniflux2/server/ui/form"
)

func (c *Controller) ShowCategories(ctx *core.Context, request *core.Request, response *core.Response) {
	args, err := c.getCommonTemplateArgs(ctx)
	if err != nil {
		response.Html().ServerError(err)
		return
	}

	user := ctx.GetLoggedUser()
	categories, err := c.store.GetCategoriesWithFeedCount(user.ID)
	if err != nil {
		response.Html().ServerError(err)
		return
	}

	response.Html().Render("categories", args.Merge(tplParams{
		"categories": categories,
		"total":      len(categories),
		"menu":       "categories",
	}))
}

func (c *Controller) ShowCategoryEntries(ctx *core.Context, request *core.Request, response *core.Response) {
	user := ctx.GetLoggedUser()
	offset := request.GetQueryIntegerParam("offset", 0)

	args, err := c.getCommonTemplateArgs(ctx)
	if err != nil {
		response.Html().ServerError(err)
		return
	}

	category, err := c.getCategoryFromURL(ctx, request, response)
	if err != nil {
		return
	}

	builder := c.store.GetEntryQueryBuilder(user.ID, user.Timezone)
	builder.WithCategoryID(category.ID)
	builder.WithOrder(model.DefaultSortingOrder)
	builder.WithDirection(model.DefaultSortingDirection)
	builder.WithOffset(offset)
	builder.WithLimit(NbItemsPerPage)

	entries, err := builder.GetEntries()
	if err != nil {
		response.Html().ServerError(err)
		return
	}

	count, err := builder.CountEntries()
	if err != nil {
		response.Html().ServerError(err)
		return
	}

	response.Html().Render("category_entries", args.Merge(tplParams{
		"category":   category,
		"entries":    entries,
		"total":      count,
		"pagination": c.getPagination(ctx.GetRoute("categoryEntries", "categoryID", category.ID), count, offset),
		"menu":       "categories",
	}))
}

func (c *Controller) CreateCategory(ctx *core.Context, request *core.Request, response *core.Response) {
	args, err := c.getCommonTemplateArgs(ctx)
	if err != nil {
		response.Html().ServerError(err)
		return
	}

	response.Html().Render("create_category", args.Merge(tplParams{
		"menu": "categories",
	}))
}

func (c *Controller) SaveCategory(ctx *core.Context, request *core.Request, response *core.Response) {
	user := ctx.GetLoggedUser()
	args, err := c.getCommonTemplateArgs(ctx)
	if err != nil {
		response.Html().ServerError(err)
		return
	}

	categoryForm := form.NewCategoryForm(request.GetRequest())
	if err := categoryForm.Validate(); err != nil {
		response.Html().Render("create_category", args.Merge(tplParams{
			"errorMessage": err.Error(),
		}))
		return
	}

	duplicateCategory, err := c.store.GetCategoryByTitle(user.ID, categoryForm.Title)
	if err != nil {
		response.Html().ServerError(err)
		return
	}

	if duplicateCategory != nil {
		response.Html().Render("create_category", args.Merge(tplParams{
			"errorMessage": "This category already exists.",
		}))
		return
	}

	category := model.Category{Title: categoryForm.Title, UserID: user.ID}
	err = c.store.CreateCategory(&category)
	if err != nil {
		log.Println(err)
		response.Html().Render("create_category", args.Merge(tplParams{
			"errorMessage": "Unable to create this category.",
		}))
		return
	}

	response.Redirect(ctx.GetRoute("categories"))
}

func (c *Controller) EditCategory(ctx *core.Context, request *core.Request, response *core.Response) {
	user := ctx.GetLoggedUser()

	category, err := c.getCategoryFromURL(ctx, request, response)
	if err != nil {
		log.Println(err)
		return
	}

	args, err := c.getCategoryFormTemplateArgs(ctx, user, category, nil)
	if err != nil {
		response.Html().ServerError(err)
		return
	}

	response.Html().Render("edit_category", args)
}

func (c *Controller) UpdateCategory(ctx *core.Context, request *core.Request, response *core.Response) {
	user := ctx.GetLoggedUser()

	category, err := c.getCategoryFromURL(ctx, request, response)
	if err != nil {
		log.Println(err)
		return
	}

	categoryForm := form.NewCategoryForm(request.GetRequest())
	args, err := c.getCategoryFormTemplateArgs(ctx, user, category, categoryForm)
	if err != nil {
		response.Html().ServerError(err)
		return
	}

	if err := categoryForm.Validate(); err != nil {
		response.Html().Render("edit_category", args.Merge(tplParams{
			"errorMessage": err.Error(),
		}))
		return
	}

	if c.store.AnotherCategoryExists(user.ID, category.ID, categoryForm.Title) {
		response.Html().Render("edit_category", args.Merge(tplParams{
			"errorMessage": "This category already exists.",
		}))
		return
	}

	err = c.store.UpdateCategory(categoryForm.Merge(category))
	if err != nil {
		log.Println(err)
		response.Html().Render("edit_category", args.Merge(tplParams{
			"errorMessage": "Unable to update this category.",
		}))
		return
	}

	response.Redirect(ctx.GetRoute("categories"))
}

func (c *Controller) RemoveCategory(ctx *core.Context, request *core.Request, response *core.Response) {
	user := ctx.GetLoggedUser()

	category, err := c.getCategoryFromURL(ctx, request, response)
	if err != nil {
		return
	}

	if err := c.store.RemoveCategory(user.ID, category.ID); err != nil {
		response.Html().ServerError(err)
		return
	}

	response.Redirect(ctx.GetRoute("categories"))
}

func (c *Controller) getCategoryFromURL(ctx *core.Context, request *core.Request, response *core.Response) (*model.Category, error) {
	categoryID, err := request.GetIntegerParam("categoryID")
	if err != nil {
		response.Html().BadRequest(err)
		return nil, err
	}

	user := ctx.GetLoggedUser()
	category, err := c.store.GetCategory(user.ID, categoryID)
	if err != nil {
		response.Html().ServerError(err)
		return nil, err
	}

	if category == nil {
		response.Html().NotFound()
		return nil, errors.New("Category not found")
	}

	return category, nil
}

func (c *Controller) getCategoryFormTemplateArgs(ctx *core.Context, user *model.User, category *model.Category, categoryForm *form.CategoryForm) (tplParams, error) {
	args, err := c.getCommonTemplateArgs(ctx)
	if err != nil {
		return nil, err
	}

	if categoryForm == nil {
		args["form"] = form.CategoryForm{
			Title: category.Title,
		}
	} else {
		args["form"] = categoryForm
	}

	args["category"] = category
	args["menu"] = "categories"
	return args, nil
}