item_status.go

package ui

import (
	"bytes"
	"fmt"
	"strings"
	"time"

	"github.com/RasmusLindroth/go-mastodon"
	"github.com/RasmusLindroth/tut/api"
	"github.com/RasmusLindroth/tut/config"
	"github.com/RasmusLindroth/tut/util"
	"github.com/rivo/tview"
)

type Toot struct {
	Visibility         string
	Boosted            bool
	BoostedDisplayName string
	BoostedAcct        string
	Bookmarked         bool
	AccountDisplayName string
	Account            string
	Spoiler            bool
	CWText             string
	SpoilerText        string
	ShowSpoiler        bool
	CWlabel            string
	ContentText        string
	Width              int
	HasExtra           bool
	Poll               Poll
	Media              []Media
	Card               Card
	Replies            int
	Boosts             int
	Favorites          int
	Edited             bool
	Lang               string
	Controls           string
}

type Poll struct {
	ID         string
	ExpiresAt  time.Time
	Expired    bool
	Multiple   bool
	VotesCount int64
	Options    []PollOption
	Voted      bool
}

type PollOption struct {
	Title      string
	VotesCount int64
	Percent    string
}

type Media struct {
	Type        string
	Description string
	URL         string
}

type Card struct {
	Type        string
	Title       string
	Description string
	URL         string
}

type DisplayTootData struct {
	Toot  Toot
	Style config.Style
}

