ALApplicationTableDataSource.m

#import "ALApplicationTableDataSource.h"

#import "ALApplicationList.h"

#import <UIKit/UIKit2.h>
#import <CoreGraphics/CoreGraphics.h>

const NSString *ALSectionDescriptorTitleKey = @"title";
const NSString *ALSectionDescriptorFooterTitleKey = @"footer-title";
const NSString *ALSectionDescriptorPredicateKey = @"predicate";
const NSString *ALSectionDescriptorCellClassNameKey = @"cell-class-name";
const NSString *ALSectionDescriptorIconSizeKey = @"icon-size";
const NSString *ALSectionDescriptorItemsKey = @"items";
const NSString *ALSectionDescriptorSuppressHiddenAppsKey = @"suppress-hidden-apps";
const NSString *ALSectionDescriptorVisibilityPredicateKey = @"visibility-predicate";

const NSString *ALItemDescriptorTextKey = @"text";
const NSString *ALItemDescriptorDetailTextKey = @"detail-text";
const NSString *ALItemDescriptorImageKey = @"image";

static NSInteger DictionaryTextComparator(id a, id b, void *context)
{
	return [[(NSDictionary *)context objectForKey:a] localizedCaseInsensitiveCompare:[(NSDictionary *)context objectForKey:b]];
}

@implementation ALApplicationTableDataSource

static NSArray *hiddenDisplayIdentifiers;

+ (void)initialize
{
	if ((self == [ALApplicationTableDataSource class])) {
		hiddenDisplayIdentifiers = [[NSArray alloc] initWithObjects:
		                            @"com.apple.AdSheet",
		                            @"com.apple.AdSheetPhone",
		                            @"com.apple.AdSheetPad",
		                            @"com.apple.DataActivation",
		                            @"com.apple.DemoApp",
		                            @"com.apple.fieldtest",
		                            @"com.apple.iosdiagnostics",
		                            @"com.apple.iphoneos.iPodOut",
		                            @"com.apple.TrustMe",
		                            @"com.apple.WebSheet",
		                            @"com.apple.springboard",
                                            @"com.apple.purplebuddy",
		                            nil];
	}
}

+ (NSArray *)standardSectionDescriptors
{
	NSNumber *iconSize = [NSNumber numberWithUnsignedInteger:ALApplicationIconSizeSmall];
	return [NSArray arrayWithObjects:
		[NSDictionary dictionaryWithObjectsAndKeys:
			@"System Applications", ALSectionDescriptorTitleKey,
			@"isSystemApplication = TRUE", ALSectionDescriptorPredicateKey,
			@"UITableViewCell", ALSectionDescriptorCellClassNameKey,
			iconSize, ALSectionDescriptorIconSizeKey,
			(id)kCFBooleanTrue, ALSectionDescriptorSuppressHiddenAppsKey,
		nil],
		[NSDictionary dictionaryWithObjectsAndKeys:
			@"User Applications", ALSectionDescriptorTitleKey,
			@"isSystemApplication = FALSE", ALSectionDescriptorPredicateKey,
			@"UITableViewCell", ALSectionDescriptorCellClassNameKey,
			iconSize, ALSectionDescriptorIconSizeKey,
			(id)kCFBooleanTrue, ALSectionDescriptorSuppressHiddenAppsKey,
		nil],
	nil];
}

+ (id)dataSource
{
	return [[[self alloc] init] autorelease];
}

- (id)init
{
	if ((self = [super init])) {
		appList = [[ALApplicationList sharedApplicationList] retain];
		_displayIdentifiers = [[NSMutableArray alloc] init];
		_displayNames = [[NSMutableArray alloc] init];
		_defaultImage = [[appList iconOfSize:ALApplicationIconSizeSmall forDisplayIdentifier:@"com.apple.WebSheet"] retain];
		[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(iconLoadedFromNotification:) name:ALIconLoadedNotification object:nil];
	}
	return self;
}

- (void)dealloc
{
	[[NSNotificationCenter defaultCenter] removeObserver:self];
	[_localizationBundle release];
	[_tableView release];
	[_displayIdentifiers release];
	[_displayNames release];
	[_defaultImage release];
	[appList release];
	[super dealloc];
}

@synthesize sectionDescriptors = _sectionDescriptors;
@synthesize tableView = _tableView;
@synthesize localizationBundle = _localizationBundle;

- (void)_insertSectionDescriptor:(NSDictionary *)descriptor atIndex:(NSInteger)index
{
	NSArray *items = [descriptor objectForKey:@"items"];
	if (items) {
		[_displayIdentifiers insertObject:items atIndex:index];
		[_displayNames insertObject:[NSNull null] atIndex:index];
	} else {
		NSString *predicateText = [descriptor objectForKey:ALSectionDescriptorPredicateKey];
		NSDictionary *applications;
		if (predicateText)
			applications = [appList applicationsFilteredUsingPredicate:[NSPredicate predicateWithFormat:predicateText]];
		else
			applications = [appList applications];
		NSMutableArray *displayIdentifiers = [[applications allKeys] mutableCopy];
		if ([[descriptor objectForKey:ALSectionDescriptorSuppressHiddenAppsKey] boolValue]) {
			for (NSString *displayIdentifier in hiddenDisplayIdentifiers)
				[displayIdentifiers removeObject:displayIdentifier];
		}
		[displayIdentifiers sortUsingFunction:DictionaryTextComparator context:applications];
		[_displayIdentifiers insertObject:displayIdentifiers atIndex:index];
		[displayIdentifiers release];
		NSMutableArray *displayNames = [[NSMutableArray alloc] init];
		for (NSString *displayId in displayIdentifiers)
			[displayNames addObject:[applications objectForKey:displayId]];
		[_displayNames insertObject:displayNames atIndex:index];
		[displayNames release];
	}
}

