feat: add organizationId to onSuccess callback

[nicknisi]

This PR allows the organizationId returned the userManagement call to be returned directly to the onSuccess callback.

This can help enable flows such as initializing an organization for a user where not present on signup.

This is a port of workos/authkit-remix#80.

[greptile-apps]

Greptile Overview

Greptile Summary

Added organizationId to the onSuccess callback data in the authentication flow, enabling developers to access the organization ID immediately after successful authentication.

Changes:

• Extracted organizationId from WorkOS authenticateWithCode response
• Added organizationId: string | null to AuthLoaderSuccessData interface
• Passed organizationId to the onSuccess callback handler

Impact:
This change enables use cases like initializing organization data for new users during signup. The implementation is consistent with the existing pattern used in RefreshSuccessOptions interface (line 39 in src/interfaces.ts) and follows the same null-coalescing pattern used for other optional fields like impersonator and oauthTokens.

Confidence Score: 5/5

• This PR is safe to merge with minimal risk
• The changes are straightforward and consistent with existing patterns in the codebase. The organizationId field is already used throughout the codebase in interfaces like RefreshSuccessOptions, UserInfo, and AuthorizedData. The implementation follows the same null-coalescing pattern (?? null) used for other optional fields. No security concerns, no breaking changes, and no custom instruction violations.
• No files require special attention

Important Files Changed

File Analysis

Filename	Score	Overview
src/authkit-callback-route.ts	5/5	Added organizationId extraction from authentication response and passed it to onSuccess callback
src/interfaces.ts	5/5	Added `organizationId: string

Sequence Diagram

sequenceDiagram
    participant Client
    participant AuthLoader
    participant WorkOS
    participant OnSuccess
    participant Session
    
    Client->>AuthLoader: GET /callback?code=xxx&state=xxx
    AuthLoader->>WorkOS: authenticateWithCode(clientId, code)
    WorkOS-->>AuthLoader: {accessToken, refreshToken, user, impersonator, oauthTokens, organizationId}
    AuthLoader->>AuthLoader: encryptSession({accessToken, refreshToken, user, impersonator})
    AuthLoader->>Session: commitSession(encrypted session)
    Session-->>AuthLoader: cookie
    AuthLoader->>OnSuccess: onSuccess({accessToken, refreshToken, user, impersonator, oauthTokens, organizationId})
    OnSuccess-->>AuthLoader: void
    AuthLoader-->>Client: 302 Redirect with Set-Cookie header

Loading

_{2 files reviewed, no comments}

_{Edit Code Review Agent Settings | Greptile}