2 Commits

9 changed files with 108 additions and 68 deletions

View File

@@ -2,6 +2,7 @@
"cSpell.words": [ "cSpell.words": [
"bolthold", "bolthold",
"colly", "colly",
"htmx",
"incpatch", "incpatch",
"linkwallet", "linkwallet",
"nicetime", "nicetime",

View File

@@ -18,6 +18,11 @@ type BookmarkManager struct {
scrapeQueue chan *entity.Bookmark scrapeQueue chan *entity.Bookmark
} }
type SearchOptions struct {
Query string
Tags []string
}
func NewBookmarkManager(db *DB) *BookmarkManager { func NewBookmarkManager(db *DB) *BookmarkManager {
return &BookmarkManager{db: db, scrapeQueue: make(chan *entity.Bookmark)} return &BookmarkManager{db: db, scrapeQueue: make(chan *entity.Bookmark)}
} }
@@ -94,28 +99,12 @@ func (m *BookmarkManager) LoadBookmarkByID(id uint64) entity.Bookmark {
return ret return ret
} }
func (m *BookmarkManager) LoadBookmarksByIDs(ids []uint64) []entity.Bookmark { func (m *BookmarkManager) Search(opts SearchOptions) ([]entity.Bookmark, error) {
// log.Printf("loading %v", ids)
ret := make([]entity.Bookmark, 0, 0)
s := make([]interface{}, len(ids))
for i, v := range ids {
s[i] = v
}
err := m.db.store.Find(&ret, bolthold.Where("ID").In(s...))
if err != nil {
panic(err)
}
return ret
}
func (m *BookmarkManager) Search(query string) ([]entity.Bookmark, error) {
rets := make([]uint64, 0, 0)
// first get a list of all the ids that match our query
idsMatchingQuery := make([]uint64, 0, 0)
counts := make(map[uint64]uint8) counts := make(map[uint64]uint8)
words := content.StringToSearchWords(opts.Query)
words := content.StringToSearchWords(query)
for _, word := range words { for _, word := range words {
var wi *entity.WordIndex var wi *entity.WordIndex
@@ -133,14 +122,33 @@ func (m *BookmarkManager) Search(query string) ([]entity.Bookmark, error) {
for k, v := range counts { for k, v := range counts {
if v == uint8(len(words)) { if v == uint8(len(words)) {
rets = append(rets, k) idsMatchingQuery = append(idsMatchingQuery, k)
if len(rets) > 10 { if len(idsMatchingQuery) > 10 {
break break
} }
} }
} }
return m.LoadBookmarksByIDs(rets), nil // now we can do our search
bhQuery := bolthold.Query{}
if opts.Query != "" {
bhQuery = bolthold.Query(*bhQuery.And("ID").In(bolthold.Slice(idsMatchingQuery)...))
}
if opts.Tags != nil && len(opts.Tags) > 0 {
bhQuery = bolthold.Query(*bhQuery.And("Tags").ContainsAll(bolthold.Slice(opts.Tags)...))
}
out := []entity.Bookmark{}
err := m.db.store.ForEach(&bhQuery,
func(bm *entity.Bookmark) error {
out = append(out, *bm)
return nil
})
if err != nil {
panic(err)
}
return out, nil
} }
func (m *BookmarkManager) ScrapeAndIndex(bm *entity.Bookmark) error { func (m *BookmarkManager) ScrapeAndIndex(bm *entity.Bookmark) error {

View File

@@ -73,7 +73,7 @@ func BenchmarkOneWordSearch(b *testing.B) {
bmm := NewBookmarkManager(&dbh) bmm := NewBookmarkManager(&dbh)
b.ResetTimer() b.ResetTimer()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
bmm.Search("hello") bmm.Search(SearchOptions{Query: "hello"})
} }
} }
@@ -84,7 +84,7 @@ func BenchmarkTwoWordSearch(b *testing.B) {
bmm := NewBookmarkManager(&dbh) bmm := NewBookmarkManager(&dbh)
b.ResetTimer() b.ResetTimer()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
bmm.Search("human relate") bmm.Search(SearchOptions{Query: "human relate"})
} }
} }
@@ -95,6 +95,6 @@ func BenchmarkThreeWordSearch(b *testing.B) {
bmm := NewBookmarkManager(&dbh) bmm := NewBookmarkManager(&dbh)
b.ResetTimer() b.ResetTimer()
for i := 0; i < b.N; i++ { for i := 0; i < b.N; i++ {
bmm.Search("human wiki editor") bmm.Search(SearchOptions{Query: "human wiki editor"})
} }
} }

