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
91 lines
1.9 KiB
Go
91 lines
1.9 KiB
Go
package transmission
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/hekmon/transmissionrpc"
|
|
"github.com/rivo/tview"
|
|
"github.com/wtfutil/wtf/utils"
|
|
)
|
|
|
|
func (widget *Widget) content() (string, string, bool) {
|
|
title := widget.CommonSettings().Title
|
|
if widget.err != nil {
|
|
return title, widget.err.Error(), true
|
|
}
|
|
data := widget.torrents
|
|
if data == nil || len(data) == 0 {
|
|
return title, "No data", false
|
|
}
|
|
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 title, str, false
|
|
}
|
|
|
|
func (widget *Widget) display() {
|
|
widget.ScrollableWidget.Redraw(widget.content)
|
|
}
|
|
|
|
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
|
|
}
|