@@ -7,32 +7,66 @@ import (
77 "os"
88 "path/filepath"
99
10- twitterscraper "github.com/masa-finance/masa-twitter-scraper"
10+ twitterscraper "github.com/imperatrona/twitter-scraper"
11+
12+ "github.com/sirupsen/logrus"
1113)
1214
1315func SaveCookies (scraper * twitterscraper.Scraper , account * TwitterAccount , baseDir string ) error {
16+ logrus .Debugf ("Saving cookies for user %s" , account .Username )
1417 cookieFile := filepath .Join (baseDir , fmt .Sprintf ("%s_twitter_cookies.json" , account .Username ))
1518 cookies := scraper .GetCookies ()
19+ logrus .Debugf ("Got %d cookies to save" , len (cookies ))
20+
1621 data , err := json .Marshal (cookies )
1722 if err != nil {
1823 return fmt .Errorf ("error marshaling cookies: %v" , err )
1924 }
25+
26+ logrus .Debugf ("Writing cookies to file: %s" , cookieFile )
2027 if err = os .WriteFile (cookieFile , data , 0644 ); err != nil {
2128 return fmt .Errorf ("error saving cookies: %v" , err )
2229 }
30+ logrus .Debug ("Successfully saved cookies" )
2331 return nil
2432}
2533
2634func LoadCookies (scraper * twitterscraper.Scraper , account * TwitterAccount , baseDir string ) error {
35+ logrus .Debugf ("Loading cookies for user %s" , account .Username )
2736 cookieFile := filepath .Join (baseDir , fmt .Sprintf ("%s_twitter_cookies.json" , account .Username ))
37+
38+ logrus .Debugf ("Reading cookie file: %s" , cookieFile )
2839 data , err := os .ReadFile (cookieFile )
2940 if err != nil {
3041 return fmt .Errorf ("error reading cookies: %v" , err )
3142 }
43+
3244 var cookies []* http.Cookie
3345 if err = json .Unmarshal (data , & cookies ); err != nil {
3446 return fmt .Errorf ("error unmarshaling cookies: %v" , err )
3547 }
48+ logrus .Debugf ("Loaded %d cookies from file" , len (cookies ))
49+
50+ // Verify critical cookies are present
51+ var hasAuthToken , hasCSRFToken bool
52+ for _ , cookie := range cookies {
53+ if cookie .Name == "auth_token" {
54+ hasAuthToken = true
55+ logrus .Debug ("Found auth_token cookie" )
56+ }
57+ if cookie .Name == "ct0" {
58+ hasCSRFToken = true
59+ logrus .Debug ("Found CSRF token cookie" )
60+ }
61+ }
62+
63+ if ! hasAuthToken || ! hasCSRFToken {
64+ logrus .Debug ("Missing critical authentication cookies" )
65+ return fmt .Errorf ("missing critical authentication cookies" )
66+ }
67+
68+ logrus .Debug ("Setting cookies in scraper" )
3669 scraper .SetCookies (cookies )
70+ logrus .Debug ("Successfully loaded and set cookies" )
3771 return nil
3872}
0 commit comments