From 3f911ae321128b4ef651f08bd7923dc0301622f3 Mon Sep 17 00:00:00 2001
From: pahmiudahgede <pahmilucu123@gmail.com>
Date: Fri, 16 May 2025 01:06:55 +0700
Subject: [PATCH] feat: logout whatsapp session and remove device

---
 config/whatsapp.go                   | 40 ++++++++++++++++++++++++++--
 internal/handler/whatsapp_handler.go | 24 +++++++++++++++++
 presentation/whatsapp_route.go       | 13 +++++++++
 router/setup_routes.go.go            |  1 +
 4 files changed, 76 insertions(+), 2 deletions(-)
 create mode 100644 internal/handler/whatsapp_handler.go
 create mode 100644 presentation/whatsapp_route.go

diff --git a/config/whatsapp.go b/config/whatsapp.go
index 18d073c..1ed9dca 100644
--- a/config/whatsapp.go
+++ b/config/whatsapp.go
@@ -25,7 +25,7 @@ func InitWhatsApp() {
 	dbLog := waLog.Stdout("Database", "DEBUG", true)
 
 	dsn := fmt.Sprintf(
-		"postgres://%s:%s@%s:%s/%s?sslmode=disable",	
+		"postgres://%s:%s@%s:%s/%s?sslmode=disable",
 		os.Getenv("DB_USER"),
 		os.Getenv("DB_PASSWORD"),
 		os.Getenv("DB_HOST"),
@@ -80,7 +80,6 @@ func generateQRCode(qrString string) {
 	qrterminal.GenerateHalfBlock(qrString, qrterminal.M, os.Stdout)
 }
 
-
 func handleShutdown() {
 	sigChan := make(chan os.Signal, 1)
 	signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM)
@@ -109,3 +108,40 @@ func SendWhatsAppMessage(phone, message string) error {
 	log.Printf("WhatsApp message sent successfully to: %s", phone)
 	return nil
 }
+
+func LogoutWhatsApp() error {
+	if WhatsAppClient == nil {
+		return fmt.Errorf("WhatsApp client is not initialized")
+	}
+
+	WhatsAppClient.Disconnect()
+
+	err := removeWhatsAppDeviceFromContainer()
+	if err != nil {
+		return fmt.Errorf("failed to remove device from container: %v", err)
+	}
+
+	err = container.Close()
+	if err != nil {
+		return fmt.Errorf("failed to close database connection: %v", err)
+	}
+
+	log.Println("WhatsApp client disconnected and session cleared successfully.")
+	return nil
+}
+
+func removeWhatsAppDeviceFromContainer() error {
+	deviceStore, err := container.GetFirstDevice()
+	if err != nil {
+		return fmt.Errorf("failed to get WhatsApp device: %v", err)
+	}
+
+	if deviceStore != nil {
+		err := deviceStore.Delete()
+		if err != nil {
+			return fmt.Errorf("failed to remove device from store: %v", err)
+		}
+	}
+
+	return nil
+}
diff --git a/internal/handler/whatsapp_handler.go b/internal/handler/whatsapp_handler.go
new file mode 100644
index 0000000..e875f36
--- /dev/null
+++ b/internal/handler/whatsapp_handler.go
@@ -0,0 +1,24 @@
+package handler
+
+import (
+	"log"
+	"rijig/config"
+	"rijig/utils"
+
+	"github.com/gofiber/fiber/v2"
+)
+
+func WhatsAppHandler(c *fiber.Ctx) error {
+	userID, ok := c.Locals("userID").(string)
+	if !ok || userID == "" {
+		return utils.ErrorResponse(c, "User is not logged in or invalid session")
+	}
+
+	err := config.LogoutWhatsApp()
+	if err != nil {
+		log.Printf("Error during logout process for user %s: %v", userID, err)
+		return utils.ErrorResponse(c, err.Error())
+	}
+
+	return utils.SuccessResponse(c, nil, "Logged out successfully")
+}
diff --git a/presentation/whatsapp_route.go b/presentation/whatsapp_route.go
new file mode 100644
index 0000000..dd9c14e
--- /dev/null
+++ b/presentation/whatsapp_route.go
@@ -0,0 +1,13 @@
+package presentation
+
+import (
+	"rijig/internal/handler"
+	"rijig/middleware"
+	"rijig/utils"
+
+	"github.com/gofiber/fiber/v2"
+)
+
+func WhatsAppRouter(api fiber.Router) {
+	api.Post("/logout/whastapp", middleware.AuthMiddleware, middleware.RoleMiddleware(utils.RoleAdministrator), handler.WhatsAppHandler)
+}
diff --git a/router/setup_routes.go.go b/router/setup_routes.go.go
index 2bf1491..ecfad81 100644
--- a/router/setup_routes.go.go
+++ b/router/setup_routes.go.go
@@ -42,5 +42,6 @@ func SetupRoutes(app *fiber.App) {
 	presentation.CoverageAreaRouter(api)
 	presentation.StoreRouter(api)
 	presentation.ProductRouter(api)
+	presentation.WhatsAppRouter(api)
 
 }