Merge pull request #1 from rakshabesafe/feat/chat-autoreply

feat: Implement chat auto-reply feature

Author: rakshabesafe

API.md

@@ -705,6 +705,64 @@ curl -s -X POST -H 'Token: 1234ABCD' -H 'Content-Type: application/json' --data
 
 ---
 
+### Add/Update Auto-Reply
+* **Method:** `POST`
+* **Path:** `/chat/autoreply`
+* **Description:** Sets up an auto-reply for a specific phone number. If an auto-reply for the given phone number already exists for the user, it will be updated. If not, a new one will be created.
+* **Authentication:** Requires user token.
+* **Request Body (JSON):**
+  ```json
+  {
+    "Phone": "1234567890", // Target phone number (normalized, e.g., digits only or international format used by the system)
+    "Body": "Hello! I am currently unavailable and will get back to you soon."
+  }
+  ```
+* **Responses:**
+  * `201 Created`: If a new auto-reply was successfully created.
+    ```json
+    {
+      "code": 201,
+      "data": {
+        "detail": "Auto-reply added successfully",
+        "id": "generated-unique-id-for-the-rule"
+      },
+      "success": true
+    }
+    ```
+  * `400 Bad Request`: If `Phone` or `Body` is missing or invalid.
+  * `409 Conflict`: If an auto-reply for this phone number already exists for the user.
+  * `500 Internal Server Error`: For other server-side errors.
+
+---
+
+### Delete Auto-Reply
+* **Method:** `DELETE`
+* **Path:** `/chat/autoreply`
+* **Description:** Deletes an existing auto-reply for a specific phone number for the authenticated user.
+* **Authentication:** Requires user token.
+* **Request Body (JSON):**
+  ```json
+  {
+    "Phone": "1234567890" // Target phone number whose auto-reply rule should be deleted.
+  }
+  ```
+* **Responses:**
+  * `200 OK`: If the auto-reply was successfully deleted.
+    ```json
+    {
+      "code": 200,
+      "data": {
+        "detail": "Auto-reply deleted successfully"
+      },
+      "success": true
+    }
+    ```
+  * `400 Bad Request`: If `Phone` is missing or invalid.
+  * `404 Not Found`: If no auto-reply rule exists for the given phone number for this user.
+  * `500 Internal Server Error`: For other server-side errors.
+
+---
+
 ## Download Document
 
 Downloads a Document from a message and retrieves it Base64 media encoded. Required request parameters are: Url, MediaKey, Mimetype, FileSHA256 and FileLength

handlers.go

@@ -31,6 +31,15 @@ import (
 	"google.golang.org/protobuf/proto"
 )
 
+type AutoReplyRequest struct {
+	Phone string `json:"Phone"`
+	Body  string `json:"Body"`
+}
+
+type DeleteAutoReplyRequest struct {
+	Phone string `json:"Phone"`
+}
+
 type Values struct {
 	m map[string]string
 }
@@ -202,6 +211,103 @@ func (s *server) Connect() http.HandlerFunc {
 	}
 }
 
