add http proxy support

This commit is contained in:
JustHumanz 2020-12-03 21:42:16 +07:00
parent 111942f1b9
commit 341e1de6d0
4 changed files with 64 additions and 3 deletions

View file

@ -2,6 +2,7 @@ package twitterscraper
import ( import (
"fmt" "fmt"
"net"
"net/http" "net/http"
"strconv" "strconv"
"strings" "strings"
@ -35,13 +36,25 @@ type Profile struct {
func GetProfile(username string) (Profile, error) { func GetProfile(username string) (Profile, error) {
url := "https://mobile.twitter.com/" + username url := "https://mobile.twitter.com/" + username
client := http.DefaultClient
if HTTPProxy != nil {
client = &http.Client{
Transport: &http.Transport{
Proxy: http.ProxyURL(HTTPProxy),
DialContext: (&net.Dialer{
Timeout: 10 * time.Second,
}).DialContext,
},
}
}
req, err := http.NewRequest("GET", url, nil) req, err := http.NewRequest("GET", url, nil)
if err != nil { if err != nil {
return Profile{}, err return Profile{}, err
} }
req.Header.Set("Accept-Language", "en-US") req.Header.Set("Accept-Language", "en-US")
resp, err := http.DefaultClient.Do(req) resp, err := client.Do(req)
if resp == nil { if resp == nil {
return Profile{}, err return Profile{}, err
} }

View file

@ -4,9 +4,11 @@ import (
"context" "context"
"fmt" "fmt"
"io" "io"
"net"
"net/http" "net/http"
"net/url" "net/url"
"strings" "strings"
"time"
"github.com/PuerkitoBio/goquery" "github.com/PuerkitoBio/goquery"
) )
@ -64,6 +66,18 @@ func FetchSearchTweets(query, nextCursor string) ([]*Tweet, string, error) {
url = "https://mobile.twitter.com" + nextCursor url = "https://mobile.twitter.com" + nextCursor
} }
client := http.DefaultClient
if HTTPProxy != nil {
client = &http.Client{
Transport: &http.Transport{
Proxy: http.ProxyURL(HTTPProxy),
DialContext: (&net.Dialer{
Timeout: 10 * time.Second,
}).DialContext,
},
}
}
req, err := http.NewRequest("GET", url, nil) req, err := http.NewRequest("GET", url, nil)
if err != nil { if err != nil {
return nil, "", err return nil, "", err
@ -72,7 +86,7 @@ func FetchSearchTweets(query, nextCursor string) ([]*Tweet, string, error) {
req.Header.Set("Referer", "https://mobile.twitter.com/") req.Header.Set("Referer", "https://mobile.twitter.com/")
req.Header.Set("User-Agent", "Opera/9.80 (J2ME/MIDP; Opera Mini/5.1.21214/28.2725; U; ru) Presto/2.8.119 Version/11.10") req.Header.Set("User-Agent", "Opera/9.80 (J2ME/MIDP; Opera Mini/5.1.21214/28.2725; U; ru) Presto/2.8.119 Version/11.10")
resp, err := http.DefaultClient.Do(req) resp, err := client.Do(req)
if resp == nil { if resp == nil {
return nil, "", err return nil, "", err
} }

View file

@ -2,8 +2,10 @@ package twitterscraper
import ( import (
"fmt" "fmt"
"net"
"net/http" "net/http"
"strings" "strings"
"time"
"github.com/PuerkitoBio/goquery" "github.com/PuerkitoBio/goquery"
) )
@ -12,13 +14,25 @@ const trendsURL = "https://mobile.twitter.com/trends"
// GetTrends return list of trends. // GetTrends return list of trends.
func GetTrends() ([]string, error) { func GetTrends() ([]string, error) {
client := http.DefaultClient
if HTTPProxy != nil {
client = &http.Client{
Transport: &http.Transport{
Proxy: http.ProxyURL(HTTPProxy),
DialContext: (&net.Dialer{
Timeout: 10 * time.Second,
}).DialContext,
},
}
}
req, err := http.NewRequest("GET", trendsURL, nil) req, err := http.NewRequest("GET", trendsURL, nil)
if err != nil { if err != nil {
return nil, err return nil, err
} }
req.Header.Set("Accept-Language", "en-US") req.Header.Set("Accept-Language", "en-US")
resp, err := http.DefaultClient.Do(req) resp, err := client.Do(req)
if resp == nil { if resp == nil {
return nil, err return nil, err
} }

20
util.go
View file

@ -2,11 +2,31 @@ package twitterscraper
import ( import (
"encoding/json" "encoding/json"
"errors"
"fmt" "fmt"
"net/http" "net/http"
"net/url"
"regexp"
"strings" "strings"
) )
//HttpProxy Public variable for Http proxy
var HTTPProxy *url.URL
//SetProxy set http proxy
func SetProxy(Proxy string) error {
match, _ := regexp.MatchString("http.+", Proxy)
if !match {
return errors.New("only support http protocol")
}
urlproxy, err := url.Parse(Proxy)
if err != nil {
return err
}
HTTPProxy = urlproxy
return nil
}
func newRequest(url string) (*http.Request, error) { func newRequest(url string) (*http.Request, error) {
req, err := http.NewRequest("GET", url, nil) req, err := http.NewRequest("GET", url, nil)
if err != nil { if err != nil {