gokhttp_download

type CustomTime struct {
	*atomic.Time
}

type DownloadTask interface {
	Download(ctx context.Context) error
	Type() DownloadType
	Progress(sb *strings.Builder) error
	ResetDelta()
}

type DownloadType int

type DownloadVersion int

type FMP4Demuxer struct {
	Muxer            *StreamMuxer
	InitSegment      []byte
	AudioInitSegment []byte
	// contains filtered or unexported fields
}

type GlobalDownloadTracker struct {
	// Synchronization
	sync.WaitGroup
	GraceFulStop *atomic.Bool `json:"-"`
	// Idle stats
	IdleSince   *CustomTime    `json:"idleSince"`
	IdleValue   *atomic.Uint64 `json:"-"`
	IdleTimeout *atomic.Int64  `json:"-"`
	// MetaData
	LastTick     *CustomTime                    `json:"lastTick"`
	CurrentIP    *atomic.String                 `json:"currentIP"`
	TotalThreads *atomic.Uint64                 `json:"totalThreads"`
	Tasks        *TaskMap[string, DownloadTask] `json:"tasks"`
	// Total stats
	TotalFiles *atomic.Uint64 `json:"totalFiles"`
	TotalBytes *atomic.Uint64 `json:"totalBytes"`
	// Download stats
	DownloadedFiles *atomic.Uint64 `json:"downloadedFiles"`
	DownloadedBytes *atomic.Uint64 `json:"downloadedBytes"`
	// Per tick stats
	DeltaBytes *atomic.Uint64 `json:"deltaBytes"`
}

type HLSAudioRendition struct {
	GroupID   string
	Name      string
	Language  *string
	Channels  *string
	URI       *string
	MediaItem *m3u8.MediaItem
}

type HLSStreamSelector struct {
	VideoIndex int // index into sorted video variants (default 0 = highest)
	AudioIndex int // index into sorted audio renditions (default 0 = first match)
}

type HLSVariant struct {
	Bandwidth        int
	AverageBandwidth *int
	Resolution       *m3u8.Resolution
	Codecs           *string
	Name             *string
	FrameRate        *float64
	PlaylistItem     *m3u8.PlaylistItem
}

type SequencedBuffer struct {
	MSN  int
	Data []byte
}

type SequencedSegment struct {
	MSN     int
	Segment *m3u8.SegmentItem
}

type StreamFormat int

type StreamHLSTask struct {
	// Tracking ref
	Global  *GlobalDownloadTracker    `json:"-"`
	HClient *http.Client              `json:"-"`
	ReqOpts []gokhttp_requests.Option `json:"-"`

	// Muxers (only one is used, based on Format)
	Muxer       *StreamMuxer `json:"-"`
	FMP4Demuxer *FMP4Demuxer `json:"-"`
	TaskStats   *StreamStats `json:"taskStats"`

	TaskType     DownloadType                    `json:"taskType"`
	TaskVersion  DownloadVersion                 `json:"taskVersion"`
	FileName     *atomic.String                  `json:"fileName"`
	FileLocation *atomic.String                  `json:"fileLocation"`
	PlayListUrl  *atomic.String                  `json:"playListUrl"`
	BaseUrl      *url.URL                        `json:"-"`
	Opts         []gokhttp_requests.Option       `json:"-"`
	SegmentChan  chan SequencedSegment           `json:"-"`
	BufferChan   chan SequencedBuffer            `json:"-"`
	SegmentCache *hashmap.Map[string, time.Time] `json:"-"`
	SaveSegments bool                            `json:"-"`

	// Format detection (set during getSegments init)
	Format StreamFormat `json:"format"`

	// Stream selection
	StreamSelector *HLSStreamSelector `json:"-"`

	// Audio stream support
	HasAudioStream    bool                            `json:"-"`
	AudioPlaylistUrl  *atomic.String                  `json:"-"`
	AudioBaseUrl      *url.URL                        `json:"-"`
	AudioSegmentChan  chan SequencedSegment           `json:"-"`
	AudioBufferChan   chan SequencedBuffer            `json:"-"`
	AudioSegmentCache *hashmap.Map[string, time.Time] `json:"-"`
	// contains filtered or unexported fields
}

type StreamMuxer struct {
	F       *os.File
	Muxer   *mpeg2.TSMuxer
	Demuxer *mpeg2.TSDemuxer
	Streams map[mpeg2.TS_STREAM_TYPE]uint16
	// contains filtered or unexported fields
}

type StreamStats struct {
	DownloadedSegments *atomic.Uint64 `json:"downloadedSegments"`
	DownloadedDuration *atomic.Int64  `json:"downloadedDuration"`
	DownloadedBytes    *atomic.Uint64 `json:"downloadedBytes"`
}

type TaskMap[K string, V DownloadTask] struct {
	*hashmap.Map[K, V]
}

type ThreadedChunk struct {
	// Parent
	Parent *ThreadedDownloadTask `json:"-"`
	// Metadata
	ChunkID    *atomic.Uint64 `json:"chunkID"`
	ChunkStart uint64         `json:"-"`
	ChunkStop  uint64         `json:"-"`
	F          *os.File       `json:"-"`
	// Totals
	FileSize *atomic.Uint64 `json:"fileSize"`
	// Download stats
	DownloadedBytes *atomic.Uint64 `json:"downloadedBytes"`
	DeltaBytes      *atomic.Uint64 `json:"deltaBytes"`
}

type ThreadedDownloadTask struct {
	// Tracking ref
	Global  *GlobalDownloadTracker    `json:"-"`
	HClient *http.Client              `json:"-"`
	ReqOpts []gokhttp_requests.Option `json:"-"`
	// Metadata
	TaskType     DownloadType    `json:"taskType"`
	TaskVersion  DownloadVersion `json:"taskVersion"`
	FileName     *atomic.String  `json:"fileName"`
	FileLocation *atomic.String  `json:"fileLocation"`
	FileURL      *atomic.String  `json:"fileURL"`
	Resumable    *atomic.Bool    `json:"resumable"`
	// Totals
	TaskStats *ThreadedChunk `json:"taskStats"`
	// Chunks
	Chunks     *hashmap.Map[string, *ThreadedChunk] `json:"chunks"`
	ChunkCount *atomic.Uint64                       `json:"chunkCount"`
}