second review
This commit is contained in:
parent
23049da55a
commit
976a880312
7 changed files with 105 additions and 65 deletions
|
|
@ -534,6 +534,22 @@ class InviteResponse(BaseModel):
|
||||||
revoked_at: datetime | None
|
revoked_at: datetime | None
|
||||||
|
|
||||||
|
|
||||||
|
def build_invite_response(invite: Invite) -> InviteResponse:
|
||||||
|
"""Build an InviteResponse from an Invite with loaded relationships."""
|
||||||
|
return InviteResponse(
|
||||||
|
id=invite.id,
|
||||||
|
identifier=invite.identifier,
|
||||||
|
godfather_id=invite.godfather_id,
|
||||||
|
godfather_email=invite.godfather.email,
|
||||||
|
status=invite.status.value,
|
||||||
|
used_by_id=invite.used_by_id,
|
||||||
|
used_by_email=invite.used_by.email if invite.used_by else None,
|
||||||
|
created_at=invite.created_at,
|
||||||
|
spent_at=invite.spent_at,
|
||||||
|
revoked_at=invite.revoked_at,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
MAX_INVITE_COLLISION_RETRIES = 3
|
MAX_INVITE_COLLISION_RETRIES = 3
|
||||||
|
|
||||||
|
|
||||||
|
|
@ -544,17 +560,16 @@ async def create_invite(
|
||||||
_current_user: User = Depends(require_permission(Permission.MANAGE_INVITES)),
|
_current_user: User = Depends(require_permission(Permission.MANAGE_INVITES)),
|
||||||
):
|
):
|
||||||
"""Create a new invite for a specified godfather user."""
|
"""Create a new invite for a specified godfather user."""
|
||||||
# Validate godfather exists and get their info
|
# Validate godfather exists
|
||||||
result = await db.execute(
|
result = await db.execute(
|
||||||
select(User.id, User.email).where(User.id == data.godfather_id)
|
select(User.id).where(User.id == data.godfather_id)
|
||||||
)
|
)
|
||||||
godfather_row = result.one_or_none()
|
godfather_id = result.scalar_one_or_none()
|
||||||
if not godfather_row:
|
if not godfather_id:
|
||||||
raise HTTPException(
|
raise HTTPException(
|
||||||
status_code=status.HTTP_400_BAD_REQUEST,
|
status_code=status.HTTP_400_BAD_REQUEST,
|
||||||
detail="Godfather user not found",
|
detail="Godfather user not found",
|
||||||
)
|
)
|
||||||
godfather_id, godfather_email = godfather_row
|
|
||||||
|
|
||||||
# Try to create invite with retry on collision
|
# Try to create invite with retry on collision
|
||||||
invite: Invite | None = None
|
invite: Invite | None = None
|
||||||
|
|
@ -568,7 +583,7 @@ async def create_invite(
|
||||||
db.add(invite)
|
db.add(invite)
|
||||||
try:
|
try:
|
||||||
await db.commit()
|
await db.commit()
|
||||||
await db.refresh(invite)
|
await db.refresh(invite, ["godfather"])
|
||||||
break
|
break
|
||||||
except IntegrityError:
|
except IntegrityError:
|
||||||
await db.rollback()
|
await db.rollback()
|
||||||
|
|
@ -578,19 +593,12 @@ async def create_invite(
|
||||||
detail="Failed to generate unique invite code. Please try again.",
|
detail="Failed to generate unique invite code. Please try again.",
|
||||||
)
|
)
|
||||||
|
|
||||||
assert invite is not None # We either succeeded or raised an exception above
|
if invite is None:
|
||||||
return InviteResponse(
|
raise HTTPException(
|
||||||
id=invite.id,
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||||||
identifier=invite.identifier,
|
detail="Failed to create invite",
|
||||||
godfather_id=invite.godfather_id,
|
)
|
||||||
godfather_email=godfather_email,
|
return build_invite_response(invite)
|
||||||
status=invite.status.value,
|
|
||||||
used_by_id=invite.used_by_id,
|
|
||||||
used_by_email=None,
|
|
||||||
created_at=invite.created_at,
|
|
||||||
spent_at=invite.spent_at,
|
|
||||||
revoked_at=invite.revoked_at,
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class UserInviteResponse(BaseModel):
|
class UserInviteResponse(BaseModel):
|
||||||
|
|
@ -693,20 +701,7 @@ async def list_all_invites(
|
||||||
invites = result.scalars().all()
|
invites = result.scalars().all()
|
||||||
|
|
||||||
# Build responses using preloaded relationships
|
# Build responses using preloaded relationships
|
||||||
records = []
|
records = [build_invite_response(invite) for invite in invites]
|
||||||
for invite in invites:
|
|
||||||
records.append(InviteResponse(
|
|
||||||
id=invite.id,
|
|
||||||
identifier=invite.identifier,
|
|
||||||
godfather_id=invite.godfather_id,
|
|
||||||
godfather_email=invite.godfather.email,
|
|
||||||
status=invite.status.value,
|
|
||||||
used_by_id=invite.used_by_id,
|
|
||||||
used_by_email=invite.used_by.email if invite.used_by else None,
|
|
||||||
created_at=invite.created_at,
|
|
||||||
spent_at=invite.spent_at,
|
|
||||||
revoked_at=invite.revoked_at,
|
|
||||||
))
|
|
||||||
|
|
||||||
return PaginatedInviteRecords(
|
return PaginatedInviteRecords(
|
||||||
records=records,
|
records=records,
|
||||||
|
|
@ -744,16 +739,4 @@ async def revoke_invite(
|
||||||
await db.commit()
|
await db.commit()
|
||||||
await db.refresh(invite)
|
await db.refresh(invite)
|
||||||
|
|
||||||
# Use preloaded relationships (selectin loading)
|
return build_invite_response(invite)
|
||||||
return InviteResponse(
|
|
||||||
id=invite.id,
|
|
||||||
identifier=invite.identifier,
|
|
||||||
godfather_id=invite.godfather_id,
|
|
||||||
godfather_email=invite.godfather.email,
|
|
||||||
status=invite.status.value,
|
|
||||||
used_by_id=invite.used_by_id,
|
|
||||||
used_by_email=invite.used_by.email if invite.used_by else None,
|
|
||||||
created_at=invite.created_at,
|
|
||||||
spent_at=invite.spent_at,
|
|
||||||
revoked_at=invite.revoked_at,
|
|
||||||
)
|
|
||||||
|
|
|
||||||
|
|
@ -206,7 +206,7 @@ class Invite(Base):
|
||||||
godfather: Mapped[User] = relationship(
|
godfather: Mapped[User] = relationship(
|
||||||
"User",
|
"User",
|
||||||
foreign_keys=[godfather_id],
|
foreign_keys=[godfather_id],
|
||||||
lazy="selectin",
|
lazy="joined",
|
||||||
)
|
)
|
||||||
|
|
||||||
# User who used this invite (null until spent)
|
# User who used this invite (null until spent)
|
||||||
|
|
@ -216,7 +216,7 @@ class Invite(Base):
|
||||||
used_by: Mapped[User | None] = relationship(
|
used_by: Mapped[User | None] = relationship(
|
||||||
"User",
|
"User",
|
||||||
foreign_keys=[used_by_id],
|
foreign_keys=[used_by_id],
|
||||||
lazy="selectin",
|
lazy="joined",
|
||||||
)
|
)
|
||||||
|
|
||||||
# Timestamps
|
# Timestamps
|
||||||
|
|
|
||||||
|
|
@ -80,7 +80,7 @@ export default function AdminInvitesPage() {
|
||||||
|
|
||||||
const handleCreateInvite = async () => {
|
const handleCreateInvite = async () => {
|
||||||
if (!newGodfatherId) {
|
if (!newGodfatherId) {
|
||||||
setCreateError("Please enter a godfather user ID");
|
setCreateError("Please select a godfather");
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -104,6 +104,7 @@ export default function AdminInvitesPage() {
|
||||||
const handleRevoke = async (inviteId: number) => {
|
const handleRevoke = async (inviteId: number) => {
|
||||||
try {
|
try {
|
||||||
await api.post(`/api/admin/invites/${inviteId}/revoke`);
|
await api.post(`/api/admin/invites/${inviteId}/revoke`);
|
||||||
|
setError(null);
|
||||||
fetchInvites(page, statusFilter);
|
fetchInvites(page, statusFilter);
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
setError(err instanceof Error ? err.message : "Failed to revoke invite");
|
setError(err instanceof Error ? err.message : "Failed to revoke invite");
|
||||||
|
|
@ -328,16 +329,6 @@ const pageStyles: Record<string, React.CSSProperties> = {
|
||||||
fontSize: "0.8rem",
|
fontSize: "0.8rem",
|
||||||
color: "rgba(255, 255, 255, 0.5)",
|
color: "rgba(255, 255, 255, 0.5)",
|
||||||
},
|
},
|
||||||
input: {
|
|
||||||
fontFamily: "'DM Sans', system-ui, sans-serif",
|
|
||||||
fontSize: "0.9rem",
|
|
||||||
padding: "0.75rem",
|
|
||||||
background: "rgba(255, 255, 255, 0.05)",
|
|
||||||
border: "1px solid rgba(255, 255, 255, 0.1)",
|
|
||||||
borderRadius: "8px",
|
|
||||||
color: "#fff",
|
|
||||||
maxWidth: "300px",
|
|
||||||
},
|
|
||||||
select: {
|
select: {
|
||||||
fontFamily: "'DM Sans', system-ui, sans-serif",
|
fontFamily: "'DM Sans', system-ui, sans-serif",
|
||||||
fontSize: "0.9rem",
|
fontSize: "0.9rem",
|
||||||
|
|
|
||||||
|
|
@ -19,7 +19,8 @@ export default function SignupWithCodePage() {
|
||||||
router.replace("/");
|
router.replace("/");
|
||||||
} else {
|
} else {
|
||||||
// Redirect to signup with code as query param
|
// Redirect to signup with code as query param
|
||||||
router.replace(`/signup?code=${encodeURIComponent(code)}`);
|
// Invite codes only contain [a-z0-9-] so no encoding needed
|
||||||
|
router.replace(`/signup?code=${code}`);
|
||||||
}
|
}
|
||||||
}, [user, isLoading, code, router]);
|
}, [user, isLoading, code, router]);
|
||||||
|
|
||||||
|
|
@ -37,4 +38,3 @@ export default function SignupWithCodePage() {
|
||||||
</main>
|
</main>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,4 +0,0 @@
|
||||||
{
|
|
||||||
"status": "passed",
|
|
||||||
"failedTests": []
|
|
||||||
}
|
|
||||||
|
|
@ -0,0 +1,35 @@
|
||||||
|
# Page snapshot
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
- generic [active] [ref=e1]:
|
||||||
|
- main [ref=e2]:
|
||||||
|
- generic [ref=e4]:
|
||||||
|
- generic [ref=e5]:
|
||||||
|
- heading "Welcome back" [level=1] [ref=e6]
|
||||||
|
- paragraph [ref=e7]: Sign in to your account
|
||||||
|
- generic [ref=e8]:
|
||||||
|
- generic [ref=e9]: Failed to fetch
|
||||||
|
- generic [ref=e10]:
|
||||||
|
- generic [ref=e11]: Email
|
||||||
|
- textbox "Email" [ref=e12]:
|
||||||
|
- /placeholder: you@example.com
|
||||||
|
- text: admin@example.com
|
||||||
|
- generic [ref=e13]:
|
||||||
|
- generic [ref=e14]: Password
|
||||||
|
- textbox "Password" [ref=e15]:
|
||||||
|
- /placeholder: ••••••••
|
||||||
|
- text: admin123
|
||||||
|
- button "Sign in" [ref=e16] [cursor=pointer]
|
||||||
|
- paragraph [ref=e17]:
|
||||||
|
- text: Don't have an account?
|
||||||
|
- link "Sign up" [ref=e18] [cursor=pointer]:
|
||||||
|
- /url: /signup
|
||||||
|
- status [ref=e19]:
|
||||||
|
- generic [ref=e20]:
|
||||||
|
- img [ref=e22]
|
||||||
|
- generic [ref=e24]:
|
||||||
|
- text: Static route
|
||||||
|
- button "Hide static indicator" [ref=e25] [cursor=pointer]:
|
||||||
|
- img [ref=e26]
|
||||||
|
- alert [ref=e29]
|
||||||
|
```
|
||||||
|
|
@ -0,0 +1,35 @@
|
||||||
|
# Page snapshot
|
||||||
|
|
||||||
|
```yaml
|
||||||
|
- generic [active] [ref=e1]:
|
||||||
|
- main [ref=e2]:
|
||||||
|
- generic [ref=e4]:
|
||||||
|
- generic [ref=e5]:
|
||||||
|
- heading "Welcome back" [level=1] [ref=e6]
|
||||||
|
- paragraph [ref=e7]: Sign in to your account
|
||||||
|
- generic [ref=e8]:
|
||||||
|
- generic [ref=e9]: Failed to fetch
|
||||||
|
- generic [ref=e10]:
|
||||||
|
- generic [ref=e11]: Email
|
||||||
|
- textbox "Email" [ref=e12]:
|
||||||
|
- /placeholder: you@example.com
|
||||||
|
- text: admin@example.com
|
||||||
|
- generic [ref=e13]:
|
||||||
|
- generic [ref=e14]: Password
|
||||||
|
- textbox "Password" [ref=e15]:
|
||||||
|
- /placeholder: ••••••••
|
||||||
|
- text: admin123
|
||||||
|
- button "Sign in" [ref=e16] [cursor=pointer]
|
||||||
|
- paragraph [ref=e17]:
|
||||||
|
- text: Don't have an account?
|
||||||
|
- link "Sign up" [ref=e18] [cursor=pointer]:
|
||||||
|
- /url: /signup
|
||||||
|
- status [ref=e19]:
|
||||||
|
- generic [ref=e20]:
|
||||||
|
- img [ref=e22]
|
||||||
|
- generic [ref=e24]:
|
||||||
|
- text: Static route
|
||||||
|
- button "Hide static indicator" [ref=e25] [cursor=pointer]:
|
||||||
|
- img [ref=e26]
|
||||||
|
- alert [ref=e29]
|
||||||
|
```
|
||||||
Loading…
Add table
Add a link
Reference in a new issue