1 files changed, 97 insertions, 0 deletions
diff --git a/server/api/controller/category.go b/server/api/controller/category.go
new file mode 100644
index 00000000..01aa14b1
--- /dev/null
+++ b/server/api/controller/category.go
@@ -0,0 +1,97 @@
+// 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 api
+
+import (
+	"errors"
+	"github.com/miniflux/miniflux2/server/api/payload"
+	"github.com/miniflux/miniflux2/server/core"
+)
+
+// CreateCategory is the API handler to create a new category.
+func (c *Controller) CreateCategory(ctx *core.Context, request *core.Request, response *core.Response) {
+	category, err := payload.DecodeCategoryPayload(request.GetBody())
+	if err != nil {
+		response.Json().BadRequest(err)
+		return
+	}
+
+	category.UserID = ctx.GetUserID()
+	if err := category.ValidateCategoryCreation(); err != nil {
+		response.Json().ServerError(err)
+		return
+	}
+
+	err = c.store.CreateCategory(category)
+	if err != nil {
+		response.Json().ServerError(errors.New("Unable to create this category"))
+		return
+	}
+
+	response.Json().Created(category)
+}
+
+// UpdateCategory is the API handler to update a category.
+func (c *Controller) UpdateCategory(ctx *core.Context, request *core.Request, response *core.Response) {
+	categoryID, err := request.GetIntegerParam("categoryID")
+	if err != nil {
+		response.Json().BadRequest(err)
+		return
+	}
+
+	category, err := payload.DecodeCategoryPayload(request.GetBody())
+	if err != nil {
+		response.Json().BadRequest(err)
+		return
+	}
+
+	category.UserID = ctx.GetUserID()
+	category.ID = categoryID
+	if err := category.ValidateCategoryModification(); err != nil {
+		response.Json().BadRequest(err)
+		return
+	}
+
+	err = c.store.UpdateCategory(category)
+	if err != nil {
+		response.Json().ServerError(errors.New("Unable to update this category"))
+		return
+	}
+
+	response.Json().Created(category)
+}
+
+// GetCategories is the API handler to get a list of categories for a given user.
+func (c *Controller) GetCategories(ctx *core.Context, request *core.Request, response *core.Response) {
+	categories, err := c.store.GetCategories(ctx.GetUserID())
+	if err != nil {
+		response.Json().ServerError(errors.New("Unable to fetch categories"))
+		return
+	}
+
+	response.Json().Standard(categories)
+}
+
+// RemoveCategory is the API handler to remove a category.
+func (c *Controller) RemoveCategory(ctx *core.Context, request *core.Request, response *core.Response) {
+	userID := ctx.GetUserID()
+	categoryID, err := request.GetIntegerParam("categoryID")
+	if err != nil {
+		response.Json().BadRequest(err)
+		return
+	}
+
+	if !c.store.CategoryExists(userID, categoryID) {
+		response.Json().NotFound(errors.New("Category not found"))
+		return
+	}
+
+	if err := c.store.RemoveCategory(userID, categoryID); err != nil {
+		response.Json().ServerError(errors.New("Unable to remove this category"))
+		return
+	}
+
+	response.Json().NoContent()
+}