wtf/modules/twitterstats/widget.go

package twitterstats

import (
	"fmt"

	"github.com/rivo/tview"
	"github.com/wtfutil/wtf/view"
)

type Widget struct {
	view.TextWidget

	client   *Client
	settings *Settings
}

func NewWidget(app *tview.Application, pages *tview.Pages, settings *Settings) *Widget {
	widget := Widget{
		TextWidget: view.NewTextWidget(app, settings.common),

		client:   NewClient(settings),
		settings: settings,
	}

	widget.View.SetBorderPadding(1, 1, 1, 1)
	widget.View.SetWrap(true)
	widget.View.SetWordWrap(true)

	return &widget
}

func (widget *Widget) Refresh() {
	widget.Redraw(widget.content)
}

func (widget *Widget) content() (string, string, bool) {
	// Add header row
	str := fmt.Sprintf(
		"[%s]%-12s %10s %8s[white]\n",
		widget.settings.common.Colors.Subheading,
		"Username",
		"Followers",
		"Tweets",
	)

	stats := widget.client.GetStats()

	// Add rows for each of the followed usernames
	for i, username := range widget.client.screenNames {
		str += fmt.Sprintf(
			"%-12s %10d %8d\n",
			username,
			stats[i].FollowerCount,
			stats[i].TweetCount,
		)
	}

	return "Twitter Stats", str, true
}