balena-cli/lib/utils/visuals/drive-list.ts
Pagan Gazzard 98152c0b09 Lazy-load chalk
Change-type: patch
2020-02-28 18:34:54 +00:00

36 lines
957 B
TypeScript

import * as _sdk from 'etcher-sdk';
import { getChalk } from '../lazy';
import { CustomDynamicList } from './custom-dynamic-list';
export class DriveList extends CustomDynamicList<
_sdk.sourceDestination.BlockDevice
> {
constructor(private scanner: _sdk.scanner.Scanner) {
super(
'Select a drive',
`${getChalk().red('x')} No available drives were detected, plug one in!`,
);
const refresh = this.refresh.bind(this);
scanner.on('attach', refresh);
scanner.on('detach', refresh);
}
protected *getThings() {
const sdk: typeof _sdk = require('etcher-sdk');
for (const drive of this.scanner.drives) {
if (drive instanceof sdk.sourceDestination.BlockDevice) {
yield drive;
}
}
}
protected format(drive: _sdk.sourceDestination.BlockDevice) {
const size =
drive.size != null
? `${(drive.size / 1e9).toFixed(1).toString()} GB`
: 'Unknown size';
return `${drive.device} (${size}) - ${drive.description}`;
}
}