Added project files
This commit is contained in:
parent
84547f6dcb
commit
94a5aff260
23 changed files with 1042 additions and 1 deletions
252
core/gtv_stream.go
Normal file
252
core/gtv_stream.go
Normal file
|
|
@ -0,0 +1,252 @@
|
|||
// Copyright (c) 2025, Julian Müller (ChaoticByte)
|
||||
|
||||
package core
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/signal"
|
||||
"sort"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
const MaxRetries = 5
|
||||
|
||||
type Chapter struct {
|
||||
Index int `json:"index"`
|
||||
Title string `json:"title"`
|
||||
Offset time.Duration `json:"offset"`
|
||||
}
|
||||
|
||||
type StreamEpisode struct {
|
||||
Episode string `json:"episode"`
|
||||
Formats []VideoFormat `json:"formats"`
|
||||
Title string `json:"title"`
|
||||
// ProposedFilename string `json:"proposed_filename"`
|
||||
PlaylistUrl string `json:"playlist_url"`
|
||||
Chapters []Chapter `json:"chapters"`
|
||||
}
|
||||
|
||||
func (ep *StreamEpisode) GetFormatByName(formatName string) (VideoFormat, error) {
|
||||
var idx int
|
||||
var err error = nil
|
||||
if formatName == "auto" {
|
||||
// at the moment, the best format is always the first -> 0
|
||||
return ep.Formats[idx], nil
|
||||
} else {
|
||||
formatFound := false
|
||||
for i, f := range ep.Formats {
|
||||
if f.Name == formatName {
|
||||
idx = i
|
||||
formatFound = true
|
||||
}
|
||||
}
|
||||
if !formatFound {
|
||||
err = &FormatNotFoundError{FormatName: formatName}
|
||||
}
|
||||
return ep.Formats[idx], err
|
||||
}
|
||||
}
|
||||
|
||||
func (ep *StreamEpisode) GetProposedFilename(chapterIdx int) string {
|
||||
if chapterIdx >= 0 && chapterIdx < len(ep.Chapters) {
|
||||
return fmt.Sprintf("GTV%04s - %v. %s.ts", ep.Episode, chapterIdx+1, sanitizeUnicodeFilename(ep.Chapters[chapterIdx].Title))
|
||||
} else {
|
||||
return sanitizeUnicodeFilename(ep.Title) + ".ts"
|
||||
}
|
||||
}
|
||||
|
||||
func (ep *StreamEpisode) Download(args Arguments, ui UserInterface, interruptChan chan os.Signal) error {
|
||||
// Set automatic values
|
||||
if args.OutputFile == "" {
|
||||
args.OutputFile = ep.GetProposedFilename(args.ChapterIdx)
|
||||
}
|
||||
if args.ChapterIdx >= 0 {
|
||||
if args.StartDuration < 0 {
|
||||
args.StartDuration = time.Duration(ep.Chapters[args.ChapterIdx].Offset)
|
||||
}
|
||||
if args.StopDuration < 0 && args.ChapterIdx+1 < len(ep.Chapters) {
|
||||
// next chapter is stop
|
||||
args.StopDuration = time.Duration(ep.Chapters[args.ChapterIdx+1].Offset)
|
||||
}
|
||||
}
|
||||
//
|
||||
var err error
|
||||
var nextChunk int = 0
|
||||
var videoFile *os.File
|
||||
var infoFile *os.File
|
||||
var infoFilename string
|
||||
if !args.Overwrite && !args.ContinueDl {
|
||||
if _, err := os.Stat(args.OutputFile); err == nil {
|
||||
return &FileExistsError{Filename: args.OutputFile}
|
||||
}
|
||||
}
|
||||
videoFile, err = os.OpenFile(args.OutputFile, os.O_RDWR|os.O_CREATE, 0660)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer videoFile.Close()
|
||||
if args.Overwrite {
|
||||
videoFile.Truncate(0)
|
||||
}
|
||||
// always seek to the end
|
||||
videoFile.Seek(0, io.SeekEnd)
|
||||
// info file
|
||||
infoFilename = args.OutputFile + ".dl-info"
|
||||
if args.ContinueDl {
|
||||
infoFileData, err := os.ReadFile(infoFilename)
|
||||
if err != nil {
|
||||
return errors.New("could not access download info file, can't continue download")
|
||||
}
|
||||
i, err := strconv.ParseInt(string(infoFileData), 10, 32)
|
||||
nextChunk = int(i)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
infoFile, err = os.OpenFile(infoFilename, os.O_RDWR|os.O_CREATE, 0660)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
infoFile.Truncate(0)
|
||||
infoFile.Seek(0, io.SeekStart)
|
||||
_, err = infoFile.Write([]byte(strconv.Itoa(nextChunk)))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// download
|
||||
format, _ := ep.GetFormatByName(args.FormatName) // we don't have to check the error, as it was already checked by CliRun()
|
||||
chunklist, err := GetStreamChunkList(format)
|
||||
chunklist = chunklist.Cut(args.StartDuration, args.StopDuration)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
var bufferDt float64
|
||||
var progress float32
|
||||
var actualRate float64
|
||||
keyboardInterrupt := false
|
||||
signal.Notify(interruptChan, os.Interrupt)
|
||||
go func() {
|
||||
// Handle Interrupts
|
||||
<-interruptChan
|
||||
keyboardInterrupt = true
|
||||
ui.DownloadProgress(progress, actualRate, false, false, 0, ep.Title)
|
||||
ui.Aborted()
|
||||
}()
|
||||
for i, chunk := range chunklist.Chunks {
|
||||
if i < nextChunk {
|
||||
continue
|
||||
}
|
||||
var time1 int64
|
||||
var data []byte
|
||||
retries := 0
|
||||
for {
|
||||
if keyboardInterrupt {
|
||||
break
|
||||
}
|
||||
time1 = time.Now().UnixNano()
|
||||
ui.DownloadProgress(progress, actualRate, false, true, retries, ep.Title)
|
||||
data, err = httpGet(chunklist.BaseUrl+"/"+chunk, []http.Header{ApiHeadersBase, ApiHeadersVideoAdditional}, time.Second*5)
|
||||
if err != nil {
|
||||
if retries == MaxRetries {
|
||||
return err
|
||||
}
|
||||
retries++
|
||||
continue
|
||||
}
|
||||
break
|
||||
}
|
||||
if keyboardInterrupt {
|
||||
break
|
||||
}
|
||||
var dtDownload float64 = float64(time.Now().UnixNano()-time1) / 1000000000.0
|
||||
rate := float64(len(data)) / dtDownload
|
||||
actualRate = rate - max(rate-args.Ratelimit, 0)
|
||||
progress = float32(i+1) / float32(len(chunklist.Chunks))
|
||||
delayNow := bufferDt > RatelimitDelayAfter
|
||||
ui.DownloadProgress(progress, actualRate, delayNow, false, retries, ep.Title)
|
||||
if delayNow {
|
||||
bufferDt = 0
|
||||
// this simulates that the buffering is finished and the player is playing
|
||||
time.Sleep(time.Duration(RatelimitDelay * float64(time.Second)))
|
||||
} else if rate > args.Ratelimit {
|
||||
// slow down, we are too fast.
|
||||
deferTime := (rate - args.Ratelimit) / args.Ratelimit * dtDownload
|
||||
time.Sleep(time.Duration(deferTime * float64(time.Second)))
|
||||
}
|
||||
videoFile.Write(data)
|
||||
nextChunk++
|
||||
infoFile.Truncate(0)
|
||||
infoFile.Seek(0, io.SeekStart)
|
||||
infoFile.Write([]byte(strconv.Itoa(nextChunk)))
|
||||
var dtIteration float64 = float64(time.Now().UnixNano()-time1) / 1000000000.0
|
||||
if !delayNow {
|
||||
bufferDt += dtIteration
|
||||
}
|
||||
}
|
||||
infoFile.Close()
|
||||
if !keyboardInterrupt {
|
||||
err := os.Remove(infoFilename)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func GetStreamEpisode(episode string) (StreamEpisode, error) {
|
||||
ep := StreamEpisode{}
|
||||
ep.Episode = episode
|
||||
info_data, err := httpGet(
|
||||
fmt.Sprintf(ApiBaseurlStreamEpisodeInfo, episode),
|
||||
[]http.Header{ApiHeadersBase, ApiHeadersMetaAdditional},
|
||||
time.Second*10,
|
||||
)
|
||||
if err != nil {
|
||||
return ep, err
|
||||
}
|
||||
// Title
|
||||
json.Unmarshal(info_data, &ep)
|
||||
ep.Title = strings.ToValidUTF8(ep.Title, "")
|
||||
// Sort Chapters, correct offset and set index
|
||||
sort.Slice(ep.Chapters, func(i int, j int) bool {
|
||||
return ep.Chapters[i].Offset < ep.Chapters[j].Offset
|
||||
})
|
||||
for i := range ep.Chapters {
|
||||
ep.Chapters[i].Offset = ep.Chapters[i].Offset * time.Second
|
||||
ep.Chapters[i].Index = i
|
||||
}
|
||||
// Formats
|
||||
playlist_url_data, err := httpGet(
|
||||
fmt.Sprintf(ApiBaseurlStreamEpisodePlInfo, episode),
|
||||
[]http.Header{ApiHeadersBase, ApiHeadersMetaAdditional},
|
||||
time.Second*10,
|
||||
)
|
||||
if err != nil {
|
||||
return ep, err
|
||||
}
|
||||
json.Unmarshal(playlist_url_data, &ep)
|
||||
playlist_data, err := httpGet(
|
||||
ep.PlaylistUrl,
|
||||
[]http.Header{ApiHeadersBase, ApiHeadersMetaAdditional},
|
||||
time.Second*10,
|
||||
)
|
||||
ep.Formats = parseAvailFormatsFromM3u8(string(playlist_data))
|
||||
return ep, err
|
||||
}
|
||||
|
||||
func GetStreamChunkList(video VideoFormat) (ChunkList, error) {
|
||||
baseUrl := video.Url[:strings.LastIndex(video.Url, "/")]
|
||||
data, err := httpGet(video.Url, []http.Header{ApiHeadersBase, ApiHeadersMetaAdditional}, time.Second*10)
|
||||
if err != nil {
|
||||
return ChunkList{}, err
|
||||
}
|
||||
chunklist, err := parseChunkListFromM3u8(string(data), baseUrl)
|
||||
return chunklist, err
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue