mirror of
https://github.com/idanoo/autobrr
synced 2025-07-23 16:59:12 +00:00
861f30c144
* fix(actions): error on disabled client * fix(actions): sql scan args * refactor: download client cache for actions * fix: tests client store * fix: tests client store and int conversion * fix: tests revert findbyid ctx timeout * fix: tests row.err * feat: add logging to download client cache
108 lines
3.1 KiB
Go
108 lines
3.1 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/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
|
|
bus EventBus.Bus
|
|
|
|
httpClient *http.Client
|
|
}
|
|
|
|
func NewService(log logger.Logger, repo domain.ActionRepo, clientSvc download_client.Service, bus EventBus.Bus) Service {
|
|
s := &service{
|
|
log: log.With().Str("module", "action").Logger(),
|
|
repo: repo,
|
|
clientSvc: clientSvc,
|
|
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)
|
|
}
|