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
|
package schedule
import (
"errors"
"fmt"
"os"
"path/filepath"
"slices"
"strings"
"codeberg.org/snonux/gos/internal/colour"
"codeberg.org/snonux/gos/internal/config"
"codeberg.org/snonux/gos/internal/entry"
"codeberg.org/snonux/gos/internal/oi"
"codeberg.org/snonux/gos/internal/platforms"
"codeberg.org/snonux/gos/internal/prompt"
)
var (
ErrNothingToSchedule = errors.New("nothing to schedule")
ErrNothingQueued = errors.New("nothing queued")
)
func Run(args config.Args, platform platforms.Platform) (entry.Entry, error) {
dir := fmt.Sprintf("%s/db/platforms/%s", args.GosDir, platform.String())
stats, err := newStats(args.GosDir, platform.String(), args.Lookback, args.Target, args.PauseDays, args.MaxDaysQueued, args.Config)
if err != nil {
return entry.Zero, err
}
stats.RenderTable(platform)
if platform != "noop" && stats.queued < args.MinQueued {
_ = prompt.Acknowledge(
fmt.Sprintf("There are only %d messages queued for %s - time to fill it up!",
stats.queued, platform),
)
}
en, err := selectEntry(dir)
if err != nil && !errors.Is(err, oi.ErrNotFound) {
return en, err
}
if !en.HasTag("now") && stats.targetHit() {
return entry.Zero, ErrNothingToSchedule
}
return en, nil
}
/**
* Select a random entry, but in this order:
* 1. Any antry with the now tag
* 2. Any entry with the prio tag
* 3. Any entry with the soon tag
* 4. Any other entry
*/
func selectEntry(dir string) (en entry.Entry, err error) {
tagsToTry := []string{"now", "prio", "soon", ""}
for _, tag := range tagsToTry {
if en, err = selectRandomEntry(dir, tag); err == nil {
return
}
if !errors.Is(err, oi.ErrNotFound) {
return
}
}
err = ErrNothingQueued
return
}
// Select a random queed entry with a given tag. If the tag is the empty string,
// then select any random qeued entry.
func selectRandomEntry(dir, tag string) (entry.Entry, error) {
return oi.ReadDirRandom(dir, func(file os.DirEntry) (entry.Entry, bool) {
// Is there a ".TAG." in the file name?
if tag != "" && !slices.Contains(strings.Split(file.Name(), "."), tag) {
return entry.Zero, false
}
en, err := entry.New(filepath.Join(dir, file.Name()))
if err != nil {
colour.Infoln(err)
return entry.Zero, false
}
return en, en.State == entry.Queued
})
}
|
