mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
14e7619075
Have all instances take a function Update the remaining modules to take this into account Numerous smaller refactors to make some widgets work more or less the same
144 lines
3.4 KiB
Go
144 lines
3.4 KiB
Go
package twitter
|
|
|
|
import (
|
|
"fmt"
|
|
"html"
|
|
"regexp"
|
|
|
|
"github.com/dustin/go-humanize"
|
|
"github.com/rivo/tview"
|
|
"github.com/wtfutil/wtf/utils"
|
|
"github.com/wtfutil/wtf/view"
|
|
)
|
|
|
|
type Widget struct {
|
|
view.KeyboardWidget
|
|
view.MultiSourceWidget
|
|
view.TextWidget
|
|
|
|
client *Client
|
|
idx int
|
|
settings *Settings
|
|
sources []string
|
|
}
|
|
|
|
func NewWidget(app *tview.Application, pages *tview.Pages, settings *Settings) *Widget {
|
|
widget := Widget{
|
|
KeyboardWidget: view.NewKeyboardWidget(app, pages, settings.common),
|
|
MultiSourceWidget: view.NewMultiSourceWidget(settings.common, "screenName", "screenNames"),
|
|
TextWidget: view.NewTextWidget(app, settings.common, true),
|
|
|
|
idx: 0,
|
|
settings: settings,
|
|
}
|
|
|
|
widget.initializeKeyboardControls()
|
|
widget.View.SetInputCapture(widget.InputCapture)
|
|
|
|
widget.SetDisplayFunction(widget.display)
|
|
|
|
widget.client = NewClient(settings)
|
|
|
|
widget.View.SetBorderPadding(1, 1, 1, 1)
|
|
widget.View.SetWrap(true)
|
|
widget.View.SetWordWrap(true)
|
|
|
|
widget.KeyboardWidget.SetView(widget.View)
|
|
|
|
return &widget
|
|
}
|
|
|
|
/* -------------------- Exported Functions -------------------- */
|
|
|
|
// Refresh is called on the interval and refreshes the data
|
|
func (widget *Widget) Refresh() {
|
|
widget.display()
|
|
}
|
|
|
|
func (widget *Widget) HelpText() string {
|
|
return widget.KeyboardWidget.HelpText()
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) display() {
|
|
widget.Redraw(widget.content)
|
|
}
|
|
|
|
func (widget *Widget) content() (string, string, bool) {
|
|
widget.client.screenName = widget.CurrentSource()
|
|
tweets := widget.client.Tweets()
|
|
|
|
title := fmt.Sprintf("Twitter - [green]@%s[white]", widget.CurrentSource())
|
|
|
|
if len(tweets) == 0 {
|
|
str := fmt.Sprintf("\n\n\n%s", utils.CenterText("[lightblue]No Tweets[white]", 50))
|
|
return title, str, true
|
|
}
|
|
|
|
_, _, width, _ := widget.View.GetRect()
|
|
str := widget.settings.common.SigilStr(len(widget.Sources), widget.Idx, width-2) + "\n"
|
|
for _, tweet := range tweets {
|
|
str += widget.format(tweet)
|
|
}
|
|
|
|
return title, str, true
|
|
}
|
|
|
|
// If the tweet's Username is the same as the account we're watching, no
|
|
// need to display the username
|
|
func (widget *Widget) displayName(tweet Tweet) string {
|
|
if widget.CurrentSource() == tweet.User.ScreenName {
|
|
return ""
|
|
}
|
|
return tweet.User.ScreenName
|
|
}
|
|
|
|
func (widget *Widget) formatText(text string) string {
|
|
result := text
|
|
|
|
// Convert HTML entities
|
|
result = html.UnescapeString(result)
|
|
|
|
// RT indicator
|
|
rtRegExp := regexp.MustCompile(`^RT`)
|
|
result = rtRegExp.ReplaceAllString(result, "[olive]${0}[white::-]")
|
|
|
|
// @name mentions
|
|
atRegExp := regexp.MustCompile(`@[0-9A-Za-z_]*`)
|
|
result = atRegExp.ReplaceAllString(result, "[lightblue]${0}[white]")
|
|
|
|
// HTTP(S) links
|
|
linkRegExp := regexp.MustCompile(`http[s:\/.0-9A-Za-z]*`)
|
|
result = linkRegExp.ReplaceAllString(result, "[lightblue::u]${0}[white::-]")
|
|
|
|
// Hash tags
|
|
hashRegExp := regexp.MustCompile(`#[0-9A-Za-z_]*`)
|
|
result = hashRegExp.ReplaceAllString(result, "[yellow]${0}[white]")
|
|
|
|
return result
|
|
}
|
|
|
|
func (widget *Widget) format(tweet Tweet) string {
|
|
body := widget.formatText(tweet.Text)
|
|
name := widget.displayName(tweet)
|
|
|
|
var attribution string
|
|
if name == "" {
|
|
attribution = humanize.Time(tweet.Created())
|
|
} else {
|
|
attribution = fmt.Sprintf(
|
|
"%s, %s",
|
|
name,
|
|
humanize.Time(tweet.Created()),
|
|
)
|
|
}
|
|
|
|
return fmt.Sprintf("%s\n[grey]%s[white]\n\n", body, attribution)
|
|
}
|
|
func (widget *Widget) currentSourceURI() string {
|
|
|
|
src := "https://twitter.com/" + widget.CurrentSource()
|
|
return src
|
|
}
|