mirror of
https://github.com/twofas/2fas-server.git
synced 2024-12-12 20:20:19 +01:00
89 lines
3.1 KiB
Go
89 lines
3.1 KiB
Go
package command
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/google/uuid"
|
|
"github.com/twofas/2fas-server/internal/api/browser_extension/domain"
|
|
"github.com/twofas/2fas-server/internal/api/mobile/adapters"
|
|
mobile_domain "github.com/twofas/2fas-server/internal/api/mobile/domain"
|
|
"github.com/twofas/2fas-server/internal/common/websocket"
|
|
)
|
|
|
|
type BrowserExtensionHasBeenPairedWithDevice struct {
|
|
Event string `json:"event"`
|
|
BrowserExtensionId string `json:"browser_extension_id"`
|
|
DeviceId string `json:"device_id"`
|
|
DevicePublicKey string `json:"device_public_key"`
|
|
}
|
|
|
|
type BrowserExtensionHasNotBeenPairedWithDevice struct {
|
|
Event string `json:"event"`
|
|
BrowserExtensionId string `json:"browser_extension_id"`
|
|
DeviceId string `json:"device_id"`
|
|
Reason string `json:"reason"`
|
|
}
|
|
|
|
func NewBrowserExtensionHasBeenPairedWithDevice(deviceId, devicePublicKey string, extId uuid.UUID) *BrowserExtensionHasBeenPairedWithDevice {
|
|
return &BrowserExtensionHasBeenPairedWithDevice{
|
|
Event: "browser_extensions.pairing.success",
|
|
BrowserExtensionId: extId.String(),
|
|
DeviceId: deviceId,
|
|
DevicePublicKey: devicePublicKey,
|
|
}
|
|
}
|
|
|
|
func NewBrowserExtensionHasNotBeenPairedWithDevice(err error, deviceId string, extId uuid.UUID) *BrowserExtensionHasNotBeenPairedWithDevice {
|
|
return &BrowserExtensionHasNotBeenPairedWithDevice{
|
|
Event: "browser_extensions.pairing.failure",
|
|
BrowserExtensionId: extId.String(),
|
|
DeviceId: deviceId,
|
|
Reason: err.Error(),
|
|
}
|
|
}
|
|
|
|
type PairMobileWithBrowserExtension struct {
|
|
DeviceId string `uri:"device_id" validate:"required"`
|
|
ExtensionId string `json:"extension_id" validate:"required,uuid4"`
|
|
DevicePublicKey string `json:"device_public_key" validate:"required,lte=768"`
|
|
}
|
|
|
|
type PairMobileWithExtensionHandler struct {
|
|
BrowserExtensionsRepository domain.BrowserExtensionRepository
|
|
MobileDeviceExtensionsRepository mobile_domain.MobileDeviceExtensionsRepository
|
|
MobileApplicationExtensionsService *adapters.DeviceExtensionsService
|
|
WebsocketClient *websocket.WebsocketApiClient
|
|
}
|
|
|
|
func (h *PairMobileWithExtensionHandler) Handle(cmd *PairMobileWithBrowserExtension) error {
|
|
deviceId, _ := uuid.Parse(cmd.DeviceId)
|
|
extensionId, _ := uuid.Parse(cmd.ExtensionId)
|
|
|
|
browserExtension, err := h.BrowserExtensionsRepository.FindById(extensionId)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
mobileDeviceExtension, _ := h.MobileDeviceExtensionsRepository.FindById(deviceId, extensionId)
|
|
|
|
if mobileDeviceExtension != nil {
|
|
return mobile_domain.ExtensionHasAlreadyBeenPairedError{ExtensionId: extensionId.String()}
|
|
}
|
|
|
|
err = h.MobileApplicationExtensionsService.PairDeviceWithBrowserExtension(cmd.DeviceId, extensionId)
|
|
|
|
websocketUri := fmt.Sprintf("browser_extensions/%s", browserExtension.Id.String())
|
|
|
|
if err != nil {
|
|
message := NewBrowserExtensionHasNotBeenPairedWithDevice(err, cmd.DeviceId, extensionId)
|
|
|
|
h.WebsocketClient.SendMessage(websocketUri, message)
|
|
|
|
return err
|
|
}
|
|
|
|
message := NewBrowserExtensionHasBeenPairedWithDevice(cmd.DeviceId, cmd.DevicePublicKey, extensionId)
|
|
|
|
return h.WebsocketClient.SendMessage(websocketUri, message)
|
|
}
|