@scalekit-sdk/node 2.2.0-beta.1 → 2.2.0

package/src/role.ts

@@ -1,8 +1,10 @@
-import { Empty, PartialMessage } from "@bufbuild/protobuf";
-import { PromiseClient } from "@connectrpc/connect";
+import type { MessageShape } from "@bufbuild/protobuf";
+import { create } from "@bufbuild/protobuf";
+import { EmptySchema } from "@bufbuild/protobuf/wkt";
+import type { Client } from "@connectrpc/connect";
 import GrpcConnect from "./connect";
 import CoreClient from "./core";
-import { RolesService } from "./pkg/grpc/scalekit/v1/roles/roles_connect";
+import { RolesService } from "./pkg/grpc/scalekit/v1/roles/roles_pb";
 import {
   CreateRoleRequest,
   CreateRoleResponse,
@@ -22,6 +24,8 @@ import {
   ListOrganizationRolesRequest,
   ListOrganizationRolesResponse,
   DeleteOrganizationRoleRequest,
+  DeleteRoleRequestSchema,
+  DeleteOrganizationRoleRequestSchema,
   GetRoleUsersCountRequest,
   GetRoleUsersCountResponse,
   GetOrganizationRoleUsersCountRequest,
@@ -52,7 +56,7 @@ import {
  * @see {@link https://docs.scalekit.com/apis/#tag/roles | Role API Documentation}
  */
 export default class RoleClient {
-  private client: PromiseClient<typeof RolesService>;
+  private client: Client<typeof RolesService>;
 
   constructor(
     private readonly grpcConnect: GrpcConnect,
@@ -174,7 +178,7 @@ export default class RoleClient {
    * @param {string} roleName - Role to delete
    * @param {string} [reassignRoleName] - Target role for user migration
    *
-   * @returns {Promise<Empty>} Empty response on success
+   * @returns {Promise<MessageShape<typeof EmptySchema>>} Empty response on success
    *
    * @example
    * await scalekitClient.role.deleteRole('old_role', 'new_role');
@@ -184,11 +188,11 @@ export default class RoleClient {
   async deleteRole(
     roleName: string,
     reassignRoleName?: string
-  ): Promise<Empty> {
-    const request: PartialMessage<DeleteRoleRequest> = { roleName };
-    if (reassignRoleName) {
-      request.reassignRoleName = reassignRoleName;
-    }
+  ): Promise<MessageShape<typeof EmptySchema>> {
+    const request = create(DeleteRoleRequestSchema, {
+      roleName,
+      reassignRoleName,
+    });
 
     return this.coreClient.connectExec(this.client.deleteRole, request);
   }
@@ -345,7 +349,7 @@ export default class RoleClient {
    * @param {string} roleName - Role to delete
    * @param {string} [reassignRoleName] - Target role for user migration
    *
-   * @returns {Promise<Empty>} Empty response on success
+   * @returns {Promise<MessageShape<typeof EmptySchema>>} Empty response on success
    *
    * @example
    * await scalekitClient.role.deleteOrganizationRole(
@@ -360,14 +364,12 @@ export default class RoleClient {
     orgId: string,
     roleName: string,
     reassignRoleName?: string
-  ): Promise<Empty> {
-    const request: PartialMessage<DeleteOrganizationRoleRequest> = {
+  ): Promise<MessageShape<typeof EmptySchema>> {
+    const request = create(DeleteOrganizationRoleRequestSchema, {
       orgId,
       roleName,
-    };
-    if (reassignRoleName) {
-      request.reassignRoleName = reassignRoleName;
-    }
+      reassignRoleName,
+    });
 
     return this.coreClient.connectExec(
       this.client.deleteOrganizationRole,
@@ -442,7 +444,7 @@ export default class RoleClient {
    * @param {string} orgId - Organization identifier
    * @param {string} roleName - Role to remove inheritance from
    *
-   * @returns {Promise<Empty>} Empty response on success
+   * @returns {Promise<MessageShape<typeof EmptySchema>>} Empty response on success
    *
    * @example
    * await scalekitClient.role.deleteOrganizationRoleBase('org_123456', 'custom_role');
@@ -452,7 +454,7 @@ export default class RoleClient {
   async deleteOrganizationRoleBase(
     orgId: string,
     roleName: string
-  ): Promise<Empty> {
+  ): Promise<MessageShape<typeof EmptySchema>> {
     return this.coreClient.connectExec(this.client.deleteOrganizationRoleBase, {
       orgId,
       roleName,

package/src/scalekit.ts

@@ -5,8 +5,6 @@ import GrpcConnect from './connect';
 import ConnectionClient from './connection';
 import { IdTokenClaimToUserMap } from './constants/user';
 import CoreClient from './core';
-import ToolsClient from './tools';
-import ConnectedAccountsClient from './connected-accounts';
 import DirectoryClient from './directory';
 import DomainClient from './domain';
 import AuthClient from './auth';
@@ -62,8 +60,6 @@ export default class ScalekitClient {
   readonly organization: OrganizationClient;
   readonly connection: ConnectionClient;
   readonly domain: DomainClient;
-  readonly tools: ToolsClient;
-  readonly connectedAccounts: ConnectedAccountsClient;
   readonly directory: DirectoryClient;
   readonly passwordless: PasswordlessClient;
   readonly user: UserClient;
@@ -92,11 +88,6 @@ export default class ScalekitClient {
     );
     this.connection = new ConnectionClient(this.grpcConnect, this.coreClient);
     this.domain = new DomainClient(this.grpcConnect, this.coreClient);
-    this.tools = new ToolsClient(this.grpcConnect, this.coreClient);
-    this.connectedAccounts = new ConnectedAccountsClient(
-      this.grpcConnect,
-      this.coreClient
-    );
     this.directory = new DirectoryClient(this.grpcConnect, this.coreClient);
     this.passwordless = new PasswordlessClient(
       this.grpcConnect,

package/src/session.ts

@@ -1,23 +1,24 @@
-import { PartialMessage } from "@bufbuild/protobuf";
-import { PromiseClient } from "@connectrpc/connect";
+import { create } from "@bufbuild/protobuf";
+import { timestampFromDate } from "@bufbuild/protobuf/wkt";
+import type { Client } from "@connectrpc/connect";
 import GrpcConnect from "./connect";
 import CoreClient from "./core";
-import { SessionService } from "./pkg/grpc/scalekit/v1/sessions/sessions_connect";
 import {
+  SessionService,
   SessionDetailsRequest,
   SessionDetails,
   UserSessionDetailsRequest,
   UserSessionDetails,
-  UserSessionFilter,
+  UserSessionDetailsRequestSchema,
+  UserSessionFilterSchema,
   RevokeSessionRequest,
   RevokeSessionResponse,
   RevokeAllUserSessionsRequest,
   RevokeAllUserSessionsResponse,
 } from "./pkg/grpc/scalekit/v1/sessions/sessions_pb";
-import { Timestamp } from "@bufbuild/protobuf";
 
 export default class SessionClient {
-  private client: PromiseClient<typeof SessionService>;
+  private client: Client<typeof SessionService>;
 
   constructor(
     private readonly grpcConnect: GrpcConnect,
@@ -145,35 +146,20 @@ export default class SessionClient {
       };
     }
   ): Promise<UserSessionDetails> {
-    const request: PartialMessage<UserSessionDetailsRequest> = {
-      userId,
-    };
-
-    if (options?.pageSize !== undefined) {
-      request.pageSize = options.pageSize;
-    }
-
-    if (options?.pageToken) {
-      request.pageToken = options.pageToken;
-    }
-
-    if (options?.filter) {
-      const filter = new UserSessionFilter();
+    const filter = options?.filter
+      ? create(UserSessionFilterSchema, {
+          status: options.filter.status,
+          startTime: options.filter.startTime ? timestampFromDate(options.filter.startTime) : undefined,
+          endTime: options.filter.endTime ? timestampFromDate(options.filter.endTime) : undefined,
+        })
+      : undefined;
 
-      if (options.filter.status) {
-        filter.status = options.filter.status;
-      }
-
-      if (options.filter.startTime) {
-        filter.startTime = Timestamp.fromDate(options.filter.startTime);
-      }
-
-      if (options.filter.endTime) {
-        filter.endTime = Timestamp.fromDate(options.filter.endTime);
-      }
-
-      request.filter = filter;
-    }
+    const request = create(UserSessionDetailsRequestSchema, {
+      userId,
+      pageSize: options?.pageSize,
+      pageToken: options?.pageToken,
+      filter,
+    });
 
     return this.coreClient.connectExec(this.client.getUserSessions, request);
   }

package/src/user.ts

@@ -11,11 +11,13 @@
  *
  * @see {@link https://docs.scalekit.com/apis/#tag/users | User API Documentation}
  */
-import { Empty, PartialMessage } from "@bufbuild/protobuf";
-import { PromiseClient } from "@connectrpc/connect";
+import type { MessageShape } from "@bufbuild/protobuf";
+import { create } from "@bufbuild/protobuf";
+import { EmptySchema } from "@bufbuild/protobuf/wkt";
+import type { Client } from "@connectrpc/connect";
 import GrpcConnect from "./connect";
 import CoreClient from "./core";
-import { UserService } from "./pkg/grpc/scalekit/v1/users/users_connect";
+import { UserService } from "./pkg/grpc/scalekit/v1/users/users_pb";
 import {
   CreateUserAndMembershipRequest,
   CreateUserAndMembershipResponse,
@@ -30,8 +32,13 @@ import {
   UpdateUser,
   CreateUser,
   CreateUserProfile,
+  CreateUserSchema,
+  CreateUserProfileSchema,
+  UpdateUserSchema,
+  CreateUserAndMembershipRequestSchema,
   CreateMembershipRequest,
   CreateMembershipResponse,
+  CreateMembershipRequestSchema,
   DeleteMembershipRequest,
   UpdateMembershipRequest,
   UpdateMembershipResponse,
@@ -39,8 +46,11 @@ import {
   ListOrganizationUsersResponse,
   CreateMembership,
   UpdateMembership,
+  CreateMembershipSchema,
+  UpdateMembershipSchema,
   ResendInviteRequest,
   ResendInviteResponse,
+  ResendInviteRequestSchema,
 } from "./pkg/grpc/scalekit/v1/users/users_pb";
 import {
   CreateUserRequest,
@@ -48,7 +58,7 @@ import {
 } from "./types/user";
 
 export default class UserClient {
-  private client: PromiseClient<typeof UserService>;
+  private client: Client<typeof UserService>;
 
   constructor(
     private readonly grpcConnect: GrpcConnect,
@@ -125,10 +135,10 @@ export default class UserClient {
       throw new Error("email is required");
     }
 
-    const user = new CreateUser({
+    const user = create(CreateUserSchema, {
       email: options.email,
       userProfile: options.userProfile
-        ? new CreateUserProfile({
+        ? create(CreateUserProfileSchema, {
             firstName: options.userProfile.firstName,
             lastName: options.userProfile.lastName,
           })
@@ -136,14 +146,11 @@ export default class UserClient {
       metadata: options.metadata,
     });
 
-    const request: PartialMessage<CreateUserAndMembershipRequest> = {
+    const request = create(CreateUserAndMembershipRequestSchema, {
       organizationId,
       user,
-    };
-
-    if (options.sendInvitationEmail !== undefined) {
-      request.sendInvitationEmail = options.sendInvitationEmail;
-    }
+      sendInvitationEmail: options.sendInvitationEmail,
+    });
 
     const response = await this.coreClient.connectExec(
       this.client.createUserAndMembership,
@@ -382,7 +389,7 @@ export default class UserClient {
     userId: string,
     options: UpdateUserRequestType
   ): Promise<UpdateUserResponse> {
-    const updateUser = new UpdateUser({
+    const updateUser = create(UpdateUserSchema, {
       userProfile: options.userProfile
         ? {
             firstName: options.userProfile.firstName,
@@ -411,7 +418,7 @@ export default class UserClient {
    *
    * @param {string} userId - The Scalekit user identifier to delete (format: "usr_...")
    *
-   * @returns {Promise<Empty>} Empty response on successful deletion
+   * @returns {Promise<MessageShape<typeof EmptySchema>>} Empty response on successful deletion
    *
    * @throws {Error} If the user is not found or deletion fails
    *
@@ -441,7 +448,7 @@ export default class UserClient {
    * @see {@link deleteMembership} - Remove user from a specific organization only
    * @see {@link getUser} - Check if user exists before deletion
    */
-  async deleteUser(userId: string): Promise<Empty> {
+  async deleteUser(userId: string): Promise<MessageShape<typeof EmptySchema>> {
     return this.coreClient.connectExec(this.client.deleteUser, {
       identities: {
         case: "id",
@@ -525,23 +532,20 @@ export default class UserClient {
       sendInvitationEmail?: boolean;
     } = {}
   ): Promise<CreateMembershipResponse> {
-    const membership = new CreateMembership({
+    const membership = create(CreateMembershipSchema, {
       roles: options.roles?.map((role) => ({ name: role })) || [],
       metadata: options.metadata || {},
     });
 
-    const request: PartialMessage<CreateMembershipRequest> = {
+    const request = create(CreateMembershipRequestSchema, {
       organizationId,
       identities: {
         case: "id",
         value: userId,
       },
       membership,
-    };
-
-    if (options.sendInvitationEmail !== undefined) {
-      request.sendInvitationEmail = options.sendInvitationEmail;
-    }
+      sendInvitationEmail: options.sendInvitationEmail,
+    });
 
     return this.coreClient.connectExec(this.client.createMembership, request);
   }
@@ -557,7 +561,7 @@ export default class UserClient {
    * @param {string} organizationId - The organization ID to remove the user from (format: "org_...")
    * @param {string} userId - The user ID to remove (format: "usr_...")
    *
-   * @returns {Promise<Empty>} Empty response on successful removal
+   * @returns {Promise<MessageShape<typeof EmptySchema>>} Empty response on successful removal
    *
    * @throws {Error} If the user or organization is not found
    * @throws {Error} If the membership doesn't exist
@@ -591,7 +595,7 @@ export default class UserClient {
   async deleteMembership(
     organizationId: string,
     userId: string
-  ): Promise<Empty> {
+  ): Promise<MessageShape<typeof EmptySchema>> {
     return this.coreClient.connectExec(this.client.deleteMembership, {
       organizationId,
       identities: {
@@ -682,7 +686,7 @@ export default class UserClient {
       metadata?: Record<string, string>;
     } = {}
   ): Promise<UpdateMembershipResponse> {
-    const membership = new UpdateMembership({
+    const membership = create(UpdateMembershipSchema, {
       roles: options.roles?.map((role) => ({ name: role })) || [],
       metadata: options.metadata || {},
     });
@@ -815,7 +819,7 @@ export default class UserClient {
       throw new Error("userId is required");
     }
 
-    const request = new ResendInviteRequest({
+    const request = create(ResendInviteRequestSchema, {
       organizationId,
       id: userId,
     });

package/src/webauthn.ts

@@ -1,18 +1,19 @@
-import { PromiseClient } from '@connectrpc/connect';
+import { create } from '@bufbuild/protobuf';
+import type { Client } from '@connectrpc/connect';
 import GrpcConnect from './connect';
 import CoreClient from './core';
-import { WebAuthnService } from './pkg/grpc/scalekit/v1/auth/webauthn_connect';
+import { WebAuthnService } from './pkg/grpc/scalekit/v1/auth/webauthn_pb';
 import {
-  ListCredentialsRequest,
+  ListCredentialsRequestSchema,
+  UpdateCredentialRequestSchema,
+  DeleteCredentialRequestSchema,
   ListCredentialsResponse,
-  UpdateCredentialRequest,
   UpdateCredentialResponse,
-  DeleteCredentialRequest,
   DeleteCredentialResponse
 } from './pkg/grpc/scalekit/v1/auth/webauthn_pb';
 
 export default class WebAuthnClient {
-  private client: PromiseClient<typeof WebAuthnService>;
+  private client: Client<typeof WebAuthnService>;
 
   constructor(
     private readonly grpcConnect: GrpcConnect,
@@ -33,7 +34,7 @@ export default class WebAuthnClient {
       throw new Error('userId must be a non-empty string');
     }
 
-    const request = new ListCredentialsRequest({
+    const request = create(ListCredentialsRequestSchema, {
       userId
     });
 
@@ -62,7 +63,7 @@ export default class WebAuthnClient {
       throw new Error('displayName must be a non-empty string');
     }
 
-    const request = new UpdateCredentialRequest({
+    const request = create(UpdateCredentialRequestSchema, {
       credentialId,
       displayName
     });
@@ -85,7 +86,7 @@ export default class WebAuthnClient {
       throw new Error('credentialId must be a non-empty string');
     }
 
-    const request = new DeleteCredentialRequest({
+    const request = create(DeleteCredentialRequestSchema, {
       credentialId
     });
 

package/tests/utils/test-data.ts

@@ -1,8 +1,17 @@
+import { create } from '@bufbuild/protobuf';
 import { CreateUserRequest, UpdateUserRequest } from '../../src/types/user';
 import { TemplateType } from '../../src/pkg/grpc/scalekit/v1/auth/passwordless_pb';
 import { DomainType } from '../../src/pkg/grpc/scalekit/v1/domains/domains_pb';
-import { CreateRole, UpdateRole, CreateOrganizationRole } from '../../src/pkg/grpc/scalekit/v1/roles/roles_pb';
-import { CreatePermission } from '../../src/pkg/grpc/scalekit/v1/roles/roles_pb';
+import {
+  CreateRole,
+  UpdateRole,
+  CreateOrganizationRole,
+  CreatePermission,
+  CreateRoleSchema,
+  UpdateRoleSchema,
+  CreateOrganizationRoleSchema,
+  CreatePermissionSchema,
+} from '../../src/pkg/grpc/scalekit/v1/roles/roles_pb';
 
 /**
  * Test data generation utilities to reduce redundancy across test files
@@ -200,7 +209,7 @@ export class TestDataGenerator {
   static generateRoleData(overrides: Partial<CreateRole> = {}): CreateRole {
     const uniqueId = this.generateUniqueId();
     
-    return new CreateRole({
+    return create(CreateRoleSchema, {
       name: `test_role_${uniqueId}`,
       displayName: `Test Role ${uniqueId}`,
       description: `Test role description ${uniqueId}`,
@@ -214,12 +223,12 @@ export class TestDataGenerator {
    */
   static generateRoleUpdateData(overrides: Partial<UpdateRole> = {}): UpdateRole {
     const uniqueId = this.generateUniqueId();
-    
-    return new UpdateRole({
+    const init = {
       displayName: `Updated Role ${uniqueId}`,
       description: `Updated role description ${uniqueId}`,
       ...overrides
-    });
+    };
+    return create(UpdateRoleSchema, init as Parameters<typeof create>[1]);
   }
 
   /**
@@ -228,7 +237,7 @@ export class TestDataGenerator {
   static generateOrganizationRoleData(overrides: Partial<CreateOrganizationRole> = {}): CreateOrganizationRole {
     const uniqueId = this.generateUniqueId();
     
-    return new CreateOrganizationRole({
+    return create(CreateOrganizationRoleSchema, {
       name: `test_org_role_${uniqueId}`,
       displayName: `Test Organization Role ${uniqueId}`,
       description: `Test organization role description ${uniqueId}`,
@@ -243,7 +252,7 @@ export class TestDataGenerator {
   static generatePermissionData(overrides: Partial<CreatePermission> = {}): CreatePermission {
     const uniqueId = this.generateUniqueId();
     
-    return new CreatePermission({
+    return create(CreatePermissionSchema, {
       name: `test_permission_${uniqueId}`,
       description: `Test permission description ${uniqueId}`,
       ...overrides

package/lib/connected-accounts.d.ts

@@ -1,119 +0,0 @@
-import { PartialMessage } from "@bufbuild/protobuf";
-import GrpcConnect from "./connect";
-import CoreClient from "./core";
-import { AuthorizationDetails, CreateConnectedAccount, CreateConnectedAccountResponse, DeleteConnectedAccountResponse, GetConnectedAccountByIdentifierResponse, GetMagicLinkForConnectedAccountResponse, ListConnectedAccountsResponse, UpdateConnectedAccount, UpdateConnectedAccountResponse } from "./pkg/grpc/scalekit/v1/connected_accounts/connected_accounts_pb";
-/**
- * Client for managing connected accounts for third-party integrations.
- *
- * This mirrors the Python SDK `ConnectedAccountsClient` and exposes a typed,
- * ergonomic API around the `ConnectedAccountService` to:
- * - list connected accounts
- * - create/update/delete connected accounts
- * - generate magic links for authorization
- * - fetch full authentication details for a connected account
- */
-export default class ConnectedAccountsClient {
-    private readonly grpcConnect;
-    private readonly coreClient;
-    private client;
-    constructor(grpcConnect: GrpcConnect, coreClient: CoreClient);
-    /**
-     * Lists connected accounts with optional filters and pagination.
-     *
-     * @param options Optional filtering and pagination parameters
-     */
-    listConnectedAccounts(options?: {
-        organizationId?: string;
-        userId?: string;
-        connector?: string;
-        identifier?: string;
-        provider?: string;
-        pageSize?: number;
-        pageToken?: string;
-        query?: string;
-    }): Promise<ListConnectedAccountsResponse>;
-    /**
-     * Creates a new connected account.
-     *
-     * @param params Connected account creation parameters
-     */
-    createConnectedAccount(params: {
-        connector: string;
-        identifier: string;
-        connectedAccount: CreateConnectedAccount;
-        organizationId?: string;
-        userId?: string;
-    }): Promise<CreateConnectedAccountResponse>;
-    /**
-     * Gets an existing connected account by connector and identifier, or creates one if none exists.
-     * Mirrors the Python SDK `get_or_create_connected_account`. When creating, the backend may require
-     * valid authorization details; if omitted, a minimal payload is sent and the server may return
-     * a validation error.
-     *
-     * @param params Get-or-create parameters
-     * @param params.connector Connector identifier (required)
-     * @param params.identifier Connected account identifier (required)
-     * @param params.authorizationDetails Optional auth details for the create path (OAuth token or static auth)
-     * @param params.organizationId Optional organization ID
-     * @param params.userId Optional user ID
-     * @param params.apiConfig Optional API config for the create path
-     */
-    getOrCreateConnectedAccount(params: {
-        connector: string;
-        identifier: string;
-        authorizationDetails?: PartialMessage<AuthorizationDetails>;
-        organizationId?: string;
-        userId?: string;
-        apiConfig?: Record<string, unknown>;
-    }): Promise<CreateConnectedAccountResponse>;
-    /**
-     * Updates an existing connected account.
-     *
-     * You can target the account either by `connectedAccountId` or by the combination
-     * of `organizationId`/`userId`, `connector`, and `identifier`.
-     */
-    updateConnectedAccount(params: {
-        connector: string;
-        identifier: string;
-        connectedAccount: UpdateConnectedAccount;
-        organizationId?: string;
-        userId?: string;
-        connectedAccountId?: string;
-    }): Promise<UpdateConnectedAccountResponse>;
-    /**
-     * Deletes a connected account and revokes its credentials.
-     *
-     * You can target the account either by `connectedAccountId` or by the combination
-     * of `organizationId`/`userId`, `connector`, and `identifier`.
-     */
-    deleteConnectedAccount(params: {
-        connector: string;
-        identifier: string;
-        organizationId?: string;
-        userId?: string;
-        connectedAccountId?: string;
-    }): Promise<DeleteConnectedAccountResponse>;
-    /**
-     * Generates a time-limited magic link for connecting or re-authorizing a third-party account.
-     */
-    getMagicLinkForConnectedAccount(params: {
-        connector: string;
-        identifier: string;
-        organizationId?: string;
-        userId?: string;
-        connectedAccountId?: string;
-    }): Promise<GetMagicLinkForConnectedAccountResponse>;
-    /**
-     * Retrieves complete authentication details for a connected account.
-     *
-     * This method returns sensitive credential information, so ensure you protect access
-     * to this in your application.
-     */
-    getConnectedAccountByIdentifier(params: {
-        connector: string;
-        identifier: string;
-        organizationId?: string;
-        userId?: string;
-        connectedAccountId?: string;
-    }): Promise<GetConnectedAccountByIdentifierResponse>;
-}