@stack-spot/portal-network 0.1.0 → 0.3.0

package/CHANGELOG.md

@@ -0,0 +1,21 @@
+# Changelog
+
+## [0.3.0](https://github.com/stack-spot/portal-commons/compare/portal-network@v0.2.0...portal-network@v0.3.0) (2024-08-02)
+
+
+### Features
+
+* network - member ([#220](https://github.com/stack-spot/portal-commons/issues/220)) ([5bf22c2](https://github.com/stack-spot/portal-commons/commit/5bf22c29e25215052c0ad63dd6266695f437994b))
+
+
+### Bug Fixes
+
+* rewrite network client methods ([#216](https://github.com/stack-spot/portal-commons/issues/216)) ([f71bcfe](https://github.com/stack-spot/portal-commons/commit/f71bcfe53c95593e59e4cf69f23b176ab96cf02a))
+
+## [0.2.0](https://github.com/stack-spot/portal-commons/compare/portal-network-v0.1.0...portal-network@v0.2.0) (2024-07-31)
+
+
+### Features
+
+* network library ([eec8b1b](https://github.com/stack-spot/portal-commons/commit/eec8b1b24c77d055884b8ed8c2ccac452c5b16e2))
+* network: infinite query ([#214](https://github.com/stack-spot/portal-commons/issues/214)) ([aded3f7](https://github.com/stack-spot/portal-commons/commit/aded3f78665a829fd31d78766be7d5942aaa62e3))

package/package.json

@@ -1,13 +1,19 @@
 {
   "name": "@stack-spot/portal-network",
-  "version": "0.1.0",
+  "version": "0.3.0",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "exports": {
     ".": "./dist/index.js",
+    "./package.json": "./package.json",
     "./api/*": "./dist/api/*.js"
   },
+  "scripts": {
+    "build": "rimraf dist && tsc-silent -p tsconfig.build.json --suppress @/src/api/ && tsc-esm-fix --target='dist'",
+    "lint": "eslint . --ext ts,tsx --report-unused-disable-directives --max-warnings 0",
+    "generate": "tsx ./scripts/generate-apis.ts"
+  },
   "peerDependencies": {
     "@stack-spot/auth": "^5.2.0",
     "@stack-spot/opa": "^2.2.0",
@@ -47,10 +53,5 @@
   "dependencies": {
     "@oazapfts/runtime": "^1.0.3",
     "lodash": "^4.17.21"
-  },
-  "scripts": {
-    "build": "rimraf dist && tsc-silent -p tsconfig.build.json --suppress @/src/api/ && tsc-esm-fix --target='dist'",
-    "lint": "eslint . --ext ts,tsx --report-unused-disable-directives --max-warnings 0",
-    "generate": "tsx ./scripts/generate-apis.ts"
   }
 }

package/src/apis.json

@@ -1,8 +1,8 @@
 {
   "ai": {
     "url": {
-      "dev": "https://code-buddy-api.dev.stackspot.com",
-      "stg": "https://code-buddy-api.stg.stackspot.com",
+      "dev": "https://genai-code-buddy-api.dev.stackspot.com",
+      "stg": "https://genai-code-buddy-api.stg.stackspot.com",
       "prd": "https://genai-code-buddy-api.stackspot.com"
     },
     "docs": "/openapi.json"
@@ -79,14 +79,6 @@
     },
     "docs": "/v3/api-docs"
   },
-  "plugin": {
-    "url": {
-      "dev": "https://content-plugin-api.dev.stackspot.com",
-      "stg": "https://content-plugin-api.stg.stackspot.com",
-      "prd": "https://plugin-api.v1.stackspot.com"
-    },
-    "docs": "/openapi.json"
-  },
   "workflows": {
     "url": {
       "dev": "https://workflow-workflow-api.dev.stackspot.com",

package/src/client/account.ts

@@ -1,9 +1,19 @@
 import { HttpError } from '@oazapfts/runtime'
 import {
-  accountDataIsAvailable, create, createPartner,
+  accountDataIsAvailable,
+  bindToGroups, bindToRoles, create, createPartner,
+  createUser,
+  deactivateFidoCredentials,
   defaults,
-  deletePartner, getFeatures, getPartnerAccount, getPartnersSharingAllowed,
-  getPersonalClientCredentials, updatePartnerAccountAdminData, updatePartnerAccountData, validateNewPartnerData,
+  deleteMember,
+  deletePartner, enableFidoCredentials, getAccountMembers1, getAllMemberFidoCredentials, getFeatures,
+  getMemberById,
+  getMemberGroups,
+  getPartnerAccount, getPartnersSharingAllowed,
+  getPersonalClientCredentials,
+  getResources1,
+  getRoles1, removeRoleFromMember, resetPassword, updatePartnerAccountAdminData, updatePartnerAccountData, updateUser,
+  validateNewPartnerData,
   validatePartnerAssociationLimit,
 } from '../api/account'
 import apis from '../apis.json'
@@ -21,17 +31,54 @@ class AccountClient extends ReactQueryNetworkClient {
     return new DefaultAPIError(error.data, error.status, accountDictionary, error.headers)
   }
 
+  /**
+   * Gets credentials from personal service client (create if not exists one to the logged user).
+   */
   generatePersonalClientCredentials = this.mutation(getPersonalClientCredentials)
-  create = this.mutation(create)
+  /**
+   * Creates a Feature Flag
+   */
+  createFeatureFlag = this.mutation(create)
+  /**
+   * Returns information if account data (name and slug) are available to be registered
+   */
   isAvailable = this.query(accountDataIsAvailable)
+  /**
+   * Lists all Feature Flags in an Account
+   */
   featureFlags = this.query(getFeatures)
+  /**
+   * Gets Partners with whom it is allowed to share content.
+   */
   partners = this.query(getPartnersSharingAllowed)
+  /**
+   * Gets Partner by account Id
+   */
   partner = this.query(getPartnerAccount)
+  /**
+   * Creates an Account Order for Partner
+   */
   createPartner = this.mutation(createPartner)
+  /**
+   * Updates Partner Account data.
+   */
   updatePartner = this.mutation(updatePartnerAccountData)
+  /**
+   * Updates Partner Account Admin data.
+   */
   updatePartnerAdmin = this.mutation(updatePartnerAccountAdminData)
+  /**
+   * Deletes Partner
+   */
   deactivatePartner = this.mutation(deletePartner)
+  /**
+   * Validates new Partner account data
+   */
   validateNewPartnerData = this.mutation(validateNewPartnerData)
+  /**
+   * Validates the association limit for the partner. Yields `{ isValid: true }` if valid or `{ isValid: false, message: string }`
+   * otherwise.
+   */
   validatePartnerAssociationLimit = this.query({
     name: 'validatePartnerAssociationLimit',
     request: async (signal) => {
@@ -47,6 +94,66 @@ class AccountClient extends ReactQueryNetworkClient {
       }
     },
   })
+  /**
+   * Gets member by id.
+   */
+  member = this.query(getMemberById)
+  /**
+   * Gets all members (paginated).
+   */
+  allMembers = this.infiniteQuery(getAccountMembers1)
+  /**
+   * Gets member Groups (paginated).
+   */
+  memberGroups = this.infiniteQuery(getMemberGroups)
+  /**
+   * Gets member roles (paginated).
+   */
+  memberRoles = this.infiniteQuery(getRoles1)
+  /**
+   * Gets member resources.
+   */
+  memberResources = this.query(getResources1)
+  /**
+   * Gets All Fido credentials for the given member (paginated).
+   */
+  fidoCredentials = this.infiniteQuery(getAllMemberFidoCredentials, { accumulator: 'items' })
+  /**
+   * Creates member on current tenant.
+   */
+  createMember = this.mutation(createUser)
+  /**
+   * Updates member on current tenant.
+   */
+  updateMember = this.mutation(updateUser)
+  /**
+   * Adds a member to several groups.
+   */
+  addMemberToGroups = this.mutation(bindToGroups)
+  /**
+   * Attributes several roles to a member.
+   */
+  addRolesToMember = this.mutation(bindToRoles)
+  /**
+   * Removes a role from a member.
+   */
+  removeRoleFromMember = this.mutation(removeRoleFromMember)
+  /**
+   * Sends an e-mail to reset the password of the member with the provided e-mail address.
+   */
+  resetMemberPassword = this.mutation(resetPassword)
+  /**
+   * Enables Fido credentials for the given member.
+   */
+  enableFidoCredentials = this.mutation(enableFidoCredentials)
+  /**
+   * Disables Fido credentials for the given member.
+   */
+  disableFidoCredentials = this.mutation(deactivateFidoCredentials)
+  /**
+   * Removes a member from a group
+   */
+  removeMemberFromGroup = this.mutation(deleteMember)
 }
 
 export const accountClient = new AccountClient()

package/src/error/CanceledError.ts

@@ -1,5 +1,8 @@
 import { StackspotAPIError } from './StackspotAPIError'
 
+/**
+ * This error is thrown when a request is canceled by the user before it completes.
+ */
 export class CanceledError extends StackspotAPIError {
   constructor() {
     super({ status: 0, message: lang => lang === 'en' ? 'Canceled by the user' : 'Cancelado pelo usuário' })

package/src/error/DefaultAPIError.ts

@@ -34,11 +34,37 @@ function createMessage(raw: ErrorResponse, dictionary: Dictionary, language: Lan
   return `${title}\n${details?.join('\n')}`
 }
 
+/**
+ * This represents the Error thrown by most Stackspot APIs.
+ */
 export class DefaultAPIError extends StackspotAPIError {
+  /**
+   * The error object as defined by the Stackspot API. If the response doesn't match the expected format this will be undefined.
+   */
   data?: ErrorResponse
+  /**
+   * The response data, if this matches the expected format expected by a Stackspot Error, it will be the same as `data`.
+   */
   raw?: any
 
-  constructor(rawResponse: any, status: number, dictionary: Dictionary, headers?: Headers) {
+  constructor(
+    /**
+     * The response's data
+     */
+    rawResponse: any,
+    /**
+     * The response's status
+     */
+    status: number,
+    /**
+     * A dictionary for translating error codes and descriptions.
+     */
+    dictionary: Dictionary,
+    /**
+     * The response's headers.
+     */
+    headers?: Headers,
+  ) {
     super({
       status: status,
       code: rawResponse.code,

package/src/error/StackspotAPIError.ts

@@ -3,16 +3,42 @@ import { Language, getLanguage } from '@stack-spot/portal-translate'
 type InternationalizedMessage = (language: Language) => string
 
 interface ErrorProperties {
+  /**
+   * The response's status.
+   */
   status: number,
+  /**
+   * The response's headers.
+   */
   headers?: Headers,
+  /**
+   * The error code defined by the API, if any.
+   */
   code?: string,
+  /**
+   * The error message: may be a simple string or a function that returns a string depending on the language passed as parameter.
+   * 
+   * If not provided, will attempt to use the error code or 'unknown'.
+   */
   message?: InternationalizedMessage | string,
+  /**
+   * The error's stack trace, if any.
+   */
   stack?: string,
 }
 
 export class StackspotAPIError extends Error {
+  /**
+   * The response's status.
+   */
   status: number
+  /**
+   * The response's headers.
+   */
   headers: Headers | undefined
+  /**
+   * The error code defined by the API, if any.
+   */
   code: string | undefined
   private intl?: InternationalizedMessage
 
@@ -25,7 +51,13 @@ export class StackspotAPIError extends Error {
     this.intl = typeof message === 'string' ? () => message : message
   }
 
+  /**
+   * Translates the error to the given language.
+   * @param language 
+   * @returns an error message
+   */
   translate(language: Language = getLanguage()) {
-    return this.intl?.(language) ?? this.message
+    const unknown = language === 'en' ? 'unknown error' : 'erro desconhecido'
+    return this.intl?.(language) ?? this.message ?? this.code ?? (this.status === 0 ? unknown : `${this.status}`)
   }
 }

package/src/index.ts

@@ -3,3 +3,4 @@ export { DefaultAPIError } from './error/DefaultAPIError'
 export { StackspotAPIError } from './error/StackspotAPIError'
 export { NetworkClient } from './network/NetworkClient'
 export { queryClient } from './network/react-query-client'
+export { useExtendedList } from './utils/use-extended-list'

package/src/network/AutoInfiniteQuery.ts

@@ -0,0 +1,104 @@
+/* eslint-disable react-hooks/rules-of-hooks */
+
+import {
+  InfiniteData, QueryFunctionContext, QueryKey,
+  UseInfiniteQueryResult,
+  useInfiniteQuery,
+  useSuspenseInfiniteQuery,
+} from '@tanstack/react-query'
+import { StackspotAPIError } from '../error/StackspotAPIError'
+import { AutoQuery } from './AutoQuery'
+import { queryClient } from './react-query-client'
+import { AutoQueryObjectParams, InfiniteQueryObject, InfiniteQueryOptions, UseInfiniteQueryObjectOptions } from './types'
+
+export class AutoInfiniteQuery<Variables, Result, PageParamName extends keyof Variables, Accumulator extends keyof Result | ''>
+  extends AutoQuery<Variables, Result>
+  implements InfiniteQueryObject<Variables, Result, Accumulator>
+{
+  private options: Required<InfiniteQueryOptions<Variables, Result, PageParamName, Accumulator | ''>>
+
+  constructor(
+    params: AutoQueryObjectParams<Variables, Result>,
+    options: InfiniteQueryOptions<Variables, Result, PageParamName, Accumulator>,
+  ) {
+    super(params)
+    this.options = {
+      ...options,
+      accumulator: options.accumulator ?? '',
+      defaultVariables: options.defaultVariables ?? {},
+    }
+  }
+
+  private createInfiniteQueryFn(
+    variables: Variables | undefined,
+  ): (context: QueryFunctionContext<QueryKey, Variables[PageParamName]>) => Promise<Result> {
+    return async ({ pageParam }) => {
+      const paginatedVariables = {
+        ...variables,
+        [this.options.pageParamName]: pageParam ?? variables?.[this.options.pageParamName],
+      } as Variables
+      if (!this.currentRequests.has(paginatedVariables)) this.currentRequests.set(paginatedVariables, this.callFn(paginatedVariables))
+      const result = await this.currentRequests.get(paginatedVariables)
+      this.currentRequests.delete(paginatedVariables)
+      return result!
+    }
+  }
+
+  private getListFromData(data: InfiniteData<Result> | undefined) {
+    return data?.pages.map(
+      page => this.options.accumulator ? page[this.options.accumulator as keyof Result] : page,
+    ).flat() as Accumulator extends keyof Result ? Result[Accumulator] : Result
+  }
+
+  private useInfiniteQueryResult(suspense: boolean, variables = {} as Variables, options?: UseInfiniteQueryObjectOptions) {
+    variables = { ...variables, ...this.options.defaultVariables }
+    const use = suspense ? useSuspenseInfiniteQuery : useInfiniteQuery
+    return use<any, any, any, any, any>({
+      ...options,
+      queryKey: ['infinite', ...this.getKey(variables)],
+      queryFn: this.createInfiniteQueryFn(variables),
+      initialPageParam: this.options.initialPageParam,
+      getNextPageParam: (lastPage, allPages, lastPageParam, allPageParams) =>
+        this.options.getNextPageParam({ variables, lastPage, allPages, lastPageParam, allPageParams }),
+    }, queryClient) as UseInfiniteQueryResult<InfiniteData<Result>, StackspotAPIError>
+  }
+
+  useInfiniteQuery(
+    variables?: Variables,
+    options?: UseInfiniteQueryObjectOptions,
+  ): [
+    Accumulator extends keyof Result ? Result[Accumulator] : Result,
+    UseInfiniteQueryResult<InfiniteData<Result>, StackspotAPIError>,
+  ] {
+    const result = this.useInfiniteQueryResult(true, variables, options)
+    return [
+      this.getListFromData(result.data),
+      result,
+    ]
+  }
+
+  useStatefulInfiniteQuery(
+    variables?: Variables,
+    options?: UseInfiniteQueryObjectOptions,
+  ): [
+    Accumulator extends keyof Result ? Result[Accumulator] : Result,
+    boolean,
+    StackspotAPIError | undefined | null,
+    UseInfiniteQueryResult<InfiniteData<Result>, StackspotAPIError>,
+  ] {
+    const result = this.useInfiniteQueryResult(false, variables, options)
+    return [
+      this.getListFromData(result.data),
+      result.isPending,
+      result.error,
+      result,
+    ]
+  }
+
+  async invalidate(variables?: Partial<Variables> | undefined): Promise<void> {
+    await Promise.all([
+      super.invalidate(variables),
+      queryClient.invalidateQueries({ queryKey: ['infinite', ...this.getKey(variables)] }),
+    ])
+  }
+}

package/src/network/AutoOperation.ts

@@ -61,7 +61,13 @@ export abstract class AutoOperation<Variables> implements OperationObject<Variab
       ? [options?: Omit<UseQueryOptions, 'queryFn' | 'queryKey'>]
       : [variables?: Partial<Variables>, options?: Omit<UseQueryOptions, 'queryFn' | 'queryKey'>]
   ) {
-    const [variables, options] = args.length === 2 ? args : [undefined, args[0]]
+    /* `this.fn` is a oazapfts function, i.e. it has arity 1 or 2. If it accepts variables, its arity is 2: the 1st parameter is the
+    variables and the 2nd is the RequestOpts. If it doesn't accept variables, its arity is one: it accepts only the RequestOpts.
+    We can use this information to determine what the type of `args` actually is at runtime. If variables are accepted, than the 1st
+    argument is the variables and the 2nd is the query options, otherwise, it has a single argument, which is the query options. */
+    const [variables, options] = this.fn.length > 1
+      ? args as [Variables, Omit<UseQueryOptions, 'queryFn' | 'queryKey'> | undefined]
+      : [undefined, args[0] as Omit<UseQueryOptions, 'queryFn' | 'queryKey'> | undefined]
     const result = useQuery({
       ...options,
       queryKey: this.getPermissionKey(variables as Variables),

package/src/network/AutoQuery.ts

@@ -10,7 +10,8 @@ export class AutoQuery<Variables, Result> extends AutoOperation<Variables> imple
   /**
    * Prevents the same request from being triggered more than once if subsequent calls are made before the first ends.
    */
-  private currentRequests = new Map<Variables | undefined, Promise<Result>>()
+  protected currentRequests = new Map<Variables | undefined, Promise<Result>>()
+  
   constructor(params: AutoQueryObjectParams<Variables, Result>) {
     super(params)
   }

package/src/network/ManualInfiniteQuery.ts

@@ -0,0 +1,95 @@
+/* eslint-disable react-hooks/rules-of-hooks */
+
+import { InfiniteData, QueryFunctionContext, QueryKey, UseInfiniteQueryResult, useInfiniteQuery, useSuspenseInfiniteQuery } from '@tanstack/react-query'
+import { StackspotAPIError } from '../error/StackspotAPIError'
+import { ManualQuery } from './ManualQuery'
+import { queryClient } from './react-query-client'
+import { FullOperationConfig, InfiniteQueryConfig, InfiniteQueryObject, UseInfiniteQueryObjectOptions } from './types'
+
+export class ManualInfiniteQuery<
+  Variables extends Record<string, any>,
+  Result,
+  PageParamName extends keyof Variables,
+  Accumulator extends keyof Result | ''
+> extends ManualQuery<Variables, Result> implements InfiniteQueryObject<Variables, Result, Accumulator> {
+  constructor(config: InfiniteQueryConfig<Variables, Result, PageParamName, Accumulator> & { apiName: string }) {
+    super(config as FullOperationConfig<any, any>)
+  }
+
+  private getConfig() {
+    return this.config as unknown as InfiniteQueryConfig<Variables, Result, PageParamName, Accumulator>
+  }
+
+  private createInfiniteQueryFn(
+    variables: Variables,
+  ): (context: QueryFunctionContext<QueryKey, Variables[PageParamName]>) => Promise<Result> {
+    return async ({ pageParam }) => {
+      const paginatedVariables = {
+        ...variables,
+        [this.getConfig().pageParamName]: pageParam ?? variables?.[this.getConfig().pageParamName],
+      } as Variables
+      if (!this.currentRequests.has(paginatedVariables)) this.currentRequests.set(paginatedVariables, this.makeRequest(paginatedVariables))
+      const result = await this.currentRequests.get(paginatedVariables)
+      this.currentRequests.delete(paginatedVariables)
+      return result!
+    }
+  }
+
+  private getListFromData(data: InfiniteData<Result> | undefined) {
+    return data?.pages.map(
+      page => this.getConfig().accumulator ? page[this.getConfig().accumulator as keyof Result] : page,
+    ).flat() as Accumulator extends keyof Result ? Result[Accumulator] : Result
+  }
+
+  private useInfiniteQueryResult(suspense: boolean, variables = {} as Variables, options?: UseInfiniteQueryObjectOptions) {
+    variables = { ...variables, ...this.getConfig().defaultVariables }
+    const use = suspense ? useSuspenseInfiniteQuery : useInfiniteQuery
+    return use<any, any, any, any, any>({
+      ...options,
+      queryKey: ['infinite', ...this.getKey(variables)],
+      queryFn: this.createInfiniteQueryFn(variables),
+      initialPageParam: this.getConfig().initialPageParam,
+      getNextPageParam: (lastPage, allPages, lastPageParam, allPageParams) => 
+        this.getConfig().getNextPageParam({ variables, lastPage, allPages, lastPageParam, allPageParams }),
+    }, queryClient) as UseInfiniteQueryResult<InfiniteData<Result>, StackspotAPIError>
+  }
+
+  useInfiniteQuery(
+    variables?: Variables,
+    options?: UseInfiniteQueryObjectOptions,
+  ): [
+    Accumulator extends keyof Result ? Result[Accumulator] : Result,
+    UseInfiniteQueryResult<InfiniteData<Result>, StackspotAPIError>,
+  ] {
+    const result = this.useInfiniteQueryResult(true, variables, options)
+    return [
+      this.getListFromData(result.data),
+      result,
+    ]
+  }
+
+  useStatefulInfiniteQuery(
+    variables?: Variables,
+    options?: UseInfiniteQueryObjectOptions,
+  ): [
+    Accumulator extends keyof Result ? Result[Accumulator] : Result,
+    boolean,
+    StackspotAPIError | undefined | null,
+    UseInfiniteQueryResult<InfiniteData<Result>, StackspotAPIError>,
+  ] {
+    const result = this.useInfiniteQueryResult(false, variables, options)
+    return [
+      this.getListFromData(result.data),
+      result.isPending,
+      result.error,
+      result,
+    ]
+  }
+
+  async invalidate(variables?: Partial<Variables> | undefined): Promise<void> {
+    await Promise.all([
+      super.invalidate(variables),
+      queryClient.invalidateQueries({ queryKey: ['infinite', ...this.getKey(variables)] }),
+    ])
+  }
+}

package/src/network/ManualQuery.ts

@@ -13,13 +13,13 @@ export class ManualQuery<
   /**
    * Prevents the same request from being triggered more than once if subsequent calls are made before the first ends.
    */
-  private currentRequests = new Map<Record<string, any> | undefined, Promise<Result>>()
+  protected currentRequests = new Map<Record<string, any> | undefined, Promise<Result>>()
 
   constructor(config: FullOperationConfig<Variables extends void ? [AbortSignal] : [AbortSignal, Variables], Result>) {
     super(config)
   }
 
-  private async makeRequest(variables: Record<string, any> | undefined) {
+  protected async makeRequest(variables: Record<string, any> | undefined) {
     const abortController = new AbortController()
     this.abortMap.set(variables, [abortController])
     try {