Add url option

README.md

@@ -28,6 +28,7 @@ Flags:
       --to string           find tweets sent in reply to a certain user
       --top                 find top tweets
       --until string        find tweets until a certain day (e.g. 2020-09-06)
+      --url string          find tweets containing a certain url (e.g. www.example.com)
       --user-agent string   set custom user-agent
   -v, --version             version for get-old-tweets
 ```

cmd/root.go

@@ -34,6 +34,7 @@ var (
 	lang      string
 	filter    string
 	geocode   string
+	url       string
 	top       bool
 	userAgent string
 )
@@ -52,6 +53,7 @@ var rootCmd = &cobra.Command{
 			Lang:    lang,
 			Filter:  filter,
 			Geocode: geocode,
+			Url:     url,
 		}
 		if q.IsEmpty() {
 			fmt.Fprintln(os.Stderr, "Error: One or more queries are required")
@@ -100,6 +102,7 @@ func init() {
 	rootCmd.Flags().StringVarP(&to, "to", "", "", "find tweets sent in reply to a certain user")
 	rootCmd.Flags().BoolVarP(&top, "top", "", false, "find top tweets")
 	rootCmd.Flags().StringVarP(&until, "until", "", "", "find tweets until a certain day (e.g. 2020-09-06)")
+	rootCmd.Flags().StringVarP(&url, "url", "", "", "find tweets containing a certain url (e.g. www.example.com)")
 	rootCmd.Flags().StringVarP(&userAgent, "user-agent", "", "", "set custom user-agent")
 	rootCmd.MarkFlagRequired("out")
 }

twitter/query.go

@@ -27,6 +27,7 @@ type Query struct {
 	Lang    string
 	Filter  string
 	Geocode string
+	Url     string
 }
 
 func (q *Query) Encode() string {
@@ -64,6 +65,10 @@ func (q *Query) Encode() string {
 		ss = append(ss, "geocode:"+q.Geocode)
 	}
 
+	if len(q.Url) != 0 {
+		ss = append(ss, "url:"+q.Url)
+	}
+
 	return strings.Join(ss[:], " ")
 }
 

twitter/query_test.go

@@ -31,6 +31,7 @@ func TestEncode(t *testing.T) {
 		lang     string
 		filter   string
 		geocode  string
+		url      string
 		expected string
 	}{
 		"none": {
@@ -42,6 +43,7 @@ func TestEncode(t *testing.T) {
 			lang:     "",
 			filter:   "",
 			geocode:  "",
+			url:      "",
 			expected: "",
 		},
 		"all": {
@@ -53,7 +55,8 @@ func TestEncode(t *testing.T) {
 			lang:     "ja",
 			filter:   "verified",
 			geocode:  "35.6851508,139.7526768,0.1km",
-			expected: "foo bar since:2020-09-06 until:2020-09-07 from:foo to:bar lang:ja filter:verified geocode:35.6851508,139.7526768,0.1km",
+			url:      "www.example.com",
+			expected: "foo bar since:2020-09-06 until:2020-09-07 from:foo to:bar lang:ja filter:verified geocode:35.6851508,139.7526768,0.1km url:www.example.com",
 		},
 	}
 
@@ -68,6 +71,7 @@ func TestEncode(t *testing.T) {
 				Lang:    e.lang,
 				Filter:  e.filter,
 				Geocode: e.geocode,
+				Url:     e.url,
 			}
 
 			actual := q.Encode()
@@ -90,6 +94,7 @@ func TestIsEmpty(t *testing.T) {
 		lang     string
 		filter   string
 		geocode  string
+		url      string
 		expected bool
 	}{
 		"none": {
@@ -101,6 +106,7 @@ func TestIsEmpty(t *testing.T) {
 			lang:     "",
 			filter:   "",
 			geocode:  "",
+			url:      "",
 			expected: true,
 		},
 		"all": {
@@ -112,6 +118,7 @@ func TestIsEmpty(t *testing.T) {
 			lang:     "ja",
 			filter:   "verified",
 			geocode:  "35.6851508,139.7526768,0.1km",
+			url:      "www.example.com",
 			expected: false,
 		},
 	}
@@ -127,6 +134,7 @@ func TestIsEmpty(t *testing.T) {
 				Lang:    e.lang,
 				Filter:  e.filter,
 				Geocode: e.geocode,
+				Url:     e.url,
 			}
 
 			actual := q.IsEmpty()