Mike
49 lines
1.5 KiB
TypeScript
49 lines
1.5 KiB
TypeScript
import { userActiveWorkspaceSlugQuery } from '~/lib/user/graphql/queries'
|
|
import { useQuery, useMutation } from '@vue/apollo-composable'
|
|
import { setActiveWorkspaceMutation } from '~/lib/user/graphql/mutations'
|
|
import { modifyObjectField, getCacheId } from '~/lib/common/helpers/graphql'
|
|
import type { MaybeNullOrUndefined } from '@speckle/shared'
|
|
|
|
export const useActiveWorkspaceSlug = () => {
|
|
const { result } = useQuery(userActiveWorkspaceSlugQuery)
|
|
|
|
const activeWorkspaceSlug = computed(
|
|
() => result.value?.activeUser?.activeWorkspace?.slug
|
|
)
|
|
|
|
return activeWorkspaceSlug
|
|
}
|
|
|
|
export const useSetActiveWorkspace = () => {
|
|
const { activeUser } = useActiveUser()
|
|
const { mutate } = useMutation(setActiveWorkspaceMutation, {
|
|
update: (cache, { data }) => {
|
|
if (!activeUser.value) return
|
|
|
|
const newWorkspace = data?.activeUserMutations?.setActiveWorkspace
|
|
|
|
modifyObjectField(
|
|
cache,
|
|
getCacheId('User', activeUser.value.id),
|
|
'activeWorkspace',
|
|
({ helpers: { ref } }) => {
|
|
// Handle case where workspace is set to null (personal projects)
|
|
return newWorkspace ? ref('LimitedWorkspace', newWorkspace.id) : null
|
|
}
|
|
)
|
|
}
|
|
})
|
|
|
|
const setActiveWorkspace = async (
|
|
options:
|
|
| { id: MaybeNullOrUndefined<string> }
|
|
| { slug: MaybeNullOrUndefined<string> }
|
|
) => {
|
|
return await mutate('slug' in options ? { slug: options.slug } : { id: options.id })
|
|
}
|
|
|
|
return {
|
|
setActiveWorkspace
|
|
}
|
|
}
|