Files
speckle-server/packages/frontend-2/middleware/requireValidModel.ts
T
Kristaps Fabians Geikins 489b8f547d fix(fe2): SSR memory leak caused by vue apollo (#3790)
* fix(fe2): SSR memory leak caused by vue apollo

* linting fixes
2025-01-13 10:13:59 +02:00

88 lines
2.5 KiB
TypeScript

import { ProjectVisibility } from '~/lib/common/generated/gql/graphql'
import { WorkspaceSsoErrorCodes } from '~/lib/workspaces/helpers/types'
import { useApolloClientFromNuxt } from '~~/lib/common/composables/graphql'
import {
convertThrowIntoFetchResult,
getFirstErrorMessage
} from '~~/lib/common/helpers/graphql'
import { projectModelCheckQuery } from '~~/lib/projects/graphql/queries'
/**
* Used in project page to validate that project ID refers to a valid project and redirects to 404 if not
*/
export default defineNuxtRouteMiddleware(async (to) => {
const projectId = to.params.id as string
const modelId = to.params.modelId as string
const client = useApolloClientFromNuxt()
const { data, errors } = await client
.query({
query: projectModelCheckQuery,
variables: { projectId, modelId }
})
.catch(convertThrowIntoFetchResult)
// If project succesfully resolved, move on
if (data?.project?.model?.id) return
const isForbidden = (errors || []).find((e) => e.extensions?.['code'] === 'FORBIDDEN')
const isProjectNotFound = (errors || []).find(
(e) => e.extensions?.['code'] === 'STREAM_NOT_FOUND'
)
const isModelNotFound = (errors || []).find(
(e) => e.extensions?.['code'] === 'BRANCH_NOT_FOUND'
)
// Check if project exists and model is valid
if (!data?.project?.model?.id) {
if (errors?.length) {
const errMsg = getFirstErrorMessage(errors)
return abortNavigation(errMsg)
}
return
}
// If project is public or link shareable, allow access
if (
data.project.visibility === ProjectVisibility.Public ||
data.project.visibility === ProjectVisibility.Unlisted
) {
return
}
// Check for SSO session error
const ssoSessionError = (errors || []).find(
(e) => e.extensions?.['code'] === WorkspaceSsoErrorCodes.SESSION_MISSING_OR_EXPIRED
)
if (ssoSessionError) {
const workspaceSlug = ssoSessionError.message
return navigateTo(`/workspaces/${workspaceSlug}/sso/session-error`)
}
if (isForbidden) {
return abortNavigation(
createError({
statusCode: 403,
message: 'You do not have access to this project'
})
)
}
if (isProjectNotFound) {
return abortNavigation(
createError({ statusCode: 404, message: 'Project not found' })
)
}
if (isModelNotFound) {
return abortNavigation(createError({ statusCode: 404, message: 'Model not found' }))
}
if (errors?.length) {
const errMsg = getFirstErrorMessage(errors)
return abortNavigation(errMsg)
}
})