b3eaf2249d
Add new methods GetIDString() & SetIDString() & ToFormatInterface()
Change the prototype of the fixture function
(cherry picked from commit d7b263ff8b)
138 lines
3 KiB
Go
138 lines
3 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 {
|
|
g *Forgejo
|
|
project *ProjectProvider
|
|
}
|
|
|
|
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 *Label) *Label {
|
|
l := label.Label
|
|
l.RepoID = project.GetID()
|
|
if err := issues_model.NewLabel(ctx, &l); err != nil {
|
|
panic(err)
|
|
}
|
|
return o.Get(ctx, user, project, LabelConverter(&l))
|
|
}
|
|
|
|
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(project.GetID(), l.GetID()); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
return l
|
|
}
|