From e6a97b667598ca4a94ee58fff9e171c284769c39 Mon Sep 17 00:00:00 2001 From: Ethan Koenig Date: Tue, 31 Oct 2017 19:25:14 -0700 Subject: [PATCH] Use api.TrackedTime in API (#2807) * Use api.TrackedTime in API * APIFormat() method * Rename function --- models/issue_tracked_time.go | 13 ++++++ routers/api/v1/repo/issue_tracked_time.go | 50 +++++++++++++++-------- 2 files changed, 47 insertions(+), 16 deletions(-) diff --git a/models/issue_tracked_time.go b/models/issue_tracked_time.go index b1c82b6207..3b2360f68f 100644 --- a/models/issue_tracked_time.go +++ b/models/issue_tracked_time.go @@ -7,6 +7,8 @@ package models import ( "time" + api "code.gitea.io/sdk/gitea" + "github.com/go-xorm/builder" ) @@ -25,6 +27,17 @@ func (t *TrackedTime) AfterLoad() { t.Created = time.Unix(t.CreatedUnix, 0).Local() } +// APIFormat converts TrackedTime to API format +func (t *TrackedTime) APIFormat() *api.TrackedTime { + return &api.TrackedTime{ + ID: t.ID, + IssueID: t.IssueID, + UserID: t.UserID, + Time: t.Time, + Created: t.Created, + } +} + // FindTrackedTimesOptions represent the filters for tracked times. If an ID is 0 it will be ignored. type FindTrackedTimesOptions struct { IssueID int64 diff --git a/routers/api/v1/repo/issue_tracked_time.go b/routers/api/v1/repo/issue_tracked_time.go index 964fc11ddb..3962a7b0e2 100644 --- a/routers/api/v1/repo/issue_tracked_time.go +++ b/routers/api/v1/repo/issue_tracked_time.go @@ -10,6 +10,14 @@ import ( api "code.gitea.io/sdk/gitea" ) +func trackedTimesToAPIFormat(trackedTimes []*models.TrackedTime) []*api.TrackedTime { + apiTrackedTimes := make([]*api.TrackedTime, len(trackedTimes)) + for i, trackedTime := range trackedTimes { + apiTrackedTimes[i] = trackedTime.APIFormat() + } + return apiTrackedTimes +} + // ListTrackedTimes list all the tracked times of an issue func ListTrackedTimes(ctx *context.APIContext) { // swagger:route GET /repos/{username}/{reponame}/issues/{issue}/times repository issueTrackedTimes @@ -35,11 +43,13 @@ func ListTrackedTimes(ctx *context.APIContext) { return } - if trackedTimes, err := models.GetTrackedTimes(models.FindTrackedTimesOptions{IssueID: issue.ID}); err != nil { + trackedTimes, err := models.GetTrackedTimes(models.FindTrackedTimesOptions{IssueID: issue.ID}) + if err != nil { ctx.Error(500, "GetTrackedTimesByIssue", err) - } else { - ctx.JSON(200, &trackedTimes) + return } + apiTrackedTimes := trackedTimesToAPIFormat(trackedTimes) + ctx.JSON(200, &apiTrackedTimes) } // AddTime adds time manual to the given issue @@ -73,13 +83,12 @@ func AddTime(ctx *context.APIContext, form api.AddTimeOption) { ctx.Status(403) return } - var tt *models.TrackedTime - if tt, err = models.AddTime(ctx.User, issue, form.Time); err != nil { + trackedTime, err := models.AddTime(ctx.User, issue, form.Time) + if err != nil { ctx.Error(500, "AddTime", err) return } - ctx.JSON(200, tt) - + ctx.JSON(200, trackedTime.APIFormat()) } // ListTrackedTimesByUser lists all tracked times of the user @@ -111,11 +120,15 @@ func ListTrackedTimesByUser(ctx *context.APIContext) { ctx.Status(404) return } - if trackedTimes, err := models.GetTrackedTimes(models.FindTrackedTimesOptions{UserID: user.ID, RepositoryID: ctx.Repo.Repository.ID}); err != nil { + trackedTimes, err := models.GetTrackedTimes(models.FindTrackedTimesOptions{ + UserID: user.ID, + RepositoryID: ctx.Repo.Repository.ID}) + if err != nil { ctx.Error(500, "GetTrackedTimesByUser", err) - } else { - ctx.JSON(200, &trackedTimes) + return } + apiTrackedTimes := trackedTimesToAPIFormat(trackedTimes) + ctx.JSON(200, &apiTrackedTimes) } // ListTrackedTimesByRepository lists all tracked times of the user @@ -133,11 +146,14 @@ func ListTrackedTimesByRepository(ctx *context.APIContext) { ctx.JSON(400, struct{ Message string }{Message: "time tracking disabled"}) return } - if trackedTimes, err := models.GetTrackedTimes(models.FindTrackedTimesOptions{RepositoryID: ctx.Repo.Repository.ID}); err != nil { + trackedTimes, err := models.GetTrackedTimes(models.FindTrackedTimesOptions{ + RepositoryID: ctx.Repo.Repository.ID}) + if err != nil { ctx.Error(500, "GetTrackedTimesByUser", err) - } else { - ctx.JSON(200, &trackedTimes) + return } + apiTrackedTimes := trackedTimesToAPIFormat(trackedTimes) + ctx.JSON(200, &apiTrackedTimes) } // ListMyTrackedTimes lists all tracked times of the current user @@ -150,9 +166,11 @@ func ListMyTrackedTimes(ctx *context.APIContext) { // Responses: // 200: TrackedTimes // 500: error - if trackedTimes, err := models.GetTrackedTimes(models.FindTrackedTimesOptions{UserID: ctx.User.ID}); err != nil { + trackedTimes, err := models.GetTrackedTimes(models.FindTrackedTimesOptions{UserID: ctx.User.ID}) + if err != nil { ctx.Error(500, "GetTrackedTimesByUser", err) - } else { - ctx.JSON(200, &trackedTimes) + return } + apiTrackedTimes := trackedTimesToAPIFormat(trackedTimes) + ctx.JSON(200, &apiTrackedTimes) }