//
//  WPCategoryTree.m
//  WordPress
//
//  Created by JanakiRam on 30/01/09.

#import "WPCategoryTree.h"

@implementation WPCategoryTree

- (id)initWithParent:(id)aParent {
    if (self = [super init]) {
        parent = [aParent copy];
        children = [[NSMutableArray alloc] init];
    }

    return self;
}

- (void)getChildrenFromObjects:(NSArray *)collection {
    int i, count = [collection count];

    for (i = 0; i < count; i++) {
        NSDictionary *category = [collection objectAtIndex:i];

        if ([[category valueForKey:@"parentId"] intValue] ==[[parent valueForKey:@"categoryId"] intValue]) {
            WPCategoryTree *child = [[WPCategoryTree alloc] initWithParent:category];
            [child getChildrenFromObjects:collection];
            [children addObject:child];
            [child release];
        }
    }
}

- (void)dealloc {
    [parent release];
    [children release];
    [super dealloc];
}

- (NSArray *)getAllObjects {
    NSMutableArray *allObjects = [NSMutableArray array];
    int i, count = [children count];

    if (parent)
        [allObjects addObject:parent];

    for (i = 0; i < count; i++) {
        [allObjects addObjectsFromArray:[[children objectAtIndex:i] getAllObjects]];
    }

    return allObjects;
}

@end