Add UB connection

1 files changed, 4 insertions, 0 deletions

diff --git a/lib/model/device_connection_factory.dart b/lib/model/device_connection_factory.dart
index 92b2348..49e1887 100644
--- a/lib/model/device_connection_factory.dart
+++ b/lib/model/device_connection_factory.dart
@@ -13,16 +13,20 @@
 // You should have received a copy of the GNU General Public License
 // along with this program.  If not, see <https://www.gnu.org/licenses/>.
 
+import 'package:uvok_epaper_badge/model/connection/universal_ble_connection.dart';
 import 'package:uvok_epaper_badge/model/device/device.dart';
 import 'package:uvok_epaper_badge/model/connection/device_connection.dart';
 import 'package:uvok_epaper_badge/model/device/flutter_blue_plus_device.dart';
 import 'package:uvok_epaper_badge/model/connection/flutter_blue_plus_device_connection.dart';
 import 'package:uvok_epaper_badge/model/connection/mock_device_connection.dart';
+import 'package:uvok_epaper_badge/model/device/universal_ble_device.dart';
 
 class DeviceConnectionFactory {
   static DeviceConnection createConnection(Device device) {
     if (device is FlutterBluePlusDevice) {
       return FlutterBluePlusDeviceConnection(device: device);
+    } else if (device is UniversalBleDevice) {
+      return UniversalBleConnection(device);
     }
 
     // switch (device.type) {