Kristaps Fabians Geikins
70 lines
2.1 KiB
TypeScript
70 lines
2.1 KiB
TypeScript
/* eslint-disable @typescript-eslint/no-explicit-any */
|
|
import type { AuthContext } from '@/modules/shared/authz'
|
|
import type DataLoader from 'dataloader'
|
|
import { ApolloServerErrorCode } from '@apollo/server/errors'
|
|
import { GraphQLError } from 'graphql'
|
|
import {
|
|
BadRequestError,
|
|
ForbiddenError,
|
|
InvalidArgumentError,
|
|
NotFoundError,
|
|
UnauthorizedError
|
|
} from '@/modules/shared/errors'
|
|
import type { Optional } from '@speckle/shared'
|
|
import type { Knex } from 'knex'
|
|
|
|
/**
|
|
* All dataloaders must at the very least follow this type
|
|
*/
|
|
export type ModularizedDataLoadersConstraint = {
|
|
[group: string]: Optional<{
|
|
[loader: string]: DataLoader<any, any> | { clearAll: () => unknown }
|
|
}>
|
|
}
|
|
|
|
export type RequestDataLoadersBuilder<T extends ModularizedDataLoadersConstraint> =
|
|
(params: {
|
|
ctx: AuthContext
|
|
createLoader: <K, V, C = K>(
|
|
batchLoadFn: DataLoader.BatchLoadFn<K, V>,
|
|
options?: DataLoader.Options<K, V, C>
|
|
) => DataLoader<K, V, C>
|
|
deps: {
|
|
db: Knex
|
|
}
|
|
}) => T
|
|
|
|
export const defineRequestDataloaders = <T extends ModularizedDataLoadersConstraint>(
|
|
builder: RequestDataLoadersBuilder<T>
|
|
): RequestDataLoadersBuilder<T> => {
|
|
return builder
|
|
}
|
|
|
|
export const simpleTupleCacheKey = (key: [string, string]) => `${key[0]}:${key[1]}`
|
|
|
|
/**
|
|
* Is a lower significance error, caused by user error (and thus - not a bug in our code)
|
|
*/
|
|
export const isUserGraphqlError = (error: GraphQLError): boolean => {
|
|
const userCodes = [
|
|
ForbiddenError.code,
|
|
UnauthorizedError.code,
|
|
BadRequestError.code,
|
|
NotFoundError.code,
|
|
InvalidArgumentError.code,
|
|
ApolloServerErrorCode.BAD_REQUEST,
|
|
ApolloServerErrorCode.BAD_USER_INPUT,
|
|
ApolloServerErrorCode.GRAPHQL_PARSE_FAILED,
|
|
ApolloServerErrorCode.GRAPHQL_VALIDATION_FAILED,
|
|
ApolloServerErrorCode.OPERATION_RESOLUTION_FAILURE,
|
|
ApolloServerErrorCode.PERSISTED_QUERY_NOT_FOUND,
|
|
ApolloServerErrorCode.PERSISTED_QUERY_NOT_SUPPORTED
|
|
]
|
|
const code = error.extensions?.code as string
|
|
return userCodes.includes(code)
|
|
}
|
|
|
|
export const isGraphQLError = (error: unknown): error is GraphQLError => {
|
|
return error instanceof GraphQLError
|
|
}
|