mirror of
https://github.com/idanoo/autobrr
synced 2025-07-22 16:29:12 +00:00
fix(downloadclients): remove from filter actions on delete (#891)
fix(downloadclients): properly delete from filter actions
This commit is contained in:
parent
91e4f5b435
commit
64900c4d56
3 changed files with 59 additions and 6 deletions
|
@ -41,12 +41,15 @@ func (r *ActionRepo) FindByFilterID(ctx context.Context, filterID int) ([]*domai
|
|||
}
|
||||
|
||||
for _, action := range actions {
|
||||
if action.ClientID != 0 {
|
||||
if action.ClientID > 0 {
|
||||
client, err := r.attachDownloadClient(ctx, tx, action.ClientID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
action.Client = *client
|
||||
|
||||
if client != nil {
|
||||
action.Client = client
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -180,6 +183,10 @@ func (r *ActionRepo) attachDownloadClient(ctx context.Context, tx *Tx, clientID
|
|||
var settingsJsonStr string
|
||||
|
||||
if err := row.Scan(&client.ID, &client.Name, &client.Type, &client.Enabled, &client.Host, &client.Port, &client.TLS, &client.TLSSkipVerify, &client.Username, &client.Password, &settingsJsonStr); err != nil {
|
||||
if err == sql.ErrNoRows {
|
||||
r.log.Warn().Msgf("no download client with id %d", clientID)
|
||||
return nil, nil
|
||||
}
|
||||
return nil, errors.Wrap(err, "error scanning row")
|
||||
}
|
||||
|
||||
|
@ -608,7 +615,6 @@ func (r *ActionRepo) StoreFilterActions(ctx context.Context, actions []*domain.A
|
|||
|
||||
if err := tx.Commit(); err != nil {
|
||||
return nil, errors.Wrap(err, "error updating filter actions")
|
||||
|
||||
}
|
||||
|
||||
return actions, nil
|
||||
|
|
|
@ -254,6 +254,31 @@ func (r *DownloadClientRepo) Update(ctx context.Context, client domain.DownloadC
|
|||
}
|
||||
|
||||
func (r *DownloadClientRepo) Delete(ctx context.Context, clientID int) error {
|
||||
tx, err := r.db.BeginTx(ctx, &sql.TxOptions{Isolation: sql.LevelWriteCommitted})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
defer tx.Rollback()
|
||||
|
||||
if err := r.delete(ctx, tx, clientID); err != nil {
|
||||
return errors.Wrap(err, "error deleting download client: %d", clientID)
|
||||
}
|
||||
|
||||
if err := r.deleteClientFromAction(ctx, tx, clientID); err != nil {
|
||||
return errors.Wrap(err, "error deleting download client: %d", clientID)
|
||||
}
|
||||
|
||||
if err := tx.Commit(); err != nil {
|
||||
return errors.Wrap(err, "error deleting download client: %d", clientID)
|
||||
}
|
||||
|
||||
r.log.Info().Msgf("delete download client: %d", clientID)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *DownloadClientRepo) delete(ctx context.Context, tx *Tx, clientID int) error {
|
||||
queryBuilder := r.db.squirrel.
|
||||
Delete("client").
|
||||
Where(sq.Eq{"id": clientID})
|
||||
|
@ -263,7 +288,7 @@ func (r *DownloadClientRepo) Delete(ctx context.Context, clientID int) error {
|
|||
return errors.Wrap(err, "error building query")
|
||||
}
|
||||
|
||||
res, err := r.db.handler.ExecContext(ctx, query, args...)
|
||||
res, err := tx.ExecContext(ctx, query, args...)
|
||||
if err != nil {
|
||||
return errors.Wrap(err, "error executing query")
|
||||
}
|
||||
|
@ -276,7 +301,29 @@ func (r *DownloadClientRepo) Delete(ctx context.Context, clientID int) error {
|
|||
return errors.New("no rows affected")
|
||||
}
|
||||
|
||||
r.log.Info().Msgf("delete download client: %d", clientID)
|
||||
r.log.Debug().Msgf("delete download client: %d", clientID)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *DownloadClientRepo) deleteClientFromAction(ctx context.Context, tx *Tx, clientID int) error {
|
||||
var err error
|
||||
|
||||
queryBuilder := r.db.squirrel.
|
||||
Update("action").
|
||||
Set("enabled", false).
|
||||
Set("client_id", 0).
|
||||
Where(sq.Eq{"client_id": clientID}).
|
||||
Suffix("RETURNING filter_id").RunWith(tx)
|
||||
|
||||
// return values
|
||||
var filterID int
|
||||
|
||||
if err = queryBuilder.QueryRowContext(ctx).Scan(&filterID); err != nil {
|
||||
return errors.Wrap(err, "error executing query")
|
||||
}
|
||||
|
||||
r.log.Debug().Msgf("deleting download client %d from action for filter %d", clientID, filterID)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
|
|
@ -49,7 +49,7 @@ type Action struct {
|
|||
WebhookHeaders []string `json:"webhook_headers,omitempty"`
|
||||
FilterID int `json:"filter_id,omitempty"`
|
||||
ClientID int32 `json:"client_id,omitempty"`
|
||||
Client DownloadClient `json:"client,omitempty"`
|
||||
Client *DownloadClient `json:"client,omitempty"`
|
||||
}
|
||||
|
||||
// ParseMacros parse all macros on action
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue