Shakesearch submission

main.go

@@ -9,6 +9,8 @@ import (
 	"log"
 	"net/http"
 	"os"
+	"strconv"
+	"strings"
 )
 
 func main() {
@@ -48,10 +50,38 @@ func handleSearch(searcher Searcher) func(w http.ResponseWriter, r *http.Request
 			w.Write([]byte("missing search query in URL params"))
 			return
 		}
+		page, ok := r.URL.Query()["p"]
+		if !ok || len(page[0]) < 1 {
+			page = []string{"1"}
+		}
+
+		pageNum, err := strconv.ParseInt(page[0], 10, 64)
+		if err != nil || pageNum < 1 {
+			w.WriteHeader(http.StatusBadRequest)
+			w.Write([]byte("invalid page number"))
+			return
+		}
+		pageIdx := pageNum - 1
+		pageSize := 20
 		results := searcher.Search(query[0])
+
+		start := int(pageIdx) * pageSize
+		end := (int(pageIdx) + 1) * pageSize
+
+		// Check the bounds before slicing
+		if start > len(results) {
+			start = len(results)
+		}
+		if end > len(results) {
+			end = len(results)
+		}
+
+		// Slice the results
+		results = results[start:end]
+
 		buf := &bytes.Buffer{}
 		enc := json.NewEncoder(buf)
-		err := enc.Encode(results)
+		err = enc.Encode(results)
 		if err != nil {
 			w.WriteHeader(http.StatusInternalServerError)
 			w.Write([]byte("encoding failure"))
@@ -68,11 +98,12 @@ func (s *Searcher) Load(filename string) error {
 		return fmt.Errorf("Load: %w", err)
 	}
 	s.CompleteWorks = string(dat)
-	s.SuffixArray = suffixarray.New(dat)
+	s.SuffixArray = suffixarray.New([]byte(strings.ToLower(s.CompleteWorks)))
 	return nil
 }
 
 func (s *Searcher) Search(query string) []string {
+	query = strings.ToLower(query)
 	idxs := s.SuffixArray.Lookup([]byte(query), -1)
 	results := []string{}
 	for _, idx := range idxs {

static/app.js

@@ -1,9 +1,22 @@
 const Controller = {
+  currentPage: 1,
   search: (ev) => {
     ev.preventDefault();
+    Controller.currentPage = 1;
     const form = document.getElementById("form");
     const data = Object.fromEntries(new FormData(form));
-    const response = fetch(`/search?q=${data.query}`).then((response) => {
+    fetch(`/search?q=${data.query}`).then((response) => {
+      response.json().then((results) => {
+        Controller.updateTable(results);
+      });
+    });
+  },
+
+  loadMore: (ev) => {
+    ev.preventDefault();
+    Controller.currentPage = Controller.currentPage + 1;
+    const searchQuery = document.getElementById("query").value;
+    fetch(`/search?q=${searchQuery}&p=${Controller.currentPage}`).then((response) => {
       response.json().then((results) => {
         Controller.updateTable(results);
       });
@@ -16,9 +29,16 @@ const Controller = {
     for (let result of results) {
       rows.push(`<tr><td>${result}</td></tr>`);
     }
-    table.innerHTML = rows;
+    if (Controller.currentPage > 1) {
+      table.innerHTML = table.innerHTML + rows.join("");
+    } else {
+      table.innerHTML = rows.join("");
+    }
   },
 };
 
 const form = document.getElementById("form");
 form.addEventListener("submit", Controller.search);
+
+const loadMore = document.getElementById("load-more");
+loadMore.addEventListener("click", Controller.loadMore);

static/index.html

@@ -21,7 +21,7 @@
     <tbody id="table-body"></tbody>
   </table>
   </p>
-  <button>Load More</button>
+  <button id="load-more">Load More</button>
   <script src="app.js"></script>
 </body>