From 8b9f456c4a6ccabb26ff77179ddb942fe7b51c37 Mon Sep 17 00:00:00 2001
From: tylerslaton <mtslaton1@gmail.com>
Date: Fri, 22 Mar 2024 12:03:57 -0400
Subject: [PATCH] feat: add pagination to search API and UI

Signed-off-by: tylerslaton <mtslaton1@gmail.com>
---
 src/lib/db.ts                | 22 +++++++++++-----------
 src/pages/search.vue         | 33 ++++++++++++++++++++++++---------
 src/server/api/search.get.ts |  6 ++++--
 3 files changed, 39 insertions(+), 22 deletions(-)

diff --git a/src/lib/db.ts b/src/lib/db.ts
index 511e8aa..f792ffc 100644
--- a/src/lib/db.ts
+++ b/src/lib/db.ts
@@ -87,22 +87,22 @@ export async function removeToolForUrlIfExists(url: string): Promise<Tool[]> {
   return toolEntry.content as Tool[]
 }
 
-export async function getToolsForQuery(query: string): Promise<Record<string, Tool[]>> {
-  const toolEntries = await prisma.toolEntry.findMany({ where: { reference: { contains: query } } })
+export async function getToolsForQuery(query: string, page: number, pageSize: number): Promise<{ tools: Record<string, Tool[]>, totalCount: number }> {
+  const skip = (page - 1) * pageSize
+  const toolEntries = await prisma.toolEntry.findMany({
+    where: { reference: { contains: query } },
+    take: pageSize,
+    skip: skip > 0 ? skip : undefined,
+  })
 
   const tools: Record<string, Tool[]> = {}
 
   for (const entry of toolEntries) {
     const parsedTool = entry.content as Tool[]
-
-    for (const tool of parsedTool) {
-      if (tools[entry.reference]) {
-        tools[entry.reference].push(tool)
-      } else {
-        tools[entry.reference] = [tool]
-      }
-    }
+    tools[entry.reference] = tools[entry.reference] || []
+    tools[entry.reference].push(...parsedTool)
   }
 
-  return tools
+  const totalCount = await prisma.toolEntry.count({ where: { reference: { contains: query } } })
+  return { tools, totalCount }
 }
diff --git a/src/pages/search.vue b/src/pages/search.vue
index 26fae04..ac0546d 100644
--- a/src/pages/search.vue
+++ b/src/pages/search.vue
@@ -1,48 +1,58 @@
 <script setup lang="ts">
 import type { Tool } from '@/lib/types'
 
+const pageSize = 10
+
 const route = useRoute()
 const searchResults = ref({} as Record<string, Tool[]>)
 const error = ref({ status: 0, message: '' })
 const loading = ref(true)
+const page = ref(1)
+const totalItems = ref(0)
 
 definePageMeta({
   middleware: [
     (_from, to) => {
-      // The computed value isn't available yet in a middleware
-      const q = `${ to.query.q }`.trim()
+      const q = `${to.query.q}`.trim()
 
       if (!q) {
         return navigateTo('/')
       }
 
       if (/^(https?:\/\/)?(www\.)?github\.com\/[\w-]+\/[\w-]+(\/[\w-]+)*$/.test(q)) {
-        return navigateTo(`/${ q }`, { replace: true })
+        return navigateTo(`/${q}`, { replace: true })
       }
     },
   ],
 })
 
 const q = computed(() => {
-  return `${ route.query.q }`.trim()
+  return `${route.query.q}`.trim()
 })
 
 async function fetchData() {
-  const results = await fetch(`/api/search?q=${ q.value }`)
+  const results = await fetch(`/api/search?q=${ q.value }&page=${page.value}`)
 
   if (!results.ok) {
     error.value = { status: results.status, message: results.statusText }
     loading.value = false
-
     return
   }
 
-  searchResults.value = await results.json() as Record<string, Tool[]>
+  const { tools, totalCount } = await results.json()
+  console.log(totalCount)
+  searchResults.value = tools
+  totalItems.value = totalCount
   loading.value = false
 }
 
-watch(() => route.query, () => fetchData())
+watch([() => route.query, page], () => fetchData())
 onMounted(() => fetchData())
+
+async function onPageChange(newPage: number) {
+  page.value = newPage
+}
+
 </script>
 
 <template>
@@ -50,7 +60,7 @@ onMounted(() => fetchData())
   <Error v-else-if="error?.status || !searchResults" :title="`${error?.status || 0}`" :message="error?.message || 'Unknown'" />
   <div v-else>
     <h1 class="text-2xl font-semibold mb-8">
-      <template v-if="q === '.'">
+      <template v-if="q === '.' || q === ''">
         All Tools
       </template>
       <template v-else-if="q === ('sys.')">
@@ -61,6 +71,8 @@ onMounted(() => fetchData())
       </template>
     </h1>
 
+    <UPagination class="my-8" v-if="totalItems > pageSize" v-model="page" :page-count="10" :total="totalItems" @change="onPageChange" />
+
     <MiniCard v-for="(tools, url) in searchResults" :key="url" class="mb-4">
       <template #header>
         <UButton variant="ghost" class="text-xl font-semibold block w-full" :to="`/${url}`">
@@ -77,5 +89,8 @@ onMounted(() => fetchData())
         </p>
       </div>
     </MiniCard>
+
+    <UPagination class="my-8" v-if="totalItems > pageSize" v-model="page" :page-count="10" :total="totalItems" @change="onPageChange" />
+
   </div>
 </template>
diff --git a/src/server/api/search.get.ts b/src/server/api/search.get.ts
index a8fa51e..83fb3ea 100644
--- a/src/server/api/search.get.ts
+++ b/src/server/api/search.get.ts
@@ -1,7 +1,9 @@
 import * as db from '@/lib/db'
 
+const pageSize = 10
+
 export default defineEventHandler(async (event) => {
   setResponseHeader(event, 'Content-Type', 'application/json')
-
-  return await db.getToolsForQuery(getQuery(event).q as string)
+  const {q, page} = getQuery(event)
+  return await db.getToolsForQuery(q as string, page as number, pageSize)
 })