mirror of
https://github.com/idanoo/autobrr
synced 2025-07-25 01:39:13 +00:00
0391629862
* chore: update copyright year in license headers * Revert "chore: update copyright year in license headers" This reverts commit 3e58129c431b9a491089ce36b908f9bb6ba38ed3. * chore: update copyright year in license headers * fix: sort go imports * fix: add missing license headers
111 lines
3.3 KiB
Go
111 lines
3.3 KiB
Go
// Copyright (c) 2021 - 2025, 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) 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) (rejections []string, err 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) 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)
|
|
}
|