From 04dd9d51f30951ffd444cf4ea1393c1c30e0b68e Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Wed, 25 Mar 2020 14:23:41 +0100
Subject: [PATCH] Correctly set Channel.ID in DB.StoreChannel

---
 db.go | 31 ++++++++++++++++++++++---------
 1 file changed, 22 insertions(+), 9 deletions(-)

diff --git a/db.go b/db.go
index 61c9780..d67fca9 100644
--- a/db.go
+++ b/db.go
@@ -33,6 +33,13 @@ type Network struct {
 	SASL     SASL
 }
 
+func (net *Network) GetName() string {
+	if net.Name != "" {
+		return net.Name
+	}
+	return net.Addr
+}
+
 type Channel struct {
 	ID   int64
 	Name string
@@ -225,8 +232,21 @@ func (db *DB) StoreChannel(networkID int64, ch *Channel) error {
 	defer db.lock.Unlock()
 
 	key := toStringPtr(ch.Key)
-	_, err := db.db.Exec(`INSERT OR REPLACE INTO Channel(network, name, key)
-		VALUES (?, ?, ?)`, networkID, ch.Name, key)
+
+	var err error
+	if ch.ID != 0 {
+		_, err = db.db.Exec(`UPDATE Channel
+			SET network = ?, name = ?, key = ?
+			WHERE id = ?`, networkID, ch.Name, key, ch.ID)
+	} else {
+		var res sql.Result
+		res, err = db.db.Exec(`INSERT INTO Channel(network, name, key)
+			VALUES (?, ?, ?)`, networkID, ch.Name, key)
+		if err != nil {
+			return err
+		}
+		ch.ID, err = res.LastInsertId()
+	}
 	return err
 }
 
@@ -237,10 +257,3 @@ func (db *DB) DeleteChannel(networkID int64, name string) error {
 	_, err := db.db.Exec("DELETE FROM Channel WHERE network = ? AND name = ?", networkID, name)
 	return err
 }
-
-func (net *Network) GetName() string {
-	if net.Name != "" {
-		return net.Name
-	}
-	return net.Addr
-}