mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2025-02-06 22:37:24 +01:00
162 lines
3.5 KiB
Go
162 lines
3.5 KiB
Go
|
// SPDX-License-Identifier: MIT
|
||
|
|
||
|
package driver
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"fmt"
|
||
|
|
||
|
"code.gitea.io/gitea/models/db"
|
||
|
issues_model "code.gitea.io/gitea/models/issues"
|
||
|
|
||
|
"lab.forgefriends.org/friendlyforgeformat/gof3/format"
|
||
|
"lab.forgefriends.org/friendlyforgeformat/gof3/util"
|
||
|
)
|
||
|
|
||
|
type Label struct {
|
||
|
issues_model.Label
|
||
|
}
|
||
|
|
||
|
func LabelConverter(f *issues_model.Label) *Label {
|
||
|
return &Label{
|
||
|
Label: *f,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (o Label) GetID() int64 {
|
||
|
return o.ID
|
||
|
}
|
||
|
|
||
|
func (o Label) GetIDString() string {
|
||
|
return fmt.Sprintf("%d", o.GetID())
|
||
|
}
|
||
|
|
||
|
func (o Label) GetName() string {
|
||
|
return o.Name
|
||
|
}
|
||
|
|
||
|
func (o *Label) SetID(id int64) {
|
||
|
o.ID = id
|
||
|
}
|
||
|
|
||
|
func (o *Label) SetIDString(id string) {
|
||
|
o.SetID(util.ParseInt(id))
|
||
|
}
|
||
|
|
||
|
func (o *Label) IsNil() bool {
|
||
|
return o.ID == 0
|
||
|
}
|
||
|
|
||
|
func (o *Label) Equals(other *Label) bool {
|
||
|
return o.Name == other.Name
|
||
|
}
|
||
|
|
||
|
func (o *Label) ToFormatInterface() format.Interface {
|
||
|
return o.ToFormat()
|
||
|
}
|
||
|
|
||
|
func (o *Label) ToFormat() *format.Label {
|
||
|
return &format.Label{
|
||
|
Common: format.NewCommon(o.ID),
|
||
|
Name: o.Name,
|
||
|
Color: o.Color,
|
||
|
Description: o.Description,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (o *Label) FromFormat(label *format.Label) {
|
||
|
*o = Label{
|
||
|
Label: issues_model.Label{
|
||
|
ID: label.GetID(),
|
||
|
Name: label.Name,
|
||
|
Description: label.Description,
|
||
|
Color: label.Color,
|
||
|
},
|
||
|
}
|
||
|
}
|
||
|
|
||
|
type LabelProvider struct {
|
||
|
BaseProviderWithProjectProvider
|
||
|
}
|
||
|
|
||
|
func (o *LabelProvider) ToFormat(ctx context.Context, label *Label) *format.Label {
|
||
|
return label.ToFormat()
|
||
|
}
|
||
|
|
||
|
func (o *LabelProvider) FromFormat(ctx context.Context, m *format.Label) *Label {
|
||
|
var label Label
|
||
|
label.FromFormat(m)
|
||
|
return &label
|
||
|
}
|
||
|
|
||
|
func (o *LabelProvider) GetObjects(ctx context.Context, user *User, project *Project, page int) []*Label {
|
||
|
labels, err := issues_model.GetLabelsByRepoID(ctx, project.GetID(), "", db.ListOptions{Page: page, PageSize: o.g.perPage})
|
||
|
if err != nil {
|
||
|
panic(fmt.Errorf("error while listing labels: %v", err))
|
||
|
}
|
||
|
|
||
|
r := util.ConvertMap[*issues_model.Label, *Label](labels, LabelConverter)
|
||
|
if o.project != nil {
|
||
|
o.project.labels = util.NewNameIDMap[*Label](r)
|
||
|
}
|
||
|
return r
|
||
|
}
|
||
|
|
||
|
func (o *LabelProvider) ProcessObject(ctx context.Context, user *User, project *Project, label *Label) {
|
||
|
}
|
||
|
|
||
|
func (o *LabelProvider) Get(ctx context.Context, user *User, project *Project, exemplar *Label) *Label {
|
||
|
id := exemplar.GetID()
|
||
|
label, err := issues_model.GetLabelInRepoByID(ctx, project.GetID(), id)
|
||
|
if issues_model.IsErrRepoLabelNotExist(err) {
|
||
|
return &Label{}
|
||
|
}
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
return LabelConverter(label)
|
||
|
}
|
||
|
|
||
|
func (o *LabelProvider) Put(ctx context.Context, user *User, project *Project, label, existing *Label) *Label {
|
||
|
l := label.Label
|
||
|
l.RepoID = project.GetID()
|
||
|
|
||
|
var result *Label
|
||
|
|
||
|
if existing == nil || existing.IsNil() {
|
||
|
if err := issues_model.NewLabel(ctx, &l); err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
result = LabelConverter(&l)
|
||
|
} else {
|
||
|
var u issues_model.Label
|
||
|
u.ID = existing.GetID()
|
||
|
cols := make([]string, 0, 10)
|
||
|
|
||
|
if l.Name != existing.Name {
|
||
|
u.Name = l.Name
|
||
|
cols = append(cols, "name")
|
||
|
}
|
||
|
|
||
|
if len(cols) > 0 {
|
||
|
if _, err := db.GetEngine(ctx).ID(existing.ID).Cols(cols...).Update(u); err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
result = existing
|
||
|
}
|
||
|
|
||
|
return o.Get(ctx, user, project, result)
|
||
|
}
|
||
|
|
||
|
func (o *LabelProvider) Delete(ctx context.Context, user *User, project *Project, label *Label) *Label {
|
||
|
l := o.Get(ctx, user, project, label)
|
||
|
if !l.IsNil() {
|
||
|
if err := issues_model.DeleteLabel(ctx, project.GetID(), l.GetID()); err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
}
|
||
|
return l
|
||
|
}
|