package rest
import (
"fmt"
"github.com/aldinokemal/go-whatsapp-web-multidevice/config"
domainApp "github.com/aldinokemal/go-whatsapp-web-multidevice/domains/app"
"github.com/aldinokemal/go-whatsapp-web-multidevice/infrastructure/whatsapp"
"github.com/aldinokemal/go-whatsapp-web-multidevice/pkg/utils"
"github.com/gofiber/fiber/v2"
)
type App struct {
Service domainApp.IAppUsecase
}
func InitRestApp(app fiber.Router, service domainApp.IAppUsecase) App {
rest := App{Service: service}
app.Get("/app/login", rest.Login)
app.Get("/app/login-with-code", rest.LoginWithCode)
app.Get("/app/logout", rest.Logout)
app.Get("/app/reconnect", rest.Reconnect)
app.Get("/app/devices", rest.Devices)
app.Get("/app/status", rest.ConnectionStatus)
return App{Service: service}
}
func (handler *App) Login(c *fiber.Ctx) error {
response, err := handler.Service.Login(c.UserContext())
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Login success",
Results: map[string]any{
"qr_link": fmt.Sprintf("%s://%s%s/%s", c.Protocol(), c.Hostname(), config.AppBasePath, response.ImagePath),
"qr_duration": response.Duration,
},
})
}
func (handler *App) LoginWithCode(c *fiber.Ctx) error {
pairCode, err := handler.Service.LoginWithCode(c.UserContext(), c.Query("phone"))
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Login with code success",
Results: map[string]any{
"pair_code": pairCode,
},
})
}
func (handler *App) Logout(c *fiber.Ctx) error {
err := handler.Service.Logout(c.UserContext())
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Success logout",
Results: nil,
})
}
func (handler *App) Reconnect(c *fiber.Ctx) error {
err := handler.Service.Reconnect(c.UserContext())
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Reconnect success",
Results: nil,
})
}
func (handler *App) Devices(c *fiber.Ctx) error {
devices, err := handler.Service.FetchDevices(c.UserContext())
utils.PanicIfNeeded(err)
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Fetch device success",
Results: devices,
})
}
func (handler *App) ConnectionStatus(c *fiber.Ctx) error {
isConnected, isLoggedIn, deviceID := whatsapp.GetConnectionStatus()
return c.JSON(utils.ResponseData{
Status: 200,
Code: "SUCCESS",
Message: "Connection status retrieved",
Results: map[string]any{
"is_connected": isConnected,
"is_logged_in": isLoggedIn,
"device_id": deviceID,
},
})
}
