feat: better typesafety

amandesai01 · Jul 8, 2024 · 0e8fc75 · 0e8fc75
1 parent 8bc99d2
commit 0e8fc75
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/runtime/composables/useOramaInstance.ts b/src/runtime/composables/useOramaInstance.ts
@@ -1,8 +1,9 @@
 import { useNuxtApp } from "nuxt/app";
 import { DEFAULT_KEY } from "../constants";
 import type { NuxtOramaProvider } from "../interfaces";
+import type { AnyOrama } from "@orama/orama";
 
-export default function useOramaInstance(id?: string) {
+export default function useOramaInstance<T extends AnyOrama>(id?: string) {
   const key = id || DEFAULT_KEY;
   const nuxtApp = useNuxtApp();
 
@@ -18,5 +19,5 @@ export default function useOramaInstance(id?: string) {
     }
   }
 
-  return oramaInstance;
+  return oramaInstance as T;
 }
diff --git a/src/runtime/composables/useOramaSearch.ts b/src/runtime/composables/useOramaSearch.ts
@@ -9,8 +9,8 @@ import { wrapPromiseToRef } from "../utils";
 import { ref, type Ref } from "#imports";
 import useOramaInstance from "./useOramaInstance";
 
-export default function useOramaSearch(id?: string) {
-  const oramaInstance = useOramaInstance(id);
+export default function useOramaSearch<T extends AnyOrama>(id?: string) {
+  const oramaInstance = useOramaInstance<T>(id);
 
   const searchResults: Ref<Results<any> | null> = ref(null);
   const searchPending: Ref<boolean> = ref(false);
@@ -35,14 +35,14 @@ export default function useOramaSearch(id?: string) {
   };
 
   const insertMultipleWrapped = (
-    docs: PartialSchemaDeep<any>[],
+    docs: PartialSchemaDeep<TypedDocument<T>>[],
     batchSize?: number,
     language?: string,
     skipHooks?: boolean,
     timeout?: number
   ) => {
     return wrapPromiseToRef(
-      insertMultipleOrama(
+      insertMultipleOrama<T>(
         oramaInstance,
         docs,
         batchSize,