fix: server/utils types

Intevel · Mar 19, 2024 · aeafc47 · aeafc47
1 parent ce75616
commit aeafc47
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/runtime/server/utils/use-directus.ts b/src/runtime/server/utils/use-directus.ts
@@ -5,6 +5,12 @@ import {
   rest,
   staticToken
 } from '@directus/sdk'
+import type {
+  DirectusClient,
+  RestClient,
+  StaticTokenClient,
+  WebSocketClient
+} from '@directus/sdk'
 import { WebSocket } from 'ws'
 import type {
   DirectusClientOptions,
@@ -18,7 +24,7 @@ import { useRuntimeConfig } from '#imports'
  *
  * @returns createDirectus.
  */
-export const useDirectus = <T extends object = any>(options?: Partial<DirectusClientOptions>) => {
+export const useDirectus = <T extends object = any>(options?: Partial<DirectusClientOptions>): DirectusClient<T> => {
   const { url } = useRuntimeConfig().public.directus
 
   const defaultOptions: DirectusClientOptions = {
@@ -45,7 +51,7 @@ export const useDirectus = <T extends object = any>(options?: Partial<DirectusCl
  *
  * @returns A Directus REST client.
  */
-export const useDirectusRest = <T extends object = any>(options?: Partial<Omit<DirectusRestConfig, 'authConfig'>>) => {
+export const useDirectusRest = <T extends object = any>(options?: Partial<Omit<DirectusRestConfig, 'authConfig'>>): DirectusClient<T> & RestClient<T> & StaticTokenClient<T> => {
   const { staticToken: privateToken } = useRuntimeConfig().directus
 
   const defaultOptions: Partial<Omit<DirectusRestConfig, 'authConfig'>> = {
@@ -75,7 +81,7 @@ export const useDirectusRest = <T extends object = any>(options?: Partial<Omit<D
  *
  * @returns A Directus Realtime client.
  */
-export const useDirectusRealtime = <T extends object = any>(options?: Partial<DirectusRealtimeConfig>) => {
+export const useDirectusRealtime = <T extends object = any>(options?: Partial<DirectusRealtimeConfig>): DirectusClient<T> & WebSocketClient<T> & StaticTokenClient<T> => {
   const { staticToken: privateToken } = useRuntimeConfig().directus
 
   const defaultOptions: Partial<Omit<DirectusRealtimeConfig, 'authConfig'>> = {