torrserver/server/tgbot/upload/torrfile.go

package upload

import (
	"errors"
	"fmt"
	"path/filepath"

	"github.com/anacrolix/torrent"

	sets "server/settings"
	"server/log"
	"server/tgbot/config"
	"server/torr"
	"server/torr/state"
	"server/torr/storage/torrstor"
)

var ERR_STOPPED = errors.New("stopped")

type TorrFile struct {
	hash   string
	name   string
	wrk    *Worker
	offset int64
	size   int64
	id     int

	reader *torrstor.Reader
}

func NewTorrFile(wrk *Worker, stFile *state.TorrentFileStat) (*TorrFile, error) {
	uid := int64(0)
	if wrk.c != nil && wrk.c.Sender() != nil {
		uid = wrk.c.Sender().ID
	}
	if config.Cfg != nil && config.Cfg.HostTG != "" && stFile.Length > 2*1024*1024*1024 {
		return nil, errors.New(tr(uid, "upload_file_too_large_2gb"))
	}
	if (config.Cfg == nil || config.Cfg.HostTG == "") && stFile.Length > 50*1024*1024 {
		return nil, errors.New(tr(uid, "upload_file_too_large_50mb"))
	}

	tf := new(TorrFile)
	tf.hash = wrk.torrentHash
	tf.name = filepath.Base(stFile.Path)
	tf.wrk = wrk
	tf.size = stFile.Length

	t := torr.GetTorrent(wrk.torrentHash)
	t.WaitInfo()

	files := t.Files()
	var file *torrent.File
	for _, tfile := range files {
		if tfile.Path() == stFile.Path {
			file = tfile
			break
		}
	}
	if file == nil {
		return nil, fmt.Errorf("file with id %v not found", stFile.Id)
	}
	if int64(sets.MaxSize) > 0 && file.Length() > int64(sets.MaxSize) {
		log.TLogln("tg upload err size", file.DisplayPath(), "max", sets.MaxSize)
		return nil, fmt.Errorf("file size exceeded max allowed %d bytes", sets.MaxSize)
	}

	reader := t.NewReader(file)
	if reader == nil {
		return nil, errors.New("cannot create torrent reader")
	}
	if sets.BTsets != nil && sets.BTsets.ResponsiveMode {
		reader.SetResponsive()
	}
	tf.reader = reader

	return tf, nil
}

func (t *TorrFile) Read(p []byte) (n int, err error) {
	if t.wrk.isCancelled {
		return 0, ERR_STOPPED
	}
	n, err = t.reader.Read(p)
	t.offset += int64(n)
	return
}

func (t *TorrFile) Remaining() int64 {
	return t.size - t.offset
}

func (t *TorrFile) Close() {
	if t.reader != nil {
		t.reader.Close()
		t.reader = nil
	}
}