mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
115 lines
2.5 KiB
Go
115 lines
2.5 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(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() {
|
|
if widget.stories == nil {
|
|
return
|
|
}
|
|
|
|
title := fmt.Sprintf("%s - %s stories", widget.CommonSettings().Title, widget.settings.storyType)
|
|
widget.Redraw(title, widget.contentFrom(widget.stories), false)
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) contentFrom(stories []Story) string {
|
|
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 str
|
|
}
|
|
|
|
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))
|
|
}
|
|
}
|