Initial commit: docker compose config

server/tgbot/list.go

@@ -0,0 +1,139 @@
+package tgbot
+
+import (
+	"strconv"
+	"strings"
+
+	"github.com/dustin/go-humanize"
+	tele "gopkg.in/telebot.v4"
+	"server/log"
+	"server/torr"
+)
+
+const listPageSize = 5
+
+func list(c tele.Context) error {
+	args := c.Args()
+	compact := len(args) > 0 && strings.ToLower(args[0]) == "compact"
+	return sendListPage(c, 0, compact)
+}
+
+func sendListPage(c tele.Context, page int, compact bool) error {
+	torrents := torr.ListTorrent()
+	if len(torrents) == 0 {
+		return c.Send(tr(c.Sender().ID, "no_torrents"))
+	}
+
+	totalPages := (len(torrents) + listPageSize - 1) / listPageSize
+	if page < 0 {
+		page = 0
+	}
+	if page >= totalPages {
+		page = totalPages - 1
+	}
+	start := page * listPageSize
+	end := start + listPageSize
+	if end > len(torrents) {
+		end = len(torrents)
+	}
+	pageTorrents := torrents[start:end]
+
+	uid := c.Sender().ID
+	for _, t := range pageTorrents {
+		hash := t.Hash().HexString()
+		var rows [][]tele.InlineButton
+		if compact {
+			rows = [][]tele.InlineButton{
+				{
+					tele.InlineButton{Text: tr(uid, "btn_files"), Unique: "files", Data: hash},
+					tele.InlineButton{Text: tr(uid, "btn_status"), Unique: "fstatus", Data: hash},
+					tele.InlineButton{Text: tr(uid, "btn_delete"), Unique: "delete", Data: hash},
+				},
+			}
+		} else {
+			rows = [][]tele.InlineButton{
+				{
+					tele.InlineButton{Text: tr(uid, "btn_files"), Unique: "files", Data: hash},
+					tele.InlineButton{Text: tr(uid, "btn_delete"), Unique: "delete", Data: hash},
+					tele.InlineButton{Text: tr(uid, "btn_status"), Unique: "fstatus", Data: hash},
+					tele.InlineButton{Text: tr(uid, "btn_m3u"), Unique: "fm3u", Data: hash},
+				},
+				{
+					tele.InlineButton{Text: tr(uid, "btn_link"), Unique: "flink", Data: hash},
+					tele.InlineButton{Text: tr(uid, "btn_drop"), Unique: "fdrop", Data: hash},
+				},
+			}
+		}
+		torrKbd := &tele.ReplyMarkup{InlineKeyboard: rows}
+		msg := "<b>" + escapeHtml(t.Title) + "</b>"
+		if t.Size > 0 {
+			msg += " <i>" + humanize.IBytes(uint64(t.Size)) + "</i>"
+		}
+		msg += "\n<code>" + hash + "</code>"
+		if err := c.Send(msg, torrKbd); err != nil {
+			log.TLogln("tg list send err", err)
+			return err
+		}
+	}
+
+	compactStr := "0"
+	if compact {
+		compactStr = "1"
+	}
+	navRow := []tele.InlineButton{}
+	if totalPages > 1 {
+		if page > 0 {
+			navRow = append(navRow, tele.InlineButton{Text: "◀️", Unique: "flist", Data: strconv.Itoa(page-1) + "|" + compactStr})
+		}
+		navRow = append(navRow, tele.InlineButton{Text: strconv.Itoa(page+1) + "/" + strconv.Itoa(totalPages), Unique: "fnop", Data: ""})
+		if page < totalPages-1 {
+			navRow = append(navRow, tele.InlineButton{Text: "▶️", Unique: "flist", Data: strconv.Itoa(page+1) + "|" + compactStr})
+		}
+	}
+	navRow = append(navRow, tele.InlineButton{Text: "🔄", Unique: "frefresh", Data: strconv.Itoa(page) + "|" + compactStr})
+	if len(navRow) > 1 || totalPages == 1 {
+		if err := c.Send(tr(uid, "page")+" "+strconv.Itoa(page+1)+"/"+strconv.Itoa(totalPages), &tele.ReplyMarkup{InlineKeyboard: [][]tele.InlineButton{navRow}}); err != nil {
+			log.TLogln("tg list nav err", err)
+			return err
+		}
+	}
+	return nil
+}
+
+func callbackListPage(c tele.Context, data string) error {
+	parts := strings.Split(data, "|")
+	page := 0
+	compact := false
+	if len(parts) > 0 && parts[0] != "" {
+		if p, err := strconv.Atoi(parts[0]); err == nil {
+			page = p
+		}
+	}
+	if len(parts) > 1 && parts[1] == "1" {
+		compact = true
+	}
+	_ = c.Respond(&tele.CallbackResponse{})
+	if c.Callback().Message != nil {
+		_ = c.Bot().Delete(c.Callback().Message)
+	}
+	return sendListPage(c, page, compact)
+}
+
+func callbackListRefresh(c tele.Context, data string) error {
+	parts := strings.Split(data, "|")
+	page := 0
+	compact := false
+	if len(parts) > 0 && parts[0] != "" {
+		if p, err := strconv.Atoi(parts[0]); err == nil {
+			page = p
+		}
+	}
+	if len(parts) > 1 && parts[1] == "1" {
+		compact = true
+	}
+	_ = c.Respond(&tele.CallbackResponse{Text: "🔄"})
+	if c.Callback().Message != nil {
+		_ = c.Bot().Delete(c.Callback().Message)
+	}
+	return sendListPage(c, page, compact)
+}