func drawStatus(tv *TutView, item api.Item, status *mastodon.Status, main *tview.TextView, controls *tview.Flex, ft config.FeedType, isHistory bool, additional string) {
	controls.Clear()
	filtered, _, phrase, _ := item.Filtered(ft)
	if filtered {
		var output string
		if tv.tut.Config.General.ShowFilterPhrase {
			output = fmt.Sprintf("Filtered by phrase: %s\n\n", tview.Escape(phrase))
		} else {
			output = "Filtered.\n\n"
		}
		ctrl := NewControl(tv.tut.Config, tv.tut.Config.Input.StatusShowFiltered, true)
		output += ctrl.Label
		if main != nil {
			if additional != "" {
				additional = fmt.Sprintf("%s\n\n", config.SublteText(tv.tut.Config, additional))
			}
			main.SetText(additional + output)
		}
		return
	}

	showSensitive := item.ShowCW()

	var strippedContent string
	var strippedSpoiler string

	so := status
	if status.Reblog != nil {
		status = status.Reblog
	}

	strippedContent, _ = util.CleanHTMLStyled(status.Content)

	width := 0
	if main != nil {
		_, _, width, _ = main.GetInnerRect()
	}
	cwToggle := NewControl(tv.tut.Config, tv.tut.Config.Input.StatusToggleCW, true)
	toot := Toot{
		Width:              width,
		ContentText:        strippedContent,
		Boosted:            so.Reblog != nil,
		BoostedDisplayName: tview.Escape(so.Account.DisplayName),
		BoostedAcct:        tview.Escape(so.Account.Acct),
		ShowSpoiler:        showSensitive,
		CWlabel:            cwToggle.Label,
	}
	for _, lang := range util.Languages {
		if status.Language == lang.Code {
			toot.Lang = lang.English
			break
		}
	}

	toot.AccountDisplayName = tview.Escape(status.Account.DisplayName)
	toot.Account = tview.Escape(status.Account.Acct)
	toot.Bookmarked = false
	if status.Bookmarked != nil {
		toot.Bookmarked = status.Bookmarked.(bool)
	}
	toot.Visibility = status.Visibility
	toot.Spoiler = status.Sensitive
	toot.Edited = status.CreatedAt.Before(status.EditedAt)

	if status.Poll != nil {
		p := *status.Poll
		toot.Poll = Poll{
			ID:         string(p.ID),
			ExpiresAt:  p.ExpiresAt,
			Expired:    p.Expired,
			Multiple:   p.Multiple,
			VotesCount: p.VotesCount,
			Voted:      p.Voted,
			Options:    []PollOption{},
		}
		for _, item := range p.Options {
			percent := 0.0
			if p.VotesCount > 0 {
				percent = float64(item.VotesCount) / float64(p.VotesCount) * 100
			}

			o := PollOption{
				Title:      tview.Escape(item.Title),
				VotesCount: item.VotesCount,
				Percent:    fmt.Sprintf("%.2f", percent),
			}
			toot.Poll.Options = append(toot.Poll.Options, o)
		}

	} else {
		toot.Poll = Poll{}
	}

	if status.Sensitive {
		strippedSpoiler, _ = util.CleanHTMLStyled(status.SpoilerText)
	}

	toot.CWText = strippedSpoiler
	toot.SpoilerText = toot.CWText

	media := []Media{}
	for _, att := range status.MediaAttachments {
		m := Media{
			Type:        att.Type,
			Description: tview.Escape(att.Description),
			URL:         att.URL,
		}
		media = append(media, m)
	}
	toot.Media = media

	if status.Card != nil {
		toot.Card = Card{
			Type:        status.Card.Type,
			Title:       tview.Escape(strings.TrimSpace(status.Card.Title)),
			Description: tview.Escape(strings.TrimSpace(status.Card.Description)),
			URL:         status.Card.URL,
		}
	} else {
		toot.Card = Card{}
	}

	toot.HasExtra = len(status.MediaAttachments) > 0 || status.Card != nil || status.Poll != nil
	toot.Replies = int(status.RepliesCount)
	toot.Boosts = int(status.ReblogsCount)
	toot.Favorites = int(status.FavouritesCount)

	if main != nil {
		main.ScrollToBeginning()
	}

	var info []Control
	statusFavorited, statusBoosted, statusBookmarked := false, false, false
	if status.Favourited != nil {
		statusFavorited = status.Favourited.(bool)
	}
	if status.Reblogged != nil {
		statusBoosted = status.Reblogged.(bool)
	}
	if status.Bookmarked != nil {
		statusBookmarked = status.Bookmarked.(bool)
	}

	if statusFavorited && !isHistory {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusFavorite, false))
	} else if !statusFavorited && !isHistory {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusFavorite, true))
	}
	if statusBoosted && !isHistory {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusBoost, false))
	} else if !statusBoosted && !isHistory {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusBoost, true))
	}
	if !isHistory {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusThread, true))
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusReply, true))
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusViewFocus, true))
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusUser, true))
	}
	if len(status.MediaAttachments) > 0 {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusMedia, true))
	}
	_, _, _, length := item.URLs()
	if length > 0 {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusLinks, true))
	}
	info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusAvatar, true))
	if status.Account.ID == tv.tut.Client.Me.ID && !isHistory {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusEdit, true))
	}
	if status.Account.ID == tv.tut.Client.Me.ID && !isHistory {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusDelete, true))
	}

	if !statusBookmarked && !isHistory {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusBookmark, true))
	} else if statusBookmarked && !isHistory {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusBookmark, false))
	}
	if !isHistory {
		info = append(info, NewControl(tv.tut.Config, tv.tut.Config.Input.StatusYank, true))
	}

	for i, item := range info {
		if i < len(info)-1 {
			controls.AddItem(NewControlButton(tv, item), item.Len+1, 0, false)
		} else {
			controls.AddItem(NewControlButton(tv, item), item.Len, 0, false)
		}
	}

	td := DisplayTootData{
		Toot:  toot,
		Style: tv.tut.Config.Style,
	}
	var output bytes.Buffer
	err := tv.tut.Config.Templates.Toot.ExecuteTemplate(&output, "toot.tmpl", td)
	if err != nil {
		panic(err)
	}

	if main != nil {
		if additional != "" {
			additional = fmt.Sprintf("%s\n\n", config.SublteText(tv.tut.Config, additional))
		}
		main.SetText(additional + output.String())
	}
}