package database import ( "fmt" "html" "strings" "git.eeqj.de/sneak/feta/toot" "github.com/google/uuid" hstg "github.com/grokify/html-strip-tags-go" _ "github.com/jinzhu/gorm/dialects/postgres" ) func (m *Manager) TootInsertHashCacheSize() uint { m.cachelock.Lock() defer m.cachelock.Unlock() return uint(m.recentlyInsertedTootHashCache.Len()) } func (m *Manager) TootExists(t *toot.Toot) bool { var try StoredToot // check cache m.cachelock.Lock() if _, ok := m.recentlyInsertedTootHashCache.Get(t.GetHash()); ok { m.cachelock.Unlock() return true } m.cachelock.Unlock() if m.db.Where("Hash = ?", t.GetHash()).First(&try).RecordNotFound() { return false } else { return true } } func (m *Manager) StoreToot(t *toot.Toot) error { nt := new(StoredToot) nt.UUID = uuid.New() nt.ServerCreated = t.Parsed.CreatedAt nt.Original = t.Original // FIXME add better validation to the parsed stuff here nt.Acct = fmt.Sprintf("%s@%s", t.Parsed.Account.Acct, strings.ToLower(t.FromHost)) nt.URL = t.Parsed.URL nt.Content = []byte(t.Parsed.Content) // FIXME replace tags with spaces, don't just strip them, otherwise text // gets messed up. nt.TextContent = []byte(html.UnescapeString(hstg.StripTags(t.Parsed.Content))) nt.Hostname = strings.ToLower(t.FromHost) nt.Hash = t.GetHash() //TODO: detect hashtags and insert hashtag records //TODO: detect URLs and insert URL records r := m.db.Create(&nt) // put it in the cache to avoid relying on db for dedupe: m.cachelock.Lock() m.recentlyInsertedTootHashCache.Add(nt.Hash, true) m.cachelock.Unlock() //panic(fmt.Sprintf("%+v", t)) return r.Error } func (m *Manager) StoreToots(tc []*toot.Toot) error { for _, item := range tc { err := m.StoreToot(item) if err != nil { return err } } return nil }