tut/ui/linkview.go

package ui

import (
	"fmt"

	"github.com/RasmusLindroth/tut/config"
	"github.com/rivo/tview"
)

type LinkView struct {
	tutView     *TutView
	shared      *Shared
	View        *tview.Flex
	list        *tview.List
	controls    *tview.Flex
	scrollSleep *scrollSleep
}

func NewLinkView(tv *TutView) *LinkView {
	l := NewList(tv.tut.Config)
	c := NewControlView(tv.tut.Config)
	lv := &LinkView{
		tutView:  tv,
		shared:   tv.Shared,
		list:     l,
		controls: c,
	}
	lv.scrollSleep = NewScrollSleep(lv.Next, lv.Prev)
	lv.View = linkViewUI(lv)
	return lv
}

func linkViewUI(lv *LinkView) *tview.Flex {
	lv.controls.SetBorderPadding(0, 0, 1, 1)
	items := []Control{
		NewControl(lv.tutView.tut.Config, lv.tutView.tut.Config.Input.LinkOpen, true),
		NewControl(lv.tutView.tut.Config, lv.tutView.tut.Config.Input.LinkYank, true),
	}
	for _, cust := range lv.tutView.tut.Config.OpenCustom.OpenCustoms {
		items = append(items, NewControl(lv.tutView.tut.Config, cust.Key, true))
	}
	lv.controls.Clear()
	for i, item := range items {
		if i < len(items)-1 {
			lv.controls.AddItem(NewControlButton(lv.tutView, item), item.Len+1, 0, false)
		} else {
			lv.controls.AddItem(NewControlButton(lv.tutView, item), item.Len, 0, false)
		}
	}

	r := tview.NewFlex().SetDirection(tview.FlexRow)
	if lv.tutView.tut.Config.General.TerminalTitle < 2 {
		r.AddItem(lv.shared.Top.View, 1, 0, false)
	}
	r.AddItem(lv.list, 0, 1, false).
		AddItem(lv.controls, 1, 0, false).
		AddItem(lv.shared.Bottom.View, 2, 0, false)
	return r
}

func (lv *LinkView) SetLinks() {
	item, err := lv.tutView.GetCurrentItem()
	if err != nil {
		lv.list.Clear()
		return
	}
	lv.list.Clear()
	urls, mentions, tags, _ := item.URLs()

	for _, url := range urls {
		lv.list.AddItem(url.Text, "", 0, nil)
	}
	for _, mention := range mentions {
		lv.list.AddItem(mention.Acct, "", 0, nil)
	}
	for _, tag := range tags {
		lv.list.AddItem("#"+tag.Name, "", 0, nil)
	}
}

func (lv *LinkView) Next() {
	listNext(lv.list)
}

func (lv *LinkView) Prev() {
	listPrev(lv.list)
}

func (lv *LinkView) Open() {
	item, err := lv.tutView.GetCurrentItem()
	if err != nil {
		return
	}
	urls, mentions, tags, total := item.URLs()
	index := lv.list.GetCurrentItem()

	if total == 0 || index >= total {
		return
	}
	if index < len(urls) {
		openURL(lv.tutView, urls[index].URL)
		return
	}
	mIndex := index - len(urls)
	if mIndex < len(mentions) {
		u, err := lv.tutView.tut.Client.GetUserByID(mentions[mIndex].ID)
		if err != nil {
			lv.tutView.ShowError(
				fmt.Sprintf("Couldn't load user. Error:%v\n", err),
			)
			return
		}
		lv.tutView.Timeline.AddFeed(
			NewUserFeed(lv.tutView, u, config.NewTimeline(config.Timeline{
				FeedType: config.User,
			}),
			), false)
		lv.tutView.FocusMainNoHistory()
		return
	}
	tIndex := index - len(mentions) - len(urls)
	if tIndex < len(tags) {
		lv.tutView.Timeline.AddFeed(
			NewTagFeed(lv.tutView, config.NewTimeline(config.Timeline{
				FeedType:  config.Tag,
				Subaction: tags[tIndex].Name,
			})), false)
		lv.tutView.FocusMainNoHistory()
		return
	}
}

func (lv *LinkView) getURL() string {
	item, err := lv.tutView.GetCurrentItem()
	if err != nil {
		return ""
	}
	urls, mentions, tags, total := item.URLs()
	index := lv.list.GetCurrentItem()

	if total == 0 || index >= total {
		return ""
	}
	if index < len(urls) {
		return urls[index].URL
	}
	mIndex := index - len(urls)
	if mIndex < len(mentions) {
		return mentions[mIndex].URL
	}
	tIndex := index - len(mentions) - len(urls)
	if tIndex < len(tags) {
		return tags[tIndex].URL
	}
	return ""
}

func (lv *LinkView) Yank() {
	url := lv.getURL()
	if url == "" {
		return
	}
	copyToClipboard(url)
}

func (lv *LinkView) OpenCustom(c config.Custom) {
	url := lv.getURL()
	if url == "" {
		return
	}
	openCustom(lv.tutView, c.Program, c.Args, c.Terminal, url)
}