vlc-ios/VLC for iOS WatchKit Extension/InterfaceController.m

//
//  InterfaceController.m
//  VLC for iOS WatchKit Extension
//
//  Created by Carola Nitz on 22/03/15.
//  Copyright (c) 2015 VideoLAN. All rights reserved.
//

#import "InterfaceController.h"
#import <MediaLibraryKit/MediaLibraryKit.h>
#import "VLCRowController.h"
#import <MobileVLCKit/VLCTime.h>
#import "VLCDBChangeNotifier.h"

static NSString *const rowType = @"mediaRow";

@interface InterfaceController()
@property (nonatomic, strong) NSMutableArray *mediaObjects;
@end


@implementation InterfaceController

- (void)awakeWithContext:(id)context {
    [super awakeWithContext:context];
    NSLog(@"%s",__PRETTY_FUNCTION__);

    NSURL *groupURL = [[NSFileManager defaultManager] containerURLForSecurityApplicationGroupIdentifier:@"group.org.videolan.vlc-ios"];

    MLMediaLibrary *mediaLibrary = [MLMediaLibrary sharedMediaLibrary];
    mediaLibrary.libraryBasePath = groupURL.path;
    mediaLibrary.additionalPersitentStoreOptions = @{NSReadOnlyPersistentStoreOption : @YES};

    self.title = NSLocalizedString(@"LIBRARY_MUSIC", nil);
    [[VLCDBChangeNotifier sharedNotifier] addObserver:self block:^{
        [self updateData];
    }];
    [self updateData];
}

- (void)willActivate {
    // This method is called when watch view controller is about to be visible to user
    [super willActivate];
    NSLog(@"%s",__PRETTY_FUNCTION__);
    [[VLCDBChangeNotifier sharedNotifier] addObserver:self block:^{
        [self updateData];
    }];
}

- (void)didDeactivate {
    // This method is called when watch view controller is no longer visible
    [super didDeactivate];
    NSLog(@"%s",__PRETTY_FUNCTION__);

    [[VLCDBChangeNotifier sharedNotifier] removeObserver:self];
}

- (void)updateData {
    NSArray *oldObjects = self.mediaObjects;
    NSSet *oldSet = [[NSSet alloc] initWithArray:oldObjects];
    NSMutableArray *newObjects = [self mediaArray];
    NSMutableSet *newSet = [[NSMutableSet alloc] initWithArray:newObjects];

    WKInterfaceTable *table = self.table;

    [oldObjects enumerateObjectsUsingBlock:^(id obj, NSUInteger idx, BOOL *stop) {
        if (![newSet containsObject:obj]) {
            [table removeRowsAtIndexes:[NSIndexSet indexSetWithIndex:idx]];
        }
    }];
    [newObjects enumerateObjectsUsingBlock:^(id obj, NSUInteger idx, BOOL *stop) {
        if (![oldSet containsObject:obj]) {
            [table insertRowsAtIndexes:[NSIndexSet indexSetWithIndex:idx] withRowType:rowType];
        }
        [self configureTableCellAtIndex:idx withObject:obj];
    }];

    self.mediaObjects = newObjects;
}

- (void)configureTableCellAtIndex:(NSUInteger)index withObject:(MLFile *)object {
    VLCRowController *row = [self.table rowControllerAtIndex:index];
    row.titleLabel.text = object.title;
    row.durationLabel.text = [VLCTime timeWithNumber:object.duration].stringValue,
    [row.group setBackgroundImage:object.computedThumbnail];
}


- (NSMutableArray *)mediaArray {
    NSMutableArray *objects = [NSMutableArray array];
//    /* add all albums */
//    NSArray *rawAlbums = [MLAlbum allAlbums];
//    for (MLAlbum *album in rawAlbums) {
//        if (album.name.length > 0 && album.tracks.count > 1)
//            [objects addObject:album];
//    }
//
//    /* add all shows */
//    NSArray *rawShows = [MLShow allShows];
//    for (MLShow *show in rawShows) {
//        if (show.name.length > 0 && show.episodes.count > 1)
//            [objects addObject:show];
//    }
//
//    /* add all folders*/
//    NSArray *allFolders = [MLLabel allLabels];
//    for (MLLabel *folder in allFolders)
//        [objects addObject:folder];

    /* add all remaining files */
    NSArray *allFiles = [MLFile allFiles];
    for (MLFile *file in allFiles) {
        if (file.labels.count > 0) continue;

        if (!file.isShowEpisode && !file.isAlbumTrack)
            [objects addObject:file];
        else if (file.isShowEpisode) {
            if (file.showEpisode.show.episodes.count < 2)
                [objects addObject:file];
        } else if (file.isAlbumTrack) {
            if (file.albumTrack.album.tracks.count < 2)
                [objects addObject:file];
        }
    }
    return objects;
}

@end