✨ (rc): Set SerialNumber in BLE

leka · Mar 18, 2022 · 5911c1f · 5911c1f
1 parent c6d28fb
commit 5911c1f
Show file tree

Hide file tree

Showing 5 changed files with 27 additions and 3 deletions.
diff --git a/app/os/CMakeLists.txt b/app/os/CMakeLists.txt
@@ -17,11 +17,13 @@ target_sources(LekaOS
 target_link_libraries(LekaOS
 	bootutil
 	RobotKit
+	CoreMCU
 	CoreTimeout
 	CoreBattery
 	CoreQSPI
 	CoreFlashMemory
 	FileSystemKit
+	SerialNumberKit
 	FirmwareKit
 	BehaviorKit
 	CorePwm

diff --git a/app/os/main.cpp b/app/os/main.cpp
@@ -10,6 +10,7 @@
 #include "CoreBattery.h"
 #include "CoreFlashIS25LP016D.h"
 #include "CoreFlashManagerIS25LP016D.h"
+#include "CoreMCU.h"
 #include "CorePwm.h"
 #include "CoreQSPI.h"
 #include "CoreSPI.h"
@@ -21,6 +22,7 @@
 #include "QSPIFBlockDevice.h"
 #include "RobotController.h"
 #include "SDBlockDevice.h"
+#include "SerialNumberKit.h"
 #include "SlicingBlockDevice.h"
 #include "VideoKit.h"
 #include "bootutil/bootutil.h"
@@ -52,6 +54,9 @@ void start()
 
 auto sleep_timeout = CoreTimeout {};
 
+auto mcu			 = CoreMCU {};
+auto serialnumberkit = SerialNumberKit {mcu};
+
 auto sd_blockdevice = SDBlockDevice {SD_SPI_MOSI, SD_SPI_MISO, SD_SPI_SCK};
 auto fatfs			= FATFileSystem {"fs"};
 
@@ -86,7 +91,7 @@ auto coreflashmanager = CoreFlashManagerIS25LP016D(coreqspi);
 auto coreflash		  = CoreFlashIS25LP016D(coreqspi, coreflashmanager);
 auto firmwarekit	  = FirmwareKit(coreflash);
 
-auto rc = RobotController {sleep_timeout, battery, firmwarekit, behaviorkit};
+auto rc = RobotController {sleep_timeout, battery, serialnumberkit, firmwarekit, behaviorkit};
 
 void initializeSD()
 {

diff --git a/libs/RobotKit/CMakeLists.txt b/libs/RobotKit/CMakeLists.txt
@@ -19,6 +19,7 @@ target_sources(RobotKit
 
 target_link_libraries(RobotKit
 	BatteryKit
+	SerialNumberKit
 	BLEKit
 	BehaviorKit
 )

diff --git a/libs/RobotKit/include/RobotController.h b/libs/RobotKit/include/RobotController.h
@@ -13,6 +13,7 @@
 
 #include "BatteryKit.h"
 #include "BehaviorKit.h"
+#include "SerialNumberKit.h"
 #include "StateMachine.h"
 #include "interface/RobotController.h"
 #include "interface/drivers/Battery.h"
@@ -28,9 +29,11 @@ class RobotController : public interface::RobotController
 	sm_t state_machine {static_cast<interface::RobotController &>(*this)};
 
 	explicit RobotController(interface::Timeout &sleep_timeout, interface::Battery &battery,
-							 interface::FirmwareUpdate &firmware_update, BehaviorKit &behaviorkit)
+							 SerialNumberKit &serialnumberkit, interface::FirmwareUpdate &firmware_update,
+							 BehaviorKit &behaviorkit)
 		: _sleep_timeout(sleep_timeout),
 		  _battery(battery),
+		  _serialnumberkit(serialnumberkit),
 		  _firmware_update(firmware_update),
 		  _behaviorkit(behaviorkit) {};
 
@@ -103,6 +106,9 @@ class RobotController : public interface::RobotController
 
 		_ble.setServices(services);
 		_ble.init();
+
+		auto _serial_number = _serialnumberkit.getSerialNumber();
+		_service_device_information.setSerialNumber(_serial_number);
 	}
 
 	void registerOnUpdateLoadedCallback(std::function<void()> const &on_update_loaded_callback)
@@ -142,6 +148,8 @@ class RobotController : public interface::RobotController
 	BatteryKit _battery_kit {_battery};
 	uint8_t _minimal_battery_level_to_update {25};
 
+	SerialNumberKit &_serialnumberkit;
+
 	interface::FirmwareUpdate &_firmware_update;
 	std::function<void()> _on_update_loaded_callback {};
 

diff --git a/libs/RobotKit/tests/RobotController_test.cpp b/libs/RobotKit/tests/RobotController_test.cpp
@@ -10,11 +10,13 @@
 #include "CoreMotor.h"
 #include "CorePwm.h"
 #include "CoreSPI.h"
+#include "SerialNumberKit.h"
 #include "gmock/gmock.h"
 #include "gtest/gtest.h"
 #include "mocks/leka/Battery.h"
 #include "mocks/leka/FirmwareUpdate.h"
 #include "mocks/leka/LEDAnimation.h"
+#include "mocks/leka/MCU.h"
 #include "mocks/leka/PwmOut.h"
 #include "mocks/leka/Timeout.h"
 #include "mocks/mbed/DigitalOut.h"
@@ -55,6 +57,7 @@ class RobotControllerTest : public testing::Test
 		EXPECT_CALL(mbed_mock_gatt, addService).Times(AnyNumber());
 		EXPECT_CALL(mbed_mock_gap, setEventHandler).Times(AnyNumber());
 		EXPECT_CALL(mbed_mock_gatt, setEventHandler).Times(AnyNumber());
+		EXPECT_CALL(mock_mcu, getID).Times(AnyNumber());
 
 		rc.initializeComponents();
 
@@ -72,6 +75,10 @@ class RobotControllerTest : public testing::Test
 
 	mock::Timeout sleep_timeout {};
 	mock::Battery battery {};
+
+	mock::MCU mock_mcu {};
+	SerialNumberKit serialnumberkit {mock_mcu};
+
 	mock::FirmwareUpdate firmware_update {};
 
 	CoreSPI spi {NC, NC, NC, NC};
@@ -99,7 +106,8 @@ class RobotControllerTest : public testing::Test
 
 	BehaviorKit bhvkit {videokit, ledkit, motor_left, motor_right};
 
-	RobotController<bsml::sm<robot::StateMachine, bsml::testing>> rc {sleep_timeout, battery, firmware_update, bhvkit};
+	RobotController<bsml::sm<robot::StateMachine, bsml::testing>> rc {sleep_timeout, battery, serialnumberkit,
+																	  firmware_update, bhvkit};
 
 	interface::Timeout::callback_t on_sleep_timeout = {};