109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package scraper
|
|
|
|
import (
|
|
"time"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"offline_twitter/terminal_utils"
|
|
)
|
|
|
|
type UserID string
|
|
type UserHandle string
|
|
|
|
func JoinArrayOfHandles(handles []UserHandle) string {
|
|
ret := []string{}
|
|
for _, h := range handles {
|
|
ret = append(ret, string(h))
|
|
}
|
|
return strings.Join(ret, ",")
|
|
}
|
|
|
|
type User struct {
|
|
ID UserID
|
|
DisplayName string
|
|
Handle UserHandle
|
|
Bio string
|
|
FollowingCount int
|
|
FollowersCount int
|
|
Location string
|
|
Website string
|
|
JoinDate time.Time
|
|
IsPrivate bool
|
|
IsVerified bool
|
|
ProfileImageUrl string
|
|
BannerImageUrl string
|
|
PinnedTweetID TweetID
|
|
PinnedTweet *Tweet
|
|
}
|
|
|
|
func (u User) String() string {
|
|
var verified string
|
|
if u.IsVerified {
|
|
verified = "[\u2713]"
|
|
}
|
|
ret := fmt.Sprintf(
|
|
`%s%s
|
|
@%s
|
|
%s
|
|
|
|
Following: %d Followers: %d
|
|
|
|
Joined %s
|
|
%s
|
|
%s
|
|
`,
|
|
u.DisplayName,
|
|
verified,
|
|
u.Handle,
|
|
terminal_utils.WrapText(u.Bio, 60),
|
|
u.FollowingCount,
|
|
u.FollowersCount,
|
|
terminal_utils.FormatDate(u.JoinDate),
|
|
u.Location,
|
|
u.Website,
|
|
)
|
|
if u.PinnedTweet != nil {
|
|
ret += "\n" + terminal_utils.WrapText(u.PinnedTweet.Text, 60)
|
|
} else {
|
|
println("Pinned tweet id:", u.PinnedTweetID)
|
|
}
|
|
return ret
|
|
}
|
|
|
|
// Turn an APIUser, as returned from the scraper, into a properly structured User object
|
|
func ParseSingleUser(apiUser APIUser) (ret User, err error) {
|
|
ret.ID = UserID(apiUser.IDStr)
|
|
ret.DisplayName = apiUser.Name
|
|
ret.Handle = UserHandle(apiUser.ScreenName)
|
|
ret.Bio = apiUser.Description
|
|
ret.FollowingCount = apiUser.FriendsCount
|
|
ret.FollowersCount = apiUser.FollowersCount
|
|
ret.Location = apiUser.Location
|
|
if len(apiUser.Entities.URL.Urls) > 0 {
|
|
ret.Website = apiUser.Entities.URL.Urls[0].ExpandedURL
|
|
}
|
|
ret.JoinDate, err = time.Parse(time.RubyDate, apiUser.CreatedAt)
|
|
if err != nil {
|
|
return
|
|
}
|
|
ret.IsPrivate = apiUser.Protected
|
|
ret.IsVerified = apiUser.Verified
|
|
ret.ProfileImageUrl = apiUser.ProfileImageURLHTTPS
|
|
ret.BannerImageUrl = apiUser.ProfileBannerURL
|
|
if len(apiUser.PinnedTweetIdsStr) > 0 {
|
|
ret.PinnedTweetID = TweetID(apiUser.PinnedTweetIdsStr[0])
|
|
}
|
|
return
|
|
}
|
|
|
|
// Calls API#GetUser and returns the parsed result
|
|
func GetUser(handle UserHandle) (User, error) {
|
|
api := API{}
|
|
apiUser, err := api.GetUser(handle)
|
|
if err != nil {
|
|
return User{}, err
|
|
}
|
|
return ParseSingleUser(apiUser)
|
|
}
|