this repo has no description
1package spindles 2 3import ( 4 "errors" 5 "fmt" 6 "log/slog" 7 "net/http" 8 "slices" 9 "time" 10 11 "github.com/go-chi/chi/v5" 12 "tangled.sh/tangled.sh/core/api/tangled" 13 "tangled.sh/tangled.sh/core/appview/config" 14 "tangled.sh/tangled.sh/core/appview/db" 15 "tangled.sh/tangled.sh/core/appview/middleware" 16 "tangled.sh/tangled.sh/core/appview/oauth" 17 "tangled.sh/tangled.sh/core/appview/pages" 18 verify "tangled.sh/tangled.sh/core/appview/spindleverify" 19 "tangled.sh/tangled.sh/core/idresolver" 20 "tangled.sh/tangled.sh/core/rbac" 21 "tangled.sh/tangled.sh/core/tid" 22 23 comatproto "github.com/bluesky-social/indigo/api/atproto" 24 "github.com/bluesky-social/indigo/atproto/syntax" 25 lexutil "github.com/bluesky-social/indigo/lex/util" 26) 27 28type Spindles struct { 29 Db *db.DB 30 OAuth *oauth.OAuth 31 Pages *pages.Pages 32 Config *config.Config 33 Enforcer *rbac.Enforcer 34 IdResolver *idresolver.Resolver 35 Logger *slog.Logger 36} 37 38func (s *Spindles) Router() http.Handler { 39 r := chi.NewRouter() 40 41 r.With(middleware.AuthMiddleware(s.OAuth)).Get("/", s.spindles) 42 r.With(middleware.AuthMiddleware(s.OAuth)).Post("/register", s.register) 43 44 r.With(middleware.AuthMiddleware(s.OAuth)).Get("/{instance}", s.dashboard) 45 r.With(middleware.AuthMiddleware(s.OAuth)).Delete("/{instance}", s.delete) 46 47 r.With(middleware.AuthMiddleware(s.OAuth)).Post("/{instance}/retry", s.retry) 48 r.With(middleware.AuthMiddleware(s.OAuth)).Post("/{instance}/add", s.addMember) 49 r.With(middleware.AuthMiddleware(s.OAuth)).Post("/{instance}/remove", s.removeMember) 50 51 return r 52} 53 54func (s *Spindles) spindles(w http.ResponseWriter, r *http.Request) { 55 user := s.OAuth.GetUser(r) 56 all, err := db.GetSpindles( 57 s.Db, 58 db.FilterEq("owner", user.Did), 59 ) 60 if err != nil { 61 s.Logger.Error("failed to fetch spindles", "err", err) 62 w.WriteHeader(http.StatusInternalServerError) 63 return 64 } 65 66 s.Pages.Spindles(w, pages.SpindlesParams{ 67 LoggedInUser: user, 68 Spindles: all, 69 }) 70} 71 72func (s *Spindles) dashboard(w http.ResponseWriter, r *http.Request) { 73 l := s.Logger.With("handler", "dashboard") 74 75 user := s.OAuth.GetUser(r) 76 l = l.With("user", user.Did) 77 78 instance := chi.URLParam(r, "instance") 79 if instance == "" { 80 return 81 } 82 l = l.With("instance", instance) 83 84 spindles, err := db.GetSpindles( 85 s.Db, 86 db.FilterEq("instance", instance), 87 db.FilterEq("owner", user.Did), 88 db.FilterIsNot("verified", "null"), 89 ) 90 if err != nil || len(spindles) != 1 { 91 l.Error("failed to get spindle", "err", err, "len(spindles)", len(spindles)) 92 http.Error(w, "Not found", http.StatusNotFound) 93 return 94 } 95 96 spindle := spindles[0] 97 members, err := s.Enforcer.GetSpindleUsersByRole("server:member", spindle.Instance) 98 if err != nil { 99 l.Error("failed to get spindle members", "err", err) 100 http.Error(w, "Not found", http.StatusInternalServerError) 101 return 102 } 103 slices.Sort(members) 104 105 repos, err := db.GetRepos( 106 s.Db, 107 0, 108 db.FilterEq("spindle", instance), 109 ) 110 if err != nil { 111 l.Error("failed to get spindle repos", "err", err) 112 http.Error(w, "Not found", http.StatusInternalServerError) 113 return 114 } 115 116 identsToResolve := make([]string, len(members)) 117 copy(identsToResolve, members) 118 resolvedIds := s.IdResolver.ResolveIdents(r.Context(), identsToResolve) 119 didHandleMap := make(map[string]string) 120 for _, identity := range resolvedIds { 121 if !identity.Handle.IsInvalidHandle() { 122 didHandleMap[identity.DID.String()] = fmt.Sprintf("@%s", identity.Handle.String()) 123 } else { 124 didHandleMap[identity.DID.String()] = identity.DID.String() 125 } 126 } 127 128 // organize repos by did 129 repoMap := make(map[string][]db.Repo) 130 for _, r := range repos { 131 repoMap[r.Did] = append(repoMap[r.Did], r) 132 } 133 134 s.Pages.SpindleDashboard(w, pages.SpindleDashboardParams{ 135 LoggedInUser: user, 136 Spindle: spindle, 137 Members: members, 138 Repos: repoMap, 139 DidHandleMap: didHandleMap, 140 }) 141} 142 143// this endpoint inserts a record on behalf of the user to register that domain 144// 145// when registered, it also makes a request to see if the spindle declares this users as its owner, 146// and if so, marks the spindle as verified. 147// 148// if the spindle is not up yet, the user is free to retry verification at a later point 149func (s *Spindles) register(w http.ResponseWriter, r *http.Request) { 150 user := s.OAuth.GetUser(r) 151 l := s.Logger.With("handler", "register") 152 153 noticeId := "register-error" 154 defaultErr := "Failed to register spindle. Try again later." 155 fail := func() { 156 s.Pages.Notice(w, noticeId, defaultErr) 157 } 158 159 instance := r.FormValue("instance") 160 if instance == "" { 161 s.Pages.Notice(w, noticeId, "Incomplete form.") 162 return 163 } 164 l = l.With("instance", instance) 165 l = l.With("user", user.Did) 166 167 tx, err := s.Db.Begin() 168 if err != nil { 169 l.Error("failed to start transaction", "err", err) 170 fail() 171 return 172 } 173 defer func() { 174 tx.Rollback() 175 s.Enforcer.E.LoadPolicy() 176 }() 177 178 err = db.AddSpindle(tx, db.Spindle{ 179 Owner: syntax.DID(user.Did), 180 Instance: instance, 181 }) 182 if err != nil { 183 l.Error("failed to insert", "err", err) 184 fail() 185 return 186 } 187 188 err = s.Enforcer.AddSpindle(instance) 189 if err != nil { 190 l.Error("failed to create spindle", "err", err) 191 fail() 192 return 193 } 194 195 // create record on pds 196 client, err := s.OAuth.AuthorizedClient(r) 197 if err != nil { 198 l.Error("failed to authorize client", "err", err) 199 fail() 200 return 201 } 202 203 ex, _ := client.RepoGetRecord(r.Context(), "", tangled.SpindleNSID, user.Did, instance) 204 var exCid *string 205 if ex != nil { 206 exCid = ex.Cid 207 } 208 209 // re-announce by registering under same rkey 210 _, err = client.RepoPutRecord(r.Context(), &comatproto.RepoPutRecord_Input{ 211 Collection: tangled.SpindleNSID, 212 Repo: user.Did, 213 Rkey: instance, 214 Record: &lexutil.LexiconTypeDecoder{ 215 Val: &tangled.Spindle{ 216 CreatedAt: time.Now().Format(time.RFC3339), 217 }, 218 }, 219 SwapRecord: exCid, 220 }) 221 222 if err != nil { 223 l.Error("failed to put record", "err", err) 224 fail() 225 return 226 } 227 228 err = tx.Commit() 229 if err != nil { 230 l.Error("failed to commit transaction", "err", err) 231 fail() 232 return 233 } 234 235 err = s.Enforcer.E.SavePolicy() 236 if err != nil { 237 l.Error("failed to update ACL", "err", err) 238 s.Pages.HxRefresh(w) 239 return 240 } 241 242 // begin verification 243 err = verify.RunVerification(r.Context(), instance, user.Did, s.Config.Core.Dev) 244 if err != nil { 245 l.Error("verification failed", "err", err) 246 s.Pages.HxRefresh(w) 247 return 248 } 249 250 _, err = verify.MarkVerified(s.Db, s.Enforcer, instance, user.Did) 251 if err != nil { 252 l.Error("failed to mark verified", "err", err) 253 s.Pages.HxRefresh(w) 254 return 255 } 256 257 // ok 258 s.Pages.HxRefresh(w) 259} 260 261func (s *Spindles) delete(w http.ResponseWriter, r *http.Request) { 262 user := s.OAuth.GetUser(r) 263 l := s.Logger.With("handler", "delete") 264 265 noticeId := "operation-error" 266 defaultErr := "Failed to delete spindle. Try again later." 267 fail := func() { 268 s.Pages.Notice(w, noticeId, defaultErr) 269 } 270 271 instance := chi.URLParam(r, "instance") 272 if instance == "" { 273 l.Error("empty instance") 274 fail() 275 return 276 } 277 278 spindles, err := db.GetSpindles( 279 s.Db, 280 db.FilterEq("owner", user.Did), 281 db.FilterEq("instance", instance), 282 ) 283 if err != nil || len(spindles) != 1 { 284 l.Error("failed to retrieve instance", "err", err, "len(spindles)", len(spindles)) 285 fail() 286 return 287 } 288 289 if string(spindles[0].Owner) != user.Did { 290 l.Error("unauthorized", "user", user.Did, "owner", spindles[0].Owner) 291 s.Pages.Notice(w, noticeId, "Failed to delete spindle, unauthorized deletion attempt.") 292 return 293 } 294 295 tx, err := s.Db.Begin() 296 if err != nil { 297 l.Error("failed to start txn", "err", err) 298 fail() 299 return 300 } 301 defer func() { 302 tx.Rollback() 303 s.Enforcer.E.LoadPolicy() 304 }() 305 306 err = db.DeleteSpindle( 307 tx, 308 db.FilterEq("owner", user.Did), 309 db.FilterEq("instance", instance), 310 ) 311 if err != nil { 312 l.Error("failed to delete spindle", "err", err) 313 fail() 314 return 315 } 316 317 // delete from enforcer 318 if spindles[0].Verified != nil { 319 err = s.Enforcer.RemoveSpindle(instance) 320 if err != nil { 321 l.Error("failed to update ACL", "err", err) 322 fail() 323 return 324 } 325 } 326 327 client, err := s.OAuth.AuthorizedClient(r) 328 if err != nil { 329 l.Error("failed to authorize client", "err", err) 330 fail() 331 return 332 } 333 334 _, err = client.RepoDeleteRecord(r.Context(), &comatproto.RepoDeleteRecord_Input{ 335 Collection: tangled.SpindleNSID, 336 Repo: user.Did, 337 Rkey: instance, 338 }) 339 if err != nil { 340 // non-fatal 341 l.Error("failed to delete record", "err", err) 342 } 343 344 err = tx.Commit() 345 if err != nil { 346 l.Error("failed to delete spindle", "err", err) 347 fail() 348 return 349 } 350 351 err = s.Enforcer.E.SavePolicy() 352 if err != nil { 353 l.Error("failed to update ACL", "err", err) 354 s.Pages.HxRefresh(w) 355 return 356 } 357 358 shouldRedirect := r.Header.Get("shouldRedirect") 359 if shouldRedirect == "true" { 360 s.Pages.HxRedirect(w, "/spindles") 361 return 362 } 363 364 w.Write([]byte{}) 365} 366 367func (s *Spindles) retry(w http.ResponseWriter, r *http.Request) { 368 user := s.OAuth.GetUser(r) 369 l := s.Logger.With("handler", "retry") 370 371 noticeId := "operation-error" 372 defaultErr := "Failed to verify spindle. Try again later." 373 fail := func() { 374 s.Pages.Notice(w, noticeId, defaultErr) 375 } 376 377 instance := chi.URLParam(r, "instance") 378 if instance == "" { 379 l.Error("empty instance") 380 fail() 381 return 382 } 383 l = l.With("instance", instance) 384 l = l.With("user", user.Did) 385 386 spindles, err := db.GetSpindles( 387 s.Db, 388 db.FilterEq("owner", user.Did), 389 db.FilterEq("instance", instance), 390 ) 391 if err != nil || len(spindles) != 1 { 392 l.Error("failed to retrieve instance", "err", err, "len(spindles)", len(spindles)) 393 fail() 394 return 395 } 396 397 if string(spindles[0].Owner) != user.Did { 398 l.Error("unauthorized", "user", user.Did, "owner", spindles[0].Owner) 399 s.Pages.Notice(w, noticeId, "Failed to verify spindle, unauthorized verification attempt.") 400 return 401 } 402 403 // begin verification 404 err = verify.RunVerification(r.Context(), instance, user.Did, s.Config.Core.Dev) 405 if err != nil { 406 l.Error("verification failed", "err", err) 407 408 if errors.Is(err, verify.FetchError) { 409 s.Pages.Notice(w, noticeId, err.Error()) 410 return 411 } 412 413 if e, ok := err.(*verify.OwnerMismatch); ok { 414 s.Pages.Notice(w, noticeId, e.Error()) 415 return 416 } 417 418 fail() 419 return 420 } 421 422 rowId, err := verify.MarkVerified(s.Db, s.Enforcer, instance, user.Did) 423 if err != nil { 424 l.Error("failed to mark verified", "err", err) 425 s.Pages.Notice(w, noticeId, err.Error()) 426 return 427 } 428 429 verifiedSpindle, err := db.GetSpindles( 430 s.Db, 431 db.FilterEq("id", rowId), 432 ) 433 if err != nil || len(verifiedSpindle) != 1 { 434 l.Error("failed get new spindle", "err", err) 435 s.Pages.HxRefresh(w) 436 return 437 } 438 439 shouldRefresh := r.Header.Get("shouldRefresh") 440 if shouldRefresh == "true" { 441 s.Pages.HxRefresh(w) 442 return 443 } 444 445 w.Header().Set("HX-Reswap", "outerHTML") 446 s.Pages.SpindleListing(w, pages.SpindleListingParams{verifiedSpindle[0]}) 447} 448 449func (s *Spindles) addMember(w http.ResponseWriter, r *http.Request) { 450 user := s.OAuth.GetUser(r) 451 l := s.Logger.With("handler", "addMember") 452 453 instance := chi.URLParam(r, "instance") 454 if instance == "" { 455 l.Error("empty instance") 456 http.Error(w, "Not found", http.StatusNotFound) 457 return 458 } 459 l = l.With("instance", instance) 460 l = l.With("user", user.Did) 461 462 spindles, err := db.GetSpindles( 463 s.Db, 464 db.FilterEq("owner", user.Did), 465 db.FilterEq("instance", instance), 466 ) 467 if err != nil || len(spindles) != 1 { 468 l.Error("failed to retrieve instance", "err", err, "len(spindles)", len(spindles)) 469 http.Error(w, "Not found", http.StatusNotFound) 470 return 471 } 472 473 noticeId := fmt.Sprintf("add-member-error-%d", spindles[0].Id) 474 defaultErr := "Failed to add member. Try again later." 475 fail := func() { 476 s.Pages.Notice(w, noticeId, defaultErr) 477 } 478 479 if string(spindles[0].Owner) != user.Did { 480 l.Error("unauthorized", "user", user.Did, "owner", spindles[0].Owner) 481 s.Pages.Notice(w, noticeId, "Failed to add member, unauthorized attempt.") 482 return 483 } 484 485 member := r.FormValue("member") 486 if member == "" { 487 l.Error("empty member") 488 s.Pages.Notice(w, noticeId, "Failed to add member, empty form.") 489 return 490 } 491 l = l.With("member", member) 492 493 memberId, err := s.IdResolver.ResolveIdent(r.Context(), member) 494 if err != nil { 495 l.Error("failed to resolve member identity to handle", "err", err) 496 s.Pages.Notice(w, noticeId, "Failed to add member, identity resolution failed.") 497 return 498 } 499 if memberId.Handle.IsInvalidHandle() { 500 l.Error("failed to resolve member identity to handle") 501 s.Pages.Notice(w, noticeId, "Failed to add member, identity resolution failed.") 502 return 503 } 504 505 // write to pds 506 client, err := s.OAuth.AuthorizedClient(r) 507 if err != nil { 508 l.Error("failed to authorize client", "err", err) 509 fail() 510 return 511 } 512 513 tx, err := s.Db.Begin() 514 if err != nil { 515 l.Error("failed to start txn", "err", err) 516 fail() 517 return 518 } 519 defer func() { 520 tx.Rollback() 521 s.Enforcer.E.LoadPolicy() 522 }() 523 524 rkey := tid.TID() 525 526 // add member to db 527 if err = db.AddSpindleMember(tx, db.SpindleMember{ 528 Did: syntax.DID(user.Did), 529 Rkey: rkey, 530 Instance: instance, 531 Subject: memberId.DID, 532 }); err != nil { 533 l.Error("failed to add spindle member", "err", err) 534 fail() 535 return 536 } 537 538 if err = s.Enforcer.AddSpindleMember(instance, memberId.DID.String()); err != nil { 539 l.Error("failed to add member to ACLs") 540 fail() 541 return 542 } 543 544 _, err = client.RepoPutRecord(r.Context(), &comatproto.RepoPutRecord_Input{ 545 Collection: tangled.SpindleMemberNSID, 546 Repo: user.Did, 547 Rkey: rkey, 548 Record: &lexutil.LexiconTypeDecoder{ 549 Val: &tangled.SpindleMember{ 550 CreatedAt: time.Now().Format(time.RFC3339), 551 Instance: instance, 552 Subject: memberId.DID.String(), 553 }, 554 }, 555 }) 556 if err != nil { 557 l.Error("failed to add record to PDS", "err", err) 558 s.Pages.Notice(w, noticeId, "Failed to add record to PDS, try again later.") 559 return 560 } 561 562 if err = tx.Commit(); err != nil { 563 l.Error("failed to commit txn", "err", err) 564 fail() 565 return 566 } 567 568 if err = s.Enforcer.E.SavePolicy(); err != nil { 569 l.Error("failed to add member to ACLs", "err", err) 570 fail() 571 return 572 } 573 574 // success 575 s.Pages.HxRedirect(w, fmt.Sprintf("/spindles/%s", instance)) 576} 577 578func (s *Spindles) removeMember(w http.ResponseWriter, r *http.Request) { 579 user := s.OAuth.GetUser(r) 580 l := s.Logger.With("handler", "removeMember") 581 582 noticeId := "operation-error" 583 defaultErr := "Failed to remove member. Try again later." 584 fail := func() { 585 s.Pages.Notice(w, noticeId, defaultErr) 586 } 587 588 instance := chi.URLParam(r, "instance") 589 if instance == "" { 590 l.Error("empty instance") 591 fail() 592 return 593 } 594 l = l.With("instance", instance) 595 l = l.With("user", user.Did) 596 597 spindles, err := db.GetSpindles( 598 s.Db, 599 db.FilterEq("owner", user.Did), 600 db.FilterEq("instance", instance), 601 ) 602 if err != nil || len(spindles) != 1 { 603 l.Error("failed to retrieve instance", "err", err, "len(spindles)", len(spindles)) 604 fail() 605 return 606 } 607 608 if string(spindles[0].Owner) != user.Did { 609 l.Error("unauthorized", "user", user.Did, "owner", spindles[0].Owner) 610 s.Pages.Notice(w, noticeId, "Failed to add member, unauthorized attempt.") 611 return 612 } 613 614 member := r.FormValue("member") 615 if member == "" { 616 l.Error("empty member") 617 s.Pages.Notice(w, noticeId, "Failed to add member, empty form.") 618 return 619 } 620 l = l.With("member", member) 621 622 memberId, err := s.IdResolver.ResolveIdent(r.Context(), member) 623 if err != nil { 624 l.Error("failed to resolve member identity to handle", "err", err) 625 s.Pages.Notice(w, noticeId, "Failed to add member, identity resolution failed.") 626 return 627 } 628 if memberId.Handle.IsInvalidHandle() { 629 l.Error("failed to resolve member identity to handle") 630 s.Pages.Notice(w, noticeId, "Failed to add member, identity resolution failed.") 631 return 632 } 633 634 tx, err := s.Db.Begin() 635 if err != nil { 636 l.Error("failed to start txn", "err", err) 637 fail() 638 return 639 } 640 defer func() { 641 tx.Rollback() 642 s.Enforcer.E.LoadPolicy() 643 }() 644 645 // get the record from the DB first: 646 members, err := db.GetSpindleMembers( 647 s.Db, 648 db.FilterEq("did", user.Did), 649 db.FilterEq("instance", instance), 650 db.FilterEq("subject", memberId.DID), 651 ) 652 if err != nil || len(members) != 1 { 653 l.Error("failed to get member", "err", err) 654 fail() 655 return 656 } 657 658 // remove from db 659 if err = db.RemoveSpindleMember( 660 tx, 661 db.FilterEq("did", user.Did), 662 db.FilterEq("instance", instance), 663 db.FilterEq("subject", memberId.DID), 664 ); err != nil { 665 l.Error("failed to remove spindle member", "err", err) 666 fail() 667 return 668 } 669 670 // remove from enforcer 671 if err = s.Enforcer.RemoveSpindleMember(instance, memberId.DID.String()); err != nil { 672 l.Error("failed to update ACLs", "err", err) 673 fail() 674 return 675 } 676 677 client, err := s.OAuth.AuthorizedClient(r) 678 if err != nil { 679 l.Error("failed to authorize client", "err", err) 680 fail() 681 return 682 } 683 684 // remove from pds 685 _, err = client.RepoDeleteRecord(r.Context(), &comatproto.RepoDeleteRecord_Input{ 686 Collection: tangled.SpindleMemberNSID, 687 Repo: user.Did, 688 Rkey: members[0].Rkey, 689 }) 690 if err != nil { 691 // non-fatal 692 l.Error("failed to delete record", "err", err) 693 } 694 695 // commit everything 696 if err = tx.Commit(); err != nil { 697 l.Error("failed to commit txn", "err", err) 698 fail() 699 return 700 } 701 702 // commit everything 703 if err = s.Enforcer.E.SavePolicy(); err != nil { 704 l.Error("failed to save ACLs", "err", err) 705 fail() 706 return 707 } 708 709 // ok 710 s.Pages.HxRefresh(w) 711}