+// AddAutoReply handles adding a new auto-reply entry for a user.
+func (s *server) AddAutoReply() http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		txtid := r.Context().Value("userinfo").(Values).Get("Id")
+
+		var req AutoReplyRequest
+		if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+			s.Respond(w, r, http.StatusBadRequest, errors.New("Could not decode payload"))
+			return
+		}
+
+		if req.Phone == "" {
+			s.Respond(w, r, http.StatusBadRequest, errors.New("Missing Phone in Payload"))
+			return
+		}
+		if req.Body == "" {
+			s.Respond(w, r, http.StatusBadRequest, errors.New("Missing Body in Payload"))
+			return
+		}
+
+		newId, err := GenerateRandomID() // Assuming GenerateRandomID is accessible from migrations.go
+		if err != nil {
+			log.Error().Err(err).Msg("Failed to generate random ID for auto-reply")
+			s.Respond(w, r, http.StatusInternalServerError, errors.New("Failed to create auto-reply entry"))
+			return
+		}
+
+		_, err = s.db.Exec("INSERT INTO autoreplies (id, user_id, phone_number, reply_body) VALUES ($1, $2, $3, $4)", newId, txtid, req.Phone, req.Body)
+		if err != nil {
+			// Check for unique constraint violation (specific error code might depend on DB: PostgreSQL uses "23505")
+			// This is a simplified check; a more robust way involves checking pq.Error.Code or sqlite3.ErrConstraintUnique
+			if strings.Contains(err.Error(), "UNIQUE constraint failed") || strings.Contains(err.Error(), "duplicate key value violates unique constraint") {
+				s.Respond(w, r, http.StatusConflict, errors.New("Auto-reply for this phone number already exists for the user"))
+				return
+			}
+			log.Error().Err(err).Str("user_id", txtid).Str("phone", req.Phone).Msg("Failed to add auto-reply")
+			s.Respond(w, r, http.StatusInternalServerError, errors.New("Failed to add auto-reply"))
+			return
+		}
+
+		response := map[string]string{"detail": "Auto-reply added successfully", "id": newId}
+		responseJson, err := json.Marshal(response)
+		if err != nil {
+			log.Error().Err(err).Msg("Failed to marshal success response for AddAutoReply")
+			s.Respond(w, r, http.StatusInternalServerError, errors.New("Failed to create auto-reply entry"))
+			return
+		}
+		s.Respond(w, r, http.StatusCreated, string(responseJson))
+	}
+}
+
+// DeleteAutoReply handles deleting an auto-reply entry for a user.
+func (s *server) DeleteAutoReply() http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		txtid := r.Context().Value("userinfo").(Values).Get("Id")
+
+		var req DeleteAutoReplyRequest
+		if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
+			s.Respond(w, r, http.StatusBadRequest, errors.New("Could not decode payload"))
+			return
+		}
+
+		if req.Phone == "" {
+			s.Respond(w, r, http.StatusBadRequest, errors.New("Missing Phone in Payload"))
+			return
+		}
+
+		result, err := s.db.Exec("DELETE FROM autoreplies WHERE user_id = $1 AND phone_number = $2", txtid, req.Phone)
+		if err != nil {
+			log.Error().Err(err).Str("user_id", txtid).Str("phone", req.Phone).Msg("Failed to delete auto-reply")
+			s.Respond(w, r, http.StatusInternalServerError, errors.New("Failed to delete auto-reply"))
+			return
+		}
+
+		rowsAffected, err := result.RowsAffected()
+		if err != nil {
+			log.Error().Err(err).Str("user_id", txtid).Str("phone", req.Phone).Msg("Failed to check affected rows after delete")
+			s.Respond(w, r, http.StatusInternalServerError, errors.New("Failed to confirm deletion status"))
+			return
+		}
+
+		if rowsAffected == 0 {
+			s.Respond(w, r, http.StatusNotFound, errors.New("Auto-reply not found for this user and phone number"))
+			return
+		}
+
+		response := map[string]string{"detail": "Auto-reply deleted successfully"}
+		responseJson, err := json.Marshal(response)
+		if err != nil {
+			log.Error().Err(err).Msg("Failed to marshal success response for DeleteAutoReply")
+			s.Respond(w, r, http.StatusInternalServerError, errors.New("Failed to process deletion confirmation")) // Should ideally not happen
+			return
+		}
+		s.Respond(w, r, http.StatusOK, string(responseJson))
+	}
+}
+
 // Disconnects from Whatsapp websocket, does not log out device
 func (s *server) Disconnect() http.HandlerFunc {
 	return func(w http.ResponseWriter, r *http.Request) {

main.go

@@ -14,6 +14,11 @@ import (
 	"syscall"
 	"time"
 
+	"database/sql" // Added for sql.ErrNoRows
+	"go.mau.fi/whatsmeow" // Added for whatsmeow.SendRequestExtra
+	"go.mau.fi/whatsmeow/proto/waE2E" // Added for waE2E.Message
+	"google.golang.org/protobuf/proto" // Added for proto.String
+
 	"go.mau.fi/whatsmeow/store/sqlstore"
 	waLog "go.mau.fi/whatsmeow/util/log"
 

migrations.go

@@ -45,8 +45,41 @@ var migrations = []Migration{
 		Name:  "change_id_to_string",
 		UpSQL: changeIDToStringSQL,
 	},
+	{
+		ID:    4,
+		Name:  "add_autoreplies_table",
+		UpSQL: addAutorepliesTableSQLPostgres,
+	},
 }
 
+const addAutorepliesTableSQLPostgres = `
+DO $$
+BEGIN
+    IF NOT EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = 'autoreplies') THEN
+        CREATE TABLE autoreplies (
+            id TEXT PRIMARY KEY,
+            user_id TEXT NOT NULL REFERENCES users(id) ON DELETE CASCADE,
+            phone_number TEXT NOT NULL,
+            reply_body TEXT NOT NULL,
+            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+            UNIQUE(user_id, phone_number)
+        );
+    END IF;
+END $$;
+`
+
+const addAutorepliesTableSQLSQLite = `
+CREATE TABLE autoreplies (
+    id TEXT PRIMARY KEY,
+    user_id TEXT NOT NULL,
+    phone_number TEXT NOT NULL,
+    reply_body TEXT NOT NULL,
+    created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+    FOREIGN KEY(user_id) REFERENCES users(id) ON DELETE CASCADE,
+    UNIQUE(user_id, phone_number)
+)
+`
+
 const changeIDToStringSQL = `
 -- Migration to change ID from integer to random string
 DO $$
@@ -228,6 +261,12 @@ func applyMigration(db *sqlx.DB, migration Migration) error {
 		} else {
 			_, err = tx.Exec(migration.UpSQL)
 		}
+	} else if migration.ID == 4 {
+		if db.DriverName() == "sqlite" {
+			err = createTableIfNotExistsSQLite(tx, "autoreplies", addAutorepliesTableSQLSQLite)
+		} else {
+			_, err = tx.Exec(migration.UpSQL)
+		}
 	} else {
 		_, err = tx.Exec(migration.UpSQL)
 	}

routes.go

@@ -101,6 +101,9 @@ func (s *server) routes() {
 	s.router.Handle("/chat/send/poll", c.Then(s.SendPoll())).Methods("POST")
 	s.router.Handle("/chat/send/edit", c.Then(s.SendEditMessage())).Methods("POST")
 
+	s.router.Handle("/chat/autoreply", c.Then(s.AddAutoReply())).Methods("POST")
+	s.router.Handle("/chat/autoreply", c.Then(s.DeleteAutoReply())).Methods("DELETE")
+
 	s.router.Handle("/user/presence", c.Then(s.SendPresence())).Methods("POST")
 	s.router.Handle("/user/info", c.Then(s.GetUser())).Methods("POST")
 	s.router.Handle("/user/check", c.Then(s.CheckUser())).Methods("POST")