// Copyright (c) 2021 - 2025, Ludvig Lundgren and the autobrr contributors. // SPDX-License-Identifier: GPL-2.0-or-later package database import ( "context" "database/sql" "encoding/json" "github.com/autobrr/autobrr/internal/domain" "github.com/autobrr/autobrr/internal/logger" "github.com/autobrr/autobrr/pkg/errors" sq "github.com/Masterminds/squirrel" "github.com/rs/zerolog" ) type ActionRepo struct { log zerolog.Logger db *DB clientRepo domain.DownloadClientRepo } func NewActionRepo(log logger.Logger, db *DB, clientRepo domain.DownloadClientRepo) domain.ActionRepo { return &ActionRepo{ log: log.With().Str("repo", "action").Logger(), db: db, clientRepo: clientRepo, } } func (r *ActionRepo) FindByFilterID(ctx context.Context, filterID int, active *bool, withClient bool) ([]*domain.Action, error) { if withClient { return r.findByFilterIDWithClient(ctx, filterID, active) } return r.findByFilterID(ctx, filterID, active) } func (r *ActionRepo) findByFilterID(ctx context.Context, filterID int, active *bool) ([]*domain.Action, error) { queryBuilder := r.db.squirrel. Select( "a.id", "a.name", "a.type", "a.enabled", "a.exec_cmd", "a.exec_args", "a.watch_folder", "a.category", "a.tags", "a.label", "a.save_path", "a.paused", "a.ignore_rules", "a.first_last_piece_prio", "a.skip_hash_check", "a.content_layout", "a.priority", "a.limit_download_speed", "a.limit_upload_speed", "a.limit_ratio", "a.limit_seed_time", "a.reannounce_skip", "a.reannounce_delete", "a.reannounce_interval", "a.reannounce_max_attempts", "a.webhook_host", "a.webhook_type", "a.webhook_method", "a.webhook_data", "a.external_client_id", "a.external_client", "a.client_id", ). From("action a"). Where(sq.Eq{"a.filter_id": filterID}) if active != nil { queryBuilder = queryBuilder.Where(sq.Eq{"enabled": *active}) } query, args, err := queryBuilder.ToSql() if err != nil { return nil, errors.Wrap(err, "error building query") } rows, err := r.db.handler.QueryContext(ctx, query, args...) if err != nil { return nil, errors.Wrap(err, "error executing query") } defer rows.Close() actions := make([]*domain.Action, 0) for rows.Next() { var a domain.Action var execCmd, execArgs, watchFolder, category, tags, label, savePath, contentLayout, priorityLayout, webhookHost, webhookType, webhookMethod, webhookData, externalClient sql.NullString var limitUl, limitDl, limitSeedTime sql.NullInt64 var limitRatio sql.NullFloat64 var externalClientID, clientID sql.NullInt32 var paused, ignoreRules sql.NullBool if err := rows.Scan(&a.ID, &a.Name, &a.Type, &a.Enabled, &execCmd, &execArgs, &watchFolder, &category, &tags, &label, &savePath, &paused, &ignoreRules, &a.FirstLastPiecePrio, &a.SkipHashCheck, &contentLayout, &priorityLayout, &limitDl, &limitUl, &limitRatio, &limitSeedTime, &a.ReAnnounceSkip, &a.ReAnnounceDelete, &a.ReAnnounceInterval, &a.ReAnnounceMaxAttempts, &webhookHost, &webhookType, &webhookMethod, &webhookData, &externalClientID, &externalClient, &clientID); err != nil { return nil, errors.Wrap(err, "error scanning row") } a.ExecCmd = execCmd.String a.ExecArgs = execArgs.String a.WatchFolder = watchFolder.String a.Category = category.String a.Tags = tags.String a.Label = label.String a.SavePath = savePath.String a.Paused = paused.Bool a.IgnoreRules = ignoreRules.Bool a.ContentLayout = domain.ActionContentLayout(contentLayout.String) a.PriorityLayout = domain.PriorityLayout(priorityLayout.String) a.LimitDownloadSpeed = limitDl.Int64 a.LimitUploadSpeed = limitUl.Int64 a.LimitRatio = limitRatio.Float64 a.LimitSeedTime = limitSeedTime.Int64 a.WebhookHost = webhookHost.String a.WebhookType = webhookType.String a.WebhookMethod = webhookMethod.String a.WebhookData = webhookData.String a.ExternalDownloadClientID = externalClientID.Int32 a.ExternalDownloadClient = externalClient.String a.ClientID = clientID.Int32 actions = append(actions, &a) } if err := rows.Err(); err != nil { return nil, errors.Wrap(err, "row error") } return actions, nil } func (r *ActionRepo) findByFilterIDWithClient(ctx context.Context, filterID int, active *bool) ([]*domain.Action, error) { queryBuilder := r.db.squirrel. Select( "a.id", "a.name", "a.type", "a.enabled", "a.exec_cmd", "a.exec_args", "a.watch_folder", "a.category", "a.tags", "a.label", "a.save_path", "a.paused", "a.ignore_rules", "a.first_last_piece_prio", "a.skip_hash_check", "a.content_layout", "a.priority", "a.limit_download_speed", "a.limit_upload_speed", "a.limit_ratio", "a.limit_seed_time", "a.reannounce_skip", "a.reannounce_delete", "a.reannounce_interval", "a.reannounce_max_attempts", "a.webhook_host", "a.webhook_type", "a.webhook_method", "a.webhook_data", "a.external_client_id", "a.external_client", "a.client_id", "c.id", "c.name", "c.type", "c.enabled", "c.host", "c.port", "c.tls", "c.tls_skip_verify", "c.username", "c.password", "c.settings", ). From("action a"). Join("client c ON a.client_id = c.id"). Where(sq.Eq{"a.filter_id": filterID}) if active != nil { queryBuilder = queryBuilder.Where(sq.Eq{"enabled": *active}) } query, args, err := queryBuilder.ToSql() if err != nil { return nil, errors.Wrap(err, "error building query") } rows, err := r.db.handler.QueryContext(ctx, query, args...) if err != nil { return nil, errors.Wrap(err, "error executing query") } defer rows.Close() actions := make([]*domain.Action, 0) for rows.Next() { var a domain.Action var c domain.DownloadClient var execCmd, execArgs, watchFolder, category, tags, label, savePath, contentLayout, priorityLayout, webhookHost, webhookType, webhookMethod, webhookData, externalClient sql.NullString var limitUl, limitDl, limitSeedTime sql.NullInt64 var limitRatio sql.NullFloat64 var externalClientID, clientID sql.NullInt32 var paused, ignoreRules sql.NullBool var clientClientId, clientPort sql.Null[int32] var clientName, clientType, clientHost, clientUsername, clientPassword, clientSettings sql.Null[string] var clientEnabled, clientTLS, clientTLSSkip sql.Null[bool] if err := rows.Scan(&a.ID, &a.Name, &a.Type, &a.Enabled, &execCmd, &execArgs, &watchFolder, &category, &tags, &label, &savePath, &paused, &ignoreRules, &a.FirstLastPiecePrio, &a.SkipHashCheck, &contentLayout, &priorityLayout, &limitDl, &limitUl, &limitRatio, &limitSeedTime, &a.ReAnnounceSkip, &a.ReAnnounceDelete, &a.ReAnnounceInterval, &a.ReAnnounceMaxAttempts, &webhookHost, &webhookType, &webhookMethod, &webhookData, &externalClientID, &externalClient, &clientID, &clientClientId, &clientName, &clientType, &clientEnabled, &clientHost, &clientPort, &clientTLS, &clientTLSSkip, &clientUsername, &clientPassword, &clientSettings); err != nil { return nil, errors.Wrap(err, "error scanning row") } a.ExecCmd = execCmd.String a.ExecArgs = execArgs.String a.WatchFolder = watchFolder.String a.Category = category.String a.Tags = tags.String a.Label = label.String a.SavePath = savePath.String a.Paused = paused.Bool a.IgnoreRules = ignoreRules.Bool a.ContentLayout = domain.ActionContentLayout(contentLayout.String) a.PriorityLayout = domain.PriorityLayout(priorityLayout.String) a.LimitDownloadSpeed = limitDl.Int64 a.LimitUploadSpeed = limitUl.Int64 a.LimitRatio = limitRatio.Float64 a.LimitSeedTime = limitSeedTime.Int64 a.WebhookHost = webhookHost.String a.WebhookType = webhookType.String a.WebhookMethod = webhookMethod.String a.WebhookData = webhookData.String a.ExternalDownloadClientID = externalClientID.Int32 a.ExternalDownloadClient = externalClient.String a.ClientID = clientID.Int32 c.ID = clientClientId.V c.Name = clientName.V c.Type = domain.DownloadClientType(clientType.V) c.Enabled = clientEnabled.V c.Host = clientHost.V c.Port = int(clientPort.V) c.TLS = clientTLS.V c.TLSSkipVerify = clientTLSSkip.V c.Username = clientUsername.V c.Password = clientPassword.V //c.Settings = clientSettings.String if a.ClientID > 0 { if clientSettings.Valid { if err := json.Unmarshal([]byte(clientSettings.V), &c.Settings); err != nil { return nil, errors.Wrap(err, "could not unmarshal download client settings: %v", clientSettings.V) } } a.Client = &c } actions = append(actions, &a) } if err := rows.Err(); err != nil { return nil, errors.Wrap(err, "row error") } return actions, nil } func (r *ActionRepo) FindByFilterIDTx(ctx context.Context, filterID int, active *bool) ([]*domain.Action, error) { tx, err := r.db.BeginTx(ctx, &sql.TxOptions{Isolation: sql.LevelReadCommitted}) if err != nil { return nil, err } defer tx.Rollback() actions, err := r.findByFilterIDTx(ctx, tx, filterID, active) if err != nil { return nil, err } for _, action := range actions { if action.ClientID > 0 { client, err := r.attachDownloadClient(ctx, tx, action.ClientID) if err != nil { return nil, err } if client != nil { action.Client = client } } } return actions, nil } func (r *ActionRepo) findByFilterIDTx(ctx context.Context, tx *Tx, filterID int, active *bool) ([]*domain.Action, error) { queryBuilder := r.db.squirrel. Select( "id", "name", "type", "enabled", "exec_cmd", "exec_args", "watch_folder", "category", "tags", "label", "save_path", "paused", "ignore_rules", "first_last_piece_prio", "skip_hash_check", "content_layout", "priority", "limit_download_speed", "limit_upload_speed", "limit_ratio", "limit_seed_time", "reannounce_skip", "reannounce_delete", "reannounce_interval", "reannounce_max_attempts", "webhook_host", "webhook_type", "webhook_method", "webhook_data", "external_client_id", "external_client", "client_id", ). From("action"). Where(sq.Eq{"filter_id": filterID}) if active != nil { queryBuilder = queryBuilder.Where(sq.Eq{"enabled": *active}) } query, args, err := queryBuilder.ToSql() if err != nil { return nil, errors.Wrap(err, "error building query") } rows, err := tx.QueryContext(ctx, query, args...) if err != nil { return nil, errors.Wrap(err, "error executing query") } defer rows.Close() actions := make([]*domain.Action, 0) for rows.Next() { var a domain.Action var execCmd, execArgs, watchFolder, category, tags, label, savePath, contentLayout, priorityLayout, webhookHost, webhookType, webhookMethod, webhookData, externalClient sql.NullString var limitUl, limitDl, limitSeedTime sql.NullInt64 var limitRatio sql.NullFloat64 var externalClientID, clientID sql.NullInt32 var paused, ignoreRules sql.NullBool if err := rows.Scan(&a.ID, &a.Name, &a.Type, &a.Enabled, &execCmd, &execArgs, &watchFolder, &category, &tags, &label, &savePath, &paused, &ignoreRules, &a.FirstLastPiecePrio, &a.SkipHashCheck, &contentLayout, &priorityLayout, &limitDl, &limitUl, &limitRatio, &limitSeedTime, &a.ReAnnounceSkip, &a.ReAnnounceDelete, &a.ReAnnounceInterval, &a.ReAnnounceMaxAttempts, &webhookHost, &webhookType, &webhookMethod, &webhookData, &externalClientID, &externalClient, &clientID); err != nil { return nil, errors.Wrap(err, "error scanning row") } a.ExecCmd = execCmd.String a.ExecArgs = execArgs.String a.WatchFolder = watchFolder.String a.Category = category.String a.Tags = tags.String a.Label = label.String a.SavePath = savePath.String a.Paused = paused.Bool a.IgnoreRules = ignoreRules.Bool a.ContentLayout = domain.ActionContentLayout(contentLayout.String) a.PriorityLayout = domain.PriorityLayout(priorityLayout.String) a.LimitDownloadSpeed = limitDl.Int64 a.LimitUploadSpeed = limitUl.Int64 a.LimitRatio = limitRatio.Float64 a.LimitSeedTime = limitSeedTime.Int64 a.WebhookHost = webhookHost.String a.WebhookType = webhookType.String a.WebhookMethod = webhookMethod.String a.WebhookData = webhookData.String a.ExternalDownloadClientID = externalClientID.Int32 a.ExternalDownloadClient = externalClient.String a.ClientID = clientID.Int32 actions = append(actions, &a) } if err := rows.Err(); err != nil { return nil, errors.Wrap(err, "row error") } return actions, nil } func (r *ActionRepo) attachDownloadClient(ctx context.Context, tx *Tx, clientID int32) (*domain.DownloadClient, error) { queryBuilder := r.db.squirrel. Select( "id", "name", "type", "enabled", "host", "port", "tls", "tls_skip_verify", "username", "password", "settings", ). From("client"). Where(sq.Eq{"id": clientID}) query, args, err := queryBuilder.ToSql() if err != nil { return nil, errors.Wrap(err, "error building query") } row := tx.QueryRowContext(ctx, query, args...) if err := row.Err(); err != nil { return nil, errors.Wrap(err, "error executing query") } var client domain.DownloadClient 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 errors.Is(err, sql.ErrNoRows) { r.log.Warn().Msgf("no download client with id %d", clientID) return nil, domain.ErrRecordNotFound } return nil, errors.Wrap(err, "error scanning row") } if settingsJsonStr != "" { if err := json.Unmarshal([]byte(settingsJsonStr), &client.Settings); err != nil { return nil, errors.Wrap(err, "could not unmarshal download client settings: %v", settingsJsonStr) } } return &client, nil } func (r *ActionRepo) List(ctx context.Context) ([]domain.Action, error) { queryBuilder := r.db.squirrel. Select( "id", "name", "type", "enabled", "exec_cmd", "exec_args", "watch_folder", "category", "tags", "label", "save_path", "paused", "ignore_rules", "first_last_piece_prio", "skip_hash_check", "content_layout", "priority", "limit_download_speed", "limit_upload_speed", "limit_ratio", "limit_seed_time", "reannounce_skip", "reannounce_delete", "reannounce_interval", "reannounce_max_attempts", "webhook_host", "webhook_type", "webhook_method", "webhook_data", "external_client_id", "external_client", "client_id", ). From("action") query, args, err := queryBuilder.ToSql() if err != nil { return nil, errors.Wrap(err, "error building query") } rows, err := r.db.handler.QueryContext(ctx, query, args...) if err != nil { return nil, errors.Wrap(err, "error executing query") } defer rows.Close() actions := make([]domain.Action, 0) for rows.Next() { var a domain.Action var execCmd, execArgs, watchFolder, category, tags, label, savePath, contentLayout, priorityLayout, webhookHost, webhookType, webhookMethod, webhookData, externalClient sql.NullString var limitUl, limitDl, limitSeedTime sql.NullInt64 var limitRatio sql.NullFloat64 var externalClientID, clientID sql.NullInt32 var paused, ignoreRules sql.NullBool if err := rows.Scan(&a.ID, &a.Name, &a.Type, &a.Enabled, &execCmd, &execArgs, &watchFolder, &category, &tags, &label, &savePath, &paused, &ignoreRules, &a.FirstLastPiecePrio, &a.SkipHashCheck, &contentLayout, &priorityLayout, &limitDl, &limitUl, &limitRatio, &limitSeedTime, &a.ReAnnounceSkip, &a.ReAnnounceDelete, &a.ReAnnounceInterval, &a.ReAnnounceMaxAttempts, &webhookHost, &webhookType, &webhookMethod, &webhookData, &externalClientID, &externalClient, &clientID); err != nil { return nil, errors.Wrap(err, "error scanning row") } a.Category = category.String a.Tags = tags.String a.Label = label.String a.SavePath = savePath.String a.Paused = paused.Bool a.IgnoreRules = ignoreRules.Bool a.ContentLayout = domain.ActionContentLayout(contentLayout.String) a.PriorityLayout = domain.PriorityLayout(priorityLayout.String) a.LimitDownloadSpeed = limitDl.Int64 a.LimitUploadSpeed = limitUl.Int64 a.LimitRatio = limitRatio.Float64 a.LimitSeedTime = limitSeedTime.Int64 a.WebhookHost = webhookHost.String a.WebhookType = webhookType.String a.WebhookMethod = webhookMethod.String a.WebhookData = webhookData.String a.ExternalDownloadClientID = externalClientID.Int32 a.ExternalDownloadClient = externalClient.String a.ClientID = clientID.Int32 actions = append(actions, a) if err := rows.Err(); err != nil { return nil, errors.Wrap(err, "rows error") } } return actions, nil } func (r *ActionRepo) Get(ctx context.Context, req *domain.GetActionRequest) (*domain.Action, error) { queryBuilder := r.db.squirrel. Select( "id", "name", "type", "enabled", "exec_cmd", "exec_args", "watch_folder", "category", "tags", "label", "save_path", "paused", "ignore_rules", "first_last_piece_prio", "skip_hash_check", "content_layout", "priority", "limit_download_speed", "limit_upload_speed", "limit_ratio", "limit_seed_time", "reannounce_skip", "reannounce_delete", "reannounce_interval", "reannounce_max_attempts", "webhook_host", "webhook_type", "webhook_method", "webhook_data", "external_client_id", "external_client", "client_id", "filter_id", ). From("action"). Where(sq.Eq{"id": req.Id}) query, args, err := queryBuilder.ToSql() if err != nil { return nil, errors.Wrap(err, "error building query") } row := r.db.handler.QueryRowContext(ctx, query, args...) if err := row.Err(); err != nil { return nil, errors.Wrap(err, "error executing query") } var a domain.Action var execCmd, execArgs, watchFolder, category, tags, label, savePath, contentLayout, priorityLayout, webhookHost, webhookType, webhookMethod, webhookData, externalClient sql.NullString var limitUl, limitDl, limitSeedTime sql.NullInt64 var limitRatio sql.NullFloat64 var externalClientID, clientID, filterID sql.NullInt32 var paused, ignoreRules sql.NullBool if err := row.Scan(&a.ID, &a.Name, &a.Type, &a.Enabled, &execCmd, &execArgs, &watchFolder, &category, &tags, &label, &savePath, &paused, &ignoreRules, &a.FirstLastPiecePrio, &a.SkipHashCheck, &contentLayout, &priorityLayout, &limitDl, &limitUl, &limitRatio, &limitSeedTime, &a.ReAnnounceSkip, &a.ReAnnounceDelete, &a.ReAnnounceInterval, &a.ReAnnounceMaxAttempts, &webhookHost, &webhookType, &webhookMethod, &webhookData, &externalClientID, &externalClient, &clientID, &filterID); err != nil { if errors.Is(err, sql.ErrNoRows) { return nil, domain.ErrRecordNotFound } return nil, errors.Wrap(err, "error scanning row") } a.ExecCmd = execCmd.String a.ExecArgs = execArgs.String a.WatchFolder = watchFolder.String a.Category = category.String a.Tags = tags.String a.Label = label.String a.SavePath = savePath.String a.Paused = paused.Bool a.IgnoreRules = ignoreRules.Bool a.ContentLayout = domain.ActionContentLayout(contentLayout.String) a.PriorityLayout = domain.PriorityLayout(priorityLayout.String) a.LimitDownloadSpeed = limitDl.Int64 a.LimitUploadSpeed = limitUl.Int64 a.LimitRatio = limitRatio.Float64 a.LimitSeedTime = limitSeedTime.Int64 a.WebhookHost = webhookHost.String a.WebhookType = webhookType.String a.WebhookMethod = webhookMethod.String a.WebhookData = webhookData.String a.ExternalDownloadClientID = externalClientID.Int32 a.ExternalDownloadClient = externalClient.String a.ClientID = clientID.Int32 a.FilterID = int(filterID.Int32) return &a, nil } func (r *ActionRepo) Delete(ctx context.Context, req *domain.DeleteActionRequest) error { queryBuilder := r.db.squirrel. Delete("action"). Where(sq.Eq{"id": req.ActionId}) query, args, err := queryBuilder.ToSql() if err != nil { return errors.Wrap(err, "error building query") } _, err = r.db.handler.ExecContext(ctx, query, args...) if err != nil { return errors.Wrap(err, "error executing query") } r.log.Debug().Msgf("action.delete: %v", req.ActionId) return nil } func (r *ActionRepo) DeleteByFilterID(ctx context.Context, filterID int) error { queryBuilder := r.db.squirrel. Delete("action"). Where(sq.Eq{"filter_id": filterID}) query, args, err := queryBuilder.ToSql() if err != nil { return errors.Wrap(err, "error building query") } _, err = r.db.handler.ExecContext(ctx, query, args...) if err != nil { return errors.Wrap(err, "error executing query") } r.log.Debug().Msgf("action.deleteByFilterID: %v", filterID) return nil } func (r *ActionRepo) Store(ctx context.Context, action *domain.Action) error { queryBuilder := r.db.squirrel. Insert("action"). Columns( "name", "type", "enabled", "exec_cmd", "exec_args", "watch_folder", "category", "tags", "label", "save_path", "paused", "ignore_rules", "first_last_piece_prio", "skip_hash_check", "content_layout", "priority", "limit_upload_speed", "limit_download_speed", "limit_ratio", "limit_seed_time", "reannounce_skip", "reannounce_delete", "reannounce_interval", "reannounce_max_attempts", "webhook_host", "webhook_type", "webhook_method", "webhook_data", "external_client_id", "external_client", "client_id", "filter_id", ). Values( action.Name, action.Type, action.Enabled, toNullString(action.ExecCmd), toNullString(action.ExecArgs), toNullString(action.WatchFolder), toNullString(action.Category), toNullString(action.Tags), toNullString(action.Label), toNullString(action.SavePath), action.Paused, action.IgnoreRules, action.FirstLastPiecePrio, action.SkipHashCheck, toNullString(string(action.ContentLayout)), toNullString(string(action.PriorityLayout)), toNullInt64(action.LimitUploadSpeed), toNullInt64(action.LimitDownloadSpeed), toNullFloat64(action.LimitRatio), toNullInt64(action.LimitSeedTime), action.ReAnnounceSkip, action.ReAnnounceDelete, action.ReAnnounceInterval, action.ReAnnounceMaxAttempts, toNullString(action.WebhookHost), toNullString(action.WebhookType), toNullString(action.WebhookMethod), toNullString(action.WebhookData), toNullInt32(action.ExternalDownloadClientID), toNullString(action.ExternalDownloadClient), toNullInt32(action.ClientID), toNullInt32(int32(action.FilterID)), ). Suffix("RETURNING id").RunWith(r.db.handler) // return values var retID int64 if err := queryBuilder.QueryRowContext(ctx).Scan(&retID); err != nil { return errors.Wrap(err, "error executing query") } action.ID = int(retID) r.log.Debug().Msgf("action.store: added new %d", retID) return nil } func (r *ActionRepo) Update(ctx context.Context, action domain.Action) (*domain.Action, error) { queryBuilder := r.db.squirrel. Update("action"). Set("name", action.Name). Set("type", action.Type). Set("enabled", action.Enabled). Set("exec_cmd", toNullString(action.ExecCmd)). Set("exec_args", toNullString(action.ExecArgs)). Set("watch_folder", toNullString(action.WatchFolder)). Set("category", toNullString(action.Category)). Set("tags", toNullString(action.Tags)). Set("label", toNullString(action.Label)). Set("save_path", toNullString(action.SavePath)). Set("paused", action.Paused). Set("ignore_rules", action.IgnoreRules). Set("first_last_piece_prio", action.FirstLastPiecePrio). Set("skip_hash_check", action.SkipHashCheck). Set("content_layout", toNullString(string(action.ContentLayout))). Set("priority", toNullString(string(action.PriorityLayout))). Set("limit_upload_speed", toNullInt64(action.LimitUploadSpeed)). Set("limit_download_speed", toNullInt64(action.LimitDownloadSpeed)). Set("limit_ratio", toNullFloat64(action.LimitRatio)). Set("limit_seed_time", toNullInt64(action.LimitSeedTime)). Set("reannounce_skip", action.ReAnnounceSkip). Set("reannounce_delete", action.ReAnnounceDelete). Set("reannounce_interval", action.ReAnnounceInterval). Set("reannounce_max_attempts", action.ReAnnounceMaxAttempts). Set("webhook_host", toNullString(action.WebhookHost)). Set("webhook_type", toNullString(action.WebhookType)). Set("webhook_method", toNullString(action.WebhookMethod)). Set("webhook_data", toNullString(action.WebhookData)). Set("external_client_id", toNullInt32(action.ExternalDownloadClientID)). Set("external_client", toNullString(action.ExternalDownloadClient)). Set("client_id", toNullInt32(action.ClientID)). Set("filter_id", toNullInt32(int32(action.FilterID))). Where(sq.Eq{"id": action.ID}) query, args, err := queryBuilder.ToSql() if err != nil { return nil, errors.Wrap(err, "error building query") } if _, err := r.db.handler.ExecContext(ctx, query, args...); err != nil { return nil, errors.Wrap(err, "error executing query") } r.log.Debug().Msgf("action.update: %v", action.ID) return &action, nil } func (r *ActionRepo) StoreFilterActions(ctx context.Context, filterID int64, actions []*domain.Action) ([]*domain.Action, error) { tx, err := r.db.handler.BeginTx(ctx, nil) if err != nil { return nil, errors.Wrap(err, "error begin transaction") } defer tx.Rollback() for _, action := range actions { action := action if action.ID > 0 { queryBuilder := r.db.squirrel. Update("action"). Set("name", action.Name). Set("type", action.Type). Set("enabled", action.Enabled). Set("exec_cmd", toNullString(action.ExecCmd)). Set("exec_args", toNullString(action.ExecArgs)). Set("watch_folder", toNullString(action.WatchFolder)). Set("category", toNullString(action.Category)). Set("tags", toNullString(action.Tags)). Set("label", toNullString(action.Label)). Set("save_path", toNullString(action.SavePath)). Set("paused", action.Paused). Set("ignore_rules", action.IgnoreRules). Set("first_last_piece_prio", action.FirstLastPiecePrio). Set("skip_hash_check", action.SkipHashCheck). Set("content_layout", toNullString(string(action.ContentLayout))). Set("priority", toNullString(string(action.PriorityLayout))). Set("limit_upload_speed", toNullInt64(action.LimitUploadSpeed)). Set("limit_download_speed", toNullInt64(action.LimitDownloadSpeed)). Set("limit_ratio", toNullFloat64(action.LimitRatio)). Set("limit_seed_time", toNullInt64(action.LimitSeedTime)). Set("reannounce_skip", action.ReAnnounceSkip). Set("reannounce_delete", action.ReAnnounceDelete). Set("reannounce_interval", action.ReAnnounceInterval). Set("reannounce_max_attempts", action.ReAnnounceMaxAttempts). Set("webhook_host", toNullString(action.WebhookHost)). Set("webhook_type", toNullString(action.WebhookType)). Set("webhook_method", toNullString(action.WebhookMethod)). Set("webhook_data", toNullString(action.WebhookData)). Set("external_client_id", toNullInt32(action.ExternalDownloadClientID)). Set("external_client", toNullString(action.ExternalDownloadClient)). Set("client_id", toNullInt32(action.ClientID)). Set("filter_id", toNullInt64(filterID)). Where(sq.Eq{"id": action.ID}) query, args, err := queryBuilder.ToSql() if err != nil { return nil, errors.Wrap(err, "error building query") } if _, err := tx.ExecContext(ctx, query, args...); err != nil { return nil, errors.Wrap(err, "error executing query") } r.log.Trace().Msgf("action.StoreFilterActions: update %d", action.ID) } else { queryBuilder := r.db.squirrel. Insert("action"). Columns( "name", "type", "enabled", "exec_cmd", "exec_args", "watch_folder", "category", "tags", "label", "save_path", "paused", "ignore_rules", "first_last_piece_prio", "skip_hash_check", "content_layout", "priority", "limit_upload_speed", "limit_download_speed", "limit_ratio", "limit_seed_time", "reannounce_skip", "reannounce_delete", "reannounce_interval", "reannounce_max_attempts", "webhook_host", "webhook_type", "webhook_method", "webhook_data", "external_client_id", "external_client", "client_id", "filter_id", ). Values( action.Name, action.Type, action.Enabled, toNullString(action.ExecCmd), toNullString(action.ExecArgs), toNullString(action.WatchFolder), toNullString(action.Category), toNullString(action.Tags), toNullString(action.Label), toNullString(action.SavePath), action.Paused, action.IgnoreRules, action.FirstLastPiecePrio, action.SkipHashCheck, toNullString(string(action.ContentLayout)), toNullString(string(action.PriorityLayout)), toNullInt64(action.LimitUploadSpeed), toNullInt64(action.LimitDownloadSpeed), toNullFloat64(action.LimitRatio), toNullInt64(action.LimitSeedTime), action.ReAnnounceSkip, action.ReAnnounceDelete, action.ReAnnounceInterval, action.ReAnnounceMaxAttempts, toNullString(action.WebhookHost), toNullString(action.WebhookType), toNullString(action.WebhookMethod), toNullString(action.WebhookData), toNullInt32(action.ExternalDownloadClientID), toNullString(action.ExternalDownloadClient), toNullInt32(action.ClientID), toNullInt64(filterID), ). Suffix("RETURNING id").RunWith(tx) // return values var retID int if err := queryBuilder.QueryRowContext(ctx).Scan(&retID); err != nil { return nil, errors.Wrap(err, "error executing query") } action.ID = retID r.log.Trace().Msgf("action.StoreFilterActions: store %d", action.ID) } r.log.Debug().Msgf("action.StoreFilterActions: store '%s' type: '%v' on filter: %d", action.Name, action.Type, filterID) } if err := tx.Commit(); err != nil { return nil, errors.Wrap(err, "error updating filter actions") } return actions, nil } func (r *ActionRepo) ToggleEnabled(actionID int) error { queryBuilder := r.db.squirrel. Update("action"). Set("enabled", sq.Expr("NOT enabled")). Where(sq.Eq{"id": actionID}) query, args, err := queryBuilder.ToSql() if err != nil { return errors.Wrap(err, "error building query") } result, err := r.db.handler.Exec(query, args...) if err != nil { return errors.Wrap(err, "error executing query") } if rowsAffected, err := result.RowsAffected(); err != nil { return errors.Wrap(err, "error getting rows affected") } else if rowsAffected == 0 { return domain.ErrRecordNotFound } r.log.Debug().Msgf("action.toggleEnabled: %v", actionID) return nil }