feat: add backend

This commit is contained in:
Ludvig Lundgren 2021-08-11 15:26:17 +02:00
parent bc418ff248
commit a838d994a6
68 changed files with 9561 additions and 0 deletions

View file

@ -0,0 +1,95 @@
package download_client
import (
"github.com/rs/zerolog/log"
"github.com/autobrr/autobrr/internal/domain"
"github.com/autobrr/autobrr/pkg/qbittorrent"
)
type Service interface {
List() ([]domain.DownloadClient, error)
FindByID(id int32) (*domain.DownloadClient, error)
Store(client domain.DownloadClient) (*domain.DownloadClient, error)
Delete(clientID int) error
Test(client domain.DownloadClient) error
}
type service struct {
repo domain.DownloadClientRepo
}
func NewService(repo domain.DownloadClientRepo) Service {
return &service{repo: repo}
}
func (s *service) List() ([]domain.DownloadClient, error) {
clients, err := s.repo.List()
if err != nil {
return nil, err
}
return clients, nil
}
func (s *service) FindByID(id int32) (*domain.DownloadClient, error) {
client, err := s.repo.FindByID(id)
if err != nil {
return nil, err
}
return client, nil
}
func (s *service) Store(client domain.DownloadClient) (*domain.DownloadClient, error) {
// validate data
// store
c, err := s.repo.Store(client)
if err != nil {
return nil, err
}
return c, nil
}
func (s *service) Delete(clientID int) error {
if err := s.repo.Delete(clientID); err != nil {
return err
}
log.Debug().Msgf("delete client: %v", clientID)
return nil
}
func (s *service) Test(client domain.DownloadClient) error {
// test
err := s.testConnection(client)
if err != nil {
return err
}
return nil
}
func (s *service) testConnection(client domain.DownloadClient) error {
if client.Type == "QBITTORRENT" {
qbtSettings := qbittorrent.Settings{
Hostname: client.Host,
Port: uint(client.Port),
Username: client.Username,
Password: client.Password,
SSL: client.SSL,
}
qbt := qbittorrent.NewClient(qbtSettings)
err := qbt.Login()
if err != nil {
log.Error().Err(err).Msgf("error logging into client: %v", client.Host)
return err
}
}
return nil
}