Fix(avbase): skip image tls verification (#49)

metatube-community · Jun 12, 2024 · aa5ea61 · aa5ea61
1 parent 99c62d3
commit aa5ea61
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/common/fetch/fetch.go b/common/fetch/fetch.go
@@ -1,6 +1,7 @@
 package fetch
 
 import (
+	"crypto/tls"
 	"io"
 	"net/http"
 	"net/http/cookiejar"
@@ -36,6 +37,10 @@ type Config struct {
 
 	// Custom HTTP Transport.
 	Transport http.RoundTripper
+
+	// Skip TLS verification. Applies only
+	// to *http.Transport based transport.
+	SkipVerify bool
 }
 
 type Fetcher struct {
@@ -82,6 +87,15 @@ func Default(cfg *Config) *Fetcher {
 	if cfg.Transport != nil {
 		c.HTTPClient.Transport = cfg.Transport
 	}
+	if cfg.SkipVerify {
+		if transport, ok := c.HTTPClient.Transport.(*http.Transport); ok {
+			if transport.TLSClientConfig == nil {
+				// init TLS config if is nil.
+				transport.TLSClientConfig = &tls.Config{}
+			}
+			transport.TLSClientConfig.InsecureSkipVerify = true
+		}
+	}
 	return New(c.StandardClient(), cfg)
 }
 

diff --git a/provider/avbase/avbase.go b/provider/avbase/avbase.go
@@ -12,6 +12,7 @@ import (
 
 	"github.com/gocolly/colly/v2"
 
+	"github.com/metatube-community/metatube-sdk-go/common/fetch"
 	"github.com/metatube-community/metatube-sdk-go/common/number"
 	"github.com/metatube-community/metatube-sdk-go/common/parser"
 	"github.com/metatube-community/metatube-sdk-go/common/singledo"
@@ -43,13 +44,15 @@ const (
 )
 
 type AVBase struct {
+	*fetch.Fetcher
 	*scraper.Scraper
 	single    *singledo.Single
 	providers map[string]provider.MovieProvider
 }
 
 func New() *AVBase {
 	return &AVBase{
+		Fetcher: fetch.Default(&fetch.Config{SkipVerify: true}),
 		Scraper: scraper.NewDefaultScraper(Name, baseURL, Priority,
 			scraper.WithHeaders(map[string]string{
 				"Referer": baseURL,