summaryrefslogtreecommitdiffstats
path: root/models/actions/schedule_spec_list.go
blob: 4dc43f975bfe69478a4919cf58c3f4151833eb6b (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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
// Copyright 2023 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package actions

import (
	"context"

	"code.gitea.io/gitea/models/db"
	repo_model "code.gitea.io/gitea/models/repo"
	"code.gitea.io/gitea/modules/container"

	"xorm.io/builder"
)

type SpecList []*ActionScheduleSpec

func (specs SpecList) GetScheduleIDs() []int64 {
	return container.FilterSlice(specs, func(spec *ActionScheduleSpec) (int64, bool) {
		return spec.ScheduleID, true
	})
}

func (specs SpecList) LoadSchedules(ctx context.Context) error {
	if len(specs) == 0 {
		return nil
	}

	scheduleIDs := specs.GetScheduleIDs()
	schedules, err := GetSchedulesMapByIDs(ctx, scheduleIDs)
	if err != nil {
		return err
	}
	for _, spec := range specs {
		spec.Schedule = schedules[spec.ScheduleID]
	}

	repoIDs := specs.GetRepoIDs()
	repos, err := GetReposMapByIDs(ctx, repoIDs)
	if err != nil {
		return err
	}
	for _, spec := range specs {
		spec.Repo = repos[spec.RepoID]
	}

	return nil
}

func (specs SpecList) GetRepoIDs() []int64 {
	return container.FilterSlice(specs, func(spec *ActionScheduleSpec) (int64, bool) {
		return spec.RepoID, true
	})
}

func (specs SpecList) LoadRepos(ctx context.Context) error {
	if len(specs) == 0 {
		return nil
	}

	repoIDs := specs.GetRepoIDs()
	repos, err := repo_model.GetRepositoriesMapByIDs(ctx, repoIDs)
	if err != nil {
		return err
	}
	for _, spec := range specs {
		spec.Repo = repos[spec.RepoID]
	}
	return nil
}

type FindSpecOptions struct {
	db.ListOptions
	RepoID int64
	Next   int64
}

func (opts FindSpecOptions) ToConds() builder.Cond {
	cond := builder.NewCond()
	if opts.RepoID > 0 {
		cond = cond.And(builder.Eq{"repo_id": opts.RepoID})
	}

	if opts.Next > 0 {
		cond = cond.And(builder.Lte{"next": opts.Next})
	}

	return cond
}

func (opts FindSpecOptions) ToOrders() string {
	return "`id` DESC"
}

func FindSpecs(ctx context.Context, opts FindSpecOptions) (SpecList, int64, error) {
	specs, total, err := db.FindAndCount[ActionScheduleSpec](ctx, opts)
	if err != nil {
		return nil, 0, err
	}

	if err := SpecList(specs).LoadSchedules(ctx); err != nil {
		return nil, 0, err
	}
	return specs, total, nil
}