- (void)setSectionDescriptors:(NSArray *)sectionDescriptors
{
	[_displayIdentifiers removeAllObjects];
	[_displayNames removeAllObjects];
	NSInteger i = 0;
	for (NSDictionary *descriptor in sectionDescriptors) {
		NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
		[self _insertSectionDescriptor:descriptor atIndex:i];
		[pool release];
		i++;
	}
	[_sectionDescriptors release];
	_sectionDescriptors = [sectionDescriptors mutableCopy];
	[_tableView reloadData];
}

- (void)removeSectionDescriptorsAtIndexes:(NSIndexSet *)indexSet
{
	[_sectionDescriptors removeObjectsAtIndexes:indexSet];
	[_displayIdentifiers removeObjectsAtIndexes:indexSet];
	[_displayNames removeObjectsAtIndexes:indexSet];
	[_tableView deleteSections:indexSet withRowAnimation:UITableViewRowAnimationFade];
}

- (void)removeSectionDescriptorAtIndex:(NSInteger)index
{
	[self removeSectionDescriptorsAtIndexes:[NSIndexSet indexSetWithIndex:index]];
}

- (void)insertSectionDescriptor:(NSDictionary *)sectionDescriptor atIndex:(NSInteger)index
{
	[self _insertSectionDescriptor:sectionDescriptor atIndex:index];
	[_sectionDescriptors insertObject:sectionDescriptor atIndex:index];
	[_tableView insertSections:[NSIndexSet indexSetWithIndex:index] withRowAnimation:UITableViewRowAnimationFade];
}

- (void)setLocalizationBundle:(NSBundle *)localizationBundle
{
	if (_localizationBundle != localizationBundle) {
		[_localizationBundle autorelease];
		_localizationBundle = [localizationBundle retain];
		[_tableView reloadData];
	}
}

static inline NSString *Localize(NSBundle *bundle, NSString *string)
{
	return bundle ? [bundle localizedStringForKey:string value:string table:nil] : string;
}
#define Localize(string) Localize(_localizationBundle, string)

- (id)cellDescriptorForIndexPath:(NSIndexPath *)indexPath;
{
	return [[_displayIdentifiers objectAtIndex:[indexPath section]] objectAtIndex:[indexPath row]];
}

- (NSString *)displayIdentifierForIndexPath:(NSIndexPath *)indexPath
{
	id result = [[_displayIdentifiers objectAtIndex:[indexPath section]] objectAtIndex:[indexPath row]];
	return [result isKindOfClass:[NSString class]] ? result : nil;
}

- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView
{
	if (!_tableView) {
		_tableView = [tableView retain];
		NSLog(@"ALApplicationTableDataSource warning: Assumed control over %@", tableView);
	}
	return [_displayIdentifiers count];
}

- (NSString *)tableView:(UITableView *)tableView titleForHeaderInSection:(NSInteger)section
{
	return Localize([[_sectionDescriptors objectAtIndex:section] objectForKey:ALSectionDescriptorTitleKey]);
}

- (NSString *)tableView:(UITableView *)tableView titleForFooterInSection:(NSInteger)section
{
	return Localize([[_sectionDescriptors objectAtIndex:section] objectForKey:ALSectionDescriptorFooterTitleKey]);
}

- (NSInteger)tableView:(UITableView *)table numberOfRowsInSection:(NSInteger)section
{
	return [[_displayIdentifiers objectAtIndex:section] count];
}

- (void)loadIconsFromBackground
{
	NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];
	OSSpinLockLock(&spinLock);
	while ([_iconsToLoad count]) {
		NSDictionary *userInfo = [[_iconsToLoad objectAtIndex:0] retain];
		[_iconsToLoad removeObjectAtIndex:0];
		OSSpinLockUnlock(&spinLock);
		CGImageRelease([appList copyIconOfSize:[[userInfo objectForKey:ALIconSizeKey] integerValue] forDisplayIdentifier:[userInfo objectForKey:ALDisplayIdentifierKey]]);
		[userInfo release];
		[pool drain];
		pool = [[NSAutoreleasePool alloc] init];
		OSSpinLockLock(&spinLock);
	}
	[_iconsToLoad release];
	_iconsToLoad = nil;
	OSSpinLockUnlock(&spinLock);
	[pool drain];
}

