src/darwin/CHIPTool/CHIPTool/View Controllers/DeviceSelector.m

/**
 *
 *    Copyright (c) 2021 Project CHIP Authors
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

#import "DeviceSelector.h"

#import "CHIPUIViewUtils.h"
#import "DefaultsUtils.h"

@interface DeviceSelector ()
@end

@implementation DeviceSelector {
    NSMutableArray * _deviceList;
    UIPickerView * _devicePicker;
    NSUInteger _selectedDeviceIndex;
}

- (id)init
{
    if (self = [super init]) {
        [self refreshDeviceList];
        [self setupView];
        [self setEnabled:YES];
    }
    return self;
}

- (void)refreshDeviceList
{
    uint64_t nextDeviceID = MTRGetNextAvailableDeviceID();
    _deviceList = [NSMutableArray new];
    for (uint64_t i = 0; i < nextDeviceID; i++) {
        if (MTRIsDevicePaired(i)) {
            [_deviceList addObject:[@(i) stringValue]];
        }
    }
    _selectedDeviceIndex = 0;

    // This will refresh the view with the updated device list
    [self setEnabled:self.isEnabled];
}

- (void)forSelectedDevices:(DeviceAction)action
{
    if ([self isEnabled]) {
        if ([_deviceList count] > 0) {
            uint64_t nodeId;
            NSScanner * scanner = [NSScanner scannerWithString:[_deviceList objectAtIndex:_selectedDeviceIndex]];
            [scanner scanUnsignedLongLong:&nodeId];
            action(nodeId);
        }
    } else {
        for (id device in _deviceList) {
            uint64_t nodeId;
            NSScanner * scanner = [NSScanner scannerWithString:device];
            [scanner scanUnsignedLongLong:&nodeId];
            action(nodeId);
        }
    }
}

- (void)setupView
{
    _devicePicker = [[UIPickerView alloc] initWithFrame:CGRectMake(0, 100, 0, 0)];
    self.inputView = _devicePicker;
    [_devicePicker setDataSource:self];
    [_devicePicker setDelegate:self];
    self.delegate = self;

    UIToolbar * deviceSelectButtonView = [[UIToolbar alloc] init];
    [deviceSelectButtonView sizeToFit];
    UIBarButtonItem * deviceSelectButton = [[UIBarButtonItem alloc] initWithTitle:@"Select"
                                                                            style:UIBarButtonItemStylePlain
                                                                           target:self
                                                                           action:@selector(deviceSelectClicked:)];
    UIBarButtonItem * flexible = [[UIBarButtonItem alloc] initWithBarButtonSystemItem:UIBarButtonSystemItemFlexibleSpace
                                                                               target:self
                                                                               action:nil];
    [deviceSelectButtonView setItems:[NSArray arrayWithObjects:flexible, deviceSelectButton, nil]];
    self.inputAccessoryView = deviceSelectButtonView;
}

- (void)setEnabled:(BOOL)enabled
{
    [super setEnabled:enabled];
    if (enabled == NO) {
        self.text = [_deviceList description];
    } else if ([_deviceList count] > 0) {
        self.text = [NSString stringWithFormat:@"%@", [_deviceList objectAtIndex:_selectedDeviceIndex]];
    }
}

// MARK: UIPickerView

- (void)pickerView:(UIPickerView *)pickerView didSelectRow:(NSInteger)row inComponent:(NSInteger)component
{
    if ([_deviceList count] > 0) {
        NSLog(@"%@", [_deviceList objectAtIndex:row]);
        self.text = [NSString stringWithFormat:@"%@", [_deviceList objectAtIndex:row]];
    }
}

- (NSInteger)pickerView:(UIPickerView *)pickerView numberOfRowsInComponent:(NSInteger)component
{
    return [_deviceList count];
}

- (NSInteger)numberOfComponentsInPickerView:(UIPickerView *)pickerView
{
    return 1;
}

- (NSString *)pickerView:(UIPickerView *)pickerView titleForRow:(NSInteger)row forComponent:(NSInteger)component
{
    if ([_deviceList count] > 0) {
        return [_deviceList objectAtIndex:row];
    } else {
        return [NSString new];
    }
}

- (CGFloat)pickerView:(UIPickerView *)pickerView widthForComponent:(NSInteger)component
{
    return 200;
}

- (IBAction)deviceSelectClicked:(id)sender
{
    if ([_deviceList count] > 0) {
        _selectedDeviceIndex = [_deviceList indexOfObject:self.text];
    }
    [self resignFirstResponder];
}

// MARK: MTRDeviceControllerDelegate
- (void)deviceControllerOnConnected
{
    NSLog(@"Status: Device connected");
}

- (void)deviceControllerOnError:(nonnull NSError *)error
{
}

@end