This repository has no description
1package appview
2
3import (
4 "bytes"
5 "context"
6 "database/sql"
7 "encoding/json"
8 "errors"
9 "fmt"
10 "io"
11 "log/slog"
12 "net/http"
13 "net/url"
14 "slices"
15 "strings"
16
17 "time"
18
19 "github.com/avast/retry-go/v4"
20 "github.com/bluesky-social/indigo/atproto/syntax"
21 jmodels "github.com/bluesky-social/jetstream/pkg/models"
22 "github.com/go-git/go-git/v5/plumbing"
23 "github.com/ipfs/go-cid"
24 "golang.org/x/sync/errgroup"
25 "tangled.org/core/api/tangled"
26 "tangled.org/core/appview/cache"
27 "tangled.org/core/appview/config"
28 "tangled.org/core/appview/db"
29 "tangled.org/core/appview/knotacl"
30 "tangled.org/core/appview/mentions"
31 "tangled.org/core/appview/models"
32 "tangled.org/core/appview/notify"
33 "tangled.org/core/appview/serververify"
34 "tangled.org/core/consts"
35 "tangled.org/core/idresolver"
36 "tangled.org/core/orm"
37 "tangled.org/core/rbac"
38 "tangled.org/core/repoverify"
39)
40
41type RepoPermissionChecker interface {
42 HasRepoPermissionErr(ctx context.Context, repo *models.Repo, userDid, perm string) (bool, error)
43}
44
45type Ingester struct {
46 Ctx context.Context
47 Db *db.DB
48 Enforcer *rbac.Enforcer
49 Acl RepoPermissionChecker
50 IdResolver *idresolver.Resolver
51 Cache *cache.Cache
52 Config *config.Config
53 Logger *slog.Logger
54 MentionsResolver *mentions.Resolver
55 Notifier notify.Notifier
56 Verifier repoverify.Verifier
57}
58
59type processFunc func(ctx context.Context, e *jmodels.Event) error
60
61func (i *Ingester) Ingest() processFunc {
62 return func(ctx context.Context, e *jmodels.Event) error {
63 var err error
64
65 l := i.Logger.With("kind", e.Kind)
66 switch e.Kind {
67 case jmodels.EventKindAccount:
68 // TODO: sync account state to db
69 if e.Account.Active {
70 break
71 }
72 // TODO: revoke sessions by DID
73 if *e.Account.Status == "deactivated" {
74 err = i.IdResolver.InvalidateIdent(ctx, e.Account.Did)
75 }
76 case jmodels.EventKindIdentity:
77 err = i.IdResolver.InvalidateIdent(ctx, e.Identity.Did)
78 case jmodels.EventKindCommit:
79 l = l.With(
80 "nsid", e.Commit.Collection,
81 "did", e.Did,
82 "rkey", e.Commit.RKey,
83 "op", e.Commit.Operation,
84 )
85 switch e.Commit.Collection {
86 case tangled.GraphFollowNSID:
87 err = i.ingestFollow(e, l)
88 case tangled.GraphVouchNSID:
89 err = i.ingestVouch(ctx, e, l)
90 case tangled.FeedStarNSID:
91 err = i.ingestStar(ctx, e, l)
92 case tangled.FeedReactionNSID:
93 err = i.ingestReaction(e, l)
94 case tangled.PublicKeyNSID:
95 err = i.ingestPublicKey(e, l)
96 case tangled.RepoArtifactNSID:
97 err = i.ingestArtifact(ctx, e, l)
98 case tangled.ActorProfileNSID:
99 err = i.ingestProfile(ctx, e, l)
100 case tangled.SpindleMemberNSID:
101 err = i.ingestSpindleMember(ctx, e, l)
102 case tangled.SpindleNSID:
103 err = i.ingestSpindle(ctx, e, l)
104 case tangled.KnotMemberNSID:
105 err = i.ingestKnotMember(ctx, e, l)
106 case tangled.KnotNSID:
107 err = i.ingestKnot(ctx, e, l)
108 case tangled.StringNSID:
109 err = i.ingestString(e, l)
110 case tangled.RepoIssueNSID:
111 err = i.ingestIssue(ctx, e, l)
112 case tangled.RepoIssueStateNSID:
113 err = i.ingestState(ctx, e, l, issueStateSpec)
114 case tangled.RepoPullNSID:
115 err = i.ingestPull(ctx, e, l)
116 case tangled.RepoPullStatusNSID:
117 err = i.ingestState(ctx, e, l, pullStatusSpec)
118 case tangled.FeedCommentNSID:
119 err = i.ingestComment(e, l)
120 case tangled.RepoIssueCommentNSID:
121 err = i.ingestIssueComment(e, l)
122 case tangled.RepoPullCommentNSID:
123 err = i.ingestPullComment(e, l)
124 case tangled.LabelDefinitionNSID:
125 err = i.ingestLabelDefinition(e, l)
126 case tangled.LabelOpNSID:
127 err = i.ingestLabelOp(ctx, e, l)
128 case tangled.RepoNSID:
129 err = i.ingestRepo(ctx, e, l)
130 }
131 }
132
133 if err != nil {
134 l.Warn("failed to ingest record, skipping", "err", err)
135 }
136
137 lastTimeUs := e.TimeUS + 1
138 if saveErr := i.Db.SaveLastTimeUs(lastTimeUs); saveErr != nil {
139 l.Error("failed to save cursor", "err", saveErr)
140 }
141
142 return nil
143 }
144}
145
146func (i *Ingester) resolveRepoRef(ref string) (*models.Repo, error) {
147 if strings.HasPrefix(ref, "did:") {
148 return db.GetRepoByDid(i.Db, ref)
149 }
150 return db.GetRepoByAtUri(i.Db, ref)
151}
152
153func (i *Ingester) resolveOldFormatStar(raw json.RawMessage, star *models.Star, l *slog.Logger) (bool, error) {
154 var legacy struct {
155 Subject *string `json:"subject"`
156 SubjectDid *string `json:"subjectDid"`
157 }
158 if err := json.Unmarshal(raw, &legacy); err != nil {
159 return false, err
160 }
161
162 switch {
163 case legacy.SubjectDid != nil:
164 repo, err := i.resolveRepoRef(*legacy.SubjectDid)
165 if err != nil {
166 l.Warn("skipping old-format star for unknown repo", "subjectDid", *legacy.SubjectDid)
167 return false, nil
168 }
169 star.SubjectType = models.StarSubjectRepo
170 star.Subject = repo.RepoDid
171 return true, nil
172
173 case legacy.Subject != nil:
174 uri, err := syntax.ParseATURI(*legacy.Subject)
175 if err != nil {
176 return false, fmt.Errorf("invalid old-format star subject: %w", err)
177 }
178 switch uri.Collection().String() {
179 case tangled.RepoNSID:
180 repo, err := db.GetRepoByAtUri(i.Db, uri.String())
181 if err != nil {
182 l.Warn("skipping old-format star for unknown repo", "subject", *legacy.Subject)
183 return false, nil
184 }
185 star.SubjectType = models.StarSubjectRepo
186 star.Subject = repo.RepoDid
187 return true, nil
188 default:
189 star.SubjectType = models.StarSubjectString
190 star.Subject = *legacy.Subject
191 return true, nil
192 }
193
194 default:
195 return false, fmt.Errorf("old-format star has neither subject nor subjectDid")
196 }
197}
198
199func (i *Ingester) ingestStar(ctx context.Context, e *jmodels.Event, l *slog.Logger) error {
200 var err error
201 did := e.Did
202
203 l = l.With("handler", "ingestStar")
204
205 switch e.Commit.Operation {
206 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
207 raw := json.RawMessage(e.Commit.Record)
208 record := tangled.FeedStar{}
209 unmarshalErr := json.Unmarshal(raw, &record)
210
211 createdAt, parseErr := time.Parse(time.RFC3339, record.CreatedAt)
212 if parseErr != nil {
213 createdAt = time.Now()
214 }
215
216 star := models.Star{
217 Did: did,
218 Rkey: e.Commit.RKey,
219 Created: createdAt,
220 }
221
222 switch {
223 case unmarshalErr != nil:
224 resolved, resolveErr := i.resolveOldFormatStar(raw, &star, l)
225 if resolveErr != nil {
226 l.Error("invalid record", "newFmtErr", unmarshalErr, "oldFmtErr", resolveErr)
227 return unmarshalErr
228 }
229 if !resolved {
230 return nil
231 }
232
233 case record.Subject == nil:
234 return fmt.Errorf("star record has nil subject")
235
236 case record.Subject.FeedStar_Repo != nil:
237 repo, repoErr := i.resolveRepoRef(record.Subject.FeedStar_Repo.Did)
238 if repoErr != nil {
239 l.Warn("skipping star for unknown repo", "did", record.Subject.FeedStar_Repo.Did)
240 return nil
241 }
242 star.SubjectType = models.StarSubjectRepo
243 star.Subject = repo.RepoDid
244
245 case record.Subject.FeedStar_String != nil:
246 star.SubjectType = models.StarSubjectString
247 star.Subject = record.Subject.FeedStar_String.Uri
248
249 default:
250 return fmt.Errorf("star record has empty subject union")
251 }
252
253 err = db.UpsertStar(i.Db, star)
254 case jmodels.CommitOperationDelete:
255 err = db.DeleteStarByRkey(i.Db, did, e.Commit.RKey)
256 }
257
258 if err != nil {
259 return fmt.Errorf("failed to %s star record: %w", e.Commit.Operation, err)
260 }
261 l.Info("processed star", "operation", e.Commit.Operation, "rkey", e.Commit.RKey)
262
263 l.Info("ingested record")
264 return nil
265}
266
267func (i *Ingester) ingestFollow(e *jmodels.Event, l *slog.Logger) error {
268 var err error
269 did := e.Did
270
271 l = l.With("handler", "ingestFollow")
272
273 switch e.Commit.Operation {
274 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
275 raw := json.RawMessage(e.Commit.Record)
276 record := tangled.GraphFollow{}
277 err = json.Unmarshal(raw, &record)
278 if err != nil {
279 l.Error("invalid record", "err", err)
280 return err
281 }
282 _, err := syntax.ParseDID(record.Subject)
283 if err != nil {
284 l.Error("invalid record. subject is invalid DID", "err", err)
285 return err
286 }
287
288 followedAt, err := time.Parse(time.RFC3339, record.CreatedAt)
289 if err != nil {
290 err = fmt.Errorf("createdAt is invalid datetime: %w", err)
291 l.Error("invalid record", "err", err)
292 return err
293 }
294
295 err = db.UpsertFollow(i.Db, models.Follow{
296 UserDid: did,
297 SubjectDid: record.Subject,
298 Rkey: e.Commit.RKey,
299 FollowedAt: followedAt,
300 })
301 case jmodels.CommitOperationDelete:
302 err = db.DeleteFollowByRkey(i.Db, did, e.Commit.RKey)
303 }
304
305 if err != nil {
306 return fmt.Errorf("failed to %s follow record: %w", e.Commit.Operation, err)
307 }
308 l.Info("processed follow", "operation", e.Commit.Operation, "rkey", e.Commit.RKey)
309
310 l.Info("ingested record")
311 return nil
312}
313
314func (i *Ingester) ingestVouch(ctx context.Context, e *jmodels.Event, l *slog.Logger) error {
315 var err error
316 did := e.Did
317
318 l = l.With("handler", "ingestVouch")
319
320 switch e.Commit.Operation {
321 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
322 raw := json.RawMessage(e.Commit.Record)
323 record := tangled.GraphVouch{}
324 err = json.Unmarshal(raw, &record)
325 if err != nil {
326 l.Error("invalid record", "err", err)
327 return err
328 }
329
330 // rkey is the subject_did being vouched for/denounced
331 subjectDID := e.Commit.RKey
332
333 _, err = syntax.ParseDID(subjectDID)
334 if err != nil {
335 l.Error("invalid subject_did in rkey", "err", err, "rkey", subjectDID)
336 return fmt.Errorf("invalid subject_did: %w", err)
337 }
338
339 if did == subjectDID {
340 l.Warn("attempted self-vouch", "did", did)
341 return fmt.Errorf("cannot vouch for self")
342 }
343
344 subjectId, err := i.IdResolver.ResolveIdent(ctx, subjectDID)
345 if err != nil {
346 return err
347 }
348
349 if subjectId.Handle.IsInvalidHandle() {
350 return err
351 }
352
353 kind, err := models.ParseVouchKind(record.Kind)
354 if err != nil {
355 l.Error("invalid kind", "kind", kind)
356 return fmt.Errorf("invalid kind: %s", kind)
357 }
358
359 recordCid, err := cid.Parse(e.Commit.CID)
360 if err != nil {
361 l.Error("invalid cid", "err", err, "cid", e.Commit.CID)
362 return fmt.Errorf("invalid cid: %w", err)
363 }
364
365 var evidences []syntax.ATURI
366 for _, raw := range record.Evidences {
367 uri, parseErr := syntax.ParseATURI(raw)
368 if parseErr != nil {
369 l.Warn("invalid evidence AT-URI, skipping", "uri", raw, "err", parseErr)
370 continue
371 }
372 evidences = append(evidences, uri)
373 }
374
375 tx, txErr := i.Db.Begin()
376 if txErr != nil {
377 return fmt.Errorf("failed to start transaction: %w", txErr)
378 }
379
380 addErr := db.AddVouch(tx, &models.Vouch{
381 Did: syntax.DID(did),
382 SubjectDid: subjectId.DID,
383 Cid: recordCid,
384 Kind: kind,
385 Reason: record.Reason,
386 Evidences: evidences,
387 })
388 if addErr != nil {
389 tx.Rollback()
390 err = addErr
391 } else {
392 err = tx.Commit()
393 }
394
395 case jmodels.CommitOperationDelete:
396 err = db.DeleteVouchByRkey(i.Db, did, e.Commit.RKey)
397 }
398
399 if err != nil {
400 return fmt.Errorf("failed to %s vouch record: %w", e.Commit.Operation, err)
401 }
402
403 l.Info("ingested record")
404 return nil
405}
406
407func (i *Ingester) ingestPublicKey(e *jmodels.Event, l *slog.Logger) error {
408 did := e.Did
409 var err error
410
411 l = l.With("handler", "ingestPublicKey")
412
413 switch e.Commit.Operation {
414 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
415 l.Debug("processing add of pubkey")
416 raw := json.RawMessage(e.Commit.Record)
417 record := tangled.PublicKey{}
418 err = json.Unmarshal(raw, &record)
419 if err != nil {
420 l.Error("invalid record", "err", err)
421 return err
422 }
423 pubKey, err := models.PublicKeyFromRecord(syntax.DID(did), syntax.RecordKey(e.Commit.RKey), record)
424 if err != nil {
425 l.Error("invalid record", "err", err)
426 return err
427 }
428 if err := pubKey.Validate(); err != nil {
429 l.Error("invalid record", "err", err)
430 return err
431 }
432
433 err = db.UpsertPublicKey(i.Db, pubKey)
434 case jmodels.CommitOperationDelete:
435 l.Debug("processing delete of pubkey")
436 err = db.DeletePublicKeyByRkey(i.Db, did, e.Commit.RKey)
437 }
438
439 if err != nil {
440 return fmt.Errorf("failed to %s pubkey record: %w", e.Commit.Operation, err)
441 }
442 l.Info("processed pubkey", "operation", e.Commit.Operation, "rkey", e.Commit.RKey)
443
444 l.Info("ingested record")
445 return nil
446}
447
448func (i *Ingester) ingestArtifact(ctx context.Context, e *jmodels.Event, l *slog.Logger) error {
449 did := e.Did
450 var err error
451
452 l = l.With("handler", "ingestArtifact")
453
454 switch e.Commit.Operation {
455 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
456 raw := json.RawMessage(e.Commit.Record)
457 record := tangled.RepoArtifact{}
458 err = json.Unmarshal(raw, &record)
459 if err != nil {
460 l.Error("invalid record", "err", err)
461 return err
462 }
463
464 var repo *models.Repo
465 if record.RepoDid != nil && *record.RepoDid != "" {
466 repo, err = db.GetRepoByDid(i.Db, *record.RepoDid)
467 if err != nil && !errors.Is(err, sql.ErrNoRows) {
468 return fmt.Errorf("failed to look up repo by DID %s: %w", *record.RepoDid, err)
469 }
470 }
471 if repo == nil && record.Repo != nil {
472 repoAt, parseErr := syntax.ParseATURI(*record.Repo)
473 if parseErr != nil {
474 return parseErr
475 }
476 repo, err = db.GetRepoByAtUri(i.Db, repoAt.String())
477 if err != nil {
478 return err
479 }
480 }
481 if repo == nil {
482 return fmt.Errorf("artifact record has neither valid repoDid nor repo field")
483 }
484
485 allowed, permErr := i.Acl.HasRepoPermissionErr(ctx, repo, did, "repo:push")
486 if permErr != nil {
487 l.Warn("ingesting artifact without permission check", "did", did, "repo", repo.RepoIdentifier(), "err", permErr)
488 } else if !allowed {
489 l.Info("skipping unauthorized artifact", "did", did, "repo", repo.RepoIdentifier())
490 return nil
491 }
492
493 repoDid := repo.RepoDid
494 if repoDid == "" && record.RepoDid != nil {
495 repoDid = *record.RepoDid
496 }
497 if repoDid != "" && (record.RepoDid == nil || *record.RepoDid == "") && record.Repo != nil {
498 if enqErr := db.EnqueuePdsRecordMigration(ctx, i.Db, "add-repo-did", syntax.DID(did), syntax.NSID(tangled.RepoArtifactNSID), syntax.RecordKey(e.Commit.RKey)); enqErr != nil {
499 l.Warn("failed to enqueue PDS rewrite for artifact", "err", enqErr, "did", did, "repoDid", repoDid)
500 }
501 }
502
503 createdAt, parseErr := time.Parse(time.RFC3339, record.CreatedAt)
504 if parseErr != nil {
505 createdAt = time.Now()
506 }
507
508 artifact := models.Artifact{
509 Did: did,
510 Rkey: e.Commit.RKey,
511 RepoDid: syntax.DID(repo.RepoDid),
512 Tag: plumbing.Hash(record.Tag),
513 CreatedAt: createdAt,
514 BlobCid: cid.Cid(record.Artifact.Ref),
515 Name: record.Name,
516 Size: uint64(record.Artifact.Size),
517 MimeType: record.Artifact.MimeType,
518 }
519
520 err = db.AddArtifact(i.Db, artifact)
521 case jmodels.CommitOperationDelete:
522 err = db.DeleteArtifact(i.Db, orm.FilterEq("did", did), orm.FilterEq("rkey", e.Commit.RKey))
523 }
524
525 if err != nil {
526 return fmt.Errorf("failed to %s artifact record: %w", e.Commit.Operation, err)
527 }
528
529 l.Info("ingested record")
530 return nil
531}
532
533func (i *Ingester) ingestProfile(ctx context.Context, e *jmodels.Event, l *slog.Logger) error {
534 did := e.Did
535 var err error
536
537 l = l.With("handler", "ingestProfile")
538
539 if e.Commit.RKey != "self" {
540 return fmt.Errorf("ingestProfile only ingests `self` record")
541 }
542
543 switch e.Commit.Operation {
544 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
545 raw := json.RawMessage(e.Commit.Record)
546 record := tangled.ActorProfile{}
547 err = json.Unmarshal(raw, &record)
548 if err != nil {
549 l.Error("invalid record", "err", err)
550 return err
551 }
552
553 avatar := ""
554 if record.Avatar != nil {
555 avatar = record.Avatar.Ref.String()
556 }
557
558 description := ""
559 if record.Description != nil {
560 description = *record.Description
561 }
562
563 includeBluesky := record.Bluesky
564
565 pronouns := ""
566 if record.Pronouns != nil {
567 pronouns = *record.Pronouns
568 }
569
570 location := ""
571 if record.Location != nil {
572 location = *record.Location
573 }
574
575 var links [5]string
576 for i, l := range record.Links {
577 if i < 5 {
578 links[i] = l
579 }
580 }
581
582 var stats [2]models.VanityStat
583 for i, s := range record.Stats {
584 if i < 2 {
585 stats[i].Kind = models.ParseVanityStatKind(s)
586 }
587 }
588
589 var pinned [6]string
590 for i, r := range record.PinnedRepositories {
591 if i < 6 {
592 pinned[i] = r
593 }
594 }
595
596 var preferredHandle syntax.Handle
597 if record.PreferredHandle != nil {
598 if h, err := syntax.ParseHandle(*record.PreferredHandle); err == nil {
599 ident, identErr := i.IdResolver.ResolveIdent(ctx, did)
600 if identErr == nil && slices.Contains(ident.AlsoKnownAs, "at://"+string(h)) {
601 preferredHandle = h
602 }
603 }
604 }
605
606 profile := models.Profile{
607 Did: did,
608 Avatar: avatar,
609 Description: description,
610 IncludeBluesky: includeBluesky,
611 Location: location,
612 Links: links,
613 Stats: stats,
614 PinnedRepos: pinned,
615 Pronouns: pronouns,
616 PreferredHandle: preferredHandle,
617 }
618
619 err = db.ValidateProfile(i.Db, &profile)
620 if err != nil {
621 return fmt.Errorf("invalid profile record")
622 }
623
624 err = db.UpsertProfile(i.Db, &profile)
625 if err != nil {
626 return fmt.Errorf("upserting profile: %w", err)
627 }
628
629 if i.Cache != nil {
630 pipe := i.Cache.Pipeline()
631 didKey := fmt.Sprintf(cache.PreferredHandleByDid, did)
632 if preferredHandle != "" {
633 pipe.Set(ctx, didKey, string(preferredHandle), cache.PreferredHandleTTL)
634 pipe.Set(ctx, fmt.Sprintf(cache.PreferredHandleByHandle, string(preferredHandle)), did, cache.PreferredHandleTTL)
635 } else {
636 pipe.Del(ctx, didKey)
637 }
638 if _, execErr := pipe.Exec(ctx); execErr != nil {
639 l.Warn("failed to update preferred handle cache", "err", execErr)
640 }
641 }
642 case jmodels.CommitOperationDelete:
643 tx, beginErr := i.Db.Begin()
644 if beginErr != nil {
645 return fmt.Errorf("failed to start transaction: %w", beginErr)
646 }
647
648 priorHandle, phErr := db.GetPreferredHandle(tx, did)
649 if phErr != nil && !errors.Is(phErr, sql.ErrNoRows) {
650 l.Warn("failed to read prior preferred handle", "err", phErr)
651 }
652
653 err = db.DeleteProfile(tx, did)
654 if err == nil && i.Cache != nil {
655 pipe := i.Cache.Pipeline()
656 pipe.Del(ctx, fmt.Sprintf(cache.PreferredHandleByDid, did))
657 if priorHandle != "" {
658 pipe.Del(ctx, fmt.Sprintf(cache.PreferredHandleByHandle, string(priorHandle)))
659 }
660 if _, execErr := pipe.Exec(ctx); execErr != nil {
661 l.Warn("failed to evict preferred handle cache", "err", execErr)
662 }
663 }
664 }
665
666 if err != nil {
667 return fmt.Errorf("failed to %s profile record: %w", e.Commit.Operation, err)
668 }
669
670 l.Info("ingested record")
671 return nil
672}
673
674func (i *Ingester) ingestSpindleMember(ctx context.Context, e *jmodels.Event, l *slog.Logger) error {
675 did := e.Did
676 var err error
677
678 l = l.With("handler", "ingestSpindleMember")
679
680 switch e.Commit.Operation {
681 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
682 raw := json.RawMessage(e.Commit.Record)
683 record := tangled.SpindleMember{}
684 err = json.Unmarshal(raw, &record)
685 if err != nil {
686 l.Error("invalid record", "err", err)
687 return err
688 }
689
690 // only spindle owner can invite to spindles
691 ok, err := i.Enforcer.IsSpindleInviteAllowed(did, record.Instance)
692 if err != nil {
693 return fmt.Errorf("failed to check invite permission: %w", err)
694 }
695 if !ok {
696 if verifyErr := i.verifySpindle(ctx, record.Instance, did); verifyErr != nil {
697 return fmt.Errorf("invite denied and verify failed: %w", verifyErr)
698 }
699 ok, err = i.Enforcer.IsSpindleInviteAllowed(did, record.Instance)
700 if err != nil {
701 return fmt.Errorf("failed to re-check invite permission: %w", err)
702 }
703 if !ok {
704 return fmt.Errorf("invite denied for did %s on spindle %s", did, record.Instance)
705 }
706 }
707
708 memberId, err := i.IdResolver.ResolveIdent(ctx, record.Subject)
709 if err != nil {
710 return err
711 }
712
713 if memberId.Handle.IsInvalidHandle() {
714 return fmt.Errorf("invalid handle for member %s", record.Subject)
715 }
716
717 existing, err := db.GetSpindleMembers(i.Db,
718 orm.FilterEq("did", did),
719 orm.FilterEq("rkey", e.Commit.RKey),
720 )
721 if err != nil {
722 return fmt.Errorf("failed to look up existing member: %w", err)
723 }
724 if len(existing) > 1 {
725 return fmt.Errorf("multiple spindle members with rkey %s", e.Commit.RKey)
726 }
727
728 tx, err := i.Db.Begin()
729 if err != nil {
730 return fmt.Errorf("failed to start txn: %w", err)
731 }
732 committed := false
733 defer func() {
734 if committed {
735 return
736 }
737 tx.Rollback()
738 i.Enforcer.E.LoadPolicy()
739 }()
740
741 if len(existing) == 1 {
742 prev := existing[0]
743 if prev.Instance != record.Instance || prev.Subject != memberId.DID {
744 if err = db.RemoveSpindleMember(tx,
745 orm.FilterEq("did", did),
746 orm.FilterEq("rkey", e.Commit.RKey),
747 ); err != nil {
748 return fmt.Errorf("failed to remove stale row: %w", err)
749 }
750 if err = i.Enforcer.RemoveSpindleMember(prev.Instance, prev.Subject.String()); err != nil {
751 return fmt.Errorf("failed to remove stale ACL: %w", err)
752 }
753 }
754 }
755
756 if err = db.AddSpindleMember(tx, models.SpindleMember{
757 Did: syntax.DID(did),
758 Rkey: e.Commit.RKey,
759 Instance: record.Instance,
760 Subject: memberId.DID,
761 }); err != nil {
762 return fmt.Errorf("failed to add to db: %w", err)
763 }
764
765 if err = i.Enforcer.AddSpindleMember(record.Instance, memberId.DID.String()); err != nil {
766 return fmt.Errorf("failed to update ACLs: %w", err)
767 }
768
769 if err = tx.Commit(); err != nil {
770 return fmt.Errorf("failed to commit txn: %w", err)
771 }
772
773 if err = i.Enforcer.E.SavePolicy(); err != nil {
774 return fmt.Errorf("failed to save ACLs: %w", err)
775 }
776 committed = true
777
778 l.Info("upserted spindle member")
779 case jmodels.CommitOperationDelete:
780 rkey := e.Commit.RKey
781
782 // get record from db first
783 members, err := db.GetSpindleMembers(
784 i.Db,
785 orm.FilterEq("did", did),
786 orm.FilterEq("rkey", rkey),
787 )
788 if err != nil || len(members) != 1 {
789 return fmt.Errorf("failed to get member: %w, len(members) = %d", err, len(members))
790 }
791 member := members[0]
792
793 tx, err := i.Db.Begin()
794 if err != nil {
795 return fmt.Errorf("failed to start txn: %w", err)
796 }
797 committed := false
798 defer func() {
799 if committed {
800 return
801 }
802 tx.Rollback()
803 i.Enforcer.E.LoadPolicy()
804 }()
805
806 // remove record by rkey && update enforcer
807 if err = db.RemoveSpindleMember(
808 tx,
809 orm.FilterEq("did", did),
810 orm.FilterEq("rkey", rkey),
811 ); err != nil {
812 return fmt.Errorf("failed to remove from db: %w", err)
813 }
814
815 // update enforcer
816 err = i.Enforcer.RemoveSpindleMember(member.Instance, member.Subject.String())
817 if err != nil {
818 return fmt.Errorf("failed to update ACLs: %w", err)
819 }
820
821 if err = tx.Commit(); err != nil {
822 return fmt.Errorf("failed to commit txn: %w", err)
823 }
824
825 if err = i.Enforcer.E.SavePolicy(); err != nil {
826 return fmt.Errorf("failed to save ACLs: %w", err)
827 }
828 committed = true
829
830 l.Info("removed spindle member")
831 }
832
833 return nil
834}
835
836func (i *Ingester) ingestSpindle(ctx context.Context, e *jmodels.Event, l *slog.Logger) error {
837 did := e.Did
838 var err error
839
840 l = l.With("handler", "ingestSpindle")
841
842 switch e.Commit.Operation {
843 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
844 raw := json.RawMessage(e.Commit.Record)
845 record := tangled.Spindle{}
846 err = json.Unmarshal(raw, &record)
847 if err != nil {
848 l.Error("invalid record", "err", err)
849 return err
850 }
851
852 instance := e.Commit.RKey
853
854 err := db.AddSpindle(i.Db, models.Spindle{
855 Owner: syntax.DID(did),
856 Instance: instance,
857 })
858 if err != nil {
859 l.Error("failed to add spindle to db", "err", err, "instance", instance)
860 return err
861 }
862
863 if err := i.verifySpindle(ctx, instance, did); err != nil {
864 l.Warn("failed to verify spindle", "instance", instance, "did", did, "err", err)
865 }
866
867 l.Info("ingested record", "instance", instance)
868 return nil
869
870 case jmodels.CommitOperationDelete:
871 instance := e.Commit.RKey
872
873 // get record from db first
874 spindles, err := db.GetSpindles(
875 ctx,
876 i.Db,
877 orm.FilterEq("owner", did),
878 orm.FilterEq("instance", instance),
879 )
880 if err != nil || len(spindles) != 1 {
881 return fmt.Errorf("failed to get spindles: %w, len(spindles) = %d", err, len(spindles))
882 }
883 spindle := spindles[0]
884
885 tx, err := i.Db.Begin()
886 if err != nil {
887 return fmt.Errorf("failed to start txn: %w", err)
888 }
889 defer func() {
890 tx.Rollback()
891 i.Enforcer.E.LoadPolicy()
892 }()
893
894 // remove spindle members first
895 err = db.RemoveSpindleMember(
896 tx,
897 orm.FilterEq("owner", did),
898 orm.FilterEq("instance", instance),
899 )
900 if err != nil {
901 return fmt.Errorf("failed to remove spindle members: %w", err)
902 }
903
904 err = db.DeleteSpindle(
905 tx,
906 orm.FilterEq("owner", did),
907 orm.FilterEq("instance", instance),
908 )
909 if err != nil {
910 return fmt.Errorf("failed to delete spindle: %w", err)
911 }
912
913 if spindle.Verified != nil {
914 err = i.Enforcer.RemoveSpindle(instance)
915 if err != nil {
916 return fmt.Errorf("failed to remove spindle from enforcer: %w", err)
917 }
918 }
919
920 err = tx.Commit()
921 if err != nil {
922 return fmt.Errorf("failed to commit txn: %w", err)
923 }
924
925 err = i.Enforcer.E.SavePolicy()
926 if err != nil {
927 return fmt.Errorf("failed to save ACLs: %w", err)
928 }
929
930 l.Info("ingested record", "instance", instance)
931 }
932
933 return nil
934}
935
936func (i *Ingester) ingestString(e *jmodels.Event, l *slog.Logger) error {
937 did := e.Did
938 rkey := e.Commit.RKey
939
940 var err error
941
942 l = l.With("handler", "ingestString")
943
944 switch e.Commit.Operation {
945 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
946 raw := json.RawMessage(e.Commit.Record)
947 record := tangled.String{}
948 err = json.Unmarshal(raw, &record)
949 if err != nil {
950 l.Error("invalid record", "err", err)
951 return err
952 }
953
954 string := models.StringFromRecord(did, rkey, record)
955
956 if err = string.Validate(); err != nil {
957 l.Error("invalid record", "err", err)
958 return err
959 }
960
961 if err = db.AddString(i.Db, string); err != nil {
962 l.Error("failed to add string", "err", err)
963 return err
964 }
965
966 l.Info("ingested record")
967 return nil
968
969 case jmodels.CommitOperationDelete:
970 if err := db.DeleteString(
971 i.Db,
972 orm.FilterEq("did", did),
973 orm.FilterEq("rkey", rkey),
974 ); err != nil {
975 l.Error("failed to delete", "err", err)
976 return fmt.Errorf("failed to delete string record: %w", err)
977 }
978
979 l.Info("ingested record")
980 return nil
981 }
982
983 return nil
984}
985
986func (i *Ingester) ingestKnotMember(ctx context.Context, e *jmodels.Event, l *slog.Logger) error {
987 did := e.Did
988 var err error
989
990 l = l.With("handler", "ingestKnotMember")
991
992 switch e.Commit.Operation {
993 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
994 raw := json.RawMessage(e.Commit.Record)
995 record := tangled.KnotMember{}
996 err = json.Unmarshal(raw, &record)
997 if err != nil {
998 l.Error("invalid record", "err", err)
999 return err
1000 }
1001
1002 // only knot owner can invite to knots
1003 ok, err := i.Enforcer.IsKnotInviteAllowed(did, record.Domain)
1004 if err != nil {
1005 return fmt.Errorf("failed to check invite permission: %w", err)
1006 }
1007 if !ok {
1008 if verifyErr := i.verifyKnot(ctx, record.Domain, did); verifyErr != nil {
1009 return fmt.Errorf("invite denied and verify failed: %w", verifyErr)
1010 }
1011 ok, err = i.Enforcer.IsKnotInviteAllowed(did, record.Domain)
1012 if err != nil {
1013 return fmt.Errorf("failed to re-check invite permission: %w", err)
1014 }
1015 if !ok {
1016 return fmt.Errorf("invite denied for did %s on knot %s", did, record.Domain)
1017 }
1018 }
1019
1020 memberId, err := i.IdResolver.ResolveIdent(ctx, record.Subject)
1021 if err != nil {
1022 return err
1023 }
1024
1025 if memberId.Handle.IsInvalidHandle() {
1026 return fmt.Errorf("invalid handle for member %s", record.Subject)
1027 }
1028
1029 existing, err := db.GetKnotMembers(i.Db,
1030 orm.FilterEq("did", did),
1031 orm.FilterEq("rkey", e.Commit.RKey),
1032 )
1033 if err != nil {
1034 return fmt.Errorf("failed to look up existing member: %w", err)
1035 }
1036 if len(existing) > 1 {
1037 return fmt.Errorf("multiple knot members with rkey %s", e.Commit.RKey)
1038 }
1039
1040 tx, err := i.Db.Begin()
1041 if err != nil {
1042 return fmt.Errorf("failed to start txn: %w", err)
1043 }
1044 committed := false
1045 defer func() {
1046 if committed {
1047 return
1048 }
1049 tx.Rollback()
1050 i.Enforcer.E.LoadPolicy()
1051 }()
1052
1053 if len(existing) == 1 {
1054 prev := existing[0]
1055 if prev.Domain != record.Domain || prev.Subject != memberId.DID {
1056 if err = db.RemoveKnotMember(tx,
1057 orm.FilterEq("did", did),
1058 orm.FilterEq("rkey", e.Commit.RKey),
1059 ); err != nil {
1060 return fmt.Errorf("failed to remove stale row: %w", err)
1061 }
1062 if err = i.Enforcer.RemoveKnotMember(prev.Domain, prev.Subject.String()); err != nil {
1063 return fmt.Errorf("failed to remove stale ACL: %w", err)
1064 }
1065 }
1066 }
1067
1068 if err = db.AddKnotMember(tx, models.KnotMember{
1069 Did: syntax.DID(did),
1070 Rkey: e.Commit.RKey,
1071 Domain: record.Domain,
1072 Subject: memberId.DID,
1073 }); err != nil {
1074 return fmt.Errorf("failed to add to db: %w", err)
1075 }
1076
1077 if err = i.Enforcer.AddKnotMember(record.Domain, memberId.DID.String()); err != nil {
1078 return fmt.Errorf("failed to update ACLs: %w", err)
1079 }
1080
1081 if err = tx.Commit(); err != nil {
1082 return fmt.Errorf("failed to commit txn: %w", err)
1083 }
1084
1085 if err = i.Enforcer.E.SavePolicy(); err != nil {
1086 return fmt.Errorf("failed to save ACLs: %w", err)
1087 }
1088 committed = true
1089
1090 l.Info("upserted knot member")
1091 case jmodels.CommitOperationDelete:
1092 rkey := e.Commit.RKey
1093
1094 members, err := db.GetKnotMembers(
1095 i.Db,
1096 orm.FilterEq("did", did),
1097 orm.FilterEq("rkey", rkey),
1098 )
1099 if err != nil {
1100 return fmt.Errorf("failed to look up knot member with rkey %s: %w", rkey, err)
1101 }
1102 if len(members) == 0 {
1103 l.Info("knot member already removed", "rkey", rkey)
1104 return nil
1105 }
1106 if len(members) > 1 {
1107 return fmt.Errorf("multiple knot members with rkey %s", rkey)
1108 }
1109 member := members[0]
1110
1111 tx, err := i.Db.Begin()
1112 if err != nil {
1113 return fmt.Errorf("failed to start txn: %w", err)
1114 }
1115 committed := false
1116 defer func() {
1117 if committed {
1118 return
1119 }
1120 tx.Rollback()
1121 i.Enforcer.E.LoadPolicy()
1122 }()
1123
1124 if err = db.RemoveKnotMember(
1125 tx,
1126 orm.FilterEq("did", did),
1127 orm.FilterEq("rkey", rkey),
1128 ); err != nil {
1129 return fmt.Errorf("failed to remove from db: %w", err)
1130 }
1131
1132 if err = i.Enforcer.RemoveKnotMember(member.Domain, member.Subject.String()); err != nil {
1133 return fmt.Errorf("failed to update ACLs: %w", err)
1134 }
1135
1136 if err = tx.Commit(); err != nil {
1137 return fmt.Errorf("failed to commit txn: %w", err)
1138 }
1139
1140 if err = i.Enforcer.E.SavePolicy(); err != nil {
1141 return fmt.Errorf("failed to save ACLs: %w", err)
1142 }
1143 committed = true
1144
1145 l.Info("removed knot member")
1146 }
1147
1148 return nil
1149}
1150
1151func (i *Ingester) ingestKnot(ctx context.Context, e *jmodels.Event, l *slog.Logger) error {
1152 did := e.Did
1153 var err error
1154
1155 l = l.With("handler", "ingestKnot")
1156
1157 switch e.Commit.Operation {
1158 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
1159 raw := json.RawMessage(e.Commit.Record)
1160 record := tangled.Knot{}
1161 err = json.Unmarshal(raw, &record)
1162 if err != nil {
1163 l.Error("invalid record", "err", err)
1164 return err
1165 }
1166
1167 domain := e.Commit.RKey
1168
1169 err := db.AddKnot(i.Db, domain, did)
1170 if err != nil {
1171 l.Error("failed to add knot to db", "err", err, "domain", domain)
1172 return err
1173 }
1174
1175 if err := i.verifyKnot(ctx, domain, did); err != nil {
1176 l.Warn("failed to verify knot", "domain", domain, "did", did, "err", err)
1177 }
1178
1179 l.Info("ingested record", "domain", domain)
1180 return nil
1181
1182 case jmodels.CommitOperationDelete:
1183 domain := e.Commit.RKey
1184
1185 // get record from db first
1186 registrations, err := db.GetRegistrations(
1187 i.Db,
1188 orm.FilterEq("domain", domain),
1189 orm.FilterEq("did", did),
1190 )
1191 if err != nil {
1192 return fmt.Errorf("failed to get registration: %w", err)
1193 }
1194 if len(registrations) != 1 {
1195 return fmt.Errorf("got incorrect number of registrations: %d, expected 1", len(registrations))
1196 }
1197 registration := registrations[0]
1198
1199 tx, err := i.Db.Begin()
1200 if err != nil {
1201 return fmt.Errorf("failed to start txn: %w", err)
1202 }
1203 defer func() {
1204 tx.Rollback()
1205 i.Enforcer.E.LoadPolicy()
1206 }()
1207
1208 err = db.RemoveKnotMember(
1209 tx,
1210 orm.FilterEq("did", did),
1211 orm.FilterEq("domain", domain),
1212 )
1213 if err != nil {
1214 return fmt.Errorf("failed to remove knot members: %w", err)
1215 }
1216
1217 err = db.DeleteKnot(
1218 tx,
1219 orm.FilterEq("did", did),
1220 orm.FilterEq("domain", domain),
1221 )
1222 if err != nil {
1223 return fmt.Errorf("failed to delete knot: %w", err)
1224 }
1225
1226 err = db.RemoveReposByKnot(tx, domain)
1227 if err != nil {
1228 return fmt.Errorf("failed to remove repos by knot: %w", err)
1229 }
1230
1231 if registration.Registered != nil {
1232 err = i.Enforcer.RemoveKnot(domain)
1233 if err != nil {
1234 return fmt.Errorf("failed to remove knot from enforcer: %w", err)
1235 }
1236 }
1237
1238 err = tx.Commit()
1239 if err != nil {
1240 return fmt.Errorf("failed to commit txn: %w", err)
1241 }
1242
1243 err = i.Enforcer.E.SavePolicy()
1244 if err != nil {
1245 return fmt.Errorf("failed to save ACLs: %w", err)
1246 }
1247
1248 l.Info("ingested record", "domain", domain)
1249 }
1250
1251 return nil
1252}
1253
1254const (
1255 verifyAttempts = 4
1256 verifyMinDelay = 1 * time.Second
1257 verifyMaxDelay = 5 * time.Second
1258)
1259
1260func (i *Ingester) verifyKnot(ctx context.Context, domain, did string) error {
1261 regs, err := db.GetRegistrations(i.Db,
1262 orm.FilterEq("domain", domain),
1263 orm.FilterEq("did", did),
1264 )
1265 if err != nil {
1266 return fmt.Errorf("look up registration: %w", err)
1267 }
1268 if len(regs) != 1 {
1269 return fmt.Errorf("no registration for %s by %s", domain, did)
1270 }
1271 if regs[0].Registered != nil {
1272 return nil
1273 }
1274
1275 err = retry.Do(
1276 func() error { return serververify.RunVerification(ctx, domain, did, i.Config.Core.Dev) },
1277 retry.Context(ctx),
1278 retry.Attempts(verifyAttempts),
1279 retry.Delay(verifyMinDelay),
1280 retry.MaxDelay(verifyMaxDelay),
1281 retry.DelayType(retry.BackOffDelay),
1282 retry.LastErrorOnly(true),
1283 )
1284 if err != nil {
1285 return fmt.Errorf("verify: %w", err)
1286 }
1287 return serververify.MarkKnotVerified(i.Db, i.Enforcer, domain, did)
1288}
1289
1290func (i *Ingester) verifySpindle(ctx context.Context, instance, did string) error {
1291 spindles, err := db.GetSpindles(ctx, i.Db,
1292 orm.FilterEq("instance", instance),
1293 orm.FilterEq("owner", did),
1294 )
1295 if err != nil {
1296 return fmt.Errorf("look up spindle: %w", err)
1297 }
1298 if len(spindles) != 1 {
1299 return fmt.Errorf("no spindle for %s by %s", instance, did)
1300 }
1301 if spindles[0].Verified != nil {
1302 return nil
1303 }
1304
1305 err = retry.Do(
1306 func() error { return serververify.RunVerification(ctx, instance, did, i.Config.Core.Dev) },
1307 retry.Context(ctx),
1308 retry.Attempts(verifyAttempts),
1309 retry.Delay(verifyMinDelay),
1310 retry.MaxDelay(verifyMaxDelay),
1311 retry.DelayType(retry.BackOffDelay),
1312 retry.LastErrorOnly(true),
1313 )
1314 if err != nil {
1315 return fmt.Errorf("verify: %w", err)
1316 }
1317 _, err = serververify.MarkSpindleVerified(i.Db, i.Enforcer, instance, did)
1318 return err
1319}
1320
1321const sweepConcurrency = 4
1322
1323func (i *Ingester) SweepPendingVerifications() {
1324 l := i.Logger.With("handler", "SweepPendingVerifications")
1325
1326 var g errgroup.Group
1327 g.SetLimit(sweepConcurrency)
1328
1329 regs, err := db.GetRegistrations(i.Db, orm.FilterIs("registered", nil))
1330 if err != nil {
1331 l.Error("failed to list unverified knots", "err", err)
1332 } else {
1333 for _, reg := range regs {
1334 g.Go(func() error {
1335 if err := i.verifyKnot(i.Ctx, reg.Domain, reg.ByDid); err != nil {
1336 l.Warn("verify knot failed", "domain", reg.Domain, "did", reg.ByDid, "err", err)
1337 }
1338 return nil
1339 })
1340 }
1341 }
1342
1343 spindles, err := db.GetSpindles(i.Ctx, i.Db, orm.FilterIs("verified", nil))
1344 if err != nil {
1345 l.Error("failed to list unverified spindles", "err", err)
1346 g.Wait()
1347 return
1348 }
1349 for _, s := range spindles {
1350 g.Go(func() error {
1351 if err := i.verifySpindle(i.Ctx, s.Instance, s.Owner.String()); err != nil {
1352 l.Warn("verify spindle failed", "instance", s.Instance, "owner", s.Owner, "err", err)
1353 }
1354 return nil
1355 })
1356 }
1357 g.Wait()
1358}
1359
1360func (i *Ingester) ingestIssue(ctx context.Context, e *jmodels.Event, l *slog.Logger) error {
1361 did := e.Did
1362 rkey := e.Commit.RKey
1363
1364 var err error
1365
1366 l = l.With("handler", "ingestIssue")
1367
1368 switch e.Commit.Operation {
1369 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
1370 raw := json.RawMessage(e.Commit.Record)
1371 record := tangled.RepoIssue{}
1372 err = json.Unmarshal(raw, &record)
1373 if err != nil {
1374 l.Error("invalid record", "err", err)
1375 return err
1376 }
1377
1378 issue := models.IssueFromRecord(did, rkey, record)
1379
1380 if issue.RepoDid == "" {
1381 return fmt.Errorf("issue record has no repo field")
1382 }
1383 if _, err := syntax.ParseDID(string(issue.RepoDid)); err != nil {
1384 return fmt.Errorf("issue record repo field is not a valid DID: %w", err)
1385 }
1386
1387 if err := issue.Validate(); err != nil {
1388 return fmt.Errorf("failed to validate issue: %w", err)
1389 }
1390
1391 if record.Repo != "" && !strings.HasPrefix(record.Repo, "did:") {
1392 repo, repoErr := db.GetRepoByAtUri(i.Db, record.Repo)
1393 if repoErr == nil && repo.RepoDid != "" {
1394 if enqErr := db.EnqueuePdsRecordMigration(ctx, i.Db, "add-repo-did", syntax.DID(did), syntax.NSID(tangled.RepoIssueNSID), syntax.RecordKey(e.Commit.RKey)); enqErr != nil {
1395 l.Warn("failed to enqueue PDS rewrite for issue", "err", enqErr, "did", did, "repoDid", repo.RepoDid)
1396 }
1397 }
1398 }
1399
1400 tx, err := i.Db.BeginTx(ctx, nil)
1401 if err != nil {
1402 l.Error("failed to begin transaction", "err", err)
1403 return err
1404 }
1405 defer tx.Rollback()
1406
1407 err = db.PutIssue(tx, &issue)
1408 if err != nil {
1409 l.Error("failed to create issue", "err", err)
1410 return err
1411 }
1412
1413 if err := db.ResolveIssueState(tx, issue.AtUri()); err != nil {
1414 l.Error("failed to resolve issue state", "err", err)
1415 return err
1416 }
1417
1418 err = tx.Commit()
1419 if err != nil {
1420 l.Error("failed to commit txn", "err", err)
1421 return err
1422 }
1423
1424 i.drainPendingState(ctx, issue.AtUri(), l)
1425
1426 l.Info("ingested record")
1427 return nil
1428
1429 case jmodels.CommitOperationDelete:
1430 tx, err := i.Db.BeginTx(ctx, nil)
1431 if err != nil {
1432 l.Error("failed to begin transaction", "err", err)
1433 return err
1434 }
1435 defer tx.Rollback()
1436
1437 if err := db.DeleteIssues(
1438 tx,
1439 did,
1440 rkey,
1441 ); err != nil {
1442 l.Error("failed to delete", "err", err)
1443 return fmt.Errorf("failed to delete issue record: %w", err)
1444 }
1445 if err := tx.Commit(); err != nil {
1446 l.Error("failed to commit txn", "err", err)
1447 return err
1448 }
1449
1450 l.Info("ingested record")
1451 return nil
1452 }
1453
1454 return nil
1455}
1456
1457func (i *Ingester) ingestPull(ctx context.Context, e *jmodels.Event, l *slog.Logger) error {
1458 did := e.Did
1459 rkey := e.Commit.RKey
1460
1461 var err error
1462
1463 l = l.With("handler", "ingestPull")
1464
1465 switch e.Commit.Operation {
1466 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
1467 raw := json.RawMessage(e.Commit.Record)
1468 record := tangled.RepoPull{}
1469 err = json.Unmarshal(raw, &record)
1470 if err != nil {
1471 l.Error("invalid record", "err", err)
1472 return err
1473 }
1474
1475 ownerId, err := i.IdResolver.ResolveIdent(ctx, did)
1476 if err != nil {
1477 l.Error("failed to resolve did", "err", err)
1478 return err
1479 }
1480
1481 // go through and fetch all blobs in parallel
1482 blobs := make([]io.Reader, len(record.Rounds))
1483
1484 g, gctx := errgroup.WithContext(ctx)
1485
1486 for idx, b := range record.Rounds {
1487 g.Go(func() error {
1488 // for some reason, a blob is empty
1489 if b.PatchBlob == nil {
1490 return fmt.Errorf("missing patchBlob in round %d", idx)
1491 }
1492
1493 ownerPds := ownerId.PDSEndpoint()
1494 url, _ := url.Parse(fmt.Sprintf("%s/xrpc/com.atproto.sync.getBlob", ownerPds))
1495 q := url.Query()
1496 q.Set("cid", b.PatchBlob.Ref.String())
1497 q.Set("did", did)
1498 url.RawQuery = q.Encode()
1499
1500 req, err := http.NewRequestWithContext(gctx, http.MethodGet, url.String(), nil)
1501 if err != nil {
1502 l.Error("failed to create request")
1503 return err
1504 }
1505 req.Header.Set("Content-Type", "application/json")
1506
1507 resp, err := http.DefaultClient.Do(req)
1508 if err != nil {
1509 l.Error("failed to make request")
1510 return err
1511 }
1512 defer resp.Body.Close()
1513
1514 var buf bytes.Buffer
1515 if _, err := io.Copy(&buf, io.LimitReader(resp.Body, 16<<20)); err != nil {
1516 return fmt.Errorf("failed to read blob in round %d: %w", idx, err)
1517 }
1518 blobs[idx] = &buf
1519
1520 return nil
1521 })
1522 }
1523
1524 if err := g.Wait(); err != nil {
1525 return err
1526 }
1527
1528 pull, err := models.PullFromRecord(did, rkey, record, blobs)
1529 if err != nil {
1530 return fmt.Errorf("failed to parse pull from record: %w", err)
1531 }
1532 if err := pull.Validate(); err != nil {
1533 return fmt.Errorf("failed to validate pull: %w", err)
1534 }
1535 if pull.DependentOn != nil {
1536 if err := func() error {
1537 dependentPull, err := db.GetPull(
1538 i.Db,
1539 orm.FilterEq("dependent_on", pull.DependentOn.String()),
1540 )
1541 if errors.Is(err, sql.ErrNoRows) {
1542 return nil
1543 }
1544 if err != nil {
1545 return fmt.Errorf("failed to fetch pulls with same dependency: %w", err)
1546 }
1547 if dependentPull.AtUri() == pull.AtUri() {
1548 return nil
1549 }
1550 return fmt.Errorf("another pull already depends on %s, which would form a DAG, this is presently disallowed", pull.DependentOn.String())
1551 }(); err != nil {
1552 return fmt.Errorf("failed to validate pull stack: %w", err)
1553 }
1554 }
1555
1556 tx, err := i.Db.BeginTx(ctx, nil)
1557 if err != nil {
1558 l.Error("failed to begin transaction", "err", err)
1559 return err
1560 }
1561 defer tx.Rollback()
1562
1563 err = db.PutPull(tx, pull)
1564 if err != nil {
1565 l.Error("failed to create pull", "err", err)
1566 return err
1567 }
1568
1569 if err := db.ResolvePullStatus(tx, pull.AtUri()); err != nil {
1570 l.Error("failed to resolve pull status", "err", err)
1571 return err
1572 }
1573
1574 err = tx.Commit()
1575 if err != nil {
1576 l.Error("failed to commit txn", "err", err)
1577 return err
1578 }
1579
1580 i.drainPendingState(ctx, pull.AtUri(), l)
1581
1582 l.Info("ingested record")
1583 return nil
1584
1585 case jmodels.CommitOperationDelete:
1586 tx, err := i.Db.BeginTx(ctx, nil)
1587 if err != nil {
1588 l.Error("failed to begin transaction", "err", err)
1589 return err
1590 }
1591 defer tx.Rollback()
1592
1593 if err := db.AbandonPulls(
1594 tx,
1595 orm.FilterEq("owner_did", did),
1596 orm.FilterEq("rkey", rkey),
1597 ); err != nil {
1598 l.Error("failed to abandon", "err", err)
1599 return fmt.Errorf("failed to abandon pull record: %w", err)
1600 }
1601 if err := tx.Commit(); err != nil {
1602 l.Error("failed to commit txn", "err", err)
1603 return err
1604 }
1605
1606 l.Info("ingested record")
1607 return nil
1608 }
1609
1610 return nil
1611}
1612
1613func (i *Ingester) authorizeStateRecord(ctx context.Context, repo *models.Repo, subjectAuthorDid, recordAuthorDid string, l *slog.Logger) (bool, error) {
1614 if recordAuthorDid == subjectAuthorDid {
1615 return true, nil
1616 }
1617 if recordAuthorDid == consts.TangledDid {
1618 return true, nil
1619 }
1620
1621 ok, err := i.Acl.HasRepoPermissionErr(ctx, repo, recordAuthorDid, "repo:push")
1622 if err != nil {
1623 if errors.Is(err, knotacl.ErrKnotUnreachable) {
1624 l.Warn("ingesting state record without permission check", "did", recordAuthorDid, "err", err)
1625 return true, nil
1626 }
1627 return false, err
1628 }
1629 return ok, nil
1630}
1631
1632type stateIngestSpec struct {
1633 subjectNSID string
1634 parse func(did, rkey string, raw json.RawMessage) (models.StateRecord, error)
1635 findSubject func(e db.Execer, subject syntax.ATURI) (repo *models.Repo, authorDid string, found bool, err error)
1636 put func(tx *sql.Tx, rec models.StateRecord) (syntax.ATURI, error)
1637 resolve func(tx *sql.Tx, subject syntax.ATURI) error
1638 recompute func(tx *sql.Tx, subject syntax.ATURI) error
1639 del func(tx *sql.Tx, did, rkey string) (syntax.ATURI, error)
1640}
1641
1642var issueStateSpec = stateIngestSpec{
1643 subjectNSID: tangled.RepoIssueNSID,
1644 parse: func(did, rkey string, raw json.RawMessage) (models.StateRecord, error) {
1645 record := tangled.RepoIssueState{}
1646 if err := json.Unmarshal(raw, &record); err != nil {
1647 return models.StateRecord{}, fmt.Errorf("invalid record: %w", err)
1648 }
1649 return models.IssueStateFromRecord(did, rkey, record)
1650 },
1651 findSubject: func(e db.Execer, subject syntax.ATURI) (*models.Repo, string, bool, error) {
1652 issues, err := db.GetIssues(e, orm.FilterEq("at_uri", subject))
1653 if err != nil {
1654 return nil, "", false, err
1655 }
1656 if len(issues) != 1 || issues[0].Repo == nil {
1657 return nil, "", false, nil
1658 }
1659 return issues[0].Repo, issues[0].Did, true, nil
1660 },
1661 put: db.PutIssueState,
1662 resolve: db.ResolveIssueState,
1663 recompute: db.RecomputeIssueState,
1664 del: db.DeleteIssueState,
1665}
1666
1667var pullStatusSpec = stateIngestSpec{
1668 subjectNSID: tangled.RepoPullNSID,
1669 parse: func(did, rkey string, raw json.RawMessage) (models.StateRecord, error) {
1670 record := tangled.RepoPullStatus{}
1671 if err := json.Unmarshal(raw, &record); err != nil {
1672 return models.StateRecord{}, fmt.Errorf("invalid record: %w", err)
1673 }
1674 return models.PullStatusFromRecord(did, rkey, record)
1675 },
1676 findSubject: func(e db.Execer, subject syntax.ATURI) (*models.Repo, string, bool, error) {
1677 pulls, err := db.GetPulls(e, orm.FilterEq("at_uri", subject))
1678 if err != nil {
1679 return nil, "", false, err
1680 }
1681 if len(pulls) != 1 || pulls[0].Repo == nil {
1682 return nil, "", false, nil
1683 }
1684 return pulls[0].Repo, pulls[0].OwnerDid, true, nil
1685 },
1686 put: db.PutPullStatus,
1687 resolve: db.ResolvePullStatus,
1688 recompute: db.RecomputePullStatus,
1689 del: db.DeletePullStatus,
1690}
1691
1692func (i *Ingester) ingestState(ctx context.Context, e *jmodels.Event, l *slog.Logger, spec stateIngestSpec) error {
1693 did := e.Did
1694 rkey := e.Commit.RKey
1695 nsid := e.Commit.Collection
1696
1697 l = l.With("handler", "ingestState", "nsid", nsid)
1698
1699 switch e.Commit.Operation {
1700 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
1701 return i.applyStateRecord(ctx, did, rkey, nsid, e.Commit.Record, spec, l)
1702 case jmodels.CommitOperationDelete:
1703 return i.deleteStateRecord(ctx, did, rkey, nsid, spec, l)
1704 }
1705
1706 return nil
1707}
1708
1709func (i *Ingester) applyStateRecord(ctx context.Context, did, rkey, nsid string, raw []byte, spec stateIngestSpec, l *slog.Logger) error {
1710 rec, err := spec.parse(did, rkey, json.RawMessage(raw))
1711 if err != nil {
1712 return err
1713 }
1714 if string(rec.Subject.Collection()) != spec.subjectNSID {
1715 return fmt.Errorf("state subject is not %s: %s", spec.subjectNSID, rec.Subject)
1716 }
1717
1718 repo, authorDid, found, err := spec.findSubject(i.Db, rec.Subject)
1719 if err != nil {
1720 return fmt.Errorf("failed to look up state subject: %w", err)
1721 }
1722 if !found {
1723 return i.parkStateRecord(ctx, did, rkey, nsid, rec.Subject, raw, l)
1724 }
1725
1726 authorized, err := i.authorizeStateRecord(ctx, repo, authorDid, did, l)
1727 if err != nil {
1728 return i.parkStateRecord(ctx, did, rkey, nsid, rec.Subject, raw, l)
1729 }
1730
1731 tx, err := i.Db.BeginTx(ctx, nil)
1732 if err != nil {
1733 return err
1734 }
1735 defer tx.Rollback()
1736
1737 if err := db.UnparkStateRecord(tx, did, rkey, nsid); err != nil {
1738 return fmt.Errorf("failed to unpark state record: %w", err)
1739 }
1740
1741 if !authorized {
1742 if err := tx.Commit(); err != nil {
1743 return err
1744 }
1745 l.Warn("dropped unauthorized state record", "did", did, "rkey", rkey, "subject", rec.Subject)
1746 return nil
1747 }
1748
1749 priorSubject, err := spec.put(tx, rec)
1750 if err != nil {
1751 return fmt.Errorf("failed to put state record: %w", err)
1752 }
1753 if err := spec.resolve(tx, rec.Subject); err != nil {
1754 return fmt.Errorf("failed to resolve state: %w", err)
1755 }
1756 if priorSubject != "" {
1757 if err := spec.recompute(tx, priorSubject); err != nil {
1758 return fmt.Errorf("failed to recompute prior subject state: %w", err)
1759 }
1760 }
1761
1762 if err := tx.Commit(); err != nil {
1763 return err
1764 }
1765
1766 l.Info("ingested record")
1767 return nil
1768}
1769
1770func (i *Ingester) deleteStateRecord(ctx context.Context, did, rkey, nsid string, spec stateIngestSpec, l *slog.Logger) error {
1771 tx, err := i.Db.BeginTx(ctx, nil)
1772 if err != nil {
1773 return err
1774 }
1775 defer tx.Rollback()
1776
1777 if err := db.UnparkStateRecord(tx, did, rkey, nsid); err != nil {
1778 return fmt.Errorf("failed to unpark state record: %w", err)
1779 }
1780 subject, err := spec.del(tx, did, rkey)
1781 if err != nil {
1782 return fmt.Errorf("failed to delete state record: %w", err)
1783 }
1784 if subject != "" {
1785 if err := spec.recompute(tx, subject); err != nil {
1786 return fmt.Errorf("failed to recompute state: %w", err)
1787 }
1788 }
1789
1790 if err := tx.Commit(); err != nil {
1791 return err
1792 }
1793
1794 l.Info("ingested record")
1795 return nil
1796}
1797
1798func (i *Ingester) parkStateRecord(ctx context.Context, did, rkey, nsid string, subject syntax.ATURI, raw []byte, l *slog.Logger) error {
1799 tx, err := i.Db.BeginTx(ctx, nil)
1800 if err != nil {
1801 return err
1802 }
1803 defer tx.Rollback()
1804
1805 if err := db.ParkStateRecord(tx, db.PendingStateRecord{
1806 Did: did,
1807 Rkey: rkey,
1808 Nsid: nsid,
1809 Subject: subject,
1810 Record: raw,
1811 }); err != nil {
1812 return fmt.Errorf("failed to park state record: %w", err)
1813 }
1814
1815 if err := tx.Commit(); err != nil {
1816 return err
1817 }
1818
1819 l.Info("parked record for retry", "subject", subject, "nsid", nsid)
1820 return nil
1821}
1822
1823func (i *Ingester) drainPendingState(ctx context.Context, subject syntax.ATURI, l *slog.Logger) {
1824 pending, err := db.PendingStateRecordsForSubject(i.Db, subject)
1825 if err != nil {
1826 l.Error("failed to load pending records", "err", err, "subject", subject)
1827 return
1828 }
1829 for _, p := range pending {
1830 if err := i.reapplyPendingRecord(ctx, p, l); err != nil {
1831 l.Error("failed to drain pending record", "err", err, "did", p.Did, "rkey", p.Rkey, "nsid", p.Nsid)
1832 }
1833 }
1834}
1835
1836func (i *Ingester) drainPendingLabelOps(l *slog.Logger) {
1837 subjects, err := db.PendingStateSubjectsForNsid(i.Db, tangled.LabelOpNSID)
1838 if err != nil {
1839 l.Error("failed to list pending label op subjects", "err", err)
1840 return
1841 }
1842 for _, subject := range subjects {
1843 i.drainPendingState(i.Ctx, subject, l)
1844 }
1845}
1846
1847func (i *Ingester) reapplyPendingRecord(ctx context.Context, p db.PendingStateRecord, l *slog.Logger) error {
1848 switch p.Nsid {
1849 case tangled.RepoIssueStateNSID:
1850 return i.applyStateRecord(ctx, p.Did, p.Rkey, p.Nsid, p.Record, issueStateSpec, l)
1851 case tangled.RepoPullStatusNSID:
1852 return i.applyStateRecord(ctx, p.Did, p.Rkey, p.Nsid, p.Record, pullStatusSpec, l)
1853 case tangled.LabelOpNSID:
1854 return i.applyLabelOpRecord(ctx, p.Did, p.Rkey, p.Record, l)
1855 default:
1856 return fmt.Errorf("no reapply handler for parked nsid: %s", p.Nsid)
1857 }
1858}
1859
1860const (
1861 pendingStateReconcileInterval = time.Hour
1862 pendingStateRecordTTL = 7 * 24 * time.Hour
1863)
1864
1865func (i *Ingester) StartPendingStateReconciler() {
1866 i.ReconcilePendingState()
1867
1868 ticker := time.NewTicker(pendingStateReconcileInterval)
1869 defer ticker.Stop()
1870 for {
1871 select {
1872 case <-i.Ctx.Done():
1873 return
1874 case <-ticker.C:
1875 i.ReconcilePendingState()
1876 }
1877 }
1878}
1879
1880func (i *Ingester) ReconcilePendingState() {
1881 l := i.Logger.With("handler", "reconcilePendingState")
1882
1883 subjects, err := db.DistinctPendingStateSubjects(i.Db)
1884 if err != nil {
1885 l.Error("failed to list pending state subjects", "err", err)
1886 }
1887 for _, subject := range subjects {
1888 i.drainPendingState(i.Ctx, subject, l)
1889 }
1890
1891 cutoff := time.Now().Add(-pendingStateRecordTTL).UTC().Format(time.RFC3339)
1892 evicted, err := db.EvictStalePendingStateRecords(i.Db, cutoff)
1893 if err != nil {
1894 l.Error("failed to evict stale pending state records", "err", err)
1895 return
1896 }
1897 if evicted > 0 {
1898 l.Warn("evicted stale pending state records", "count", evicted, "olderThan", cutoff)
1899 }
1900}
1901
1902// ingestIssueComment ingests legacy sh.tangled.repo.issue.comment deletions
1903func (i *Ingester) ingestIssueComment(e *jmodels.Event, l *slog.Logger) error {
1904 l = l.With("handler", "ingestIssueComment")
1905
1906 switch e.Commit.Operation {
1907 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
1908 // no-op. sh.tangled.repo.issue.comment is deprecated
1909
1910 case jmodels.CommitOperationDelete:
1911 if err := db.PurgeComments(
1912 i.Db,
1913 orm.FilterEq("did", e.Did),
1914 orm.FilterEq("collection", e.Commit.Collection),
1915 orm.FilterEq("rkey", e.Commit.RKey),
1916 ); err != nil {
1917 return fmt.Errorf("failed to delete comment record: %w", err)
1918 }
1919 }
1920
1921 l.Info("ingested record")
1922 return nil
1923}
1924
1925// ingestPullComment ingests legacy sh.tangled.repo.pull.comment deletions
1926func (i *Ingester) ingestPullComment(e *jmodels.Event, l *slog.Logger) error {
1927 l = l.With("handler", "ingestPullComment")
1928
1929 switch e.Commit.Operation {
1930 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
1931 // no-op. sh.tangled.repo.pull.comment is deprecated
1932
1933 case jmodels.CommitOperationDelete:
1934 if err := db.PurgeComments(
1935 i.Db,
1936 orm.FilterEq("did", e.Did),
1937 orm.FilterEq("collection", e.Commit.Collection),
1938 orm.FilterEq("rkey", e.Commit.RKey),
1939 ); err != nil {
1940 return fmt.Errorf("failed to delete comment record: %w", err)
1941 }
1942 }
1943
1944 l.Info("ingested record")
1945 return nil
1946}
1947
1948func (i *Ingester) ingestComment(e *jmodels.Event, l *slog.Logger) error {
1949 did := e.Did
1950 rkey := e.Commit.RKey
1951 cid := e.Commit.CID
1952
1953 var err error
1954
1955 l = l.With("handler", "ingestComment")
1956
1957 ctx := context.Background()
1958
1959 switch e.Commit.Operation {
1960 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
1961 raw := json.RawMessage(e.Commit.Record)
1962 record := tangled.FeedComment{}
1963 err = json.Unmarshal(raw, &record)
1964 if err != nil {
1965 return fmt.Errorf("invalid record: %w", err)
1966 }
1967
1968 comment, err := models.CommentFromRecord(syntax.DID(did), syntax.RecordKey(rkey), syntax.CID(cid), record)
1969 if err != nil {
1970 return fmt.Errorf("failed to parse comment from record: %w", err)
1971 }
1972
1973 if err := comment.Validate(); err != nil {
1974 return fmt.Errorf("failed to validate comment: %w", err)
1975 }
1976
1977 var references []syntax.ATURI
1978 if comment.Body.Original != nil {
1979 _, references = i.MentionsResolver.Resolve(ctx, *comment.Body.Original)
1980 }
1981
1982 tx, err := i.Db.Begin()
1983 if err != nil {
1984 return fmt.Errorf("failed to start transaction: %w", err)
1985 }
1986 defer tx.Rollback()
1987
1988 _, err = db.PutComment(tx, comment, references)
1989 if err != nil {
1990 return fmt.Errorf("failed to create comment: %w", err)
1991 }
1992
1993 if err := tx.Commit(); err != nil {
1994 return err
1995 }
1996
1997 case jmodels.CommitOperationDelete:
1998 if err := db.DeleteComments(
1999 i.Db,
2000 orm.FilterEq("did", did),
2001 orm.FilterEq("collection", e.Commit.Collection),
2002 orm.FilterEq("rkey", rkey),
2003 ); err != nil {
2004 return fmt.Errorf("failed to delete comment record: %w", err)
2005 }
2006 }
2007
2008 l.Info("ingested record")
2009 return nil
2010}
2011
2012func (i *Ingester) ingestReaction(e *jmodels.Event, l *slog.Logger) error {
2013 did := e.Did
2014 rkey := e.Commit.RKey
2015
2016 l = l.With("handler", "ingestReaction")
2017
2018 switch e.Commit.Operation {
2019 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
2020 raw := json.RawMessage(e.Commit.Record)
2021 record := tangled.FeedReaction{}
2022 if err := json.Unmarshal(raw, &record); err != nil {
2023 return fmt.Errorf("invalid record: %w", err)
2024 }
2025
2026 subjectUri, err := syntax.ParseATURI(record.Subject)
2027 if err != nil {
2028 return fmt.Errorf("invalid reaction subject %q: %w", record.Subject, err)
2029 }
2030 subjectUri = models.NormalizeReactionSubject(subjectUri)
2031
2032 kind, ok := models.ParseReactionKind(record.Reaction)
2033 if !ok {
2034 return fmt.Errorf("invalid reaction kind: %q", record.Reaction)
2035 }
2036
2037 created, parseErr := time.Parse(time.RFC3339, record.CreatedAt)
2038 if parseErr != nil {
2039 created = time.Now()
2040 }
2041
2042 reaction := models.Reaction{
2043 ReactedByDid: did,
2044 Rkey: rkey,
2045 ThreadAt: subjectUri,
2046 Kind: kind,
2047 Created: created,
2048 }
2049 if err := db.UpsertReaction(i.Db, reaction); err != nil {
2050 return fmt.Errorf("failed to upsert reaction: %w", err)
2051 }
2052
2053 case jmodels.CommitOperationDelete:
2054 if err := db.DeleteReactionByRkey(i.Db, did, rkey); err != nil {
2055 return fmt.Errorf("failed to delete reaction record: %w", err)
2056 }
2057 }
2058
2059 l.Info("ingested record")
2060 return nil
2061}
2062
2063func (i *Ingester) ingestLabelDefinition(e *jmodels.Event, l *slog.Logger) error {
2064 did := e.Did
2065 rkey := e.Commit.RKey
2066
2067 var err error
2068
2069 l = l.With("handler", "ingestLabelDefinition")
2070
2071 switch e.Commit.Operation {
2072 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
2073 raw := json.RawMessage(e.Commit.Record)
2074 record := tangled.LabelDefinition{}
2075 err = json.Unmarshal(raw, &record)
2076 if err != nil {
2077 return fmt.Errorf("invalid record: %w", err)
2078 }
2079
2080 def, err := models.LabelDefinitionFromRecord(did, rkey, record)
2081 if err != nil {
2082 return fmt.Errorf("failed to parse labeldef from record: %w", err)
2083 }
2084
2085 if err := def.Validate(); err != nil {
2086 return fmt.Errorf("failed to validate labeldef: %w", err)
2087 }
2088
2089 _, err = db.AddLabelDefinition(i.Db, def)
2090 if err != nil {
2091 return fmt.Errorf("failed to create labeldef: %w", err)
2092 }
2093
2094 if e.Commit.Operation == jmodels.CommitOperationCreate {
2095 i.drainPendingLabelOps(l)
2096 }
2097
2098 l.Info("ingested record")
2099 return nil
2100
2101 case jmodels.CommitOperationDelete:
2102 if err := db.DeleteLabelDefinition(
2103 i.Db,
2104 orm.FilterEq("did", did),
2105 orm.FilterEq("rkey", rkey),
2106 ); err != nil {
2107 return fmt.Errorf("failed to delete labeldef record: %w", err)
2108 }
2109
2110 l.Info("ingested record")
2111 return nil
2112 }
2113
2114 return nil
2115}
2116
2117func (i *Ingester) ingestLabelOp(ctx context.Context, e *jmodels.Event, l *slog.Logger) error {
2118 l = l.With("handler", "ingestLabelOp")
2119 did := e.Did
2120 rkey := e.Commit.RKey
2121
2122 switch e.Commit.Operation {
2123 case jmodels.CommitOperationCreate, jmodels.CommitOperationUpdate:
2124 return i.applyLabelOpRecord(ctx, did, rkey, e.Commit.Record, l)
2125 case jmodels.CommitOperationDelete:
2126 return i.deleteLabelOpRecord(ctx, did, rkey, l)
2127 }
2128
2129 return nil
2130}
2131
2132func (i *Ingester) findLabelSubjectRepo(subject syntax.ATURI) (*models.Repo, bool, error) {
2133 var spec stateIngestSpec
2134 switch subject.Collection() {
2135 case tangled.RepoIssueNSID:
2136 spec = issueStateSpec
2137 case tangled.RepoPullNSID:
2138 spec = pullStatusSpec
2139 default:
2140 return nil, false, fmt.Errorf("unsupported label subject: %s", subject.Collection())
2141 }
2142 repo, _, found, err := spec.findSubject(i.Db, subject)
2143 return repo, found, err
2144}
2145
2146func (i *Ingester) applyLabelOpRecord(ctx context.Context, did, rkey string, raw []byte, l *slog.Logger) error {
2147 record := tangled.LabelOp{}
2148 if err := json.Unmarshal(raw, &record); err != nil {
2149 return fmt.Errorf("invalid record: %w", err)
2150 }
2151
2152 subject := syntax.ATURI(record.Subject)
2153 park := func() error {
2154 return i.parkStateRecord(ctx, did, rkey, tangled.LabelOpNSID, subject, raw, l)
2155 }
2156
2157 repo, found, err := i.findLabelSubjectRepo(subject)
2158 if err != nil {
2159 return err
2160 }
2161 if !found {
2162 return park()
2163 }
2164
2165 // validate permissions: only collaborators can apply labels currently
2166 //
2167 // TODO: introduce a repo:triage permission
2168 allowed, permErr := i.Acl.HasRepoPermissionErr(ctx, repo, did, "repo:push")
2169 if permErr != nil {
2170 if !errors.Is(permErr, knotacl.ErrKnotUnreachable) {
2171 return park()
2172 }
2173 l.Warn("ingesting labelop without permission check", "did", did, "err", permErr)
2174 allowed = true
2175 }
2176
2177 if !allowed {
2178 if err := i.unparkLabelOp(ctx, did, rkey); err != nil {
2179 return err
2180 }
2181 l.Warn("dropped unauthorized label op", "did", did, "rkey", rkey, "subject", subject)
2182 return nil
2183 }
2184
2185 ops := models.LabelOpsFromRecord(did, rkey, record)
2186
2187 operandKeys := make([]string, 0, len(ops))
2188 for idx := range ops {
2189 operandKeys = append(operandKeys, ops[idx].OperandKey)
2190 }
2191
2192 actx, err := db.NewLabelApplicationCtx(i.Db, orm.FilterIn("at_uri", operandKeys))
2193 if err != nil {
2194 return fmt.Errorf("failed to build label application ctx: %w", err)
2195 }
2196
2197 for idx := range ops {
2198 def, ok := actx.Defs[ops[idx].OperandKey]
2199 if !ok {
2200 return park()
2201 }
2202 if err := def.ValidateOperandValue(&ops[idx]); err != nil {
2203 return fmt.Errorf("failed to validate labelop: %w", err)
2204 }
2205 }
2206
2207 if err := i.materializeLabelOps(ctx, did, rkey, ops); err != nil {
2208 return err
2209 }
2210
2211 l.Info("ingested record")
2212 return nil
2213}
2214
2215func (i *Ingester) unparkLabelOp(ctx context.Context, did, rkey string) error {
2216 tx, err := i.Db.BeginTx(ctx, nil)
2217 if err != nil {
2218 return err
2219 }
2220 defer tx.Rollback()
2221
2222 if err := db.UnparkStateRecord(tx, did, rkey, tangled.LabelOpNSID); err != nil {
2223 return fmt.Errorf("failed to unpark label op: %w", err)
2224 }
2225
2226 return tx.Commit()
2227}
2228
2229func (i *Ingester) materializeLabelOps(ctx context.Context, did, rkey string, ops []models.LabelOp) error {
2230 tx, err := i.Db.BeginTx(ctx, nil)
2231 if err != nil {
2232 return err
2233 }
2234 defer tx.Rollback()
2235
2236 if err := db.UnparkStateRecord(tx, did, rkey, tangled.LabelOpNSID); err != nil {
2237 return fmt.Errorf("failed to unpark label op: %w", err)
2238 }
2239 if err := db.DeleteLabelOps(tx, orm.FilterEq("did", did), orm.FilterEq("rkey", rkey)); err != nil {
2240 return fmt.Errorf("failed to clear prior label ops: %w", err)
2241 }
2242 for idx := range ops {
2243 if _, err := db.AddLabelOp(tx, &ops[idx]); err != nil {
2244 return fmt.Errorf("failed to add labelop: %w", err)
2245 }
2246 }
2247 return tx.Commit()
2248}
2249
2250func (i *Ingester) deleteLabelOpRecord(ctx context.Context, did, rkey string, l *slog.Logger) error {
2251 if err := i.materializeLabelOps(ctx, did, rkey, nil); err != nil {
2252 return err
2253 }
2254 l.Info("ingested record")
2255 return nil
2256}