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
115 lines
2.6 KiB
Go
115 lines
2.6 KiB
Go
package hackernews
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/rivo/tview"
|
|
"github.com/wtfutil/wtf/utils"
|
|
"github.com/wtfutil/wtf/view"
|
|
)
|
|
|
|
type Widget struct {
|
|
view.KeyboardWidget
|
|
view.ScrollableWidget
|
|
|
|
stories []Story
|
|
settings *Settings
|
|
}
|
|
|
|
func NewWidget(app *tview.Application, pages *tview.Pages, settings *Settings) *Widget {
|
|
widget := &Widget{
|
|
KeyboardWidget: view.NewKeyboardWidget(app, pages, settings.common),
|
|
ScrollableWidget: view.NewScrollableWidget(app, settings.common, true),
|
|
|
|
settings: settings,
|
|
}
|
|
|
|
widget.SetRenderFunction(widget.Render)
|
|
widget.initializeKeyboardControls()
|
|
widget.View.SetInputCapture(widget.InputCapture)
|
|
|
|
widget.KeyboardWidget.SetView(widget.View)
|
|
|
|
return widget
|
|
}
|
|
|
|
/* -------------------- Exported Functions -------------------- */
|
|
|
|
func (widget *Widget) Refresh() {
|
|
if widget.Disabled() {
|
|
return
|
|
}
|
|
|
|
storyIds, err := GetStories(widget.settings.storyType)
|
|
if storyIds == nil {
|
|
return
|
|
}
|
|
|
|
if err != nil {
|
|
widget.Redraw(func() (string, string, bool) { return widget.CommonSettings().Title, err.Error(), true })
|
|
return
|
|
}
|
|
var stories []Story
|
|
for idx := 0; idx < widget.settings.numberOfStories; idx++ {
|
|
story, e := GetStory(storyIds[idx])
|
|
if e == nil {
|
|
stories = append(stories, story)
|
|
}
|
|
}
|
|
|
|
widget.stories = stories
|
|
widget.SetItemCount(len(stories))
|
|
|
|
widget.Render()
|
|
}
|
|
|
|
// Render sets up the widget data for redrawing to the screen
|
|
func (widget *Widget) Render() {
|
|
widget.Redraw(widget.content)
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) content() (string, string, bool) {
|
|
title := fmt.Sprintf("%s - %s stories", widget.CommonSettings().Title, widget.settings.storyType)
|
|
stories := widget.stories
|
|
if stories == nil || len(stories) == 0 {
|
|
return title, "No stories to display", false
|
|
}
|
|
var str string
|
|
|
|
for idx, story := range stories {
|
|
u, _ := url.Parse(story.URL)
|
|
|
|
row := fmt.Sprintf(
|
|
`[%s]%2d. %s [lightblue](%s)[white]`,
|
|
widget.RowColor(idx),
|
|
idx+1,
|
|
story.Title,
|
|
strings.TrimPrefix(u.Host, "www."),
|
|
)
|
|
|
|
str += utils.HighlightableHelper(widget.View, row, idx, len(story.Title))
|
|
}
|
|
|
|
return title, str, false
|
|
}
|
|
|
|
func (widget *Widget) openStory() {
|
|
sel := widget.GetSelected()
|
|
if sel >= 0 && widget.stories != nil && sel < len(widget.stories) {
|
|
story := &widget.stories[sel]
|
|
utils.OpenFile(story.URL)
|
|
}
|
|
}
|
|
|
|
func (widget *Widget) openComments() {
|
|
sel := widget.GetSelected()
|
|
if sel >= 0 && widget.stories != nil && sel < len(widget.stories) {
|
|
story := &widget.stories[sel]
|
|
utils.OpenFile(fmt.Sprintf("https://news.ycombinator.com/item?id=%d", story.ID))
|
|
}
|
|
}
|