chore: initial commit

src/lib/client/localstorage.ts

@@ -0,0 +1,48 @@
+import type { AppConfig } from '@/lib/types'
+
+const localStorageItemKey = 'chromadb-admin-config'
+
+export function getConfig(): AppConfig {
+  const config = window.localStorage.getItem(localStorageItemKey)
+  if (config) {
+    return JSON.parse(config)
+  } else {
+    return {
+      connectionString: '',
+      currentCollection: '',
+      authType: 'no_auth',
+      token: '',
+      username: '',
+      password: '',
+      tenant: 'default_tenant',
+      database: 'default_database',
+      embeddingModelUrl: '',
+      embeddingModel: 'text-embedding-3-small',
+    }
+  }
+}
+
+export function updateConfig(config: AppConfig) {
+  const stringValue = JSON.stringify(config)
+  return window.localStorage.setItem(localStorageItemKey, stringValue)
+}
+
+export function updateConnectionString(connectionString: string) {
+  const config = getConfig() || {
+    connectionString: '',
+    currentCollection: '',
+    authType: '',
+    token: '',
+    username: '',
+    password: '',
+    tenant: '',
+    database: '',
+    embeddingModelUrl: '',
+    embeddingModel: '',
+  }
+  const newConfig = {
+    ...config,
+    connectionString,
+  }
+  return updateConfig(newConfig)
+}

src/lib/client/query.ts

@@ -0,0 +1,177 @@
+import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query'
+
+import { getConfig } from '@/lib/client/localstorage'
+
+import type { AppConfig, Collection, QueryResult } from '@/lib/types'
+
+export function useGetConfig() {
+  return useQuery({
+    queryKey: ['config'],
+    queryFn: getConfig,
+    retry: false,
+  })
+}
+
+function authParamsString(config?: AppConfig) {
+  if (config?.authType === 'basic') {
+    return `&authType=basic&&username=${config.username}&password=${config.password}`
+  } else if (config?.authType === 'token') {
+    return `&authType=token&&token=${config.token}`
+  } else {
+    return ''
+  }
+}
+
+export function useGetCollections(config?: AppConfig) {
+  return useQuery({
+    queryKey: ['config', config?.connectionString, 'collections'],
+    queryFn: async (): Promise<Collection[]> => {
+      const response = await fetch(
+        `/api/collections?connectionString=${config?.connectionString}${authParamsString(config)}&tenant=${config?.tenant}&database=${config?.database}`
+      )
+      if (!response.ok) {
+        throw new Error(`API getCollections returns response code: ${response.status}, message: ${response.statusText}`)
+      }
+      return response.json()
+    },
+    enabled: !!config?.connectionString,
+    retry: false,
+  })
+}
+
+export function useGetCollectionRecords(config?: AppConfig, collectionName?: string, page?: number, query?: string) {
+  return useQuery({
+    queryKey: ['collections', collectionName, 'records', query, page],
+    queryFn: async (): Promise<QueryResult> => {
+      if (query === undefined || query === '') {
+        const response = await fetch(
+          `/api/collections/${collectionName}/records?connectionString=${config?.connectionString}&tenant=${config?.tenant}&database=${config?.database}&page=${page}&query=${query}${authParamsString(config)}`
+        )
+        return response.json()
+      } else {
+        const response = await fetch(
+          `/api/collections/${collectionName}/records?connectionString=${config?.connectionString}&tenant=${config?.tenant}&database=${config?.database}&authType=${config?.authType}${authParamsString(config)}`,
+          {
+            method: 'POST',
+            body: JSON.stringify({ query: query }),
+          }
+        )
+        return response.json()
+      }
+    },
+    enabled: !!config?.connectionString,
+    retry: false,
+  })
+}
+
+export function useDeleteRecord(collectionName: string) {
+  const queryClient = useQueryClient()
+  const config = getConfig()
+
+  return useMutation({
+    mutationFn: async (recordId: string) => {
+      const response = await fetch(
+        `/api/collections/${collectionName}/records?connectionString=${config?.connectionString}&tenant=${config?.tenant}&database=${config?.database}${authParamsString(config)}`,
+        {
+          method: 'DELETE',
+          headers: {
+            'Content-Type': 'application/json',
+          },
+          body: JSON.stringify({ id: recordId }),
+        }
+      )
+
+      if (!response.ok) {
+        const error = await response.json()
+        throw new Error(error.error || 'Failed to delete record')
+      }
+
+      return response.json()
+    },
+    onSuccess: () => {
+      // Invalidate queries to refetch data
+      queryClient.invalidateQueries({ queryKey: ['collections', collectionName, 'records'] })
+    },
+  })
+}
+
+export function useGetEmbedding() {
+  return useMutation({
+    mutationFn: async ({ text, modelUrl, model }: { text: string; modelUrl: string; model?: string }) => {
+      const response = await fetch('/api/embedding', {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+        },
+        body: JSON.stringify({ text, modelUrl, model }),
+      })
+
+      if (!response.ok) {
+        const error = await response.json()
+        throw new Error(error.error || 'Failed to get embedding')
+      }
+
+      return response.json()
+    },
+  })
+}
+
+export function useDeleteCollection() {
+  const queryClient = useQueryClient()
+  const config = getConfig()
+
+  return useMutation({
+    mutationFn: async (collectionName: string) => {
+      const response = await fetch(
+        `/api/collections?connectionString=${config?.connectionString}&tenant=${config?.tenant}&database=${config?.database}${authParamsString(config)}`,
+        {
+          method: 'DELETE',
+          headers: {
+            'Content-Type': 'application/json',
+          },
+          body: JSON.stringify({ name: collectionName }),
+        }
+      )
+
+      if (!response.ok) {
+        const error = await response.json()
+        throw new Error(error.error || 'Failed to delete collection')
+      }
+
+      return response.json()
+    },
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ['config', config?.connectionString, 'collections'] })
+    },
+  })
+}
+
+export function useRenameCollection() {
+  const queryClient = useQueryClient()
+  const config = getConfig()
+
+  return useMutation({
+    mutationFn: async ({ oldName, newName }: { oldName: string; newName: string }) => {
+      const response = await fetch(
+        `/api/collections?connectionString=${config?.connectionString}&tenant=${config?.tenant}&database=${config?.database}${authParamsString(config)}`,
+        {
+          method: 'PATCH',
+          headers: {
+            'Content-Type': 'application/json',
+          },
+          body: JSON.stringify({ oldName, newName }),
+        }
+      )
+
+      if (!response.ok) {
+        const error = await response.json()
+        throw new Error(error.error || 'Failed to rename collection')
+      }
+
+      return response.json()
+    },
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ['config', config?.connectionString, 'collections'] })
+    },
+  })
+}