Cleanup

1 files changed, 17 insertions, 1 deletions

diff --git a/lib/main.dart b/lib/main.dart
index 40a7641..e160a76 100644
--- a/lib/main.dart
+++ b/lib/main.dart
@@ -13,8 +13,10 @@
 // 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/control/flutter_blue_plus_scanner_controller.dart';
 import 'package:uvok_epaper_badge/control/mock_scanner_controller.dart';
 import 'package:uvok_epaper_badge/control/scanner_controller.dart';
+import 'package:uvok_epaper_badge/control/universal_ble_scanner_controller.dart';
 import 'package:uvok_epaper_badge/widgets/badge_app.dart';
 import 'package:flutter/material.dart';
 
@@ -23,6 +25,20 @@ import 'package:logger/logger.dart';
 var logger = Logger();
 
 void main() {
-  final ScannerController scanner = MockScannerController();
+  final ScannerController scanner;
+  final int arg = 0;
+
+  switch (arg) {
+    case 1:
+      scanner = UniversalBleScannerController(rssiLimit: -70);
+      break;
+    case 2:
+      scanner = FlutterBluePlusScannerController();
+      break;
+    default:
+      scanner = MockScannerController();
+      break;
+  }
+
   runApp(BadgeApp(selectedScanner: scanner));
 }