puppet.go 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  1. // mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
  2. // Copyright (C) 2018 Tulir Asokan
  3. //
  4. // This program is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU Affero General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. //
  9. // This program is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU Affero General Public License for more details.
  13. //
  14. // You should have received a copy of the GNU Affero General Public License
  15. // along with this program. If not, see <https://www.gnu.org/licenses/>.
  16. package database
  17. import (
  18. log "maunium.net/go/maulogger"
  19. "maunium.net/go/mautrix-whatsapp/types"
  20. )
  21. type PuppetQuery struct {
  22. db *Database
  23. log log.Logger
  24. }
  25. func (pq *PuppetQuery) CreateTable() error {
  26. _, err := pq.db.Exec(`CREATE TABLE IF NOT EXISTS puppet (
  27. jid VARCHAR(255),
  28. receiver VARCHAR(255),
  29. displayname VARCHAR(255),
  30. avatar VARCHAR(255),
  31. PRIMARY KEY(jid, receiver)
  32. )`)
  33. return err
  34. }
  35. func (pq *PuppetQuery) New() *Puppet {
  36. return &Puppet{
  37. db: pq.db,
  38. log: pq.log,
  39. }
  40. }
  41. func (pq *PuppetQuery) GetAll(receiver types.MatrixUserID) (puppets []*Puppet) {
  42. rows, err := pq.db.Query("SELECT * FROM puppet WHERE receiver=%s")
  43. if err != nil || rows == nil {
  44. return nil
  45. }
  46. defer rows.Close()
  47. for rows.Next() {
  48. puppets = append(puppets, pq.New().Scan(rows))
  49. }
  50. return
  51. }
  52. func (pq *PuppetQuery) Get(jid types.WhatsAppID, receiver types.MatrixUserID) *Puppet {
  53. row := pq.db.QueryRow("SELECT * FROM user WHERE jid=? AND receiver=?", jid, receiver)
  54. if row == nil {
  55. return nil
  56. }
  57. return pq.New().Scan(row)
  58. }
  59. type Puppet struct {
  60. db *Database
  61. log log.Logger
  62. JID types.WhatsAppID
  63. Receiver types.MatrixUserID
  64. Displayname string
  65. Avatar string
  66. }
  67. func (puppet *Puppet) Scan(row Scannable) *Puppet {
  68. err := row.Scan(&puppet.JID, &puppet.Receiver, &puppet.Displayname, &puppet.Avatar)
  69. if err != nil {
  70. puppet.log.Fatalln("Database scan failed:", err)
  71. }
  72. return puppet
  73. }
  74. func (puppet *Puppet) Insert() error {
  75. _, err := puppet.db.Exec("INSERT INTO puppet VALUES (?, ?, ?, ?)",
  76. puppet.JID, puppet.Receiver, puppet.Displayname, puppet.Avatar)
  77. return err
  78. }
  79. func (puppet *Puppet) Update() error {
  80. _, err := puppet.db.Exec("UPDATE puppet SET displayname=?, avatar=? WHERE jid=? AND receiver=?",
  81. puppet.Displayname, puppet.Avatar,
  82. puppet.JID, puppet.Receiver)
  83. return err
  84. }