From b94b5caf4c6bd28341d7636cffa7695b5566a1ad Mon Sep 17 00:00:00 2001
From: "Alexander \"Arav\" Andreev" <me@arav.top>
Date: Fri, 21 Oct 2022 04:37:28 +0400
Subject: [PATCH] Added entryID arg to Update*() methods. So IDs could be
 changed.

---
 internal/database/sqlite/database.go | 8 ++++----
 internal/guestbook/database.go       | 4 ++--
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/internal/database/sqlite/database.go b/internal/database/sqlite/database.go
index 762f3e4..a2737e6 100644
--- a/internal/database/sqlite/database.go
+++ b/internal/database/sqlite/database.go
@@ -181,14 +181,14 @@ func (d *SQLiteDatabase) NewEntry(entry *guestbook.Entry) error {
 	return nil
 }
 
-func (d *SQLiteDatabase) UpdateEntry(entry *guestbook.Entry) (bool, error) {
+func (d *SQLiteDatabase) UpdateEntry(entryID int64, entry *guestbook.Entry) (bool, error) {
 	tx, err := d.db.Begin()
 	if err != nil {
 		return false, err
 	}
 	defer tx.Rollback()
 
-	res, err := tx.Stmt(stmtUpdateEntry).Exec(entry.Name, entry.Message, entry.Website, entry.HideWebsite, entry.ID)
+	res, err := tx.Stmt(stmtUpdateEntry).Exec(entry.ID, entry.Name, entry.Message, entry.Website, entry.HideWebsite, entryID)
 	if err != nil {
 		return false, err
 	}
@@ -239,14 +239,14 @@ func (d *SQLiteDatabase) NewReply(reply *guestbook.Reply) (err error) {
 }
 
 // UpdateEntry
-func (d *SQLiteDatabase) UpdateReply(reply *guestbook.Reply) (bool, error) {
+func (d *SQLiteDatabase) UpdateReply(entryID int64, reply *guestbook.Reply) (bool, error) {
 	tx, err := d.db.Begin()
 	if err != nil {
 		return false, err
 	}
 	defer tx.Rollback()
 
-	res, err := tx.Stmt(stmtUpdateReply).Exec(reply.Created, reply.Message, reply.ID)
+	res, err := tx.Stmt(stmtUpdateReply).Exec(reply.ID, reply.Created, reply.Message, entryID)
 	if err != nil {
 		return false, err
 	}
diff --git a/internal/guestbook/database.go b/internal/guestbook/database.go
index 5c34ced..323cd0c 100644
--- a/internal/guestbook/database.go
+++ b/internal/guestbook/database.go
@@ -6,10 +6,10 @@ type Guestbook interface {
 	Entries(page, pageSize int64) ([]*Entry, error)
 	Count() (int64, error)
 	NewEntry(entry *Entry) error
-	UpdateEntry(entry *Entry) (bool, error)
+	UpdateEntry(entryID int64, entry *Entry) (bool, error)
 	DeleteEntry(entryID int64) error
 	NewReply(reply *Reply) error
-	UpdateReply(reply *Reply) (bool, error)
+	UpdateReply(entryID int64, reply *Reply) (bool, error)
 	DeleteReply(entryID int64) error
 	Close() error
 }