Additional API support for milestones (#3383)

1 files changed, 150 insertions, 0 deletions

diff --git a/routers/api/v1/repo/milestone.go b/routers/api/v1/repo/milestone.go
new file mode 100644
index 0000000000..416d9da90f
--- /dev/null
+++ b/routers/api/v1/repo/milestone.go
@@ -0,0 +1,150 @@
+// Copyright 2016 The Gogs Authors. All rights reserved.
+// Use of this source code is governed by a MIT-style
+// license that can be found in the LICENSE file.
+
+package repo
+
+import (
+	api "github.com/gogits/go-gogs-client"
+
+	"github.com/gogits/gogs/models"
+	"github.com/gogits/gogs/modules/context"
+	"github.com/gogits/gogs/routers/api/v1/convert"
+	"time"
+)
+
+func ListMilestones(ctx *context.APIContext) {
+	milestones, err := models.GetAllRepoMilestones(ctx.Repo.Repository.ID)
+	if err != nil {
+		ctx.Error(500, "GetAllRepoMilestones", err)
+		return
+	}
+
+	apiMilestones := make([]*api.Milestone, len(milestones))
+	for i := range milestones {
+		apiMilestones[i] = convert.ToMilestone(milestones[i])
+	}
+	ctx.JSON(200, &apiMilestones)
+}
+
+func GetMilestone(ctx *context.APIContext) {
+	milestone, err := models.GetRepoMilestoneByID(ctx.Repo.Repository.ID, ctx.ParamsInt64(":id"))
+	if err != nil {
+		if models.IsErrMilestoneNotExist(err) {
+			ctx.Status(404)
+		} else {
+			ctx.Error(500, "GetRepoMilestoneByID", err)
+		}
+		return
+	}
+	ctx.JSON(200, convert.ToMilestone(milestone))
+}
+
+func CreateMilestone(ctx *context.APIContext, form api.CreateMilestoneOption) {
+	if !ctx.Repo.IsWriter() {
+		ctx.Status(403)
+		return
+	}
+
+	if form.Deadline == nil {
+		defaultDeadline, _ := time.ParseInLocation("2006-01-02", "9999-12-31", time.Local)
+		form.Deadline = &defaultDeadline
+	}
+
+	milestone := &models.Milestone{
+		RepoID:   ctx.Repo.Repository.ID,
+		Name:     form.Title,
+		Content:  form.Description,
+		Deadline: *form.Deadline,
+	}
+
+	if err := models.NewMilestone(milestone); err != nil {
+		ctx.Error(500, "NewMilestone", err)
+		return
+	}
+	ctx.JSON(201, convert.ToMilestone(milestone))
+}
+
+func EditMilestone(ctx *context.APIContext, form api.EditMilestoneOption) {
+	if !ctx.Repo.IsWriter() {
+		ctx.Status(403)
+		return
+	}
+
+	milestone, err := models.GetRepoMilestoneByID(ctx.Repo.Repository.ID, ctx.ParamsInt64(":id"))
+	if err != nil {
+		if models.IsErrMilestoneNotExist(err) {
+			ctx.Status(404)
+		} else {
+			ctx.Error(500, "GetRepoMilestoneByID", err)
+		}
+		return
+	}
+
+	if len(form.Title) > 0 {
+		milestone.Name = form.Title
+	}
+	if len(form.Description) > 0 {
+		milestone.Content = form.Description
+	}
+	if !form.Deadline.IsZero() {
+		milestone.Deadline = *form.Deadline
+	}
+	if err := models.UpdateMilestone(milestone); err != nil {
+		ctx.Handle(500, "UpdateMilestone", err)
+		return
+	}
+	ctx.JSON(200, convert.ToMilestone(milestone))
+}
+
+func DeleteMilestone(ctx *context.APIContext) {
+	if !ctx.Repo.IsWriter() {
+		ctx.Status(403)
+		return
+	}
+
+	if err := models.DeleteMilestoneByID(ctx.ParamsInt64(":id")); err != nil {
+		ctx.Error(500, "DeleteMilestoneByID", err)
+		return
+	}
+	ctx.Status(204)
+}
+
+func ChangeMilestoneStatus(ctx *context.APIContext) {
+	if !ctx.Repo.IsWriter() {
+		ctx.Status(403)
+		return
+	}
+
+	m, err := models.GetMilestoneByID(ctx.ParamsInt64(":id"))
+	if err != nil {
+		if models.IsErrMilestoneNotExist(err) {
+			ctx.Handle(404, "GetMilestoneByID", err)
+		} else {
+			ctx.Handle(500, "GetMilestoneByID", err)
+		}
+		return
+	}
+
+	switch ctx.Params(":action") {
+	case "open":
+		if m.IsClosed {
+			if err = models.ChangeMilestoneStatus(m, false); err != nil {
+				ctx.Handle(500, "ChangeMilestoneStatus", err)
+				return
+			}
+		}
+		ctx.JSON(200, convert.ToMilestone(m))
+	case "close":
+		if !m.IsClosed {
+			m.ClosedDate = time.Now()
+			if err = models.ChangeMilestoneStatus(m, true); err != nil {
+				ctx.Handle(500, "ChangeMilestoneStatus", err)
+				return
+			}
+		}
+		ctx.JSON(200, convert.ToMilestone(m))
+	default:
+		ctx.Status(400)
+	}
+}