Michel2/pybricks-code
1
0
Fork 0
mirror of https://github.com/pybricks/pybricks-code.git synced 2026-09-14 10:35:11 +00:00
Code Issues Releases Activity

alerts: allow overriding key

Browse Source 
This adds an optional key parameter to alertsShowAlert() to allow
overriding the key. This allows us to call flashProgress using that
action instead of having to use the toaster directly.
This commit is contained in:
David Lechner
2022-10-19 18:26:37 -05:00
committed by David Lechner
parent fa27649fe3
commit 28cf87c356
5 changed files with 312 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/firmware/alerts/index.ts
+2
View File
@@ -3,6 +3,7 @@
import { dfuError } from './DfuError';
import { firmwareMismatch } from './FirmwareMismatch';
import { flashProgress } from './FlashProgress';
import { noDfuHub } from './NoDfuHub';
import { noDfuInterface } from './NoDfuInterface';
import { noWebUsb } from './NoWebUsb';
@@ -11,6 +12,7 @@ import { releaseButton } from './ReleaseButton';
export default {
dfuError,
firmwareMismatch,
flashProgress,
noDfuHub,
noDfuInterface,
noWebUsb,
src/firmware/sagas.test.ts
+210
View File
@@ -136,6 +136,16 @@ describe('flashFirmware', () => {
// erase first
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{ action: 'erase', progress: undefined },
'firmware.ble.progress',
),
);
action = await saga.take();
expect(action).toEqual(alertsShowAlert('firmware', 'releaseButton'));
@@ -170,6 +180,19 @@ describe('flashFirmware', () => {
action = await saga.take();
expect(action).toEqual(didProgress(offset / totalFirmwareSize));
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: offset / totalFirmwareSize,
},
'firmware.ble.progress',
),
);
// Have to be careful that a checksum request is not sent after
// last payload is sent, otherwise the hub gets confused.
@@ -194,6 +217,19 @@ describe('flashFirmware', () => {
action = await saga.take();
expect(action).toEqual(didProgress(1));
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: 1,
},
'firmware.ble.progress',
),
);
// and finally reboot the hub
action = await saga.take();
@@ -264,6 +300,16 @@ describe('flashFirmware', () => {
// erase first
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{ action: 'erase', progress: undefined },
'firmware.ble.progress',
),
);
action = await saga.take();
expect(action).toEqual(alertsShowAlert('firmware', 'releaseButton'));
@@ -298,6 +344,19 @@ describe('flashFirmware', () => {
action = await saga.take();
expect(action).toEqual(didProgress(offset / totalFirmwareSize));
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: offset / totalFirmwareSize,
},
'firmware.ble.progress',
),
);
// Have to be careful that a checksum request is not sent after
// last payload is sent, otherwise the hub gets confused.
@@ -322,6 +381,19 @@ describe('flashFirmware', () => {
action = await saga.take();
expect(action).toEqual(didProgress(1));
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: 1,
},
'firmware.ble.progress',
),
);
// and finally reboot the hub
action = await saga.take();
@@ -936,6 +1008,16 @@ describe('flashFirmware', () => {
// erase first
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{ action: 'erase', progress: undefined },
'firmware.ble.progress',
),
);
action = await saga.take();
expect(action).toEqual(alertsShowAlert('firmware', 'releaseButton'));
@@ -1035,6 +1117,16 @@ describe('flashFirmware', () => {
// erase first
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{ action: 'erase', progress: undefined },
'firmware.ble.progress',
),
);
action = await saga.take();
expect(action).toEqual(alertsShowAlert('firmware', 'releaseButton'));
@@ -1143,6 +1235,16 @@ describe('flashFirmware', () => {
// erase first
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{ action: 'erase', progress: undefined },
'firmware.ble.progress',
),
);
action = await saga.take();
expect(action).toEqual(alertsShowAlert('firmware', 'releaseButton'));
@@ -1177,6 +1279,19 @@ describe('flashFirmware', () => {
action = await saga.take();
expect(action).toEqual(didProgress(offset / totalFirmwareSize));
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: offset / totalFirmwareSize,
},
'firmware.ble.progress',
),
);
// Have to be careful that a checksum request is not sent after
// last payload is sent, otherwise the hub gets confused.
@@ -1291,6 +1406,16 @@ describe('flashFirmware', () => {
// erase first
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{ action: 'erase', progress: undefined },
'firmware.ble.progress',
),
);
action = await saga.take();
expect(action).toEqual(alertsShowAlert('firmware', 'releaseButton'));
@@ -1325,6 +1450,19 @@ describe('flashFirmware', () => {
action = await saga.take();
expect(action).toEqual(didProgress(offset / totalFirmwareSize));
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: offset / totalFirmwareSize,
},
'firmware.ble.progress',
),
);
// Have to be careful that a checksum request is not sent after
// last payload is sent, otherwise the hub gets confused.
@@ -1444,6 +1582,16 @@ describe('flashFirmware', () => {
// erase first
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{ action: 'erase', progress: undefined },
'firmware.ble.progress',
),
);
action = await saga.take();
expect(action).toEqual(alertsShowAlert('firmware', 'releaseButton'));
@@ -1479,6 +1627,19 @@ describe('flashFirmware', () => {
action = await saga.take();
expect(action).toEqual(didProgress(offset / totalFirmwareSize));
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: offset / totalFirmwareSize,
},
'firmware.ble.progress',
),
);
// Have to be careful that a checksum request is not sent after
// last payload is sent, otherwise the hub gets confused.
@@ -1503,6 +1664,19 @@ describe('flashFirmware', () => {
action = await saga.take();
expect(action).toEqual(didProgress(1));
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: 1,
},
'firmware.ble.progress',
),
);
// and finally reboot the hub
action = await saga.take();
@@ -1956,6 +2130,16 @@ describe('flashFirmware', () => {
// erase first
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{ action: 'erase', progress: undefined },
'firmware.ble.progress',
),
);
action = await saga.take();
expect(action).toEqual(alertsShowAlert('firmware', 'releaseButton'));
@@ -1989,6 +2173,19 @@ describe('flashFirmware', () => {
action = await saga.take();
expect(action).toEqual(didProgress(offset / totalFirmwareSize));
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: offset / totalFirmwareSize,
},
'firmware.ble.progress',
),
);
// Have to be careful that a checksum request is not sent after
// last payload is sent, otherwise the hub gets confused.
@@ -2012,6 +2209,19 @@ describe('flashFirmware', () => {
action = await saga.take();
expect(action).toEqual(didProgress(1));
action = await saga.take();
expect(action).toEqual(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: 1,
},
'firmware.ble.progress',
),
);
// and finally reboot the hub
action = await saga.take();
src/firmware/sagas.ts
+91 -45
View File
@@ -74,7 +74,6 @@ import {
firmwareInstallPybricks,
flashFirmware,
} from './actions';
import { flashProgress } from './alerts/FlashProgress';
import {
firmwareInstallPybricksDialogAccept,
firmwareInstallPybricksDialogCancel,
@@ -87,6 +86,8 @@ const firmwareZipMap = new Map<HubType, string>([
[HubType.MoveHub, moveHubZip],
]);
const firmwareBleProgressToastId = 'firmware.ble.progress';
/**
* Disconnects the BLE if we are connected and cancels the task (including the
* parent task).
@@ -95,7 +96,7 @@ function* disconnectAndCancel(): SagaGenerator<void> {
const toaster = (yield* getContext<ToasterRef>('toasterRef')).current;
defined(toaster);
toaster.dismiss('firmware.ble.progress');
toaster.dismiss(firmwareBleProgressToastId);
const connection = yield* select((s: RootState) => s.bootloader.connection);
@@ -376,9 +377,6 @@ function* loadFirmware(
* @param action The action that triggered this saga.
*/
function* handleFlashFirmware(action: ReturnType<typeof flashFirmware>): Generator {
const toaster = (yield* getContext<ToasterRef>('toasterRef')).current;
defined(toaster);
try {
let firmware: Uint8Array | undefined = undefined;
let deviceId: HubType | undefined = undefined;
@@ -436,12 +434,16 @@ function* handleFlashFirmware(action: ReturnType<typeof flashFirmware>): Generat
yield* put(didStart());
toaster.show(
flashProgress(() => undefined, {
action: 'erase',
progress: undefined,
}),
'firmware.ble.progress',
yield* put(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'erase',
progress: undefined,
},
firmwareBleProgressToastId,
),
);
yield* put(alertsShowAlert('firmware', 'releaseButton'));
@@ -497,12 +499,16 @@ function* handleFlashFirmware(action: ReturnType<typeof flashFirmware>): Generat
yield* put(didProgress(offset / firmware.length));
toaster.show(
flashProgress(() => undefined, {
action: 'flash',
progress: offset / firmware.length,
}),
'firmware.ble.progress',
yield* put(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: offset / firmware.length,
},
firmwareBleProgressToastId,
),
);
// we don't want to request checksum if this is the last packet since
@@ -578,12 +584,16 @@ function* handleFlashFirmware(action: ReturnType<typeof flashFirmware>): Generat
yield* put(didProgress(1));
toaster.show(
flashProgress(() => undefined, {
action: 'flash',
progress: 1,
}),
'firmware.ble.progress',
yield* put(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: 1,
},
firmwareBleProgressToastId,
),
);
// this will cause the remote device to disconnect and reboot
@@ -609,6 +619,40 @@ const dfuFirmwareStartAddress = 0x08008000;
const firmwareDfuProgressToastId = 'firmware.dfu.progress';
function* handleDfuEraseProcess(event: {
bytesSent: number;
expectedSize: number;
}): Generator {
yield* put(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'erase',
progress: event.bytesSent / event.expectedSize,
},
firmwareDfuProgressToastId,
),
);
}
function* handleDfuWriteProcess(event: {
bytesSent: number;
expectedSize: number;
}): Generator {
yield* put(
alertsShowAlert(
'firmware',
'flashProgress',
{
action: 'flash',
progress: event.bytesSent / event.expectedSize,
},
firmwareDfuProgressToastId,
),
);
}
function* handleFlashUsbDfu(action: ReturnType<typeof firmwareFlashUsbDfu>): Generator {
const defer = new Array<() => void>();
@@ -708,29 +752,31 @@ function* handleFlashUsbDfu(action: ReturnType<typeof firmwareFlashUsbDfu>): Gen
const toaster = (yield* getContext<ToasterRef>('toasterRef')).current;
defined(toaster);
defer.push(
writeProc.events.on('erase/process', (sent, total) => {
toaster.show(
flashProgress(() => undefined, {
action: 'erase',
progress: sent / total,
}),
firmwareDfuProgressToastId,
);
}),
);
const eraseProcessChan = eventChannel<{
bytesSent: number;
expectedSize: number;
}>((emit) => {
return writeProc.events.on('erase/process', (bytesSent, expectedSize) =>
emit({ bytesSent, expectedSize }),
);
});
defer.push(
writeProc.events.on('write/process', (sent, total) => {
toaster.show(
flashProgress(() => undefined, {
action: 'flash',
progress: sent / total,
}),
firmwareDfuProgressToastId,
);
}),
);
defer.push(() => eraseProcessChan.close());
yield* takeEvery(eraseProcessChan, handleDfuEraseProcess);
const writeProcessChan = eventChannel<{
bytesSent: number;
expectedSize: number;
}>((emit) => {
return writeProc.events.on('write/process', (bytesSent, expectedSize) =>
emit({ bytesSent, expectedSize }),
);
});
defer.push(() => writeProcessChan.close());
yield* takeEvery(writeProcessChan, handleDfuWriteProcess);
const endChan = eventChannel<boolean>((emit) => {
// can't emit null or undefined, so have to emit something