// Copyright 2008 Cyrus Najmabadi
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#import "UnknownFieldSet.h"

#import "CodedInputStream.h"
#import "CodedOutputStream.h"
#import "Field.h"
#import "UnknownFieldSet_Builder.h"

@interface PBUnknownFieldSet()
@property (retain) NSDictionary* fields;
@end


@implementation PBUnknownFieldSet

static PBUnknownFieldSet* defaultInstance = nil;

+ (void) initialize {
  if (self == [PBUnknownFieldSet class]) {
    defaultInstance = [PBUnknownFieldSet setWithFields:[NSMutableDictionary dictionary]];
  }
}


@synthesize fields;

- (void) dealloc {
  self.fields = nil;
}


+ (PBUnknownFieldSet*) defaultInstance {
  return defaultInstance;
}


- (id) initWithFields:(NSMutableDictionary*) fields_ {
  if ((self = [super init])) {
    self.fields = fields_;
  }

  return self;
}


+ (PBUnknownFieldSet*) setWithFields:(NSMutableDictionary*) fields {
  return [[PBUnknownFieldSet alloc] initWithFields:fields];
}


- (BOOL) hasField:(int32_t) number {
  return [fields objectForKey:[NSNumber numberWithInt:number]] != nil;
}


- (PBField*) getField:(int32_t) number {
  PBField* result = [fields objectForKey:[NSNumber numberWithInt:number]];
  return (result == nil) ? [PBField defaultInstance] : result;
}


- (void) writeToCodedOutputStream:(PBCodedOutputStream*) output {
  NSArray* sortedKeys = [fields.allKeys sortedArrayUsingSelector:@selector(compare:)];
  for (NSNumber* number in sortedKeys) {
    PBField* value = [fields objectForKey:number];
    [value writeTo:number.intValue output:output];
  }
}


- (void) writeToOutputStream:(NSOutputStream*) output {
  PBCodedOutputStream* codedOutput = [PBCodedOutputStream streamWithOutputStream:output];
  [self writeToCodedOutputStream:codedOutput];
  [codedOutput flush];
}


+ (PBUnknownFieldSet*) parseFromCodedInputStream:(PBCodedInputStream*) input {
  return [[[PBUnknownFieldSet builder] mergeFromCodedInputStream:input] build];
}


+ (PBUnknownFieldSet*) parseFromData:(NSData*) data {
  return [[[PBUnknownFieldSet builder] mergeFromData:data] build];
}


+ (PBUnknownFieldSet*) parseFromInputStream:(NSInputStream*) input {
  return [[[PBUnknownFieldSet builder] mergeFromInputStream:input] build];
}


+ (PBUnknownFieldSet_Builder*) builder {
  return [[PBUnknownFieldSet_Builder alloc] init];
}


+ (PBUnknownFieldSet_Builder*) builderWithUnknownFields:(PBUnknownFieldSet*) copyFrom {
  return [[PBUnknownFieldSet builder] mergeUnknownFields:copyFrom];
}


/** Get the number of bytes required to encode this set. */
- (int32_t) serializedSize {
  int32_t result = 0;
  for (NSNumber* number in fields) {
    result += [[fields objectForKey:number] getSerializedSize:number.intValue];
  }
  return result;
}

/**
 * Serializes the set and writes it to {@code output} using
 * {@code MessageSet} wire format.
 */
- (void) writeAsMessageSetTo:(PBCodedOutputStream*) output {
  for (NSNumber* number in fields) {
    [[fields objectForKey:number] writeAsMessageSetExtensionTo:number.intValue output:output];
  }
}


/**
 * Get the number of bytes required to encode this set using
 * {@code MessageSet} wire format.
 */
- (int32_t) serializedSizeAsMessageSet {
  int32_t result = 0;
  for (NSNumber* number in fields) {
    result += [[fields objectForKey:number] getSerializedSizeAsMessageSetExtension:number.intValue];
  }
  return result;
}


/**
 * Serializes the message to a {@code ByteString} and returns it. This is
 * just a trivial wrapper around {@link #writeTo(PBCodedOutputStream)}.
 */
- (NSData*) data {
  NSMutableData* data = [NSMutableData dataWithLength:(NSUInteger)self.serializedSize];
  PBCodedOutputStream* output = [PBCodedOutputStream streamWithData:data];

  [self writeToCodedOutputStream:output];
  return data;
}

@end