summaryrefslogtreecommitdiffstats
path: root/internal/storage/deletions.go
blob: 1a07e3cebe8d5d7f3526b7e47b7ad6555ba6b487 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
package storage

import (
	"context"
	"fmt"

	pb "northwest.io/nostr-grpc/api/nostr/v1"
)

const KindDeletion = 5

func (s *Storage) ProcessDeletion(ctx context.Context, deletionEvent *pb.Event) error {
	if deletionEvent.Kind != KindDeletion {
		return fmt.Errorf("not a deletion event")
	}

	var eventIDsToDelete []string
	for _, tag := range deletionEvent.Tags {
		if len(tag.Values) >= 2 && tag.Values[0] == "e" {
			eventIDsToDelete = append(eventIDsToDelete, tag.Values[1])
		}
	}

	if len(eventIDsToDelete) == 0 {
		return nil
	}

	tx, err := s.db.BeginTx(ctx, nil)
	if err != nil {
		return fmt.Errorf("failed to begin transaction: %w", err)
	}
	defer tx.Rollback()

	for _, eventID := range eventIDsToDelete {
		var eventPubkey string
		err := tx.QueryRowContext(ctx,
			"SELECT pubkey FROM events WHERE id = ? AND deleted = 0",
			eventID,
		).Scan(&eventPubkey)

		if err != nil {
			continue
		}

		if eventPubkey != deletionEvent.Pubkey {
			continue
		}

		_, err = tx.ExecContext(ctx,
			"UPDATE events SET deleted = 1 WHERE id = ?",
			eventID,
		)
		if err != nil {
			return fmt.Errorf("failed to mark event as deleted: %w", err)
		}

		_, err = tx.ExecContext(ctx,
			"INSERT OR IGNORE INTO deletions (deleted_event_id, deletion_event_id, pubkey, created_at) VALUES (?, ?, ?, ?)",
			eventID,
			deletionEvent.Id,
			deletionEvent.Pubkey,
			deletionEvent.CreatedAt,
		)
		if err != nil {
			return fmt.Errorf("failed to record deletion: %w", err)
		}
	}

	return tx.Commit()
}