autobrr/internal/action/deluge.go

package action

import (
	"encoding/base64"
	"errors"
	"io/ioutil"
	"time"

	"github.com/autobrr/autobrr/internal/domain"

	delugeClient "github.com/gdm85/go-libdeluge"
	"github.com/rs/zerolog/log"
)

func (s *service) deluge(action domain.Action, torrentFile string) error {
	log.Trace().Msgf("action DELUGE: %v", torrentFile)

	var err error

	// get client for action
	client, err := s.clientSvc.FindByID(action.ClientID)
	if err != nil {
		log.Error().Err(err).Msgf("error finding client: %v", action.ClientID)
		return err
	}

	if client == nil {
		return errors.New("no client found")
	}

	settings := delugeClient.Settings{
		Hostname:             client.Host,
		Port:                 uint(client.Port),
		Login:                client.Username,
		Password:             client.Password,
		DebugServerResponses: true,
		ReadWriteTimeout:     time.Second * 20,
	}

	switch client.Type {
	case "DELUGE_V1":
		err = delugeV1(settings, action, torrentFile)

	case "DELUGE_V2":
		err = delugeV2(settings, action, torrentFile)
	}

	return err
}

func delugeV1(settings delugeClient.Settings, action domain.Action, torrentFile string) error {

	deluge := delugeClient.NewV1(settings)

	// perform connection to Deluge server
	err := deluge.Connect()
	if err != nil {
		log.Error().Err(err).Msgf("error logging into client: %v", settings.Hostname)
		return err
	}

	defer deluge.Close()

	t, err := ioutil.ReadFile(torrentFile)
	if err != nil {
		log.Error().Err(err).Msgf("could not read torrent file: %v", torrentFile)
		return err
	}

	// encode file to base64 before sending to deluge
	encodedFile := base64.StdEncoding.EncodeToString(t)
	if encodedFile == "" {
		log.Error().Err(err).Msgf("could not encode torrent file: %v", torrentFile)
		return err
	}

	// set options
	options := delugeClient.Options{}

	if action.Paused {
		options.AddPaused = &action.Paused
	}
	if action.SavePath != "" {
		options.DownloadLocation = &action.SavePath
	}
	if action.LimitDownloadSpeed > 0 {
		maxDL := int(action.LimitDownloadSpeed)
		options.MaxDownloadSpeed = &maxDL
	}
	if action.LimitUploadSpeed > 0 {
		maxUL := int(action.LimitUploadSpeed)
		options.MaxUploadSpeed = &maxUL
	}

	torrentHash, err := deluge.AddTorrentFile(torrentFile, encodedFile, &options)
	if err != nil {
		log.Error().Err(err).Msgf("could not add torrent to client: %v", torrentFile)
		return err
	}

	if action.Label != "" {

		p, err := deluge.LabelPlugin()
		if err != nil {
			log.Error().Err(err).Msgf("could not load label plugin: %v", torrentFile)
			return err
		}

		if p != nil {
			// TODO first check if label exists, if not, add it, otherwise set
			err = p.SetTorrentLabel(torrentHash, action.Label)
			if err != nil {
				log.Error().Err(err).Msgf("could not set label: %v", torrentFile)
				return err
			}
		}
	}

	log.Trace().Msgf("deluge: torrent successfully added! hash: %v", torrentHash)

	return nil
}

func delugeV2(settings delugeClient.Settings, action domain.Action, torrentFile string) error {

	deluge := delugeClient.NewV2(settings)

	// perform connection to Deluge server
	err := deluge.Connect()
	if err != nil {
		log.Error().Err(err).Msgf("error logging into client: %v", settings.Hostname)
		return err
	}

	defer deluge.Close()

	t, err := ioutil.ReadFile(torrentFile)
	if err != nil {
		log.Error().Err(err).Msgf("could not read torrent file: %v", torrentFile)
		return err
	}

	// encode file to base64 before sending to deluge
	encodedFile := base64.StdEncoding.EncodeToString(t)
	if encodedFile == "" {
		log.Error().Err(err).Msgf("could not encode torrent file: %v", torrentFile)
		return err
	}

	// set options
	options := delugeClient.Options{}

	if action.Paused {
		options.AddPaused = &action.Paused
	}
	if action.SavePath != "" {
		options.DownloadLocation = &action.SavePath
	}
	if action.LimitDownloadSpeed > 0 {
		maxDL := int(action.LimitDownloadSpeed)
		options.MaxDownloadSpeed = &maxDL
	}
	if action.LimitUploadSpeed > 0 {
		maxUL := int(action.LimitUploadSpeed)
		options.MaxUploadSpeed = &maxUL
	}

	torrentHash, err := deluge.AddTorrentFile(torrentFile, encodedFile, &options)
	if err != nil {
		log.Error().Err(err).Msgf("could not add torrent to client: %v", torrentFile)
		return err
	}

	if action.Label != "" {

		p, err := deluge.LabelPlugin()
		if err != nil {
			log.Error().Err(err).Msgf("could not load label plugin: %v", torrentFile)
			return err
		}

		if p != nil {
			// TODO first check if label exists, if not, add it, otherwise set
			err = p.SetTorrentLabel(torrentHash, action.Label)
			if err != nil {
				log.Error().Err(err).Msgf("could not set label: %v", torrentFile)
				return err
			}
		}
	}

	log.Trace().Msgf("deluge: torrent successfully added! hash: %v", torrentHash)

	return nil
}