torrent.go

package services

import (
	"sync"

	"github.com/pkg/errors"
	log "github.com/sirupsen/logrus"

	"github.com/anacrolix/torrent"
)

type Torrent struct {
	t      *torrent.Torrent
	tcS    *TorrentClient
	miS    *MetaInfo
	mux    sync.Mutex
	err    error
	inited bool
}

func NewTorrent(tcS *TorrentClient, miS *MetaInfo) *Torrent {
	return &Torrent{tcS: tcS, miS: miS, inited: false}
}

func (s *Torrent) Ready() bool {
	return s.t != nil
}

func (s *Torrent) get() (*torrent.Torrent, error) {
	log.Info("Initializing Torrent")
	mi, err := s.miS.Get()
	if err != nil {
		return nil, err
	}
	cl, err := s.tcS.Get()
	if err != nil {
		return nil, err
	}
	t, err := cl.AddTorrent(mi)
	if err != nil {
		return nil, errors.Wrap(err, "Failed to add torrent")
	}
	return t, nil
}

func (s *Torrent) Get() (*torrent.Torrent, error) {
	s.mux.Lock()
	defer s.mux.Unlock()
	if s.inited {
		return s.t, s.err
	}
	s.t, s.err = s.get()
	s.inited = true
	return s.t, s.err
}