Changeset 90 in code for trunk

Timestamp:

Mar 12, 2020, 6:17:06 PM (5 years ago)

Author:

contact

Message:

Store NICK changes in the DB

Location:

trunk

Files:

• db.go (modified) (1 diff)
• downstream.go (modified) (2 diffs)
• server.go (modified) (1 diff)

trunk/db.go

@@ -118 +118 @@
 }
 
+func (db *DB) StoreNetwork(username string, network *Network) error {
+        db.lock.Lock()
+        defer db.lock.Unlock()
+
+        var netUsername, netRealname *string
+        if network.Username != "" {
+                netUsername = &network.Username
+        }
+        if network.Realname != "" {
+                netRealname = &network.Realname
+        }
+
+        var err error
+        if network.ID != 0 {
+                _, err = db.db.Exec("UPDATE Network SET addr = ?, nick = ?, username = ?, realname = ? WHERE id = ?", network.Addr, network.Nick, netUsername, netRealname, network.ID)
+        } else {
+                var res sql.Result
+                res, err = db.db.Exec("INSERT INTO Network(user, addr, nick, username, realname) VALUES (?, ?, ?, ?, ?)", username, network.Addr, network.Nick, netUsername, netRealname)
+                if err != nil {
+                        return err
+                }
+                network.ID, err = res.LastInsertId()
+        }
+        return err
+}
+
 func (db *DB) ListChannels(networkID int64) ([]Channel, error) {
         db.lock.RLock()

trunk/downstream.go

@@ -104 +104 @@
 func (dc *downstreamConn) marshalChannel(uc *upstreamConn, name string) string {
         return name
+}
+
+func (dc *downstreamConn) forEachNetwork(f func(*network)) {
+        if dc.network != nil {
+                f(dc.network)
+        } else {
+                dc.user.forEachNetwork(f)
+        }
 }
 
@@ -459 +467 @@
                 }}
         case "NICK":
+                var nick string
+                if err := parseMessageParams(msg, &nick); err != nil {
+                        return err
+                }
+
+                var err error
+                dc.forEachNetwork(func(n *network) {
+                        if err != nil {
+                                return
+                        }
+                        n.Nick = nick
+                        err = dc.srv.db.StoreNetwork(dc.user.Username, &n.Network)
+                })
+                if err != nil {
+                        return err
+                }
+
                 dc.forEachUpstream(func(uc *upstreamConn) {
                         uc.SendMessage(msg)

trunk/server.go

@@ -108 +108 @@
                 srv:  srv,
         }
+}
+
+func (u *user) forEachNetwork(f func(*network)) {
+        u.lock.Lock()
+        for _, network := range u.networks {
+                f(network)
+        }
+        u.lock.Unlock()
 }