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
|
// Used to search for packages
package packages
import (
"encoding/json"
"net/http"
"soko/pkg/app/handler/feeds"
"soko/pkg/app/layout"
"soko/pkg/database"
"soko/pkg/models"
"strings"
"github.com/go-pg/pg/v10"
)
// Search renders a template containing a list of search results
// for a given query of packages
func Search(w http.ResponseWriter, r *http.Request) {
searchTerm := getParameterValue("q", r)
if searchTerm == "" {
http.Redirect(w, r, "/", http.StatusMovedPermanently)
return
} else if strings.Contains(searchTerm, "@") {
var maintainers []models.Maintainer
database.DBCon.Model(&maintainers).Where("email = ?", searchTerm).Select()
if len(maintainers) > 0 {
http.Redirect(w, r, "/maintainer/"+searchTerm, http.StatusMovedPermanently)
return
}
} else if strings.Contains(searchTerm, "/") {
var packages []models.Package
database.DBCon.Model(&packages).Where("atom = ?", searchTerm).Select()
if len(packages) > 0 {
http.Redirect(w, r, "/packages/"+searchTerm, http.StatusMovedPermanently)
return
}
}
var packages []models.Package
query := database.DBCon.Model(&packages).
Relation("Versions")
if strings.Contains(searchTerm, "*") {
// if the query contains wildcards
wildcardSearchTerm := strings.ReplaceAll(searchTerm, "*", "%")
query = query.
WhereOr("atom LIKE ?", wildcardSearchTerm).
WhereOr("name LIKE ?", wildcardSearchTerm)
} else {
// if the query contains no wildcards do a fuzzy search
query = BuildSearchQuery(query, searchTerm).
WhereOr("atom LIKE ?", "%"+searchTerm+"%")
}
err := query.OrderExpr("name <-> ?", searchTerm).
Select()
if err != nil && err != pg.ErrNoRows {
http.Error(w, http.StatusText(http.StatusInternalServerError),
http.StatusInternalServerError)
return
}
if len(packages) == 1 {
http.Redirect(w, r, "/packages/"+packages[0].Atom, http.StatusMovedPermanently)
return
}
layout.Layout(searchTerm, "packages", search(searchTerm, packages)).Render(r.Context(), w)
}
// Search renders a template containing a list of search results
// for a given query of packages
func SearchFeed(w http.ResponseWriter, r *http.Request) {
searchTerm := getParameterValue("q", r)
searchTerm = strings.ReplaceAll(searchTerm, "*", "")
var packages []models.Package
err := BuildSearchQuery(database.DBCon.Model(&packages), searchTerm).
Relation("Versions").
OrderExpr("name <-> ?", searchTerm).
Select()
if err != nil && err != pg.ErrNoRows {
http.Error(w, http.StatusText(http.StatusInternalServerError),
http.StatusInternalServerError)
return
}
feeds.Packages(searchTerm, packages, w)
}
func BuildSearchQuery(query *pg.Query, searchString string) *pg.Query {
for _, searchTerm := range strings.Split(searchString, " ") {
if searchTerm != "" {
marshal, err := json.Marshal(searchTerm)
if err != nil {
continue
}
query = query.WhereGroup(func(q *pg.Query) (*pg.Query, error) {
return q.WhereOr("category % ?", searchTerm).
WhereOr("name % ?", searchTerm).
WhereOr("atom % ?", searchTerm).
WhereOr("maintainers @> ?", `[{"Name": `+string(marshal)+`}]`).
WhereOr("maintainers @> ?", `[{"Email": `+string(marshal)+`}]`), nil
})
}
}
return query
}
|