feat(api): add tunnel device UID update support during device merging

- Implement TunnelUpdateDeviceUID method in store interface
- Update mergeDevice to transfer tunnels along with sessions
- Add tunnel UID updates when merging devices with same MAC address
- Include TunnelStore interface in main Store composition
- Add corresponding test mocks for tunnel device UID updates
- Remove TODO comment as tunnel updates are now implemented

Author: heiytor

api/services/device.go

@@ -374,7 +374,9 @@ func (s *service) UpdateDevice(ctx context.Context, req *requests.DeviceUpdate)
 // mergeDevice merges an old device into a new device. It transfers all sessions from the old device to the new one and
 // renames the new device to preserve the old device's identity. The old device is then deleted and the namespace's device count is decremented.
 func (s *service) mergeDevice(ctx context.Context, tenantID string, oldDevice *models.Device, newDevice *models.Device) error {
-	// TODO: update tunnels as well?
+	if err := s.store.TunnelUpdateDeviceUID(ctx, tenantID, oldDevice.UID, newDevice.UID); err != nil {
+		return err
+	}
 
 	if err := s.store.SessionUpdateDeviceUID(ctx, models.UID(oldDevice.UID), models.UID(newDevice.UID)); err != nil && !errors.Is(err, store.ErrNoDocuments) {
 		return err

api/services/device_test.go

@@ -1714,6 +1714,10 @@ func TestUpdateDeviceStatus(t *testing.T) {
 					).
 					Once()
 				// Merge operations
+				storeMock.
+					On("TunnelUpdateDeviceUID", ctx, "00000000-0000-0000-0000-000000000000", "old-device", "new-device").
+					Return(nil).
+					Once()
 				storeMock.
 					On("SessionUpdateDeviceUID", ctx, models.UID("old-device"), models.UID("new-device")).
 					Return(nil).

api/store/mocks/store.go

@@ -0,0 +1,16 @@
+package mongo
+
+import (
+	"context"
+
+	"go.mongodb.org/mongo-driver/bson"
+)
+
+func (s *Store) TunnelUpdateDeviceUID(ctx context.Context, tenantID, oldUID, newUID string) error {
+	_, err := s.db.Collection("tunnels").UpdateMany(ctx, bson.M{"namespace": tenantID, "device": oldUID}, bson.M{"$set": bson.M{"device": newUID}})
+	if err != nil {
+		return FromMongoError(err)
+	}
+
+	return nil
+}

api/store/mongo/tunnel.go

@@ -15,6 +15,7 @@ type Store interface {
 	APIKeyStore
 	TransactionStore
 	SystemStore
+	TunnelStore
 
 	Options() QueryOptions
 }

api/store/store.go

@@ -0,0 +1,8 @@
+package store
+
+import "context"
+
+type TunnelStore interface {
+	// TunnelUpdateDeviceUID changes all tunnels from oldUID to newUID within the specified tenantID.
+	TunnelUpdateDeviceUID(ctx context.Context, tenantID, oldUID, newUID string) error
+}