vasyansk
477077d9e3
Error consistency: folder-level failures were counted only toward the task's done_with_errors status, not the account's error_count, so a task showed DONE_WITH_ERRORS while its only account showed 0 errors / DONE. Now folder errors increment the account counter and the account status becomes done_with_errors when errs>0. Visibility: persist accounts.last_error (migration 0002) so the failing folder / login error survives a page reload (shown red under the source login); cleared at the start of each run. Modal reset: the folder-mapping modal kept its selections across opens, so adding a second account showed the first account's mapping. It now mounts fresh per add (conditional render + key), reflecting the newly-probed folders. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01MMHQTtnQtQqL8muAXHr9kd
89 lines
2.8 KiB
Go
89 lines
2.8 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
)
|
|
|
|
type Account struct {
|
|
ID int64
|
|
TaskID int64
|
|
SrcLogin string
|
|
SrcPassEnc string
|
|
DstLogin string
|
|
DstPassEnc string
|
|
TestSrcStatus string
|
|
TestDstStatus string
|
|
Status string
|
|
Copied int64
|
|
Skipped int64
|
|
Errors int64
|
|
LastError string
|
|
}
|
|
|
|
func (s *Store) CreateAccount(ctx context.Context, a Account) (int64, error) {
|
|
var id int64
|
|
err := s.Pool.QueryRow(ctx,
|
|
`INSERT INTO accounts (task_id, src_login, src_pass_enc, dst_login, dst_pass_enc)
|
|
VALUES ($1,$2,$3,$4,$5) RETURNING id`,
|
|
a.TaskID, a.SrcLogin, a.SrcPassEnc, a.DstLogin, a.DstPassEnc).Scan(&id)
|
|
return id, err
|
|
}
|
|
|
|
// DeleteAccount removes one account (and its migrated_messages via ON DELETE CASCADE).
|
|
func (s *Store) DeleteAccount(ctx context.Context, id int64) error {
|
|
_, err := s.Pool.Exec(ctx, `DELETE FROM accounts WHERE id=$1`, id)
|
|
return err
|
|
}
|
|
|
|
func (s *Store) ListAccountsByTask(ctx context.Context, taskID int64) ([]Account, error) {
|
|
rows, err := s.Pool.Query(ctx,
|
|
`SELECT id, task_id, src_login, src_pass_enc, dst_login, dst_pass_enc,
|
|
test_src_status, test_dst_status, status, copied_count, skipped_count, error_count, last_error
|
|
FROM accounts WHERE task_id=$1 ORDER BY id`, taskID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
out := []Account{}
|
|
for rows.Next() {
|
|
var a Account
|
|
if err := rows.Scan(&a.ID, &a.TaskID, &a.SrcLogin, &a.SrcPassEnc, &a.DstLogin, &a.DstPassEnc,
|
|
&a.TestSrcStatus, &a.TestDstStatus, &a.Status, &a.Copied, &a.Skipped, &a.Errors, &a.LastError); err != nil {
|
|
return nil, err
|
|
}
|
|
out = append(out, a)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
// SetAccountError stores (or clears, with "") the last error message shown for
|
|
// an account, so it survives a page reload after the run's live log is gone.
|
|
func (s *Store) SetAccountError(ctx context.Context, id int64, msg string) error {
|
|
_, err := s.Pool.Exec(ctx, `UPDATE accounts SET last_error=$2 WHERE id=$1`, id, msg)
|
|
return err
|
|
}
|
|
|
|
// side = "src" | "dst"
|
|
func (s *Store) SetAccountTestStatus(ctx context.Context, id int64, side, status string) error {
|
|
col := "test_src_status"
|
|
if side == "dst" {
|
|
col = "test_dst_status"
|
|
}
|
|
_, err := s.Pool.Exec(ctx, fmt.Sprintf(`UPDATE accounts SET %s=$2 WHERE id=$1`, col), id, status)
|
|
return err
|
|
}
|
|
|
|
func (s *Store) SetAccountStatus(ctx context.Context, id int64, status string) error {
|
|
_, err := s.Pool.Exec(ctx, `UPDATE accounts SET status=$2 WHERE id=$1`, id, status)
|
|
return err
|
|
}
|
|
|
|
func (s *Store) IncAccountCounters(ctx context.Context, id, copied, skipped, errs int64) error {
|
|
_, err := s.Pool.Exec(ctx,
|
|
`UPDATE accounts SET copied_count=copied_count+$2,
|
|
skipped_count=skipped_count+$3, error_count=error_count+$4 WHERE id=$1`,
|
|
id, copied, skipped, errs)
|
|
return err
|
|
}
|