From 7363f676bf24fe87d15f7de3cd66a65fa0ef2dca Mon Sep 17 00:00:00 2001
From: JustHumanz <Aldinsetiawan99@gmail.com>
Date: Sun, 20 Dec 2020 00:20:27 +0700
Subject: [PATCH] add SearchLive SearchPhotos SearchVideos

---
 README.md  |  5 +++++
 scraper.go | 40 ++++++++++++++++++++++++++++++++++++++++
 search.go  |  6 ++++++
 3 files changed, 51 insertions(+)

diff --git a/README.md b/README.md
index 9cbf393..89ff0e5 100644
--- a/README.md
+++ b/README.md
@@ -65,6 +65,11 @@ func main() {
 
 The search ends if we have 50 tweets.
 
+### Search tweet in realtime
+```golang
+scraper.SearchLive(true)
+```
+
 See [Rules and filtering](https://developer.twitter.com/en/docs/tweets/rules-and-filtering/overview/standard-operators) for build standard queries.
 
 ### Get profile
diff --git a/scraper.go b/scraper.go
index 28c9fe8..d5b194d 100644
--- a/scraper.go
+++ b/scraper.go
@@ -15,6 +15,7 @@ type Scraper struct {
 	client         *http.Client
 	guestToken     string
 	includeReplies bool
+	searchMode     string
 }
 
 var defaultScraper *Scraper
@@ -26,6 +27,45 @@ func New() *Scraper {
 	}
 }
 
+// SetSearchLive enable/disable realtime search
+func (s *Scraper) SearchLive(srctype bool) *Scraper {
+	if srctype {
+		s.searchMode = "live"
+	}
+	return s
+}
+
+// SetSearchLive wrapper for default SetSearchLive
+func SearchLive(srctype bool) *Scraper {
+	return defaultScraper.SearchLive(srctype)
+}
+
+// SetSearchPhotos filter search for photos only
+func (s *Scraper) SearchPhotos(srctype bool) *Scraper {
+	if srctype {
+		s.searchMode = "image"
+	}
+	return s
+}
+
+// SetSearchPhotos wrapper for default SetSearchPhotos
+func SearchPhotos(srctype bool) *Scraper {
+	return defaultScraper.SearchPhotos(srctype)
+}
+
+// SetSearchVideos filter search for videos only
+func (s *Scraper) SearchVideos(srctype bool) *Scraper {
+	if srctype {
+		s.searchMode = "video"
+	}
+	return s
+}
+
+// SetSearchVideos wrapper for default SetSearchVideos
+func SearchVideos(srctype bool) *Scraper {
+	return defaultScraper.SearchVideos(srctype)
+}
+
 // WithReplies enable/disable load timeline with tweet replies
 func (s *Scraper) WithReplies(b bool) *Scraper {
 	s.includeReplies = b
diff --git a/search.go b/search.go
index 5a635c1..7314a6e 100644
--- a/search.go
+++ b/search.go
@@ -37,6 +37,12 @@ func (s *Scraper) FetchSearchTweets(query string, maxTweetsNbr int, cursor strin
 	if cursor != "" {
 		q.Add("cursor", cursor)
 	}
+	if s.searchMode == "live" {
+		q.Add("tweet_search_mode", s.searchMode)
+	} else {
+		q.Add("result_filter", s.searchMode)
+	}
+
 	req.URL.RawQuery = q.Encode()
 
 	var timeline timeline