mirror of
https://github.com/MichaelMure/git-bug.git
synced 2024-12-15 02:01:43 +03:00
128 lines
2.6 KiB
Go
128 lines
2.6 KiB
Go
package operations
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"io/ioutil"
|
|
"sort"
|
|
|
|
"github.com/MichaelMure/git-bug/bug"
|
|
)
|
|
|
|
// LabelChangeOperation will add or remove a set of labels
|
|
|
|
var _ bug.Operation = LabelChangeOperation{}
|
|
|
|
type LabelChangeOperation struct {
|
|
bug.OpBase
|
|
Added []bug.Label
|
|
Removed []bug.Label
|
|
}
|
|
|
|
func (op LabelChangeOperation) Apply(snapshot bug.Snapshot) bug.Snapshot {
|
|
// Add in the set
|
|
AddLoop:
|
|
for _, added := range op.Added {
|
|
for _, label := range snapshot.Labels {
|
|
if label == added {
|
|
// Already exist
|
|
continue AddLoop
|
|
}
|
|
}
|
|
|
|
snapshot.Labels = append(snapshot.Labels, added)
|
|
}
|
|
|
|
// Remove in the set
|
|
for _, removed := range op.Removed {
|
|
for i, label := range snapshot.Labels {
|
|
if label == removed {
|
|
snapshot.Labels[i] = snapshot.Labels[len(snapshot.Labels)-1]
|
|
snapshot.Labels = snapshot.Labels[:len(snapshot.Labels)-1]
|
|
}
|
|
}
|
|
}
|
|
|
|
// Sort
|
|
sort.Slice(snapshot.Labels, func(i, j int) bool {
|
|
return string(snapshot.Labels[i]) < string(snapshot.Labels[j])
|
|
})
|
|
|
|
return snapshot
|
|
}
|
|
|
|
func NewLabelChangeOperation(author bug.Person, added, removed []bug.Label) LabelChangeOperation {
|
|
return LabelChangeOperation{
|
|
OpBase: bug.NewOpBase(bug.LabelChangeOp, author),
|
|
Added: added,
|
|
Removed: removed,
|
|
}
|
|
}
|
|
|
|
// Convenience function to apply the operation
|
|
func ChangeLabels(out io.Writer, b *bug.Bug, author bug.Person, add, remove []string) error {
|
|
var added, removed []bug.Label
|
|
|
|
if out == nil {
|
|
out = ioutil.Discard
|
|
}
|
|
|
|
snap := b.Compile()
|
|
|
|
for _, str := range add {
|
|
label := bug.Label(str)
|
|
|
|
// check for duplicate
|
|
if labelExist(added, label) {
|
|
fmt.Fprintf(out, "label \"%s\" is a duplicate\n", str)
|
|
continue
|
|
}
|
|
|
|
// check that the label doesn't already exist
|
|
if labelExist(snap.Labels, label) {
|
|
fmt.Fprintf(out, "label \"%s\" is already set on this bug\n", str)
|
|
continue
|
|
}
|
|
|
|
added = append(added, label)
|
|
}
|
|
|
|
for _, str := range remove {
|
|
label := bug.Label(str)
|
|
|
|
// check for duplicate
|
|
if labelExist(removed, label) {
|
|
fmt.Fprintf(out, "label \"%s\" is a duplicate\n", str)
|
|
continue
|
|
}
|
|
|
|
// check that the label actually exist
|
|
if !labelExist(snap.Labels, label) {
|
|
fmt.Fprintf(out, "label \"%s\" doesn't exist on this bug\n", str)
|
|
continue
|
|
}
|
|
|
|
removed = append(removed, label)
|
|
}
|
|
|
|
if len(added) == 0 && len(removed) == 0 {
|
|
return fmt.Errorf("no label added or removed")
|
|
}
|
|
|
|
labelOp := NewLabelChangeOperation(author, added, removed)
|
|
|
|
b.Append(labelOp)
|
|
|
|
return nil
|
|
}
|
|
|
|
func labelExist(labels []bug.Label, label bug.Label) bool {
|
|
for _, l := range labels {
|
|
if l == label {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|