feat(clients): add rTorrent support (#421)

internal/action/rtorrent.go

@@ -0,0 +1,68 @@
+package action
+
+import (
+	"context"
+	"io/ioutil"
+
+	"github.com/autobrr/autobrr/internal/domain"
+	"github.com/autobrr/autobrr/pkg/errors"
+
+	"github.com/mrobinsn/go-rtorrent/rtorrent"
+)
+
+func (s *service) rtorrent(action domain.Action, release domain.Release) ([]string, error) {
+	s.log.Debug().Msgf("action rTorrent: %v", action.Name)
+
+	var err error
+
+	// get client for action
+	client, err := s.clientSvc.FindByID(context.TODO(), action.ClientID)
+	if err != nil {
+		s.log.Error().Stack().Err(err).Msgf("error finding client: %v", action.ClientID)
+		return nil, err
+	}
+
+	if client == nil {
+		return nil, errors.New("could not find client by id: %v", action.ClientID)
+	}
+
+	var rejections []string
+
+	if release.TorrentTmpFile == "" {
+		if err = release.DownloadTorrentFile(); err != nil {
+			s.log.Error().Err(err).Msgf("could not download torrent file for release: %v", release.TorrentName)
+			return nil, err
+		}
+	}
+
+	// create client
+	rt := rtorrent.New(client.Host, true)
+
+	tmpFile, err := ioutil.ReadFile(release.TorrentTmpFile)
+	if err != nil {
+		return nil, errors.Wrap(err, "could not read torrent file: %v", release.TorrentTmpFile)
+	}
+
+	var args []*rtorrent.FieldValue
+
+	if action.Label != "" {
+		args = append(args, &rtorrent.FieldValue{
+			Field: "d.custom1",
+			Value: action.Label,
+		})
+	}
+	if action.SavePath != "" {
+		args = append(args, &rtorrent.FieldValue{
+			Field: "d.base_path",
+			Value: action.SavePath,
+		})
+	}
+
+	if err := rt.AddTorrent(tmpFile, args...); err != nil {
+		return nil, errors.Wrap(err, "could not add torrent file: %v", release.TorrentTmpFile)
+	}
+
+	s.log.Info().Msgf("torrent with hash %v successfully added to client: '%v'", "", client.Name)
+
+	return rejections, nil
+}

internal/action/run.go

@@ -48,6 +48,9 @@ func (s *service) RunAction(action *domain.Action, release domain.Release) ([]st
 	case domain.ActionTypeQbittorrent:
 		rejections, err = s.qbittorrent(*action, release)
 
+	case domain.ActionTypeRTorrent:
+		rejections, err = s.rtorrent(*action, release)
+
 	case domain.ActionTypeTransmission:
 		rejections, err = s.transmission(*action, release)