database.go (view raw)
1//
2// Copyright (c) 2019 Ted Unangst <tedu@tedunangst.com>
3//
4// Permission to use, copy, modify, and distribute this software for any
5// purpose with or without fee is hereby granted, provided that the above
6// copyright notice and this permission notice appear in all copies.
7//
8// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15
16package main
17
18import (
19 "bytes"
20 "crypto/sha512"
21 "database/sql"
22 _ "embed"
23 "encoding/json"
24 "fmt"
25 "html/template"
26 "sort"
27 "strconv"
28 "strings"
29 "sync"
30 "time"
31
32 "humungus.tedunangst.com/r/webs/cache"
33 "humungus.tedunangst.com/r/webs/httpsig"
34 "humungus.tedunangst.com/r/webs/login"
35 "humungus.tedunangst.com/r/webs/mz"
36)
37
38//go:embed schema.sql
39var sqlSchema string
40
41func userfromrow(row *sql.Row) (*WhatAbout, error) {
42 user := new(WhatAbout)
43 var seckey, options string
44 err := row.Scan(&user.ID, &user.Name, &user.Display, &user.About, &user.Key, &seckey, &options)
45 if err == nil {
46 user.SecKey, _, err = httpsig.DecodeKey(seckey)
47 }
48 if err != nil {
49 return nil, err
50 }
51 if user.ID > 0 {
52 user.URL = fmt.Sprintf("https://%s/%s/%s", serverName, userSep, user.Name)
53 err = unjsonify(options, &user.Options)
54 if err != nil {
55 elog.Printf("error processing user options: %s", err)
56 }
57 } else {
58 user.URL = fmt.Sprintf("https://%s/%s", serverName, user.Name)
59 }
60 if user.Options.Reaction == "" {
61 user.Options.Reaction = "none"
62 }
63
64 return user, nil
65}
66
67var somenamedusers = cache.New(cache.Options{Filler: func(name string) (*WhatAbout, bool) {
68 row := stmtUserByName.QueryRow(name)
69 user, err := userfromrow(row)
70 if err != nil {
71 return nil, false
72 }
73 var marker mz.Marker
74 marker.HashLinker = ontoreplacer
75 marker.AtLinker = attoreplacer
76 user.HTAbout = template.HTML(marker.Mark(user.About))
77 user.Onts = marker.HashTags
78 return user, true
79}})
80
81var somenumberedusers = cache.New(cache.Options{Filler: func(userid int64) (*WhatAbout, bool) {
82 row := stmtUserByNumber.QueryRow(userid)
83 user, err := userfromrow(row)
84 if err != nil {
85 return nil, false
86 }
87 // don't touch attoreplacer, which introduces a loop
88 // finger -> getjunk -> keys -> users
89 return user, true
90}})
91
92func getserveruser() *WhatAbout {
93 var user *WhatAbout
94 ok := somenumberedusers.Get(serverUID, &user)
95 if !ok {
96 elog.Panicf("lost server user")
97 }
98 return user
99}
100
101func butwhatabout(name string) (*WhatAbout, error) {
102 var user *WhatAbout
103 ok := somenamedusers.Get(name, &user)
104 if !ok {
105 return nil, fmt.Errorf("no user: %s", name)
106 }
107 return user, nil
108}
109
110var honkerinvalidator cache.Invalidator
111
112func gethonkers(userid int64) []*Honker {
113 rows, err := stmtHonkers.Query(userid)
114 if err != nil {
115 elog.Printf("error querying honkers: %s", err)
116 return nil
117 }
118 defer rows.Close()
119 var honkers []*Honker
120 for rows.Next() {
121 h := new(Honker)
122 var combos, meta string
123 err = rows.Scan(&h.ID, &h.UserID, &h.Name, &h.XID, &h.Flavor, &combos, &meta)
124 if err == nil {
125 err = unjsonify(meta, &h.Meta)
126 }
127 if err != nil {
128 elog.Printf("error scanning honker: %s", err)
129 continue
130 }
131 h.Combos = strings.Split(strings.TrimSpace(combos), " ")
132 honkers = append(honkers, h)
133 }
134 return honkers
135}
136
137func getdubs(userid int64) []*Honker {
138 rows, err := stmtDubbers.Query(userid)
139 return dubsfromrows(rows, err)
140}
141
142func getnameddubs(userid int64, name string) []*Honker {
143 rows, err := stmtNamedDubbers.Query(userid, name)
144 return dubsfromrows(rows, err)
145}
146
147func dubsfromrows(rows *sql.Rows, err error) []*Honker {
148 if err != nil {
149 elog.Printf("error querying dubs: %s", err)
150 return nil
151 }
152 defer rows.Close()
153 var honkers []*Honker
154 for rows.Next() {
155 h := new(Honker)
156 err = rows.Scan(&h.ID, &h.UserID, &h.Name, &h.XID, &h.Flavor)
157 if err != nil {
158 elog.Printf("error scanning honker: %s", err)
159 return nil
160 }
161 honkers = append(honkers, h)
162 }
163 return honkers
164}
165
166func allusers() []login.UserInfo {
167 var users []login.UserInfo
168 rows, _ := opendatabase().Query("select userid, username from users where userid > 0")
169 defer rows.Close()
170 for rows.Next() {
171 var u login.UserInfo
172 rows.Scan(&u.UserID, &u.Username)
173 users = append(users, u)
174 }
175 return users
176}
177
178func getxonk(userid int64, xid string) *Honk {
179 row := stmtOneXonk.QueryRow(userid, xid)
180 return scanhonk(row)
181}
182
183func getbonk(userid int64, xid string) *Honk {
184 row := stmtOneBonk.QueryRow(userid, xid)
185 return scanhonk(row)
186}
187
188func getpublichonks() []*Honk {
189 dt := time.Now().Add(-7 * 24 * time.Hour).UTC().Format(dbtimeformat)
190 rows, err := stmtPublicHonks.Query(dt, 100)
191 return getsomehonks(rows, err)
192}
193func geteventhonks(userid int64) []*Honk {
194 rows, err := stmtEventHonks.Query(userid, 25)
195 honks := getsomehonks(rows, err)
196 sort.Slice(honks, func(i, j int) bool {
197 var t1, t2 time.Time
198 if honks[i].Time == nil {
199 t1 = honks[i].Date
200 } else {
201 t1 = honks[i].Time.StartTime
202 }
203 if honks[j].Time == nil {
204 t2 = honks[j].Date
205 } else {
206 t2 = honks[j].Time.StartTime
207 }
208 return t1.After(t2)
209 })
210 now := time.Now().Add(-24 * time.Hour)
211 for i, h := range honks {
212 t := h.Date
213 if tm := h.Time; tm != nil {
214 t = tm.StartTime
215 }
216 if t.Before(now) {
217 honks = honks[:i]
218 break
219 }
220 }
221 reversehonks(honks)
222 return honks
223}
224func gethonksbyuser(name string, includeprivate bool, wanted int64) []*Honk {
225 dt := time.Now().Add(-7 * 24 * time.Hour).UTC().Format(dbtimeformat)
226 limit := 50
227 whofore := 2
228 if includeprivate {
229 whofore = 3
230 }
231 rows, err := stmtUserHonks.Query(wanted, whofore, name, dt, limit)
232 return getsomehonks(rows, err)
233}
234func gethonksforuser(userid int64, wanted int64) []*Honk {
235 dt := time.Now().Add(-7 * 24 * time.Hour).UTC().Format(dbtimeformat)
236 rows, err := stmtHonksForUser.Query(wanted, userid, dt, userid, userid)
237 return getsomehonks(rows, err)
238}
239func gethonksforuserfirstclass(userid int64, wanted int64) []*Honk {
240 dt := time.Now().Add(-7 * 24 * time.Hour).UTC().Format(dbtimeformat)
241 rows, err := stmtHonksForUserFirstClass.Query(wanted, userid, dt, userid, userid)
242 return getsomehonks(rows, err)
243}
244
245func gethonksforme(userid int64, wanted int64) []*Honk {
246 dt := time.Now().Add(-7 * 24 * time.Hour).UTC().Format(dbtimeformat)
247 rows, err := stmtHonksForMe.Query(wanted, userid, dt, userid)
248 return getsomehonks(rows, err)
249}
250func gethonksfromlongago(userid int64, wanted int64) []*Honk {
251 now := time.Now()
252 var honks []*Honk
253 for i := 1; i <= 3; i++ {
254 dt := time.Date(now.Year()-i, now.Month(), now.Day(), now.Hour(), now.Minute(),
255 now.Second(), 0, now.Location())
256 dt1 := dt.Add(-36 * time.Hour).UTC().Format(dbtimeformat)
257 dt2 := dt.Add(12 * time.Hour).UTC().Format(dbtimeformat)
258 rows, err := stmtHonksFromLongAgo.Query(wanted, userid, dt1, dt2, userid)
259 honks = append(honks, getsomehonks(rows, err)...)
260 }
261 return honks
262}
263func getsavedhonks(userid int64, wanted int64) []*Honk {
264 rows, err := stmtHonksISaved.Query(wanted, userid)
265 return getsomehonks(rows, err)
266}
267func gethonksbyhonker(userid int64, honker string, wanted int64) []*Honk {
268 rows, err := stmtHonksByHonker.Query(wanted, userid, honker, userid)
269 return getsomehonks(rows, err)
270}
271func gethonksbyxonker(userid int64, xonker string, wanted int64) []*Honk {
272 rows, err := stmtHonksByXonker.Query(wanted, userid, xonker, xonker, userid)
273 return getsomehonks(rows, err)
274}
275func gethonksbycombo(userid int64, combo string, wanted int64) []*Honk {
276 combo = "% " + combo + " %"
277 rows, err := stmtHonksByCombo.Query(wanted, userid, userid, combo, userid, wanted, userid, combo, userid)
278 return getsomehonks(rows, err)
279}
280func gethonksbyconvoy(userid int64, convoy string, wanted int64) []*Honk {
281 rows, err := stmtHonksByConvoy.Query(wanted, userid, userid, convoy)
282 honks := getsomehonks(rows, err)
283 return honks
284}
285func gethonksbysearch(userid int64, q string, wanted int64) []*Honk {
286 var queries []string
287 var params []interface{}
288 queries = append(queries, "honks.honkid > ?")
289 params = append(params, wanted)
290 queries = append(queries, "honks.userid = ?")
291 params = append(params, userid)
292
293 terms := strings.Split(q, " ")
294 for _, t := range terms {
295 if t == "" {
296 continue
297 }
298 negate := " "
299 if t[0] == '-' {
300 t = t[1:]
301 negate = " not "
302 }
303 if t == "" {
304 continue
305 }
306 if strings.HasPrefix(t, "site:") {
307 site := t[5:]
308 site = "%" + site + "%"
309 queries = append(queries, "xid"+negate+"like ?")
310 params = append(params, site)
311 continue
312 }
313 if strings.HasPrefix(t, "honker:") {
314 honker := t[7:]
315 xid := fullname(honker, userid)
316 if xid != "" {
317 honker = xid
318 }
319 queries = append(queries, negate+"(honks.honker = ? or honks.oonker = ?)")
320 params = append(params, honker)
321 params = append(params, honker)
322 continue
323 }
324 t = "%" + t + "%"
325 queries = append(queries, "noise"+negate+"like ?")
326 params = append(params, t)
327 }
328
329 selecthonks := "select honks.honkid, honks.userid, username, what, honker, oonker, honks.xid, rid, dt, url, audience, noise, precis, format, convoy, whofore, flags from honks join users on honks.userid = users.userid "
330 where := "where " + strings.Join(queries, " and ")
331 butnotthose := " and convoy not in (select name from zonkers where userid = ? and wherefore = 'zonvoy' order by zonkerid desc limit 100)"
332 limit := " order by honks.honkid desc limit 250"
333 params = append(params, userid)
334 rows, err := opendatabase().Query(selecthonks+where+butnotthose+limit, params...)
335 honks := getsomehonks(rows, err)
336 return honks
337}
338func gethonksbyontology(userid int64, name string, wanted int64) []*Honk {
339 rows, err := stmtHonksByOntology.Query(wanted, name, userid, userid)
340 honks := getsomehonks(rows, err)
341 return honks
342}
343
344func reversehonks(honks []*Honk) {
345 for i, j := 0, len(honks)-1; i < j; i, j = i+1, j-1 {
346 honks[i], honks[j] = honks[j], honks[i]
347 }
348}
349
350func getsomehonks(rows *sql.Rows, err error) []*Honk {
351 if err != nil {
352 elog.Printf("error querying honks: %s", err)
353 return nil
354 }
355 defer rows.Close()
356 var honks []*Honk
357 for rows.Next() {
358 h := scanhonk(rows)
359 if h != nil {
360 honks = append(honks, h)
361 }
362 }
363 rows.Close()
364 donksforhonks(honks)
365 return honks
366}
367
368type RowLike interface {
369 Scan(dest ...interface{}) error
370}
371
372func scanhonk(row RowLike) *Honk {
373 h := new(Honk)
374 var dt, aud string
375 err := row.Scan(&h.ID, &h.UserID, &h.Username, &h.What, &h.Honker, &h.Oonker, &h.XID, &h.RID,
376 &dt, &h.URL, &aud, &h.Noise, &h.Precis, &h.Format, &h.Convoy, &h.Whofore, &h.Flags)
377 if err != nil {
378 if err != sql.ErrNoRows {
379 elog.Printf("error scanning honk: %s", err)
380 }
381 return nil
382 }
383 h.Date, _ = time.Parse(dbtimeformat, dt)
384 h.Audience = strings.Split(aud, " ")
385 h.Public = loudandproud(h.Audience)
386 return h
387}
388
389func donksforhonks(honks []*Honk) {
390 db := opendatabase()
391 var ids []string
392 hmap := make(map[int64]*Honk)
393 for _, h := range honks {
394 ids = append(ids, fmt.Sprintf("%d", h.ID))
395 hmap[h.ID] = h
396 }
397 idset := strings.Join(ids, ",")
398 // grab donks
399 q := fmt.Sprintf("select honkid, donks.fileid, xid, name, description, url, media, local from donks join filemeta on donks.fileid = filemeta.fileid where honkid in (%s)", idset)
400 rows, err := db.Query(q)
401 if err != nil {
402 elog.Printf("error querying donks: %s", err)
403 return
404 }
405 defer rows.Close()
406 for rows.Next() {
407 var hid int64
408 d := new(Donk)
409 err = rows.Scan(&hid, &d.FileID, &d.XID, &d.Name, &d.Desc, &d.URL, &d.Media, &d.Local)
410 if err != nil {
411 elog.Printf("error scanning donk: %s", err)
412 continue
413 }
414 d.External = !strings.HasPrefix(d.URL, serverPrefix)
415 h := hmap[hid]
416 h.Donks = append(h.Donks, d)
417 }
418 rows.Close()
419
420 // grab onts
421 q = fmt.Sprintf("select honkid, ontology from onts where honkid in (%s)", idset)
422 rows, err = db.Query(q)
423 if err != nil {
424 elog.Printf("error querying onts: %s", err)
425 return
426 }
427 defer rows.Close()
428 for rows.Next() {
429 var hid int64
430 var o string
431 err = rows.Scan(&hid, &o)
432 if err != nil {
433 elog.Printf("error scanning donk: %s", err)
434 continue
435 }
436 h := hmap[hid]
437 h.Onts = append(h.Onts, o)
438 }
439 rows.Close()
440
441 // grab meta
442 q = fmt.Sprintf("select honkid, genus, json from honkmeta where honkid in (%s)", idset)
443 rows, err = db.Query(q)
444 if err != nil {
445 elog.Printf("error querying honkmeta: %s", err)
446 return
447 }
448 defer rows.Close()
449 for rows.Next() {
450 var hid int64
451 var genus, j string
452 err = rows.Scan(&hid, &genus, &j)
453 if err != nil {
454 elog.Printf("error scanning honkmeta: %s", err)
455 continue
456 }
457 h := hmap[hid]
458 switch genus {
459 case "place":
460 p := new(Place)
461 err = unjsonify(j, p)
462 if err != nil {
463 elog.Printf("error parsing place: %s", err)
464 continue
465 }
466 h.Place = p
467 case "time":
468 t := new(Time)
469 err = unjsonify(j, t)
470 if err != nil {
471 elog.Printf("error parsing time: %s", err)
472 continue
473 }
474 h.Time = t
475 case "mentions":
476 err = unjsonify(j, &h.Mentions)
477 if err != nil {
478 elog.Printf("error parsing mentions: %s", err)
479 continue
480 }
481 case "badonks":
482 err = unjsonify(j, &h.Badonks)
483 if err != nil {
484 elog.Printf("error parsing badonks: %s", err)
485 continue
486 }
487 case "wonkles":
488 case "guesses":
489 case "oldrev":
490 default:
491 elog.Printf("unknown meta genus: %s", genus)
492 }
493 }
494 rows.Close()
495}
496
497func donksforchonks(chonks []*Chonk) {
498 db := opendatabase()
499 var ids []string
500 chmap := make(map[int64]*Chonk)
501 for _, ch := range chonks {
502 ids = append(ids, fmt.Sprintf("%d", ch.ID))
503 chmap[ch.ID] = ch
504 }
505 idset := strings.Join(ids, ",")
506 // grab donks
507 q := fmt.Sprintf("select chonkid, donks.fileid, xid, name, description, url, media, local from donks join filemeta on donks.fileid = filemeta.fileid where chonkid in (%s)", idset)
508 rows, err := db.Query(q)
509 if err != nil {
510 elog.Printf("error querying donks: %s", err)
511 return
512 }
513 defer rows.Close()
514 for rows.Next() {
515 var chid int64
516 d := new(Donk)
517 err = rows.Scan(&chid, &d.FileID, &d.XID, &d.Name, &d.Desc, &d.URL, &d.Media, &d.Local)
518 if err != nil {
519 elog.Printf("error scanning donk: %s", err)
520 continue
521 }
522 ch := chmap[chid]
523 ch.Donks = append(ch.Donks, d)
524 }
525}
526
527func savefile(name string, desc string, url string, media string, local bool, data []byte) (int64, error) {
528 fileid, _, err := savefileandxid(name, desc, url, media, local, data)
529 return fileid, err
530}
531
532func hashfiledata(data []byte) string {
533 h := sha512.New512_256()
534 h.Write(data)
535 return fmt.Sprintf("%x", h.Sum(nil))
536}
537
538func savefileandxid(name string, desc string, url string, media string, local bool, data []byte) (int64, string, error) {
539 var xid string
540 if local {
541 hash := hashfiledata(data)
542 row := stmtCheckFileData.QueryRow(hash)
543 err := row.Scan(&xid)
544 if err == sql.ErrNoRows {
545 xid = xfiltrate()
546 switch media {
547 case "image/png":
548 xid += ".png"
549 case "image/jpeg":
550 xid += ".jpg"
551 case "application/pdf":
552 xid += ".pdf"
553 case "text/plain":
554 xid += ".txt"
555 }
556 _, err = stmtSaveFileData.Exec(xid, media, hash, data)
557 if err != nil {
558 return 0, "", err
559 }
560 } else if err != nil {
561 elog.Printf("error checking file hash: %s", err)
562 return 0, "", err
563 }
564 if url == "" {
565 url = fmt.Sprintf("https://%s/d/%s", serverName, xid)
566 }
567 }
568
569 res, err := stmtSaveFile.Exec(xid, name, desc, url, media, local)
570 if err != nil {
571 return 0, "", err
572 }
573 fileid, _ := res.LastInsertId()
574 return fileid, xid, nil
575}
576
577func finddonk(url string) *Donk {
578 donk := new(Donk)
579 row := stmtFindFile.QueryRow(url)
580 err := row.Scan(&donk.FileID, &donk.XID)
581 if err == nil {
582 return donk
583 }
584 if err != sql.ErrNoRows {
585 elog.Printf("error finding file: %s", err)
586 }
587 return nil
588}
589
590func savechonk(ch *Chonk) error {
591 dt := ch.Date.UTC().Format(dbtimeformat)
592 db := opendatabase()
593 tx, err := db.Begin()
594 if err != nil {
595 elog.Printf("can't begin tx: %s", err)
596 return err
597 }
598
599 res, err := tx.Stmt(stmtSaveChonk).Exec(ch.UserID, ch.XID, ch.Who, ch.Target, dt, ch.Noise, ch.Format)
600 if err == nil {
601 ch.ID, _ = res.LastInsertId()
602 for _, d := range ch.Donks {
603 _, err := tx.Stmt(stmtSaveDonk).Exec(-1, ch.ID, d.FileID)
604 if err != nil {
605 elog.Printf("error saving donk: %s", err)
606 break
607 }
608 }
609 chatplusone(tx, ch.UserID)
610 err = tx.Commit()
611 } else {
612 tx.Rollback()
613 }
614 return err
615}
616
617func chatplusone(tx *sql.Tx, userid int64) {
618 var user *WhatAbout
619 ok := somenumberedusers.Get(userid, &user)
620 if !ok {
621 return
622 }
623 options := user.Options
624 options.ChatCount += 1
625 j, err := jsonify(options)
626 if err == nil {
627 _, err = tx.Exec("update users set options = ? where username = ?", j, user.Name)
628 }
629 if err != nil {
630 elog.Printf("error plussing chat: %s", err)
631 }
632 somenamedusers.Clear(user.Name)
633 somenumberedusers.Clear(user.ID)
634}
635
636func chatnewnone(userid int64) {
637 var user *WhatAbout
638 ok := somenumberedusers.Get(userid, &user)
639 if !ok || user.Options.ChatCount == 0 {
640 return
641 }
642 options := user.Options
643 options.ChatCount = 0
644 j, err := jsonify(options)
645 if err == nil {
646 db := opendatabase()
647 _, err = db.Exec("update users set options = ? where username = ?", j, user.Name)
648 }
649 if err != nil {
650 elog.Printf("error noneing chat: %s", err)
651 }
652 somenamedusers.Clear(user.Name)
653 somenumberedusers.Clear(user.ID)
654}
655
656func meplusone(tx *sql.Tx, userid int64) {
657 var user *WhatAbout
658 ok := somenumberedusers.Get(userid, &user)
659 if !ok {
660 return
661 }
662 options := user.Options
663 options.MeCount += 1
664 j, err := jsonify(options)
665 if err == nil {
666 _, err = tx.Exec("update users set options = ? where username = ?", j, user.Name)
667 }
668 if err != nil {
669 elog.Printf("error plussing me: %s", err)
670 }
671 somenamedusers.Clear(user.Name)
672 somenumberedusers.Clear(user.ID)
673}
674
675func menewnone(userid int64) {
676 var user *WhatAbout
677 ok := somenumberedusers.Get(userid, &user)
678 if !ok || user.Options.MeCount == 0 {
679 return
680 }
681 options := user.Options
682 options.MeCount = 0
683 j, err := jsonify(options)
684 if err == nil {
685 db := opendatabase()
686 _, err = db.Exec("update users set options = ? where username = ?", j, user.Name)
687 }
688 if err != nil {
689 elog.Printf("error noneing me: %s", err)
690 }
691 somenamedusers.Clear(user.Name)
692 somenumberedusers.Clear(user.ID)
693}
694
695func loadchatter(userid int64) []*Chatter {
696 duedt := time.Now().Add(-3 * 24 * time.Hour).UTC().Format(dbtimeformat)
697 rows, err := stmtLoadChonks.Query(userid, duedt)
698 if err != nil {
699 elog.Printf("error loading chonks: %s", err)
700 return nil
701 }
702 defer rows.Close()
703 chonks := make(map[string][]*Chonk)
704 var allchonks []*Chonk
705 for rows.Next() {
706 ch := new(Chonk)
707 var dt string
708 err = rows.Scan(&ch.ID, &ch.UserID, &ch.XID, &ch.Who, &ch.Target, &dt, &ch.Noise, &ch.Format)
709 if err != nil {
710 elog.Printf("error scanning chonk: %s", err)
711 continue
712 }
713 ch.Date, _ = time.Parse(dbtimeformat, dt)
714 chonks[ch.Target] = append(chonks[ch.Target], ch)
715 allchonks = append(allchonks, ch)
716 }
717 donksforchonks(allchonks)
718 rows.Close()
719 rows, err = stmtGetChatters.Query(userid)
720 if err != nil {
721 elog.Printf("error getting chatters: %s", err)
722 return nil
723 }
724 for rows.Next() {
725 var target string
726 err = rows.Scan(&target)
727 if err != nil {
728 elog.Printf("error scanning chatter: %s", target)
729 continue
730 }
731 if _, ok := chonks[target]; !ok {
732 chonks[target] = []*Chonk{}
733
734 }
735 }
736 var chatter []*Chatter
737 for target, chonks := range chonks {
738 chatter = append(chatter, &Chatter{
739 Target: target,
740 Chonks: chonks,
741 })
742 }
743 sort.Slice(chatter, func(i, j int) bool {
744 a, b := chatter[i], chatter[j]
745 if len(a.Chonks) == 0 || len(b.Chonks) == 0 {
746 if len(a.Chonks) == len(b.Chonks) {
747 return a.Target < b.Target
748 }
749 return len(a.Chonks) > len(b.Chonks)
750 }
751 return a.Chonks[len(a.Chonks)-1].Date.After(b.Chonks[len(b.Chonks)-1].Date)
752 })
753
754 return chatter
755}
756
757func savehonk(h *Honk) error {
758 dt := h.Date.UTC().Format(dbtimeformat)
759 aud := strings.Join(h.Audience, " ")
760
761 db := opendatabase()
762 tx, err := db.Begin()
763 if err != nil {
764 elog.Printf("can't begin tx: %s", err)
765 return err
766 }
767
768 res, err := tx.Stmt(stmtSaveHonk).Exec(h.UserID, h.What, h.Honker, h.XID, h.RID, dt, h.URL,
769 aud, h.Noise, h.Convoy, h.Whofore, h.Format, h.Precis,
770 h.Oonker, h.Flags)
771 if err == nil {
772 h.ID, _ = res.LastInsertId()
773 err = saveextras(tx, h)
774 }
775 if err == nil {
776 if h.Whofore == 1 {
777 meplusone(tx, h.UserID)
778 }
779 err = tx.Commit()
780 } else {
781 tx.Rollback()
782 }
783 if err != nil {
784 elog.Printf("error saving honk: %s", err)
785 }
786 honkhonkline()
787 return err
788}
789
790func updatehonk(h *Honk) error {
791 old := getxonk(h.UserID, h.XID)
792 oldrev := OldRevision{Precis: old.Precis, Noise: old.Noise}
793 dt := h.Date.UTC().Format(dbtimeformat)
794
795 db := opendatabase()
796 tx, err := db.Begin()
797 if err != nil {
798 elog.Printf("can't begin tx: %s", err)
799 return err
800 }
801
802 err = deleteextras(tx, h.ID, false)
803 if err == nil {
804 _, err = tx.Stmt(stmtUpdateHonk).Exec(h.Precis, h.Noise, h.Format, h.Whofore, dt, h.ID)
805 }
806 if err == nil {
807 err = saveextras(tx, h)
808 }
809 if err == nil {
810 var j string
811 j, err = jsonify(&oldrev)
812 if err == nil {
813 _, err = tx.Stmt(stmtSaveMeta).Exec(old.ID, "oldrev", j)
814 }
815 if err != nil {
816 elog.Printf("error saving oldrev: %s", err)
817 }
818 }
819 if err == nil {
820 err = tx.Commit()
821 } else {
822 tx.Rollback()
823 }
824 if err != nil {
825 elog.Printf("error updating honk %d: %s", h.ID, err)
826 }
827 return err
828}
829
830func deletehonk(honkid int64) error {
831 db := opendatabase()
832 tx, err := db.Begin()
833 if err != nil {
834 elog.Printf("can't begin tx: %s", err)
835 return err
836 }
837
838 err = deleteextras(tx, honkid, true)
839 if err == nil {
840 _, err = tx.Stmt(stmtDeleteHonk).Exec(honkid)
841 }
842 if err == nil {
843 err = tx.Commit()
844 } else {
845 tx.Rollback()
846 }
847 if err != nil {
848 elog.Printf("error deleting honk %d: %s", honkid, err)
849 }
850 return err
851}
852
853func saveextras(tx *sql.Tx, h *Honk) error {
854 for _, d := range h.Donks {
855 _, err := tx.Stmt(stmtSaveDonk).Exec(h.ID, -1, d.FileID)
856 if err != nil {
857 elog.Printf("error saving donk: %s", err)
858 return err
859 }
860 }
861 for _, o := range h.Onts {
862 _, err := tx.Stmt(stmtSaveOnt).Exec(strings.ToLower(o), h.ID)
863 if err != nil {
864 elog.Printf("error saving ont: %s", err)
865 return err
866 }
867 }
868 if p := h.Place; p != nil {
869 j, err := jsonify(p)
870 if err == nil {
871 _, err = tx.Stmt(stmtSaveMeta).Exec(h.ID, "place", j)
872 }
873 if err != nil {
874 elog.Printf("error saving place: %s", err)
875 return err
876 }
877 }
878 if t := h.Time; t != nil {
879 j, err := jsonify(t)
880 if err == nil {
881 _, err = tx.Stmt(stmtSaveMeta).Exec(h.ID, "time", j)
882 }
883 if err != nil {
884 elog.Printf("error saving time: %s", err)
885 return err
886 }
887 }
888 if m := h.Mentions; len(m) > 0 {
889 j, err := jsonify(m)
890 if err == nil {
891 _, err = tx.Stmt(stmtSaveMeta).Exec(h.ID, "mentions", j)
892 }
893 if err != nil {
894 elog.Printf("error saving mentions: %s", err)
895 return err
896 }
897 }
898 return nil
899}
900
901var baxonker sync.Mutex
902
903func addreaction(user *WhatAbout, xid string, who, react string) {
904 baxonker.Lock()
905 defer baxonker.Unlock()
906 h := getxonk(user.ID, xid)
907 if h == nil {
908 return
909 }
910 h.Badonks = append(h.Badonks, Badonk{Who: who, What: react})
911 j, _ := jsonify(h.Badonks)
912 db := opendatabase()
913 tx, _ := db.Begin()
914 _, _ = tx.Stmt(stmtDeleteOneMeta).Exec(h.ID, "badonks")
915 _, _ = tx.Stmt(stmtSaveMeta).Exec(h.ID, "badonks", j)
916 tx.Commit()
917}
918
919func deleteextras(tx *sql.Tx, honkid int64, everything bool) error {
920 _, err := tx.Stmt(stmtDeleteDonks).Exec(honkid)
921 if err != nil {
922 return err
923 }
924 _, err = tx.Stmt(stmtDeleteOnts).Exec(honkid)
925 if err != nil {
926 return err
927 }
928 if everything {
929 _, err = tx.Stmt(stmtDeleteAllMeta).Exec(honkid)
930 } else {
931 _, err = tx.Stmt(stmtDeleteSomeMeta).Exec(honkid)
932 }
933 if err != nil {
934 return err
935 }
936 return nil
937}
938
939func jsonify(what interface{}) (string, error) {
940 var buf bytes.Buffer
941 e := json.NewEncoder(&buf)
942 e.SetEscapeHTML(false)
943 e.SetIndent("", "")
944 err := e.Encode(what)
945 return buf.String(), err
946}
947
948func unjsonify(s string, dest interface{}) error {
949 d := json.NewDecoder(strings.NewReader(s))
950 err := d.Decode(dest)
951 return err
952}
953
954func getxonker(what, flav string) string {
955 var res string
956 row := stmtGetXonker.QueryRow(what, flav)
957 row.Scan(&res)
958 return res
959}
960
961func savexonker(what, value, flav, when string) {
962 stmtSaveXonker.Exec(what, value, flav, when)
963}
964
965func savehonker(user *WhatAbout, url, name, flavor, combos, mj string) (int64, error) {
966 var owner string
967 if url[0] == '#' {
968 flavor = "peep"
969 if name == "" {
970 name = url[1:]
971 }
972 owner = url
973 } else {
974 info, err := investigate(url)
975 if err != nil {
976 ilog.Printf("failed to investigate honker: %s", err)
977 return 0, err
978 }
979 url = info.XID
980 if name == "" {
981 name = info.Name
982 }
983 owner = info.Owner
984 }
985
986 var x string
987 db := opendatabase()
988 row := db.QueryRow("select xid from honkers where xid = ? and userid = ? and flavor in ('sub', 'unsub', 'peep')", url, user.ID)
989 err := row.Scan(&x)
990 if err != sql.ErrNoRows {
991 if err != nil {
992 elog.Printf("honker scan err: %s", err)
993 } else {
994 err = fmt.Errorf("it seems you are already subscribed to them")
995 }
996 return 0, err
997 }
998
999 res, err := stmtSaveHonker.Exec(user.ID, name, url, flavor, combos, owner, mj)
1000 if err != nil {
1001 elog.Print(err)
1002 return 0, err
1003 }
1004 honkerid, _ := res.LastInsertId()
1005 if flavor == "presub" {
1006 followyou(user, honkerid)
1007 }
1008 return honkerid, nil
1009}
1010
1011func cleanupdb(arg string) {
1012 db := opendatabase()
1013 days, err := strconv.Atoi(arg)
1014 var sqlargs []interface{}
1015 var where string
1016 if err != nil {
1017 honker := arg
1018 expdate := time.Now().Add(-3 * 24 * time.Hour).UTC().Format(dbtimeformat)
1019 where = "dt < ? and honker = ?"
1020 sqlargs = append(sqlargs, expdate)
1021 sqlargs = append(sqlargs, honker)
1022 } else {
1023 expdate := time.Now().Add(-time.Duration(days) * 24 * time.Hour).UTC().Format(dbtimeformat)
1024 where = "dt < ? and convoy not in (select convoy from honks where flags & 4 or whofore = 2 or whofore = 3)"
1025 sqlargs = append(sqlargs, expdate)
1026 }
1027 doordie(db, "delete from honks where flags & 4 = 0 and whofore = 0 and "+where, sqlargs...)
1028 doordie(db, "delete from donks where honkid > 0 and honkid not in (select honkid from honks)")
1029 doordie(db, "delete from onts where honkid not in (select honkid from honks)")
1030 doordie(db, "delete from honkmeta where honkid not in (select honkid from honks)")
1031
1032 doordie(db, "delete from filemeta where fileid not in (select fileid from donks)")
1033 for _, u := range allusers() {
1034 doordie(db, "delete from zonkers where userid = ? and wherefore = 'zonvoy' and zonkerid < (select zonkerid from zonkers where userid = ? and wherefore = 'zonvoy' order by zonkerid desc limit 1 offset 200)", u.UserID, u.UserID)
1035 }
1036
1037 filexids := make(map[string]bool)
1038 blobdb := openblobdb()
1039 rows, err := blobdb.Query("select xid from filedata")
1040 if err != nil {
1041 elog.Fatal(err)
1042 }
1043 for rows.Next() {
1044 var xid string
1045 err = rows.Scan(&xid)
1046 if err != nil {
1047 elog.Fatal(err)
1048 }
1049 filexids[xid] = true
1050 }
1051 rows.Close()
1052 rows, err = db.Query("select xid from filemeta")
1053 for rows.Next() {
1054 var xid string
1055 err = rows.Scan(&xid)
1056 if err != nil {
1057 elog.Fatal(err)
1058 }
1059 delete(filexids, xid)
1060 }
1061 rows.Close()
1062 tx, err := blobdb.Begin()
1063 if err != nil {
1064 elog.Fatal(err)
1065 }
1066 for xid, _ := range filexids {
1067 _, err = tx.Exec("delete from filedata where xid = ?", xid)
1068 if err != nil {
1069 elog.Fatal(err)
1070 }
1071 }
1072 err = tx.Commit()
1073 if err != nil {
1074 elog.Fatal(err)
1075 }
1076}
1077
1078var stmtHonkers, stmtDubbers, stmtNamedDubbers, stmtSaveHonker, stmtUpdateFlavor, stmtUpdateHonker *sql.Stmt
1079var stmtDeleteHonker *sql.Stmt
1080var stmtAnyXonk, stmtOneXonk, stmtPublicHonks, stmtUserHonks, stmtHonksByCombo, stmtHonksByConvoy *sql.Stmt
1081var stmtHonksByOntology, stmtHonksForUser, stmtHonksForMe, stmtSaveDub, stmtHonksByXonker *sql.Stmt
1082var stmtHonksFromLongAgo *sql.Stmt
1083var stmtHonksByHonker, stmtSaveHonk, stmtUserByName, stmtUserByNumber *sql.Stmt
1084var stmtEventHonks, stmtOneBonk, stmtFindZonk, stmtFindXonk, stmtSaveDonk *sql.Stmt
1085var stmtFindFile, stmtGetFileData, stmtSaveFileData, stmtSaveFile *sql.Stmt
1086var stmtCheckFileData *sql.Stmt
1087var stmtAddDoover, stmtGetDoovers, stmtLoadDoover, stmtZapDoover, stmtOneHonker *sql.Stmt
1088var stmtUntagged, stmtDeleteHonk, stmtDeleteDonks, stmtDeleteOnts, stmtSaveZonker *sql.Stmt
1089var stmtGetZonkers, stmtRecentHonkers, stmtGetXonker, stmtSaveXonker, stmtDeleteXonker, stmtDeleteOldXonkers *sql.Stmt
1090var stmtAllOnts, stmtSaveOnt, stmtUpdateFlags, stmtClearFlags *sql.Stmt
1091var stmtHonksForUserFirstClass *sql.Stmt
1092var stmtSaveMeta, stmtDeleteAllMeta, stmtDeleteOneMeta, stmtDeleteSomeMeta, stmtUpdateHonk *sql.Stmt
1093var stmtHonksISaved, stmtGetFilters, stmtSaveFilter, stmtDeleteFilter *sql.Stmt
1094var stmtGetTracks *sql.Stmt
1095var stmtSaveChonk, stmtLoadChonks, stmtGetChatters *sql.Stmt
1096
1097func preparetodie(db *sql.DB, s string) *sql.Stmt {
1098 stmt, err := db.Prepare(s)
1099 if err != nil {
1100 elog.Fatalf("error %s: %s", err, s)
1101 }
1102 return stmt
1103}
1104
1105func prepareStatements(db *sql.DB) {
1106 stmtHonkers = preparetodie(db, "select honkerid, userid, name, xid, flavor, combos, meta from honkers where userid = ? and (flavor = 'presub' or flavor = 'sub' or flavor = 'peep' or flavor = 'unsub') order by name")
1107 stmtSaveHonker = preparetodie(db, "insert into honkers (userid, name, xid, flavor, combos, owner, meta, folxid) values (?, ?, ?, ?, ?, ?, ?, '')")
1108 stmtUpdateFlavor = preparetodie(db, "update honkers set flavor = ?, folxid = ? where userid = ? and name = ? and xid = ? and flavor = ?")
1109 stmtUpdateHonker = preparetodie(db, "update honkers set name = ?, combos = ?, meta = ? where honkerid = ? and userid = ?")
1110 stmtDeleteHonker = preparetodie(db, "delete from honkers where honkerid = ?")
1111 stmtOneHonker = preparetodie(db, "select xid from honkers where name = ? and userid = ?")
1112 stmtDubbers = preparetodie(db, "select honkerid, userid, name, xid, flavor from honkers where userid = ? and flavor = 'dub'")
1113 stmtNamedDubbers = preparetodie(db, "select honkerid, userid, name, xid, flavor from honkers where userid = ? and name = ? and flavor = 'dub'")
1114
1115 selecthonks := "select honks.honkid, honks.userid, username, what, honker, oonker, honks.xid, rid, dt, url, audience, noise, precis, format, convoy, whofore, flags from honks join users on honks.userid = users.userid "
1116 limit := " order by honks.honkid desc limit 250"
1117 smalllimit := " order by honks.honkid desc limit ?"
1118 butnotthose := " and convoy not in (select name from zonkers where userid = ? and wherefore = 'zonvoy' order by zonkerid desc limit 100)"
1119 stmtOneXonk = preparetodie(db, selecthonks+"where honks.userid = ? and xid = ?")
1120 stmtAnyXonk = preparetodie(db, selecthonks+"where xid = ? order by honks.honkid asc")
1121 stmtOneBonk = preparetodie(db, selecthonks+"where honks.userid = ? and xid = ? and what = 'bonk' and whofore = 2")
1122 stmtPublicHonks = preparetodie(db, selecthonks+"where whofore = 2 and dt > ?"+smalllimit)
1123 stmtEventHonks = preparetodie(db, selecthonks+"where (whofore = 2 or honks.userid = ?) and what = 'event'"+smalllimit)
1124 stmtUserHonks = preparetodie(db, selecthonks+"where honks.honkid > ? and (whofore = 2 or whofore = ?) and username = ? and dt > ?"+smalllimit)
1125 myhonkers := " and honker in (select xid from honkers where userid = ? and (flavor = 'sub' or flavor = 'peep' or flavor = 'presub') and combos not like '% - %')"
1126 stmtHonksForUser = preparetodie(db, selecthonks+"where honks.honkid > ? and honks.userid = ? and dt > ?"+myhonkers+butnotthose+limit)
1127 stmtHonksForUserFirstClass = preparetodie(db, selecthonks+"where honks.honkid > ? and honks.userid = ? and dt > ? and (what <> 'tonk')"+myhonkers+butnotthose+limit)
1128 stmtHonksForMe = preparetodie(db, selecthonks+"where honks.honkid > ? and honks.userid = ? and dt > ? and whofore = 1"+butnotthose+limit)
1129 stmtHonksFromLongAgo = preparetodie(db, selecthonks+"where honks.honkid > ? and honks.userid = ? and dt > ? and dt < ? and whofore = 2"+butnotthose+limit)
1130 stmtHonksISaved = preparetodie(db, selecthonks+"where honks.honkid > ? and honks.userid = ? and flags & 4 order by honks.honkid desc")
1131 stmtHonksByHonker = preparetodie(db, selecthonks+"join honkers on (honkers.xid = honks.honker or honkers.xid = honks.oonker) where honks.honkid > ? and honks.userid = ? and honkers.name = ?"+butnotthose+limit)
1132 stmtHonksByXonker = preparetodie(db, selecthonks+" where honks.honkid > ? and honks.userid = ? and (honker = ? or oonker = ?)"+butnotthose+limit)
1133 stmtHonksByCombo = preparetodie(db, selecthonks+" where honks.honkid > ? and honks.userid = ? and honks.honker in (select xid from honkers where honkers.userid = ? and honkers.combos like ?) "+butnotthose+" union "+selecthonks+"join onts on honks.honkid = onts.honkid where honks.honkid > ? and honks.userid = ? and onts.ontology in (select xid from honkers where combos like ?)"+butnotthose+limit)
1134 stmtHonksByConvoy = preparetodie(db, selecthonks+"where honks.honkid > ? and (honks.userid = ? or (? = -1 and whofore = 2)) and convoy = ?"+limit)
1135 stmtHonksByOntology = preparetodie(db, selecthonks+"join onts on honks.honkid = onts.honkid where honks.honkid > ? and onts.ontology = ? and (honks.userid = ? or (? = -1 and honks.whofore = 2))"+limit)
1136
1137 stmtSaveMeta = preparetodie(db, "insert into honkmeta (honkid, genus, json) values (?, ?, ?)")
1138 stmtDeleteAllMeta = preparetodie(db, "delete from honkmeta where honkid = ?")
1139 stmtDeleteSomeMeta = preparetodie(db, "delete from honkmeta where honkid = ? and genus not in ('oldrev')")
1140 stmtDeleteOneMeta = preparetodie(db, "delete from honkmeta where honkid = ? and genus = ?")
1141 stmtSaveHonk = preparetodie(db, "insert into honks (userid, what, honker, xid, rid, dt, url, audience, noise, convoy, whofore, format, precis, oonker, flags) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)")
1142 stmtDeleteHonk = preparetodie(db, "delete from honks where honkid = ?")
1143 stmtUpdateHonk = preparetodie(db, "update honks set precis = ?, noise = ?, format = ?, whofore = ?, dt = ? where honkid = ?")
1144 stmtSaveOnt = preparetodie(db, "insert into onts (ontology, honkid) values (?, ?)")
1145 stmtDeleteOnts = preparetodie(db, "delete from onts where honkid = ?")
1146 stmtSaveDonk = preparetodie(db, "insert into donks (honkid, chonkid, fileid) values (?, ?, ?)")
1147 stmtDeleteDonks = preparetodie(db, "delete from donks where honkid = ?")
1148 stmtSaveFile = preparetodie(db, "insert into filemeta (xid, name, description, url, media, local) values (?, ?, ?, ?, ?, ?)")
1149 blobdb := openblobdb()
1150 stmtSaveFileData = preparetodie(blobdb, "insert into filedata (xid, media, hash, content) values (?, ?, ?, ?)")
1151 stmtCheckFileData = preparetodie(blobdb, "select xid from filedata where hash = ?")
1152 stmtGetFileData = preparetodie(blobdb, "select media, content from filedata where xid = ?")
1153 stmtFindXonk = preparetodie(db, "select honkid from honks where userid = ? and xid = ?")
1154 stmtFindFile = preparetodie(db, "select fileid, xid from filemeta where url = ? and local = 1")
1155 stmtUserByName = preparetodie(db, "select userid, username, displayname, about, pubkey, seckey, options from users where username = ? and userid > 0")
1156 stmtUserByNumber = preparetodie(db, "select userid, username, displayname, about, pubkey, seckey, options from users where userid = ?")
1157 stmtSaveDub = preparetodie(db, "insert into honkers (userid, name, xid, flavor, combos, owner, meta, folxid) values (?, ?, ?, ?, '', '', '', ?)")
1158 stmtAddDoover = preparetodie(db, "insert into doovers (dt, tries, userid, rcpt, msg) values (?, ?, ?, ?, ?)")
1159 stmtGetDoovers = preparetodie(db, "select dooverid, dt from doovers")
1160 stmtLoadDoover = preparetodie(db, "select tries, userid, rcpt, msg from doovers where dooverid = ?")
1161 stmtZapDoover = preparetodie(db, "delete from doovers where dooverid = ?")
1162 stmtUntagged = preparetodie(db, "select xid, rid, flags from (select honkid, xid, rid, flags from honks where userid = ? order by honkid desc limit 10000) order by honkid asc")
1163 stmtFindZonk = preparetodie(db, "select zonkerid from zonkers where userid = ? and name = ? and wherefore = 'zonk'")
1164 stmtGetZonkers = preparetodie(db, "select zonkerid, name, wherefore from zonkers where userid = ? and wherefore <> 'zonk'")
1165 stmtSaveZonker = preparetodie(db, "insert into zonkers (userid, name, wherefore) values (?, ?, ?)")
1166 stmtGetXonker = preparetodie(db, "select info from xonkers where name = ? and flavor = ?")
1167 stmtSaveXonker = preparetodie(db, "insert into xonkers (name, info, flavor, dt) values (?, ?, ?, ?)")
1168 stmtDeleteXonker = preparetodie(db, "delete from xonkers where name = ? and flavor = ? and dt < ?")
1169 stmtDeleteOldXonkers = preparetodie(db, "delete from xonkers where flavor = ? and dt < ?")
1170 stmtRecentHonkers = preparetodie(db, "select distinct(honker) from honks where userid = ? and honker not in (select xid from honkers where userid = ? and flavor = 'sub') order by honkid desc limit 100")
1171 stmtUpdateFlags = preparetodie(db, "update honks set flags = flags | ? where honkid = ?")
1172 stmtClearFlags = preparetodie(db, "update honks set flags = flags & ~ ? where honkid = ?")
1173 stmtAllOnts = preparetodie(db, "select ontology, count(ontology) from onts join honks on onts.honkid = honks.honkid where (honks.userid = ? or honks.whofore = 2) group by ontology")
1174 stmtGetFilters = preparetodie(db, "select hfcsid, json from hfcs where userid = ?")
1175 stmtSaveFilter = preparetodie(db, "insert into hfcs (userid, json) values (?, ?)")
1176 stmtDeleteFilter = preparetodie(db, "delete from hfcs where userid = ? and hfcsid = ?")
1177 stmtGetTracks = preparetodie(db, "select fetches from tracks where xid = ?")
1178 stmtSaveChonk = preparetodie(db, "insert into chonks (userid, xid, who, target, dt, noise, format) values (?, ?, ?, ?, ?, ?, ?)")
1179 stmtLoadChonks = preparetodie(db, "select chonkid, userid, xid, who, target, dt, noise, format from chonks where userid = ? and dt > ? order by chonkid asc")
1180 stmtGetChatters = preparetodie(db, "select distinct(target) from chonks where userid = ?")
1181}