1
0
mirror of https://github.com/taigrr/wtf synced 2025-01-18 04:03:14 -08:00
wtf/modules/transmission/display.go
2019-08-05 10:50:12 -07:00

84 lines
1.8 KiB
Go

package transmission
import (
"fmt"
"strings"
"github.com/hekmon/transmissionrpc"
"github.com/rivo/tview"
"github.com/wtfutil/wtf/utils"
)
func (widget *Widget) contentFrom(data []*transmissionrpc.Torrent) string {
str := ""
for idx, torrent := range data {
torrName := *torrent.Name
row := fmt.Sprintf(
"[%s] %s %s %s%s[white]",
widget.RowColor(idx),
widget.torrentPercentDone(torrent),
widget.torrentSeedRatio(torrent),
widget.torrentState(torrent),
tview.Escape(widget.prettyTorrentName(torrName)),
)
str += utils.HighlightableHelper(widget.View, row, idx, len(torrName))
}
return str
}
func (widget *Widget) display() {
content := widget.contentFrom(widget.torrents)
widget.ScrollableWidget.Redraw(widget.CommonSettings().Title, content, false)
}
func (widget *Widget) prettyTorrentName(name string) string {
str := strings.Replace(name, "[", "(", -1)
str = strings.Replace(str, "]", ")", -1)
return str
}
func (widget *Widget) torrentPercentDone(torrent *transmissionrpc.Torrent) string {
pctDone := *torrent.PercentDone
str := fmt.Sprintf("%3d%%↓", int(pctDone*100))
if pctDone == 0.0 {
str = "[gray::b]" + str
} else if pctDone == 1.0 {
str = "[green::b]" + str
} else {
str = "[lightblue::b]" + str
}
return str + "[white]"
}
func (widget *Widget) torrentSeedRatio(torrent *transmissionrpc.Torrent) string {
seedRatio := *torrent.UploadRatio
if seedRatio < 0 {
seedRatio = 0
}
return fmt.Sprintf("[green]%3d%%↑", int(seedRatio*100))
}
func (widget *Widget) torrentState(torrent *transmissionrpc.Torrent) string {
str := ""
switch *torrent.Status {
case transmissionrpc.TorrentStatusStopped:
str += "[gray]"
case transmissionrpc.TorrentStatusDownload:
str += "[lightblue]"
case transmissionrpc.TorrentStatusSeed:
str += "[green]"
}
return str
}