package invidious

import (
	"bytes"
	"time"

	"github.com/birabittoh/myks"
	"github.com/birabittoh/rabbitpipe"
	"github.com/sirupsen/logrus"
)

var logger = logrus.New()
var buffers = myks.New[VideoBuffer](time.Minute)
var RP = rabbitpipe.New()

type VideoBuffer struct {
	Buffer *bytes.Buffer
	Length int64
}

func GetVideoURL(video rabbitpipe.Video) string {
	for _, format := range video.FormatStreams {
		if format.Itag == "18" {
			return format.URL
		}
	}
	return ""
}

func NewVideoBuffer(b *bytes.Buffer, l int64) *VideoBuffer {
	d := new(bytes.Buffer)
	d.Write(b.Bytes())

	return &VideoBuffer{
		Buffer: d,
		Length: l,
	}
}

func (vb *VideoBuffer) Clone() *VideoBuffer {
	return NewVideoBuffer(vb.Buffer, vb.Length)
}

func (vb *VideoBuffer) ValidateLength() bool {
	return vb.Length > 0 && vb.Length == int64(vb.Buffer.Len())
}