Merge pull request #91 from windowsdeveloperwannabe/mentions

Add @user mentions to Tweet

timeline.go

@@ -27,6 +27,11 @@ type timeline struct {
 					ExpandedURL string `json:"expanded_url"`
 					URL         string `json:"url"`
 				} `json:"urls"`
+				UserMentions []struct {
+					IDStr      string `json:"id_str"`
+					Name       string `json:"name"`
+					ScreenName string `json:"screen_name"`
+				} `json:"user_mentions"`
 			} `json:"entities"`
 			ExtendedEntities struct {
 				Media []struct {
@@ -200,6 +205,14 @@ func (timeline *timeline) parseTweet(id string) *Tweet {
 			tw.Hashtags = append(tw.Hashtags, hash.Text)
 		}
 
+		for _, mention := range tweet.Entities.UserMentions {
+			tw.Mentions = append(tw.Mentions, Mention{
+				ID:       mention.IDStr,
+				Username: mention.ScreenName,
+				Name:     mention.Name,
+			})
+		}
+
 		for _, media := range tweet.ExtendedEntities.Media {
 			if media.Type == "photo" {
 				photo := Photo{

tweets_test.go

@@ -99,6 +99,23 @@ func TestGetTweet(t *testing.T) {
 	}
 }
 
+func TestTweetMentions(t *testing.T) {
+	sample := []twitterscraper.Mention{{
+		ID:       "7018222",
+		Username: "davidmcraney",
+		Name:     "David McRaney",
+	}}
+	scraper := twitterscraper.New()
+	tweet, err := scraper.GetTweet("1554522888904101890")
+	if err != nil {
+		t.Error(err)
+	} else {
+		if diff := cmp.Diff(sample, tweet.Mentions, cmpOptions...); diff != "" {
+			t.Error("Resulting tweet does not match the sample", diff)
+		}
+	}
+}
+
 func TestQuotedAndReply(t *testing.T) {
 	sample := &twitterscraper.Tweet{
 		HTML:         "The Easiest Problem Everyone Gets Wrong <br><br>[new video] --&gt; <a href=\"https://youtu.be/ytfCdqWhmdg\">https://t.co/YdaeDYmPAU</a> <br><a href=\"https://t.co/iKu4Xs6o2V\"><img src=\"https://pbs.twimg.com/media/ESsZa9AXgAIAYnF.jpg\"/></a>",

types.go

@@ -3,6 +3,13 @@ package twitterscraper
 import "time"
 
 type (
+	// Mention type.
+	Mention struct {
+		ID       string
+		Username string
+		Name     string
+	}
+
 	// Photo type.
 	Photo struct {
 		ID  string
@@ -27,6 +34,7 @@ type (
 		IsReply          bool
 		IsRetweet        bool
 		Likes            int
+		Mentions         []Mention
 		PermanentURL     string
 		Photos           []Photo
 		Place            *Place