TorrServerJellyfin/server/tgbot/upload/queue.go

package upload

import (
	"fmt"
	"github.com/dustin/go-humanize"
	tele "gopkg.in/telebot.v4"
	"server/torr"
	"strconv"
	"time"
)

type DLQueue struct {
	id        int
	c         tele.Context
	hash      string
	fileID    string
	fileName  string
	updateMsg *tele.Message
}

var (
	manager = &Manager{}
)

func Start() {
	manager.Start()
}

func ShowQueue(c tele.Context) error {
	msg := ""
	manager.queueLock.Lock()
	defer manager.queueLock.Unlock()
	if len(manager.queue) == 0 && len(manager.working) == 0 {
		return c.Send("Очередь пуста")
	}
	if len(manager.working) > 0 {
		msg += "Закачиваются:\n"
		i := 0
		for _, dlQueue := range manager.working {
			s := "#" + strconv.Itoa(i+1) + ": <code>" + dlQueue.torrentHash + "</code>\n"
			if len(msg+s) > 1024 {
				c.Send(msg)
				msg = ""
			}
			msg += s
			i++
		}
		if len(msg) > 0 {
			c.Send(msg)
			msg = ""
		}
	}
	if len(manager.queue) > 0 {
		msg = "В очереди:\n"
		for i, dlQueue := range manager.queue {
			s := "#" + strconv.Itoa(i+1) + ": <code>" + dlQueue.torrentHash + "</code>\n"
			if len(msg+s) > 1024 {
				c.Send(msg)
				msg = ""
			}
			msg += s
		}
		if len(msg) > 0 {
			c.Send(msg)
			msg = ""
		}
	}
	return nil
}

func AddRange(c tele.Context, hash string, from, to int) {
	manager.AddRange(c, hash, from, to)
}

func Cancel(id int) {
	manager.Cancel(id)
}

func updateLoadStatus(wrk *Worker, file *TorrFile, fi, fc int) {
	if wrk.msg == nil {
		return
	}
	t := torr.GetTorrent(wrk.torrentHash)
	ti := t.Status()
	if wrk.isCancelled {
		wrk.c.Bot().Edit(wrk.msg, "Остановка...")
	} else {
		wrk.c.Send(tele.UploadingVideo)
		if ti.DownloadSpeed == 0 {
			ti.DownloadSpeed = 1.0
		}
		wait := time.Duration(float64(file.Loaded())/ti.DownloadSpeed) * time.Second
		speed := humanize.Bytes(uint64(ti.DownloadSpeed)) + "/sec"
		peers := fmt.Sprintf("%v · %v/%v", ti.ConnectedSeeders, ti.ActivePeers, ti.TotalPeers)
		prc := fmt.Sprintf("%.2f%% %v / %v", float64(file.offset)*100.0/float64(file.size), humanize.Bytes(uint64(file.offset)), humanize.Bytes(uint64(file.size)))

		name := file.name
		if name == ti.Title {
			name = ""
		}

		msg := "Загрузка торрента:\n" +
			"<b>" + ti.Title + "</b>\n"
		if name != "" {
			msg += "<i>" + name + "</i>\n"
		}
		msg += "<b>Хэш:</b> <code>" + file.hash + "</code>\n"
		if file.offset < file.size {
			msg += "<b>Скорость: </b>" + speed + "\n" +
				"<b>Осталось: </b>" + wait.String() + "\n" +
				"<b>Пиры: </b>" + peers + "\n" +
				"<b>Загружено: </b>" + prc
		}
		if fc > 1 {
			msg += "\n<b>Файлов: </b>" + strconv.Itoa(fi) + "/" + strconv.Itoa(fc)
		}
		if file.offset >= file.size {
			msg += "\n<b>Завершение загрузки, это займет некоторое время</b>"
			wrk.c.Bot().Edit(wrk.msg, msg)
			return
		}

		torrKbd := &tele.ReplyMarkup{}
		torrKbd.Inline([]tele.Row{torrKbd.Row(torrKbd.Data("Отмена", "cancel", strconv.Itoa(wrk.id)))}...)
		wrk.c.Bot().Edit(wrk.msg, msg, torrKbd)
	}
}