jens@6: //
jens@6: //  MYVersionDictionary.m
jens@6: //  Ottoman
jens@6: //
jens@6: //  Created by Jens Alfke on 9/21/09.
jens@6: //  Copyright 2009 Jens Alfke. All rights reserved.
jens@6: //
jens@6: 
jens@6: #import "MYVersionDictionary.h"
jens@6: #import "MYOttoman_internal.h"
jens@6: #import <Foundation/Foundation.h>
jens@6: 
jens@6: #include "VersionDictionary.h"
jens@6: 
jens@6: using namespace Mooseyard;
jens@6: 
jens@6: 
jens@6: @interface MYDictionaryEnumerator : NSEnumerator
jens@6: {
jens@6:     Dictionary::Iterator *_iter;
jens@6:     BOOL _started;
jens@6: }
jens@6: - (id) initWithDict: (const Dictionary*)dict;
jens@6: @end
jens@6: 
jens@6: 
jens@6: 
jens@6: static id BlobToNSData (const Blob &blob, bool copy =true) {
jens@6:     if (!blob)
jens@6:         return nil;
jens@6:     else if (copy)
jens@6:         return [NSData dataWithBytes: blob.bytes length: blob.length];
jens@6:     else
jens@6:         return [NSData dataWithBytesNoCopy: (void*)blob.bytes
jens@6:                                     length: blob.length
jens@6:                               freeWhenDone: NO];
jens@6: }
jens@6: 
jens@6: static Blob BlobFromId (id object) {
jens@6:     if ([object isKindOfClass: [NSData class]])
jens@6:         return Blob([object bytes], [object length]);
jens@6:     else 
jens@6:         return Key();
jens@6: }
jens@6: 
jens@6: 
jens@6: 
jens@6: @implementation MYVersionDictionary
jens@6: 
jens@6: 
jens@6: - (id) _initWithVersionDictionary: (VersionDictionary*)dict
jens@6: {
jens@6:     self = [super init];
jens@6:     if (self != nil) {
jens@6:         if (!dict) {
jens@6:             [self release];
jens@6:             return nil;
jens@6:         }
jens@6:         _dict = dict;
jens@6:     }
jens@6:     return self;
jens@6: }
jens@6: 
jens@6: 
jens@6: - (NSUInteger)count {
jens@6:     return _dict->count();
jens@6: }
jens@6: 
jens@6: - (id)objectForKey:(id)keyObject {
jens@6:     return BlobToNSData( _dict->get( Key(BlobFromId(keyObject)) ) );
jens@6: }
jens@6: 
jens@6: - (NSEnumerator *)keyEnumerator {
jens@6:     return [[[MYDictionaryEnumerator alloc] initWithDict: _dict] autorelease];
jens@6: }
jens@6: 
jens@6: - (int) generation {
jens@6:     return _dict->generation();
jens@6: }
jens@6: 
jens@6: - (NSDate*) dateCreated {
jens@6:     return [NSDate dateWithTimeIntervalSince1970: _dict->timestamp()];
jens@6: }
jens@6: 
jens@6: - (BOOL) isLatestVersion {
jens@6:     return _dict->isLatestVersion();
jens@6: }
jens@6: 
jens@6: - (MYVersionDictionary*) previousVersion {
jens@6:     const VersionDictionary *prev = _dict->previousVersion();
jens@6:     if (prev)
jens@6:         return [[[MYVersionDictionary alloc] _initWithVersionDictionary: prev] autorelease];
jens@6:     else
jens@6:         return nil;
jens@6: }
jens@6: 
jens@6: @end
jens@6: 
jens@6: 
jens@6: 
jens@6: 
jens@6: @implementation MYCurrentVersionDictionary
jens@6: 
jens@6: 
jens@6: - (id) _initWithOverlayDictionary: (OverlayDictionary*)dict
jens@6: {
jens@6:     self = [super init];
jens@6:     if (self != nil) {
jens@6:         if (!dict) {
jens@6:             [self release];
jens@6:             return nil;
jens@6:         }
jens@6:         _dict = dict;
jens@6:     }
jens@6:     return self;
jens@6: }
jens@6: 
jens@6: 
jens@6: - (NSUInteger)count {
jens@6:     return _dict->count();
jens@6: }
jens@6: 
jens@6: - (id)objectForKey:(id)keyObject {
jens@6:     Blob keyBlob = BlobFromId(keyObject);
jens@6:     return keyBlob ? BlobToNSData( _dict->get( Key(keyBlob) ) ) :nil;
jens@6: }
jens@6: 
jens@6: - (NSEnumerator *)keyEnumerator {
jens@6:     return [[[MYDictionaryEnumerator alloc] initWithDict: _dict] autorelease];
jens@6: }
jens@6: 
jens@6: - (void)setObject:(id)anObject forKey:(id)aKey {
jens@6:     Key key = Key(BlobFromId(aKey));
jens@6:     NSAssert1(key, @"Invalid %@ key; only NSData supported", [aKey class]);
jens@6:     Blob value = BlobFromId(anObject);
jens@6:     NSAssert1(key, @"Invalid %@ value; only NSData supported", [aKey class]);
jens@6:     _dict->put(key, value);
jens@6: }
jens@6: 
jens@6: - (void)removeObjectForKey:(id)aKey {
jens@6:     Key key = Key(BlobFromId(aKey));
jens@6:     NSAssert1(key, @"Invalid key class %@; only NSData supported", [aKey class]);
jens@6:     _dict->remove(key);
jens@6: }
jens@6: 
jens@6: @end
jens@6: 
jens@6: 
jens@6: 
jens@6: 
jens@6: @implementation MYDictionaryEnumerator
jens@6: 
jens@6: - (id) initWithDict: (Dictionary*)dict
jens@6: {
jens@6:     self = [super init];
jens@6:     if (self != nil) {
jens@6:         _iter = dict->iterate();
jens@6:     }
jens@6:     return self;
jens@6: }
jens@6: 
jens@6: - (void) dealloc
jens@6: {
jens@6:     delete _iter;
jens@6:     [super dealloc];
jens@6: }
jens@6: 
jens@6: - (void) finalize {
jens@6:     delete _iter;
jens@6:     [super finalize];
jens@6: }
jens@6: 
jens@6: - (id)nextObject {
jens@6:     if (_started) {
jens@6:         if (!_iter->next())
jens@6:             return nil;
jens@6:     } else {
jens@6:         if (!_iter->hasValue())
jens@6:             return nil;
jens@6:         _started = YES;
jens@6:     }
jens@6:     Key key = _iter->key();
jens@6:     return [NSData dataWithBytes: key.bytes length: key.length];
jens@6: }
jens@6:             
jens@6: 
jens@6: @end