Something went wrong. Try again.
Monorepo for Tangled
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446package appview
import ( "context" "database/sql" "encoding/json" "errors" "fmt" "log/slog" "slices" "strings"
"github.com/bluesky-social/indigo/atproto/syntax" jmodels "github.com/bluesky-social/jetstream/pkg/models" "tangled.org/core/api/tangled" "tangled.org/core/appview/db" "tangled.org/core/appview/models" "tangled.org/core/appview/repoverify" "tangled.org/core/orm")
func (i *Ingester) ingestRepo(ctx context.Context, e *jmodels.Event, l *slog.Logger) error { l = l.With("handler", "ingestRepo")
switch e.Commit.Operation { case jmodels.CommitOperationCreate: return i.ingestRepoCreate(ctx, e, l) case jmodels.CommitOperationUpdate: return i.ingestRepoUpdate(ctx, e, l) case jmodels.CommitOperationDelete: return i.ingestRepoDelete(ctx, e, l) default: l.Info("unknown repo operation") return nil }}
func (i *Ingester) ingestRepoCreate(ctx context.Context, e *jmodels.Event, l *slog.Logger) error { l = l.With("handler", "ingestRepoCreate")
record := tangled.Repo{} if err := json.Unmarshal(json.RawMessage(e.Commit.Record), &record); err != nil { l.Error("invalid record", "err", err) return err }
if record.RepoDid == nil || *record.RepoDid == "" { l.Info("skipping repo create from non-DID-migrated knot") return nil } repoDid := *record.RepoDid
proceed, err := i.verifyOwnership(ctx, l, repoDid, e.Did, record.Knot) if err != nil { return err } if !proceed { return nil }
existing, err := db.GetRepo(i.Db, orm.FilterEq("did", e.Did), orm.FilterEq("rkey", e.Commit.RKey), ) if err == nil { l.Info("repo row already exists, skipping create", "did", e.Did, "rkey", e.Commit.RKey) if err := i.ensureRepoOwnerPermissions(e.Did, existing.Knot, existing.RepoIdentifier()); err != nil { return fmt.Errorf("failed to ensure repo owner permissions: %w", err) } return nil } if !errors.Is(err, sql.ErrNoRows) { return fmt.Errorf("failed to check existing repo: %w", err) }
prev, err := db.GetRepoByDid(i.Db, repoDid) if err != nil && !errors.Is(err, sql.ErrNoRows) { return fmt.Errorf("failed to check existing repoDid: %w", err) }
if prev != nil { l.Info("repoDid exists under different rkey, renaming", "oldRkey", prev.Rkey, "newRkey", e.Commit.RKey)
oldRepo := *prev
tx, txErr := i.Db.Begin() if txErr != nil { return fmt.Errorf("failed to begin rename tx: %w", txErr) } defer tx.Rollback()
newName := derefString(record.Name) if newName == "" { newName = e.Commit.RKey }
if err := db.RenameRepo(tx, e.Did, prev.Rkey, e.Commit.RKey, newName); err != nil { return fmt.Errorf("failed to rename repo: %w", err) } if err := db.RecordRepoRename(tx, e.Did, prev.Rkey, repoDid); err != nil { return fmt.Errorf("failed to record rename history: %w", err) }
renamed := *prev renamed.Rkey = e.Commit.RKey renamed.Name = newName desired := repoFromRecord(&renamed, &record) if repoMetadataChanged(&renamed, &desired) { if err := applyRepoMetadata(tx, &renamed, desired); err != nil { return fmt.Errorf("failed to apply metadata after rename: %w", err) } }
if err := tx.Commit(); err != nil { return fmt.Errorf("failed to commit rename tx: %w", err) }
newRepo, err := db.GetRepo(i.Db, orm.FilterEq("did", e.Did), orm.FilterEq("rkey", e.Commit.RKey), ) if err != nil { l.Warn("failed to fetch repo after rename for notification", "err", err) return nil } if err := i.ensureRepoOwnerPermissions(e.Did, newRepo.Knot, newRepo.RepoIdentifier()); err != nil { return fmt.Errorf("failed to ensure repo owner permissions: %w", err) } i.Notifier.RenameRepo(ctx, syntax.DID(e.Did), &oldRepo, newRepo) return nil }
rkey := e.Commit.RKey name := derefString(record.Name) if name == "" { name = rkey }
repo := &models.Repo{ Did: e.Did, Name: name, Knot: record.Knot, Rkey: rkey, Description: derefString(record.Description), Website: derefString(record.Website), Topics: append([]string(nil), record.Topics...), Source: derefString(record.Source), Spindle: derefString(record.Spindle), Labels: append([]string(nil), record.Labels...), RepoDid: repoDid, }
tx, err := i.Db.Begin() if err != nil { return fmt.Errorf("failed to begin insert tx: %w", err) } defer tx.Rollback()
if err := db.AddRepo(tx, repo); err != nil { return fmt.Errorf("failed to insert repo: %w", err) } if err := tx.Commit(); err != nil { return fmt.Errorf("failed to commit insert tx: %w", err) }
if err := i.ensureRepoOwnerPermissions(e.Did, repo.Knot, repo.RepoIdentifier()); err != nil { return fmt.Errorf("failed to ensure repo owner permissions: %w", err) }
i.Notifier.NewRepo(ctx, repo) return nil}
func (i *Ingester) ensureRepoOwnerPermissions(ownerDid, knot, repo string) error { if i.Enforcer == nil { return fmt.Errorf("ingester has no RBAC enforcer configured") } if err := i.Enforcer.AddRepo(ownerDid, knot, repo); err != nil { return err } return i.Enforcer.E.SavePolicy()}
func (i *Ingester) ingestRepoUpdate(ctx context.Context, e *jmodels.Event, l *slog.Logger) error { l = l.With("handler", "ingestRepoUpdate")
record := tangled.Repo{} if err := json.Unmarshal(json.RawMessage(e.Commit.Record), &record); err != nil { l.Error("invalid record", "err", err) return err }
if record.RepoDid == nil || *record.RepoDid == "" { l.Info("skipping repo update from non-DID-migrated knot") return nil }
proceed, err := i.verifyOwnership(ctx, l, *record.RepoDid, e.Did, record.Knot) if err != nil { return err } if !proceed { return nil }
current, err := db.GetRepo(i.Db, orm.FilterEq("did", e.Did), orm.FilterEq("rkey", e.Commit.RKey), ) if err != nil { if errors.Is(err, sql.ErrNoRows) { l.Info("skipping repo update for unknown row") return nil } return fmt.Errorf("failed to fetch repo for ingest: %w", err) }
if current.RepoDid != "" && current.RepoDid != *record.RepoDid { l.Warn("rejecting repo update: repoDid is immutable", "currentRepoDid", current.RepoDid, "recordRepoDid", *record.RepoDid, ) return nil }
desired := repoFromRecord(current, &record)
if current.Source != desired.Source { l.Warn("source field changed but mutation is unsupported, ignoring", "current", current.Source, "desired", desired.Source) }
if !repoMetadataChanged(current, &desired) { return nil }
tx, err := i.Db.Begin() if err != nil { return fmt.Errorf("failed to begin tx: %w", err) } defer tx.Rollback()
if err := applyRepoMetadata(tx, current, desired); err != nil { return fmt.Errorf("failed to apply repo metadata: %w", err) } return tx.Commit()}
func (i *Ingester) ingestRepoDelete(ctx context.Context, e *jmodels.Event, l *slog.Logger) error { l = l.With("handler", "ingestRepoDelete")
repo, err := db.GetRepo(i.Db, orm.FilterEq("did", e.Did), orm.FilterEq("rkey", e.Commit.RKey), ) if err != nil { if errors.Is(err, sql.ErrNoRows) { l.Info("skipping repo delete for unknown row") return nil } return fmt.Errorf("failed to fetch repo for delete: %w", err) }
if i.Enforcer == nil { return fmt.Errorf("ingester has no RBAC enforcer configured") }
tx, err := i.Db.Begin() if err != nil { return fmt.Errorf("failed to start txn: %w", err) } committed := false defer func() { if committed { return } tx.Rollback() i.Enforcer.E.LoadPolicy() }()
if err := db.RemoveRepo(tx, e.Did, e.Commit.RKey); err != nil { return fmt.Errorf("failed to delete repo: %w", err) }
if err := i.Enforcer.WipeRepoPolicies(repo.Knot, repo.RepoIdentifier()); err != nil { return fmt.Errorf("failed to wipe repo permissions: %w", err) }
if err := tx.Commit(); err != nil { return fmt.Errorf("failed to commit txn: %w", err) }
if err := i.Enforcer.E.SavePolicy(); err != nil { return fmt.Errorf("failed to save ACLs: %w", err) } committed = true
i.Notifier.DeleteRepo(ctx, repo) l.Info("deleted repo row") return nil}
func applyRepoMetadata(tx *sql.Tx, current *models.Repo, desired models.Repo) error { if err := db.PutRepo(tx, desired); err != nil { return err }
if current.Spindle != desired.Spindle { var spindlePtr *string if desired.Spindle != "" { spindlePtr = &desired.Spindle } if err := db.UpdateSpindle(tx, desired.RepoDid, spindlePtr); err != nil { return err } }
if !labelsEqual(current.Labels, desired.Labels) { if err := reconcileLabels(tx, current, desired); err != nil { return err } }
return nil}
func reconcileLabels(tx *sql.Tx, current *models.Repo, desired models.Repo) error { added := filterOut(desired.Labels, current.Labels) removed := filterOut(current.Labels, desired.Labels)
if err := applyEach(added, func(l string) error { return db.SubscribeLabel(tx, &models.RepoLabel{ RepoDid: syntax.DID(desired.RepoDid), LabelAt: syntax.ATURI(l), }) }); err != nil { return err }
return applyEach(removed, func(l string) error { return db.UnsubscribeLabel(tx, orm.FilterEq("repo_did", desired.RepoDid), orm.FilterEq("label_at", l), ) })}
func filterOut(items, exclude []string) []string { return slices.DeleteFunc(slices.Clone(items), func(s string) bool { return slices.Contains(exclude, s) })}
func applyEach(items []string, fn func(string) error) error { for _, item := range items { if err := fn(item); err != nil { return err } } return nil}
func labelsEqual(a, b []string) bool { if len(a) != len(b) { return false } aSorted := append([]string(nil), a...) bSorted := append([]string(nil), b...) slices.Sort(aSorted) slices.Sort(bSorted) return slices.Equal(aSorted, bSorted)}
func repoFromRecord(current *models.Repo, record *tangled.Repo) models.Repo { out := *current out.Name = derefString(record.Name) if out.Name == "" { out.Name = current.Rkey } out.Knot = record.Knot out.Description = derefString(record.Description) out.Website = derefString(record.Website) out.Topics = append([]string(nil), record.Topics...) out.Spindle = derefString(record.Spindle) out.Source = derefString(record.Source) out.Labels = append([]string(nil), record.Labels...) if record.RepoDid != nil { out.RepoDid = *record.RepoDid } return out}
func repoMetadataChanged(current *models.Repo, desired *models.Repo) bool { return current.Name != desired.Name || current.Knot != desired.Knot || current.Description != desired.Description || current.Website != desired.Website || current.TopicStr() != desired.TopicStr() || current.Spindle != desired.Spindle || !labelsEqual(current.Labels, desired.Labels)}
func derefString(s *string) string { if s == nil { return "" } return *s}
func (i *Ingester) verifyOwnership(ctx context.Context, l *slog.Logger, repoDid, eventDid, recordKnot string) (bool, error) { if i.Verifier == nil { return false, fmt.Errorf("ingester has no repo ownership verifier configured") } rd, err := repoverify.NewRepoDid(repoDid) if err != nil { l.Warn("rejecting repo event: invalid repoDid on record", "repoDid", repoDid, "err", err) return false, nil } result, err := i.Verifier(ctx, rd) if err != nil { return false, fmt.Errorf("verify repo ownership: %w", err) } if result.OwnerDid == "" { l.Warn("knot lacks RepoDescribeRepo, skipping owner check; upgrade knot to 1.14+", "repoDid", repoDid, "knot", result.KnotURL.String()) } else if result.OwnerDid.String() != eventDid { l.Warn("rejecting repo event: owner mismatch", "repoDid", repoDid, "claimedOwner", eventDid, "knotOwner", result.OwnerDid.String(), "knot", result.KnotURL.String(), ) return false, nil } if !strings.EqualFold(recordKnot, result.KnotURL.Host) { l.Warn("rejecting repo event: record knot does not match DID-doc endpoint", "repoDid", repoDid, "recordKnot", recordKnot, "canonicalKnot", result.KnotURL.Host, ) return false, nil } return true, nil}