teacat/chaturbate-dvr
1
0
Fork 0
mirror of https://github.com/teacat/chaturbate-dvr.git synced 2025-10-29 16:59:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added Ability to add CloudFlare Cookie

Browse Source
This commit is contained in:
HeapOfChaos 2025-03-23 02:04:56 -04:00 committed by GitHub
parent 7541b977ff
commit d3cbe90d09
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 76 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

85
chaturbate/channel_internal.go
View File

@ -18,14 +18,30 @@ import (
// requestChannelBody requests the channel page and returns the body.
func (w *Channel) requestChannelBody() (string, error) {
transport := &http.Transport{
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
}
client := &http.Client{Transport: transport}
resp, err := client.Get(w.ChannelURL)
req, err := http.NewRequest("GET", w.ChannelURL, nil)
if err != nil {
return "", fmt.Errorf("client get: %w", err)
return "", fmt.Errorf("new request: %w", err)
}
if w.CFCookie != "" {
cookie := &http.Cookie{
Name: "cf_clearance",
Value: w.CFCookie,
}
req.AddCookie(cookie)
}
if w.UserAgent != "" {
req.Header.Set("User-Agent", w.UserAgent)
}
resp, err := client.Do(req)
if err != nil {
return "", fmt.Errorf("client do: %w", err)
}
defer resp.Body.Close()
@ -33,7 +49,7 @@ func (w *Channel) requestChannelBody() (string, error) {
if err != nil {
return "", fmt.Errorf("read body: %w", err)
}
return string(body), nil
}
@ -100,9 +116,25 @@ func (w *Channel) resolveSource(body string) (string, string, error) {
}
client := &http.Client{Transport: transport}
resp, err := client.Get(roomData.HLSSource)
req, err := http.NewRequest("GET", roomData.HLSSource, nil)
if err != nil {
return "", "", fmt.Errorf("client get: %w", err)
return "", "", fmt.Errorf("new request: %w", err)
}
if w.CFCookie != "" {
cookie := &http.Cookie{
Name: "cf_clearance",
Value: w.CFCookie,
}
req.AddCookie(cookie)
}
if w.UserAgent != "" {
req.Header.Set("User-Agent", w.UserAgent)
}
resp, err := client.Do(req)
if err != nil {
return "", "", fmt.Errorf("client do: %w", err)
}
if resp.StatusCode != http.StatusOK {
switch resp.StatusCode {
@ -196,6 +228,7 @@ func (w *Channel) resolveSource(body string) (string, string, error) {
return rootURL, sourceURL, nil
}
// mergeSegments is a async function that runs in background for the channel,
// and it merges the segments from buffer to the file.
func (w *Channel) mergeSegments() {
@ -319,9 +352,25 @@ func (w *Channel) requestChunks() ([]*m3u8.MediaSegment, float64, error) {
return nil, 0, fmt.Errorf("channel seems to be paused?")
}
resp, err := client.Get(w.sourceURL)
req, err := http.NewRequest("GET", w.sourceURL, nil)
if err != nil {
return nil, 3, fmt.Errorf("client get: %w", err)
return nil, 0, fmt.Errorf("new request: %w", err)
}
if w.CFCookie != "" {
cookie := &http.Cookie{
Name: "cf_clearance",
Value: w.CFCookie,
}
req.AddCookie(cookie)
}
if w.UserAgent != "" {
req.Header.Set("User-Agent", w.UserAgent)
}
resp, err := client.Do(req)
if err != nil {
return nil, 3, fmt.Errorf("client do: %w", err)
}
if resp.StatusCode != http.StatusOK {
switch resp.StatusCode {
@ -352,6 +401,7 @@ func (w *Channel) requestChunks() ([]*m3u8.MediaSegment, float64, error) {
return chunks, playlist.TargetDuration, nil
}
// requestSegment requests the specific single segment and put it into the buffer.
// the mergeSegments function will merge the segment from buffer to the file in the backgrond.
func (w *Channel) requestSegment(url string, index int) error {
@ -364,9 +414,25 @@ func (w *Channel) requestSegment(url string, index int) error {
return fmt.Errorf("channel seems to be paused?")
}
resp, err := client.Get(w.rootURL + url)
req, err := http.NewRequest("GET", w.rootURL+url, nil)
if err != nil {
return fmt.Errorf("client get: %w", err)
return fmt.Errorf("new request: %w", err)
}
if w.CFCookie != "" {
cookie := &http.Cookie{
Name: "cf_clearance",
Value: w.CFCookie,
}
req.AddCookie(cookie)
}
if w.UserAgent != "" {
req.Header.Set("User-Agent", w.UserAgent)
}
resp, err := client.Do(req)
if err != nil {
return fmt.Errorf("client do: %w", err)
}
if resp.StatusCode != http.StatusOK {
return fmt.Errorf("received status code %d", resp.StatusCode)
@ -387,3 +453,4 @@ func (w *Channel) requestSegment(url string, index int) error {
return nil
}