summaryrefslogtreecommitdiff
path: root/main.go
blob: 44f5bdcbcb9a92a2623a9c09db31af4cc2b6f071 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package main

import (
	"fmt"
	"html/template"
	"math"
	"net/http"
	"os"
	"strconv"

	"github.com/joho/godotenv"
	"github.com/meilisearch/meilisearch-go"
)

type SearchQuery struct {
	Query      string
	MaxResults int64
	Offset     int64
}

type SearchResponse struct {
	Success       bool
	SearchResults []interface{}
	NumResults    int
	TotalResults  int64
	MoreResults   bool
	OriginalQuery SearchQuery
	Offset        int64
	LastOffset    int64
	NumPages      int
}

func pageCount(total int, perPage int) int {
	return int(math.Ceil(float64(total) / float64(perPage)))
}

func main() {
	err := godotenv.Load(".env")
	if err != nil {
		fmt.Println("Error loading .env file")
		os.Exit(1)
	}
	meili_host, ok := os.LookupEnv("MEILI_HOST")
	if !ok {
		fmt.Println("Error loading MEILI_HOST from .env file")
		os.Exit(1)
	}

	client := meilisearch.NewClient(meilisearch.ClientConfig{
		Host: meili_host,
	})

	index := client.Index("fda510k")

	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		t, _ := template.ParseFiles("search.gtpl")
		t.Execute(w, nil)
	})

	searchResTemplate := template.Must(template.ParseFiles("results.gtpl"))

	http.HandleFunc("/search", func(w http.ResponseWriter, r *http.Request) {
		r.ParseForm()
		fmt.Println(r.Form)
		if r.Form["query"] != nil || r.FormValue("query") != "" {
			fmt.Println("query:", r.Form["query"])
			var myOffset int64
			if r.Form["offset"] != nil {
				offset, _ := strconv.ParseInt(r.FormValue("offset"), 10, 64)
				myOffset = offset
				if offset < 0 {
					myOffset = 0
				}
			} else {
				offset := int64(0)
				myOffset = offset
			}
			query := SearchQuery{
				Query:      r.FormValue("query"),
				MaxResults: 100,
				Offset:     myOffset,
			}

			res, err := index.Search(query.Query, &meilisearch.SearchRequest{
				Limit:  query.MaxResults,
				Offset: query.Offset,
			})

			if err != nil {
				http.Error(w, err.Error(), http.StatusInternalServerError)
				return
			}

			numPages := pageCount(int(res.EstimatedTotalHits), int(query.MaxResults))

			searchResTemplate.Execute(w, SearchResponse{
				Success:       true,
				SearchResults: res.Hits,
				NumResults:    len(res.Hits) + int(query.Offset),
				TotalResults:  res.EstimatedTotalHits,
				MoreResults:   res.EstimatedTotalHits > query.MaxResults,
				OriginalQuery: query,
				Offset:        query.Offset + query.MaxResults,
				LastOffset:    query.Offset - query.MaxResults,
				NumPages:      numPages,
			})
		} else {
			fmt.Println("query is empty")
		}
	})

	fmt.Println("Listening on port 8080")
	http.ListenAndServe(":8080", nil)
}