|  | @@ -0,0 +1,219 @@
 | 
	
		
			
				|  |  | +// 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"
 | 
	
		
			
				|  |  | +)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +func GetIssueLabels(ctx *context.APIContext) {
 | 
	
		
			
				|  |  | +	issue, err := models.GetIssueByIndex(ctx.Repo.Repository.ID, ctx.ParamsInt64(":index"))
 | 
	
		
			
				|  |  | +	if err != nil {
 | 
	
		
			
				|  |  | +		if models.IsErrIssueNotExist(err) {
 | 
	
		
			
				|  |  | +			ctx.Status(404)
 | 
	
		
			
				|  |  | +		} else {
 | 
	
		
			
				|  |  | +			ctx.Error(500, "GetIssueByIndex", err)
 | 
	
		
			
				|  |  | +		}
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	apiLabels := make([]*api.Label, len(issue.Labels))
 | 
	
		
			
				|  |  | +	for i := range issue.Labels {
 | 
	
		
			
				|  |  | +		apiLabels[i] = convert.ToLabel(issue.Labels[i])
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	ctx.JSON(200, &apiLabels)
 | 
	
		
			
				|  |  | +}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +func AddIssueLabels(ctx *context.APIContext, form api.IssueLabelsOption) {
 | 
	
		
			
				|  |  | +	if !ctx.Repo.IsWriter() {
 | 
	
		
			
				|  |  | +		ctx.Status(403)
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	issue, err := models.GetIssueByIndex(ctx.Repo.Repository.ID, ctx.ParamsInt64(":index"))
 | 
	
		
			
				|  |  | +	if err != nil {
 | 
	
		
			
				|  |  | +		if models.IsErrIssueNotExist(err) {
 | 
	
		
			
				|  |  | +			ctx.Status(404)
 | 
	
		
			
				|  |  | +		} else {
 | 
	
		
			
				|  |  | +			ctx.Error(500, "GetIssueByIndex", err)
 | 
	
		
			
				|  |  | +		}
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	var labels []*models.Label
 | 
	
		
			
				|  |  | +	if labels, err = filterLabelsByRepoID(form.Labels, issue.RepoID); err != nil {
 | 
	
		
			
				|  |  | +		ctx.Error(400, "filterLabelsByRepoID", err)
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	for i := range labels {
 | 
	
		
			
				|  |  | +		if !models.HasIssueLabel(issue.ID, labels[i].ID) {
 | 
	
		
			
				|  |  | +			if err := models.NewIssueLabel(issue, labels[i]); err != nil {
 | 
	
		
			
				|  |  | +				ctx.Error(500, "NewIssueLabel", err)
 | 
	
		
			
				|  |  | +				return
 | 
	
		
			
				|  |  | +			}
 | 
	
		
			
				|  |  | +		}
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	// Refresh issue to get the updated list of labels from the DB
 | 
	
		
			
				|  |  | +	issue, err = models.GetIssueByIndex(ctx.Repo.Repository.ID, ctx.ParamsInt64(":index"))
 | 
	
		
			
				|  |  | +	if err != nil {
 | 
	
		
			
				|  |  | +		if models.IsErrIssueNotExist(err) {
 | 
	
		
			
				|  |  | +			ctx.Status(404)
 | 
	
		
			
				|  |  | +		} else {
 | 
	
		
			
				|  |  | +			ctx.Error(500, "GetIssueByIndex", err)
 | 
	
		
			
				|  |  | +		}
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	apiLabels := make([]*api.Label, len(issue.Labels))
 | 
	
		
			
				|  |  | +	for i := range issue.Labels {
 | 
	
		
			
				|  |  | +		apiLabels[i] = convert.ToLabel(issue.Labels[i])
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	ctx.JSON(200, &apiLabels)
 | 
	
		
			
				|  |  | +}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +func ReplaceIssueLabels(ctx *context.APIContext, form api.IssueLabelsOption) {
 | 
	
		
			
				|  |  | +	if !ctx.Repo.IsWriter() {
 | 
	
		
			
				|  |  | +		ctx.Status(403)
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	issue, err := models.GetIssueByIndex(ctx.Repo.Repository.ID, ctx.ParamsInt64(":index"))
 | 
	
		
			
				|  |  | +	if err != nil {
 | 
	
		
			
				|  |  | +		if models.IsErrIssueNotExist(err) {
 | 
	
		
			
				|  |  | +			ctx.Status(404)
 | 
	
		
			
				|  |  | +		} else {
 | 
	
		
			
				|  |  | +			ctx.Error(500, "GetIssueByIndex", err)
 | 
	
		
			
				|  |  | +		}
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	var labels []*models.Label
 | 
	
		
			
				|  |  | +	if labels, err = filterLabelsByRepoID(form.Labels, issue.RepoID); err != nil {
 | 
	
		
			
				|  |  | +		ctx.Error(400, "filterLabelsByRepoID", err)
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	if err := issue.ClearLabels(); err != nil {
 | 
	
		
			
				|  |  | +		ctx.Error(500, "ClearLabels", err)
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	for i := range labels {
 | 
	
		
			
				|  |  | +		if err := models.NewIssueLabel(issue, labels[i]); err != nil {
 | 
	
		
			
				|  |  | +			ctx.Error(500, "NewIssueLabel", err)
 | 
	
		
			
				|  |  | +			return
 | 
	
		
			
				|  |  | +		}
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	// Refresh issue to get the updated list of labels from the DB
 | 
	
		
			
				|  |  | +	issue, err = models.GetIssueByIndex(ctx.Repo.Repository.ID, ctx.ParamsInt64(":index"))
 | 
	
		
			
				|  |  | +	if err != nil {
 | 
	
		
			
				|  |  | +		if models.IsErrIssueNotExist(err) {
 | 
	
		
			
				|  |  | +			ctx.Status(404)
 | 
	
		
			
				|  |  | +		} else {
 | 
	
		
			
				|  |  | +			ctx.Error(500, "GetIssueByIndex", err)
 | 
	
		
			
				|  |  | +		}
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	apiLabels := make([]*api.Label, len(issue.Labels))
 | 
	
		
			
				|  |  | +	for i := range issue.Labels {
 | 
	
		
			
				|  |  | +		apiLabels[i] = convert.ToLabel(issue.Labels[i])
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	ctx.JSON(200, &apiLabels)
 | 
	
		
			
				|  |  | +}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +func DeleteIssueLabel(ctx *context.APIContext) {
 | 
	
		
			
				|  |  | +	if !ctx.Repo.IsWriter() {
 | 
	
		
			
				|  |  | +		ctx.Status(403)
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	issue, err := models.GetIssueByIndex(ctx.Repo.Repository.ID, ctx.ParamsInt64(":index"))
 | 
	
		
			
				|  |  | +	if err != nil {
 | 
	
		
			
				|  |  | +		if models.IsErrIssueNotExist(err) {
 | 
	
		
			
				|  |  | +			ctx.Status(404)
 | 
	
		
			
				|  |  | +		} else {
 | 
	
		
			
				|  |  | +			ctx.Error(500, "GetIssueByIndex", err)
 | 
	
		
			
				|  |  | +		}
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	label, err := models.GetLabelByID(ctx.ParamsInt64(":id"))
 | 
	
		
			
				|  |  | +	if err != nil {
 | 
	
		
			
				|  |  | +		if models.IsErrLabelNotExist(err) {
 | 
	
		
			
				|  |  | +			ctx.Status(400)
 | 
	
		
			
				|  |  | +		} else {
 | 
	
		
			
				|  |  | +			ctx.Error(500, "GetLabelByID", err)
 | 
	
		
			
				|  |  | +		}
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	if err := models.DeleteIssueLabel(issue, label); err != nil {
 | 
	
		
			
				|  |  | +		ctx.Error(500, "DeleteIssueLabel", err)
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	ctx.Status(204)
 | 
	
		
			
				|  |  | +}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +func ClearIssueLabels(ctx *context.APIContext) {
 | 
	
		
			
				|  |  | +	if !ctx.Repo.IsWriter() {
 | 
	
		
			
				|  |  | +		ctx.Status(403)
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	issue, err := models.GetIssueByIndex(ctx.Repo.Repository.ID, ctx.ParamsInt64(":index"))
 | 
	
		
			
				|  |  | +	if err != nil {
 | 
	
		
			
				|  |  | +		if models.IsErrIssueNotExist(err) {
 | 
	
		
			
				|  |  | +			ctx.Status(404)
 | 
	
		
			
				|  |  | +		} else {
 | 
	
		
			
				|  |  | +			ctx.Error(500, "GetIssueByIndex", err)
 | 
	
		
			
				|  |  | +		}
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	if err := issue.ClearLabels(); err != nil {
 | 
	
		
			
				|  |  | +		ctx.Error(500, "ClearLabels", err)
 | 
	
		
			
				|  |  | +		return
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	ctx.Status(204)
 | 
	
		
			
				|  |  | +}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +func filterLabelsByRepoID(labelIDs []int64, repoID int64) ([]*models.Label, error) {
 | 
	
		
			
				|  |  | +	labels := make([]*models.Label, 0, len(labelIDs))
 | 
	
		
			
				|  |  | +	errors := make([]error, 0, len(labelIDs))
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	for i := range labelIDs {
 | 
	
		
			
				|  |  | +		label, err := models.GetLabelByID(labelIDs[i])
 | 
	
		
			
				|  |  | +		if err != nil {
 | 
	
		
			
				|  |  | +			errors = append(errors, err)
 | 
	
		
			
				|  |  | +		} else if label.RepoID != repoID {
 | 
	
		
			
				|  |  | +			errors = append(errors, models.ErrLabelNotValidForRepository{label.ID, repoID})
 | 
	
		
			
				|  |  | +		} else {
 | 
	
		
			
				|  |  | +			labels = append(labels, label)
 | 
	
		
			
				|  |  | +		}
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	errorCount := len(errors)
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	if errorCount == 1 {
 | 
	
		
			
				|  |  | +		return labels, errors[0]
 | 
	
		
			
				|  |  | +	} else if errorCount > 1 {
 | 
	
		
			
				|  |  | +		return labels, models.ErrMultipleErrors{errors}
 | 
	
		
			
				|  |  | +	}
 | 
	
		
			
				|  |  | +
 | 
	
		
			
				|  |  | +	return labels, nil
 | 
	
		
			
				|  |  | +}
 |