AwfulForum.m

//
//  AwfulForum.m
//  Awful
//
//  Created by Nolan Waite on 12-05-17.
//  Copyright (c) 2012 Regular Berry Software LLC. All rights reserved.
//

#import "AwfulForum.h"
#import "AwfulCategory.h"
#import "AwfulDataStack.h"
#import "AwfulThread.h"
#import "NSManagedObject+Awful.h"

@implementation AwfulForum

+ (NSArray *)updateCategoriesAndForums:(ForumHierarchyParsedInfo *)info
{
    NSMutableDictionary *existingForums = [NSMutableDictionary new];
    for (AwfulForum *f in [AwfulForum fetchAll]) {
        existingForums[f.forumID] = f;
    }
    NSMutableDictionary *existingCategories = [NSMutableDictionary new];
    for (AwfulCategory *c in [AwfulCategory fetchAll]) {
        existingCategories[c.categoryID] = c;
    }
    NSMutableArray *allForums = [NSMutableArray new];
    int indexOfCategory = 0;
    int indexOfForum = 0;
    for (CategoryParsedInfo *categoryInfo in info.categories) {
        AwfulCategory *category = existingCategories[categoryInfo.categoryID];
        if (!category) category = [AwfulCategory insertNew];
        category.categoryID = categoryInfo.categoryID;
        category.name = categoryInfo.name;
        category.indexValue = indexOfCategory++;
        NSMutableArray *forumStack = [categoryInfo.forums mutableCopy];
        while ([forumStack count] > 0) {
            ForumParsedInfo *forumInfo = [forumStack objectAtIndex:0];
            [forumStack removeObjectAtIndex:0];
            AwfulForum *forum = existingForums[forumInfo.forumID] ?: [AwfulForum insertNew];
            forum.forumID = forumInfo.forumID;
            forum.name = forumInfo.name;
            forum.category = category;
            forum.indexValue = indexOfForum++;
            if (forumInfo.parentForum) {
                forum.parentForum = existingForums[forumInfo.parentForum.forumID];
            }
            [allForums addObject:forum];
            existingForums[forum.forumID] = forum;
            NSRange start = NSMakeRange(0, [forumInfo.subforums count]);
            [forumStack insertObjects:forumInfo.subforums
                            atIndexes:[NSIndexSet indexSetWithIndexesInRange:start]];
        }
    }
    
    if ([info.categories count] > 0) {
        NSArray *keep = [info.categories valueForKey:@"categoryID"];
        [AwfulCategory deleteAllMatchingPredicate:@"NOT (categoryID IN %@)", keep];
    }
    
    if ([allForums count] > 0) {
        NSArray *keep = [allForums valueForKey:AwfulForumAttributes.forumID];
        [AwfulForum deleteAllMatchingPredicate:@"NOT (forumID IN %@)", keep];
    }
    
    [[AwfulDataStack sharedDataStack] save];
    return [allForums count] > 0 ? allForums : [existingForums allValues];
}

+ (NSArray *)updateCategoriesAndForumsWithJSON:(NSArray *)json
{
    NSMutableArray *categories = [NSMutableArray new];
    NSMutableArray *forums = [NSMutableArray new];
    for (NSDictionary *categoryInfo in json) {
        NSString *categoryID = [categoryInfo[@"id"] stringValue];
        AwfulCategory *category = [AwfulCategory firstMatchingPredicate:@"categoryID = %@",
                                   categoryID];
        if (!category) {
            category = [AwfulCategory insertNew];
            category.categoryID = categoryID;
        }
        [category.forumsSet removeAllObjects];
        category.name = categoryInfo[@"title"];
        for (NSDictionary *forumInfo in categoryInfo[@"sub_forums"]) {
            [category.forumsSet addObject:ParseForumIntoCategoryAndArray(forumInfo, category, forums)];
        }
        [categories addObject:category];
    }
    [categories enumerateObjectsWithOptions:NSEnumerationConcurrent
                                 usingBlock:^(AwfulCategory *category, NSUInteger i, BOOL *stop)
    {
        category.indexValue = i;
    }];
    [forums enumerateObjectsWithOptions:NSEnumerationConcurrent
                             usingBlock:^(AwfulForum *forum, NSUInteger i, BOOL *stop)
    {
        forum.indexValue = i;
    }];
    [AwfulCategory deleteAllMatchingPredicate:@"NOT (self in %@)", categories];
    [AwfulForum deleteAllMatchingPredicate:@"NOT (self in %@)", forums];
    [[AwfulDataStack sharedDataStack] save];
    return forums;
}

static AwfulForum * ParseForumIntoCategoryAndArray(NSDictionary *forumInfo,
                                                   AwfulCategory *category,
                                                   NSMutableArray *forums)
{
    NSString *forumID = [forumInfo[@"id"] stringValue];
    AwfulForum *forum = [AwfulForum firstMatchingPredicate:@"forumID = %@", forumID];
    if (!forum) {
        forum = [AwfulForum insertNew];
        forum.forumID = forumID;
    }
    forum.category = category;
    forum.name = forumInfo[@"title"];
    [forums addObject:forum];
    for (NSDictionary *subforumInfo in forumInfo[@"sub_forums"]) {
        [forum.childrenSet addObject:ParseForumIntoCategoryAndArray(subforumInfo, category, forums)];
    }
    return forum;
}

@end