From 24065dddf2c79f9823699b0c4dfb776a365d949f Mon Sep 17 00:00:00 2001
From: Paul Buetow <paul@buetow.org>
Date: Mon, 21 Oct 2024 10:33:01 +0300
Subject: implement priority tag

---
 internal/schedule/schedule.go | 29 ++++++++++++++++++++++-------
 1 file changed, 22 insertions(+), 7 deletions(-)

(limited to 'internal/schedule')

diff --git a/internal/schedule/schedule.go b/internal/schedule/schedule.go
index be2b6f8..b6be6be 100644
--- a/internal/schedule/schedule.go
+++ b/internal/schedule/schedule.go
@@ -6,6 +6,7 @@ import (
 	"log"
 	"os"
 	"path/filepath"
+	"slices"
 	"strings"
 
 	"codeberg.org/snonux/gos/internal/config"
@@ -30,8 +31,27 @@ func Run(args config.Args, platform string) (entry.Entry, error) {
 		return entry.Zero, ErrNothingToSchedule
 	}
 
-	// Schedule random qeued entry for platform
-	ent, err := oi.ReadDirRandom(dir, func(file os.DirEntry) (entry.Entry, bool) {
+	// Schedule random qeued entry for platform. First, try to find one with
+	// a priority tag.
+	ent, err := selectRandomEntry(dir, "prio")
+	if errors.Is(err, oi.ErrNotFound) {
+		// No entry with priority tag found, select another one.
+		ent, err = selectRandomEntry(dir, "")
+	}
+	if err != nil {
+		return entry.Zero, fmt.Errorf("%w: %w", ErrNothingQueued, err)
+	}
+	return ent, nil
+}
+
+// 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
+		}
 		ent, err := entry.New(filepath.Join(dir, file.Name()))
 		if err != nil {
 			log.Println(err)
@@ -39,9 +59,4 @@ func Run(args config.Args, platform string) (entry.Entry, error) {
 		}
 		return ent, ent.State == entry.Queued
 	})
-
-	if err != nil {
-		return entry.Zero, fmt.Errorf("%w: %w", ErrNothingQueued, err)
-	}
-	return ent, nil
 }
-- 
cgit v1.2.3