mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 16:59:12 +00:00
bc0f4cc055
* feat: add support for proxies * fix(http): release handler * fix(migrations): define proxy early * fix(migrations): pg proxy * fix(proxy): list update delete * fix(proxy): remove log and imports * feat(irc): use proxy * feat(irc): tests * fix(web): update imports for ProxyForms.tsx * fix(database): migration * feat(proxy): test * feat(proxy): validate proxy type * feat(proxy): validate and test * feat(proxy): improve validate and test * feat(proxy): fix db schema * feat(proxy): add db tests * feat(proxy): handle http errors * fix(http): imports * feat(proxy): use proxy for indexer downloads * feat(proxy): indexerforms select proxy * feat(proxy): handle torrent download * feat(proxy): skip if disabled * feat(proxy): imports * feat(proxy): implement in Feeds * feat(proxy): update helper text indexer proxy * feat(proxy): add internal cache
111 lines
3.3 KiB
Go
111 lines
3.3 KiB
Go
// Copyright (c) 2021 - 2024, Ludvig Lundgren and the autobrr contributors.
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
package action
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/autobrr/autobrr/internal/domain"
|
|
"github.com/autobrr/autobrr/internal/download_client"
|
|
"github.com/autobrr/autobrr/internal/logger"
|
|
"github.com/autobrr/autobrr/internal/releasedownload"
|
|
"github.com/autobrr/autobrr/pkg/sharedhttp"
|
|
|
|
"github.com/asaskevich/EventBus"
|
|
"github.com/dcarbone/zadapters/zstdlog"
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
type Service interface {
|
|
Store(ctx context.Context, action domain.Action) (*domain.Action, error)
|
|
StoreFilterActions(ctx context.Context, filterID int64, actions []*domain.Action) ([]*domain.Action, error)
|
|
List(ctx context.Context) ([]domain.Action, error)
|
|
Get(ctx context.Context, req *domain.GetActionRequest) (*domain.Action, error)
|
|
FindByFilterID(ctx context.Context, filterID int, active *bool, withClient bool) ([]*domain.Action, error)
|
|
Delete(ctx context.Context, req *domain.DeleteActionRequest) error
|
|
DeleteByFilterID(ctx context.Context, filterID int) error
|
|
ToggleEnabled(actionID int) error
|
|
|
|
RunAction(ctx context.Context, action *domain.Action, release *domain.Release) ([]string, error)
|
|
}
|
|
|
|
type service struct {
|
|
log zerolog.Logger
|
|
subLogger *log.Logger
|
|
repo domain.ActionRepo
|
|
clientSvc download_client.Service
|
|
downloadSvc *releasedownload.DownloadService
|
|
bus EventBus.Bus
|
|
|
|
httpClient *http.Client
|
|
}
|
|
|
|
func NewService(log logger.Logger, repo domain.ActionRepo, clientSvc download_client.Service, downloadSvc *releasedownload.DownloadService, bus EventBus.Bus) Service {
|
|
s := &service{
|
|
log: log.With().Str("module", "action").Logger(),
|
|
repo: repo,
|
|
clientSvc: clientSvc,
|
|
downloadSvc: downloadSvc,
|
|
bus: bus,
|
|
|
|
httpClient: &http.Client{
|
|
Timeout: time.Second * 120,
|
|
Transport: sharedhttp.TransportTLSInsecure,
|
|
},
|
|
}
|
|
|
|
s.subLogger = zstdlog.NewStdLoggerWithLevel(s.log.With().Logger(), zerolog.TraceLevel)
|
|
|
|
return s
|
|
}
|
|
|
|
func (s *service) Store(ctx context.Context, action domain.Action) (*domain.Action, error) {
|
|
return s.repo.Store(ctx, action)
|
|
}
|
|
|
|
func (s *service) StoreFilterActions(ctx context.Context, filterID int64, actions []*domain.Action) ([]*domain.Action, error) {
|
|
return s.repo.StoreFilterActions(ctx, filterID, actions)
|
|
}
|
|
|
|
func (s *service) List(ctx context.Context) ([]domain.Action, error) {
|
|
return s.repo.List(ctx)
|
|
}
|
|
|
|
func (s *service) Get(ctx context.Context, req *domain.GetActionRequest) (*domain.Action, error) {
|
|
a, err := s.repo.Get(ctx, req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// optionally attach download client to action
|
|
if a.ClientID > 0 {
|
|
client, err := s.clientSvc.FindByID(ctx, a.ClientID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
a.Client = client
|
|
}
|
|
|
|
return a, nil
|
|
}
|
|
|
|
func (s *service) FindByFilterID(ctx context.Context, filterID int, active *bool, withClient bool) ([]*domain.Action, error) {
|
|
return s.repo.FindByFilterID(ctx, filterID, active, withClient)
|
|
}
|
|
|
|
func (s *service) Delete(ctx context.Context, req *domain.DeleteActionRequest) error {
|
|
return s.repo.Delete(ctx, req)
|
|
}
|
|
|
|
func (s *service) DeleteByFilterID(ctx context.Context, filterID int) error {
|
|
return s.repo.DeleteByFilterID(ctx, filterID)
|
|
}
|
|
|
|
func (s *service) ToggleEnabled(actionID int) error {
|
|
return s.repo.ToggleEnabled(actionID)
|
|
}
|