Improve logout
drop session in Twitter
This commit is contained in:
parent
e86c2fa4ce
commit
90f9a71721
3 changed files with 28 additions and 14 deletions
19
api.go
19
api.go
|
|
@ -3,7 +3,7 @@ package twitterscraper
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io/ioutil"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
@ -48,9 +48,13 @@ func (s *Scraper) RequestAPI(req *http.Request, target interface{}) error {
|
||||||
}
|
}
|
||||||
defer resp.Body.Close()
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
content, err := io.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
// private profiles return forbidden, but also data
|
// private profiles return forbidden, but also data
|
||||||
if resp.StatusCode != http.StatusOK && resp.StatusCode != http.StatusForbidden {
|
if resp.StatusCode != http.StatusOK && resp.StatusCode != http.StatusForbidden {
|
||||||
content, _ := ioutil.ReadAll(resp.Body)
|
|
||||||
return fmt.Errorf("response status %s: %s", resp.Status, content)
|
return fmt.Errorf("response status %s: %s", resp.Status, content)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -58,13 +62,10 @@ func (s *Scraper) RequestAPI(req *http.Request, target interface{}) error {
|
||||||
s.guestToken = ""
|
s.guestToken = ""
|
||||||
}
|
}
|
||||||
|
|
||||||
b, err := ioutil.ReadAll(resp.Body)
|
if target == nil {
|
||||||
if err != nil {
|
return nil
|
||||||
return err
|
|
||||||
}
|
}
|
||||||
// fmt.Println(string(b))
|
return json.Unmarshal(content, target)
|
||||||
|
|
||||||
return json.Unmarshal(b, target)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetGuestToken from Twitter API
|
// GetGuestToken from Twitter API
|
||||||
|
|
@ -81,7 +82,7 @@ func (s *Scraper) GetGuestToken() error {
|
||||||
}
|
}
|
||||||
defer resp.Body.Close()
|
defer resp.Body.Close()
|
||||||
|
|
||||||
body, err := ioutil.ReadAll(resp.Body)
|
body, err := io.ReadAll(resp.Body)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
|
||||||
13
auth.go
13
auth.go
|
|
@ -11,6 +11,7 @@ import (
|
||||||
|
|
||||||
const (
|
const (
|
||||||
loginURL = "https://api.twitter.com/1.1/onboarding/task.json"
|
loginURL = "https://api.twitter.com/1.1/onboarding/task.json"
|
||||||
|
logoutURL = "https://api.twitter.com/1.1/account/logout.json"
|
||||||
bearerToken2 = "AAAAAAAAAAAAAAAAAAAAANRILgAAAAAAnNwIzUejRCOuH5E6I8xnZz4puTs%3D1Zv7ttfk8LF81IUq16cHjhLTvJu4FA33AGWWjCpTnA"
|
bearerToken2 = "AAAAAAAAAAAAAAAAAAAAANRILgAAAAAAnNwIzUejRCOuH5E6I8xnZz4puTs%3D1Zv7ttfk8LF81IUq16cHjhLTvJu4FA33AGWWjCpTnA"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -230,11 +231,21 @@ func (s *Scraper) Login(credentials ...string) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Logout is reset session
|
// Logout is reset session
|
||||||
func (s *Scraper) Logout() {
|
func (s *Scraper) Logout() error {
|
||||||
|
req, err := http.NewRequest("POST", logoutURL, nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
err = s.RequestAPI(req, nil)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
s.isLogged = false
|
s.isLogged = false
|
||||||
s.guestToken = ""
|
s.guestToken = ""
|
||||||
s.client.Jar, _ = cookiejar.New(nil)
|
s.client.Jar, _ = cookiejar.New(nil)
|
||||||
s.setBearerToken(bearerToken)
|
s.setBearerToken(bearerToken)
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Scraper) GetCookies() []*http.Cookie {
|
func (s *Scraper) GetCookies() []*http.Cookie {
|
||||||
|
|
|
||||||
10
auth_test.go
10
auth_test.go
|
|
@ -22,13 +22,15 @@ func TestAuth(t *testing.T) {
|
||||||
t.Fatalf("Expected IsLoggedIn() = true")
|
t.Fatalf("Expected IsLoggedIn() = true")
|
||||||
}
|
}
|
||||||
cookies := scraper.GetCookies()
|
cookies := scraper.GetCookies()
|
||||||
scraper.Logout()
|
|
||||||
if scraper.IsLoggedIn() {
|
|
||||||
t.Error("Expected IsLoggedIn() = false")
|
|
||||||
}
|
|
||||||
scraper2 := twitterscraper.New()
|
scraper2 := twitterscraper.New()
|
||||||
scraper2.SetCookies(cookies)
|
scraper2.SetCookies(cookies)
|
||||||
if !scraper2.IsLoggedIn() {
|
if !scraper2.IsLoggedIn() {
|
||||||
t.Error("Expected restored IsLoggedIn() = true")
|
t.Error("Expected restored IsLoggedIn() = true")
|
||||||
}
|
}
|
||||||
|
if err := scraper.Logout(); err != nil {
|
||||||
|
t.Errorf("Logout() error = %v", err)
|
||||||
|
}
|
||||||
|
if scraper.IsLoggedIn() {
|
||||||
|
t.Error("Expected IsLoggedIn() = false")
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue