cmd/streamdata: peek at subsequent assets for parts

Some streams disconnect half-way. This collects all parts.

Signed-off-by: Yonah <contrib@gensokyo.uk>

cmd/streamdata/format.go

@@ -18,4 +18,7 @@ func printVOD(v *streamdata.VOD) {
 	} else {
 		fmt.Printf("YouTube  : %s | %s\n", v.Title, date)
 	}
+	if v.Mirror != "" {
+		fmt.Println("Mirror   :", v.Mirror)
+	}
 }

cmd/streamdata/main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"bufio"
+	"cmp"
 	"context"
 	"errors"
 	"fmt"
@@ -11,6 +12,8 @@ import (
 	"os"
 	"os/signal"
 	"path"
+	"slices"
+	"strconv"
 	"strings"
 	"syscall"
 	"time"
@@ -221,13 +224,108 @@ func main() {
 				return errors.New("list requires a channel selected")
 			}
 
-			for ident := range channel.All(&err) {
+			for ident := range channel.All(&err, true) {
 				fmt.Println(ident)
 			}
 			return
 		},
 	)
 
+	c.NewCommand(
+		"mirror",
+		"Populate mirror address of the next un-mirrored asset",
+		func([]string) (err error) {
+			if channel == nil {
+				return errors.New("mirror requires a channel selected")
+			}
+
+			idents := slices.Collect(channel.All(&err, false))
+			if err != nil {
+				return
+			}
+			slices.SortFunc(idents, func(a, b *streamdata.Ident) int {
+				return cmp.Compare(a.Serial, b.Serial)
+			})
+
+			unique := errors.New("entry already mirrored")
+			for i, ident := range idents {
+				if err = channel.Edit(ident, func(v *streamdata.VOD) (err error) {
+					if v.Mirror != "" {
+						return unique
+					}
+
+					var parts []*streamdata.VOD
+					for _, next := range idents[i+1:] {
+						var part *streamdata.VOD
+						if part, err = channel.Load(next); err != nil {
+							return
+						}
+						if part.Title != v.Title {
+							break
+						}
+						parts = append(parts, part)
+					}
+
+					pathname := path.Join(
+						flagBase,
+						"upload"+streamdata.ChannelVODSuffix,
+					)
+					if err = os.Symlink(path.Join(
+						"vod",
+						ident.String()+streamdata.ChannelVODSuffix,
+					), pathname); err != nil {
+						return
+					}
+					log.Println("pending video at", pathname)
+					defer func() {
+						removeErr := os.Remove(pathname)
+						if err == nil {
+							err = removeErr
+						}
+					}()
+
+					defer toErr(&err)
+					isattyStd()
+					br := bufio.NewReader(os.Stdin)
+
+					if len(parts) > 0 {
+						fmt.Println("assets with identical title:")
+						for j := range parts {
+							fmt.Println(idents[i+1+j])
+						}
+						if !require(promptBool(br, "Proceed? ", true)) {
+							for j, part := range parts {
+								fmt.Println(idents[i+1+j])
+								printVOD(part)
+							}
+							return errors.New(
+								strconv.Itoa(len(parts)) + " trailing parts",
+							)
+						}
+					}
+
+				retry:
+					fmt.Println(ident)
+					printVOD(v)
+					v.Mirror = require(promptTrimNoEmpty(br, "Mirror   : "))
+					if !require(promptBool(br, "Proceed? ", true)) {
+						v.Mirror = ""
+						goto retry
+					}
+					return nil
+				}); err != nil {
+					if err == unique {
+						err = nil
+						continue
+					}
+					return
+				}
+				break
+			}
+			return
+		},
+	)
+
 	c.MustParse(os.Args[1:], func(err error) {
 		if channel != nil {
 			if closeErr := channel.Close(); closeErr != nil {

streamdata.go

@@ -276,7 +276,10 @@ func (c *Channel) Path(ident *Ident) string {
 // All returns an iterator over all known [Ident] in the on-disk representation.
 // Iteration stops when encountering the first non-nil error, and its value is
 // saved to the value pointed to by errP.
-func (c *Channel) All(errP *error) iter.Seq[*Ident] {
+//
+// If reuse is true, the same value is updated every iteration and the same
+// address is yileded as a result.
+func (c *Channel) All(errP *error, reuse bool) iter.Seq[*Ident] {
 	return func(yield func(*Ident) bool) {
 		dents, err := c.root.FS().(fs.ReadDirFS).ReadDir(channelPathVOD)
 		if err != nil {
@@ -298,7 +301,11 @@ func (c *Channel) All(errP *error) iter.Seq[*Ident] {
 				return
 			}
 
-			if !yield(&ident) {
+			p := &ident
+			if !reuse {
+				p = new(ident)
+			}
+			if !yield(p) {
 				return
 			}
 		}

streamdata_test.go

@@ -303,7 +303,7 @@ func TestChannel(t *testing.T) {
 	}
 
 	var iterErr error
-	idents := slices.Collect(c.All(&iterErr))
+	idents := slices.Collect(c.All(&iterErr, false))
 	if iterErr != nil {
 		t.Fatalf("All: error = %#v", iterErr)
 	}