clipper/backend/cmd/progress-test/main.go

package main

import (
	"context"
	"database/sql"
	"io"
	"log"
	"os"

	"git.netflux.io/rob/clipper/generated/store"
	"git.netflux.io/rob/clipper/media"
	"github.com/aws/aws-sdk-go-v2/config"
	"github.com/aws/aws-sdk-go-v2/service/s3"
	"github.com/kkdai/youtube/v2"
	_ "github.com/lib/pq"
	"go.uber.org/zap"
)

const (
	// videoID = "N1BufwrE1I8"
	videoID = "LBRIsFbOoc4"
)

func main() {
	ctx := context.Background()

	// Create a store
	databaseURL := os.Getenv("DATABASE_URL")
	db, err := sql.Open("postgres", databaseURL)
	if err != nil {
		log.Fatal(err)
	}
	store := store.New(db)

	// Create an Amazon S3 service s3Client
	cfg, err := config.LoadDefaultConfig(ctx)
	if err != nil {
		log.Fatal(err)
	}
	s3Client := s3.NewFromConfig(cfg)

	// Create a Youtube client
	var youtubeClient youtube.Client

	// Create a MediaSetService
	mediaSetService := media.NewMediaSetService(store, &youtubeClient, media.S3API{S3Client: s3Client}, zap.NewNop())

	mediaSet, err := mediaSetService.Get(ctx, videoID)
	if err != nil {
		log.Fatalf("error calling fetch service: %v", err)
	}

	// Create a progressReader
	progressReader, err := mediaSetService.GetAudio(ctx, mediaSet.ID, 2_000)
	if err != nil {
		log.Fatalf("error calling fetch service: %v", err)
	}

	for {
		progress, err := progressReader.Read()
		if err != nil {
			if err != io.EOF {
				log.Printf("error reading progress: %v", err)
			}
			break
		}

		log.Printf("progress = %+v", progress)
	}

	log.Println("done")
}