summaryrefslogtreecommitdiffstats
path: root/services/f3/driver/organizations.go
blob: 98c4c1497d20ca734d65d2325e28006c51c277d3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
// 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.NewNodeID(organization.ID)
}

func newOrganizations() generic.NodeDriverInterface {
	return &organizations{}
}