static inline UITableViewCell *CellWithClassName(NSString *className, UITableView *tableView)
{
	return [tableView dequeueReusableCellWithIdentifier:className] ?: [[[NSClassFromString(className) alloc] initWithStyle:UITableViewCellStyleDefault reuseIdentifier:className] autorelease];
}

#define CellWithClassName(className) \
	CellWithClassName(className, tableView)

- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath
{
	NSUInteger section = [indexPath section];
	NSUInteger row = [indexPath row];
	id displayNames = [_displayNames objectAtIndex:section];
	NSDictionary *sectionDescriptor = [_sectionDescriptors objectAtIndex:section];
	if (displayNames == [NSNull null]) {
		NSDictionary *itemDescriptor = [[_displayIdentifiers objectAtIndex:section] objectAtIndex:row];
		UITableViewCell *cell = CellWithClassName([itemDescriptor objectForKey:ALSectionDescriptorCellClassNameKey] ?: [sectionDescriptor objectForKey:ALSectionDescriptorCellClassNameKey] ?: @"UITableViewCell");
		cell.textLabel.text = Localize([itemDescriptor objectForKey:ALItemDescriptorTextKey]);
		cell.detailTextLabel.text = Localize([itemDescriptor objectForKey:ALItemDescriptorDetailTextKey]);
		NSString *imagePath = [itemDescriptor objectForKey:ALItemDescriptorImageKey];
		UIImage *image = nil;
		if (imagePath) {
			CGFloat scale;
			if ([UIScreen instancesRespondToSelector:@selector(scale)] && ((scale = [[UIScreen mainScreen] scale]) != 1.0f))
				image = [UIImage imageWithContentsOfFile:[NSString stringWithFormat:@"%@@%gx.%@", [imagePath stringByDeletingPathExtension], scale, [imagePath pathExtension]]];
			if (!image)
				image = [UIImage imageWithContentsOfFile:imagePath];
		}
		cell.imageView.image = image;
		return cell;
	} else {
		UITableViewCell *cell = CellWithClassName([sectionDescriptor objectForKey:ALSectionDescriptorCellClassNameKey] ?: @"UITableViewCell");
		cell.textLabel.text = [displayNames objectAtIndex:row];
		CGFloat iconSize = [[sectionDescriptor objectForKey:ALSectionDescriptorIconSizeKey] floatValue];
		if (iconSize > 0) {
			NSString *displayIdentifier = [[_displayIdentifiers objectAtIndex:section] objectAtIndex:row];
			if ([appList hasCachedIconOfSize:iconSize forDisplayIdentifier:displayIdentifier]) {
				cell.imageView.image = [appList iconOfSize:iconSize forDisplayIdentifier:displayIdentifier];
				cell.indentationWidth = 10.0f;
				cell.indentationLevel = 0;
			} else {
				if (_defaultImage.size.width == iconSize) {
					cell.imageView.image = _defaultImage;
					cell.indentationWidth = 10.0f;
					cell.indentationLevel = 0;
				} else {
					cell.indentationWidth = iconSize + 7.0f;
					cell.indentationLevel = 1;
					cell.imageView.image = nil;
				}
				cell.imageView.image = _defaultImage;
				NSDictionary *userInfo = [NSDictionary dictionaryWithObjectsAndKeys:
				                          [NSNumber numberWithInteger:iconSize], ALIconSizeKey,
				                          displayIdentifier, ALDisplayIdentifierKey,
				                          nil];
				OSSpinLockLock(&spinLock);
				if (_iconsToLoad)
					[_iconsToLoad insertObject:userInfo atIndex:0];
				else {
					_iconsToLoad = [[NSMutableArray alloc] initWithObjects:userInfo, nil];
					[self performSelectorInBackground:@selector(loadIconsFromBackground) withObject:nil];
				}
				OSSpinLockUnlock(&spinLock);
			}
		} else {
			cell.imageView.image = nil;
		}
		return cell;
	}
}

- (void)iconLoadedFromNotification:(NSNotification *)notification
{
	NSDictionary *userInfo = notification.userInfo;
	NSString *displayIdentifier = [userInfo objectForKey:ALDisplayIdentifierKey];
	for (NSIndexPath *indexPath in _tableView.indexPathsForVisibleRows) {
		NSInteger section = indexPath.section;
		NSString *rowDisplayIdentifier = [[_displayIdentifiers objectAtIndex:section] objectAtIndex:indexPath.row];
		if ([rowDisplayIdentifier isEqual:displayIdentifier]) {
			UITableViewCell *cell = [_tableView cellForRowAtIndexPath:indexPath];
			UIImageView *imageView = cell.imageView;
			UIImage *image = imageView.image;
			if (!image || (image == _defaultImage)) {
				NSDictionary *sectionDescriptor = [_sectionDescriptors objectAtIndex:section];
				CGFloat iconSize = [[sectionDescriptor objectForKey:ALSectionDescriptorIconSizeKey] floatValue];
				cell.indentationLevel = 0;
				cell.indentationWidth = 10.0f;
				imageView.image = [appList iconOfSize:iconSize forDisplayIdentifier:displayIdentifier];
				[cell setNeedsLayout];
			}
		}
	}
}

@end