51 lines
1.4 KiB
Go
51 lines
1.4 KiB
Go
|
// Copyright Earl Warren <contact@earl-warren.org>
|
||
|
|
// Copyright Loïc Dachary <loic@dachary.org>
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
|
||
|
|
package driver
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"code.gitea.io/gitea/models/db"
|
||
|
|
org_model "code.gitea.io/gitea/models/organization"
|
||
|
|
user_model "code.gitea.io/gitea/models/user"
|
||
|
|
|
||
|
|
f3_tree "code.forgejo.org/f3/gof3/v3/tree/f3"
|
||
|
|
"code.forgejo.org/f3/gof3/v3/tree/generic"
|
||
|
|
)
|
||
|
|
|
||
|
|
type organizations struct {
|
||
|
|
container
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *organizations) ListPage(ctx context.Context, page int) generic.ChildrenSlice {
|
||
|
|
sess := db.GetEngine(ctx)
|
||
|
|
if page != 0 {
|
||
|
|
sess = db.SetSessionPagination(sess, &db.ListOptions{Page: page, PageSize: o.getPageSize()})
|
||
|
|
}
|
||
|
|
sess = sess.Select("`user`.*").
|
||
|
|
Where("`type`=?", user_model.UserTypeOrganization)
|
||
|
|
organizations := make([]*org_model.Organization, 0, o.getPageSize())
|
||
|
|
|
||
|
|
if err := sess.Find(&organizations); err != nil {
|
||
|
|
panic(fmt.Errorf("error while listing organizations: %v", err))
|
||
|
|
}
|
||
|
|
|
||
|
|
return f3_tree.ConvertListed(ctx, o.GetNode(), f3_tree.ConvertToAny(organizations...)...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *organizations) GetIDFromName(ctx context.Context, name string) generic.NodeID {
|
||
|
|
organization, err := org_model.GetOrgByName(ctx, name)
|
||
|
|
if err != nil {
|
||
|
|
panic(fmt.Errorf("GetOrganizationByName: %v", err))
|
||
|
|
}
|
||
|
|
|
||
|
|
return generic.NodeID(fmt.Sprintf("%d", organization.ID))
|
||
|
|
}
|
||
|
|
|
||
|
|
func newOrganizations() generic.NodeDriverInterface {
|
||
|
|
return &organizations{}
|
||
|
|
}
|