object-torrent.go

package main

import (
	"context"
	"fmt"
	"github.com/anacrolix/torrent"
	tgbotapi "github.com/krol44/telegram-bot-api"
	log "github.com/sirupsen/logrus"
	"path"
	"strconv"
	"strings"
	"time"
)

func (o *ObjectTorrent) Download() bool {
	if o.Task.Limit("torrent") {
		log.Debug("limit exceeded")
		return false
	}

	if !o.Task.AllocTorrent("torrent") {
		o.Task.App.SendLogToChannel(o.Task.Message.From, "mess", "❗️ return - error alloc")
		return false
	}

	o.Task.Torrent.Process = o.TorrentProcess

	var fileChosen *torrent.File
	for _, val := range o.Task.Torrent.Process.Files() {
		recoveryPath := val.Path() + " ~ " + strconv.FormatInt(val.Length()>>20, 10) + " MB"
		if strings.Contains(recoveryPath, o.Task.Message.Text) {
			if val.Length() > 1999e6 { // more 2 GB
				o.Task.Send(tgbotapi.NewMessage(o.Task.Message.Chat.ID,
					fmt.Sprintf("😔 "+o.Task.Lang("File is bigger 2 GB"))))
				o.Task.App.SendLogToChannel(o.Task.Message.From, "mess", "File is bigger 2 GB")

				_, err := Postgres.Exec(`DELETE FROM limits WHERE id = any 
                         (array(SELECT id FROM limits WHERE telegram_id = $1 AND type_object = $2
                                                      ORDER BY date_create DESC LIMIT 1))`,
					o.Task.Message.From.ID, "torrent")
				if err != nil {
					log.Error(err)
				}

				return false
			}
			val.SetPriority(torrent.PiecePriorityNow)
			fileChosen = val
		} else {
			val.SetPriority(torrent.PiecePriorityNone)
		}
	}

	if fileChosen == nil {
		log.Error("file chosen is empty")
		return false
	}

	// if name not correct
	o.Task.Torrent.Process.SetDisplayName(o.Task.UniqueId("temp-torrent-name"))

	o.Task.Torrent.Process.SetMaxEstablishedConns(200)

	<-o.Task.Torrent.Process.GotInfo()

	o.Task.Torrent.Name = fileChosen.DisplayPath()

	ctx, cancelProgress := context.WithCancel(context.Background())
	go func(ctx context.Context, fileChosen *torrent.File) {
		for {
			select {
			case <-ctx.Done():
				return
			default:
				if _, bo := o.Task.App.ChatsWork.StopTasks.Load(o.Task.Message.Chat.ID); bo {
					return
				}

				stat, percent := o.Task.StatDlTor(fileChosen)
				if percent == 100 {
					return
				}

				go func(ctx context.Context, t *Task) {
					select {
					case <-ctx.Done():
						return
					default:
						if time.Now().Second()%2 == 0 && t.MessageTextLast != stat {
							t.Send(tgbotapi.NewEditMessageText(t.Message.Chat.ID, t.MessageEditID, stat))
							t.MessageTextLast = stat
						}
					}
				}(ctx, o.Task)
			}
			time.Sleep(time.Second)
		}
	}(ctx, fileChosen)

	o.Task.Torrent.Process.AllowDataDownload()

	timeStartToWork := time.Now().Unix()
	maxTimeWork := int64(1800)
	var sizeCheck int

	for {
		if time.Now().Unix() > timeStartToWork+maxTimeWork {
			break
		}
		if _, bo := o.Task.App.ChatsWork.StopTasks.Load(o.Task.Message.Chat.ID); bo {
			break
		}
		if fileChosen.FileInfo().Length == fileChosen.BytesCompleted() {
			break
		}

		if sizeCheck > 60 && fileChosen.BytesCompleted() == 0 {
			maxTimeWork = 0
			break
		}
		sizeCheck++

		time.Sleep(time.Second)
	}

	cancelProgress()

	if time.Now().Unix() > timeStartToWork+maxTimeWork {
		o.Task.Send(tgbotapi.NewMessage(o.Task.Message.Chat.ID,
			fmt.Sprintf("😔 "+o.Task.Lang("Didn't have time to download, maximum 30 minutes or speed is low"))))
		o.Task.App.SendLogToChannel(o.Task.Message.From, "mess",
			"didn't have time to download torrent")

		o.Task.Torrent.Process.Drop()
		return false
	}

	if _, bo := o.Task.App.ChatsWork.StopTasks.Load(o.Task.Message.Chat.ID); bo {
		o.Task.Torrent.Process.Drop()
		return false
	}

	o.Task.Send(tgbotapi.NewEditMessageText(o.Task.Message.Chat.ID, o.Task.MessageEditID,
		"✅ "+o.Task.Lang("Torrent downloaded, wait next step")))

	pathway := path.Clean(config.DirBot + "/torrent-client/" + fileChosen.Path())

	cache := Cache{Task: o.Task}
	if cache.TrySend("video", pathway) {
		return false
	}
	if cache.TrySendThroughMd5(pathway) {
		return false
	}
	if cache.TrySend("doc", o.Task.Torrent.Name+".torrent") {
		return false
	}

	o.Task.File = pathway

	return true
}

func (o *ObjectTorrent) Convert() bool {
	var c = Convert{Task: o.Task, IsTorrent: true}

	if c.Task.IsAllowFormatForConvert(c.Task.File) {
		o.Task.FileConverted = c.Run()

		return true
	} else {
		return false
	}
}

func (o *ObjectTorrent) Send() bool {
	if path.Ext(o.Task.File) == ".flac" ||
		path.Ext(o.Task.File) == ".mp3" ||
		path.Ext(o.Task.File) == ".ogg" ||
		path.Ext(o.Task.File) == ".wav" {
		o.Task.Files = []string{o.Task.File}
		return o.Task.SendAudio()
	}

	if path.Ext(o.Task.FileConverted.FilePath) == ".mp4" {
		return o.Task.SendVideo(true)
	}

	//if o.Task.UserFromDB.Premium == 0 {
	//	o.Task.Send(tgbotapi.NewMessage(o.Task.Message.Chat.ID,
	//		fmt.Sprintf("❗️ "+o.Task.Lang("Available only for users who support us"))))
	//}
	return o.Task.SendDoc()
}

func (o *ObjectTorrent) Clean() {
	o.Task.RemoveMessageEdit()
}