View File

@@ -47,7 +47,7 @@ func TestAddRemove(t *testing.T) {
t.Errorf("scrape index returned %s", err) t.Errorf("scrape index returned %s", err)
} }
searchRes, err := bmm.Search("fox") searchRes, err := bmm.Search(SearchOptions{Query: "fox"})
if err != nil { if err != nil {
t.Errorf("search returned %s", err) t.Errorf("search returned %s", err)
} }
@@ -62,7 +62,7 @@ func TestAddRemove(t *testing.T) {
t.Errorf("scrape index returned %s", err) t.Errorf("scrape index returned %s", err)
} }
searchRes, err = bmm.Search("fox") searchRes, err = bmm.Search(SearchOptions{Query: "fox"})
if err != nil { if err != nil {
t.Errorf("search returned %s", err) t.Errorf("search returned %s", err)
} }
@@ -70,7 +70,7 @@ func TestAddRemove(t *testing.T) {
t.Error("got result when should not") t.Error("got result when should not")
} }
searchRes, err = bmm.Search("rabbit") searchRes, err = bmm.Search(SearchOptions{Query: "rabbit"})
if err != nil { if err != nil {
t.Errorf("search returned %s", err) t.Errorf("search returned %s", err)
} }
@@ -83,7 +83,7 @@ func TestAddRemove(t *testing.T) {
t.Errorf("got error when deleting: %s", err) t.Errorf("got error when deleting: %s", err)
} }
searchRes, err = bmm.Search("rabbit") searchRes, err = bmm.Search(SearchOptions{Query: "rabbit"})
if err != nil { if err != nil {
t.Errorf("search returned %s", err) t.Errorf("search returned %s", err)
} }
@@ -119,7 +119,7 @@ func TestTagIndexing(t *testing.T) {
t.Errorf("scrape index returned %s", err) t.Errorf("scrape index returned %s", err)
} }
searchRes, err := bmm.Search("fox") searchRes, err := bmm.Search(SearchOptions{Query: "fox"})
if err != nil { if err != nil {
t.Errorf("search returned %s", err) t.Errorf("search returned %s", err)
} }
@@ -133,7 +133,7 @@ func TestTagIndexing(t *testing.T) {
if err != nil { if err != nil {
t.Errorf("scrape index returned %s", err) t.Errorf("scrape index returned %s", err)
} }
searchRes, err = bmm.Search("sloth") searchRes, err = bmm.Search(SearchOptions{Query: "sloth"})
if err != nil { if err != nil {
t.Errorf("search returned %s", err) t.Errorf("search returned %s", err)
} }

View File

@@ -8,7 +8,7 @@ import (
"golang.org/x/mod/semver" "golang.org/x/mod/semver"
) )
const Tag = "v0.0.20" const Tag = "v0.0.21"
var versionInfo struct { var versionInfo struct {
Local struct { Local struct {

View File

@@ -1,38 +1,20 @@
<div class="grid-x grid-padding-x"> <div class="grid-x grid-padding-x" id="manage">
<div class="large-12 cell"> <div class="large-12 cell">
<h5>Manage links</h5> <h5>Manage links</h5>
<table> <form onsubmit="return false">
<tr> <div class="grid-x grid-padding-x">
<th>&nbsp;</th> <div class="large-12 cell">
<th>title/url</th> <label>Filter</label>
<th>tags</th> <input type="text" name="query" placeholder="" hx-post="/manage/results" hx-swap="outerHTML"
<th class="show-for-large">created</th> hx-trigger="keyup changed delay:500ms, tag_update" hx-target="#manage-results"
<th class="show-for-large">scraped</th> hx-indicator="#htmx-indicator-search" id="manage-search" />
</tr> </div>
{{ range .bookmarks }} </div>
<tr> {{ template "tags_widget.html" . }}
<th><a class="button" href="/edit/{{ .ID }}">edit</a></th> </form>
<td> {{ template "manage_results.html" . }}
<a href="{{ .URL }}">{{ .Info.Title }}</a>
<br>
<a href="{{ .URL }}">{{ niceURL .URL }}</a>
</td>
<td>
{{ range .Tags }}
<span class="label primary">{{ . }}</span>
{{ end }}
</td>
<td class="show-for-large">{{ (nicetime .TimestampCreated).HumanDuration }} ago</td>
<td class="show-for-large">{{ (nicetime .TimestampLastScraped).HumanDuration }} ago</td>
<td>
<a class="button" hx-swap="outerHTML" hx-post="/scrape/{{ .ID }}">scrape</button>
</td>
</tr>
{{ end }}
</table>
</div> </div>
</div> </div>

View File

@@ -0,0 +1,31 @@
<table id="manage-results">
<tr>
<th>&nbsp;</th>
<th>title/url</th>
<th>tags</th>
<th class="show-for-large">created</th>
<th class="show-for-large">scraped</th>
</tr>
{{ range .bookmarks }}
<tr>
<th><a class="button" href="/edit/{{ .ID }}">edit</a></th>
<td>
<a href="{{ .URL }}">{{ .Info.Title }}</a>
<br>
<a href="{{ .URL }}">{{ niceURL .URL }}</a>
</td>
<td>
{{ range .Tags }}
<span class="label primary">{{ . }}</span>
{{ end }}
</td>
<td class="show-for-large">{{ (nicetime .TimestampCreated).HumanDuration }} ago</td>
<td class="show-for-large">{{ (nicetime .TimestampLastScraped).HumanDuration }} ago</td>
<td>
<a class="button" hx-swap="outerHTML" hx-post="/scrape/{{ .ID }}">scrape</button>
</td>
</tr>
{{ end }}
</table>

View File

@@ -9,7 +9,7 @@
<input id="tag-entry" type="text" name="tag" placeholder="enter tags" /> <input id="tag-entry" type="text" name="tag" placeholder="enter tags" />
</div> </div>
<div class="small-12 large-6 cell"> <div class="small-12 large-6 cell" id="tags-list">
{{ range .tags }} {{ range .tags }}
<a href="#" <a href="#"
class="" class=""
@@ -20,7 +20,7 @@
<span class="label primary">{{ . }}</span> <span class="label primary">{{ . }}</span>
{{ end }} {{ end }}
<input type="hidden" name="tags_hidden" value="{{ .tags_hidden }}"> <input _="on load send tag_update to #manage-search" type="hidden" id="tags-hidden" name="tags_hidden" value="{{ .tags_hidden }}">
</div> </div>
</div> </div>
</div> </div>

View File

@@ -81,6 +81,7 @@ func Create(bmm *db.BookmarkManager, cmm *db.ConfigManager) *Server {
}) })
r.GET("/manage", func(c *gin.Context) { r.GET("/manage", func(c *gin.Context) {
allBookmarks, _ := bmm.ListBookmarks() allBookmarks, _ := bmm.ListBookmarks()
meta := gin.H{"page": "manage", "config": config, "bookmarks": allBookmarks} meta := gin.H{"page": "manage", "config": config, "bookmarks": allBookmarks}
c.HTML(http.StatusOK, c.HTML(http.StatusOK,
@@ -88,6 +89,23 @@ func Create(bmm *db.BookmarkManager, cmm *db.ConfigManager) *Server {
) )
}) })
r.POST("/manage/results", func(c *gin.Context) {
query := c.PostForm("query")
tags := []string{}
if c.PostForm("tags_hidden") != "" {
tags = strings.Split(c.PostForm("tags_hidden"), "|")
}
allBookmarks, _ := bmm.Search(db.SearchOptions{Query: query, Tags: tags})
meta := gin.H{"config": config, "bookmarks": allBookmarks}
log.Printf("query is %s, tags %v", query, tags)
c.HTML(http.StatusOK,
"manage_results.html", meta,
)
})
r.GET("/config", func(c *gin.Context) { r.GET("/config", func(c *gin.Context) {
meta := gin.H{"page": "config", "config": config} meta := gin.H{"page": "config", "config": config}
c.HTML(http.StatusOK, c.HTML(http.StatusOK,
@@ -107,7 +125,7 @@ func Create(bmm *db.BookmarkManager, cmm *db.ConfigManager) *Server {
r.POST("/search", func(c *gin.Context) { r.POST("/search", func(c *gin.Context) {
query := c.PostForm("query") query := c.PostForm("query")
sr, err := bmm.Search(query) sr, err := bmm.Search(db.SearchOptions{Query: query})
data := gin.H{ data := gin.H{
"results": sr, "results": sr,
"error": err, "error": err,