//

 //  BLIPMessage.m

 //  MYNetwork

 //

 //  Created by Jens Alfke on 5/10/08.

 //  Copyright 2008 Jens Alfke. All rights reserved.

 //

 #import "BLIPMessage.h"

 #import "BLIP_Internal.h"

 #import "BLIPReader.h"

 #import "BLIPWriter.h"

 #import "Logging.h"

 #import "Test.h"

 #import "ExceptionUtils.h"

 #import "Target.h"

 // From Google Toolbox For Mac <http://code.google.com/p/google-toolbox-for-mac/>

 #import "GTMNSData+zlib.h"

 @implementation BLIPMessage

 - (id) _initWithConnection: (BLIPConnection*)connection

                     isMine: (BOOL)isMine

                      flags: (BLIPMessageFlags)flags

                     number: (UInt32)msgNo

                       body: (NSData*)body

 {

     self = [super init];

     if (self != nil) {

         _connection = [connection retain];

         _isMine = isMine;

         _flags = flags;

         _number = msgNo;

         if( isMine ) {

             _body = body.copy;

             _properties = [[BLIPMutableProperties alloc] init];

             _propertiesAvailable = YES;

         } else {

             _encodedBody = body.mutableCopy;

         }

         LogTo(BLIPVerbose,@"INIT %@",self);

     }

     return self;

 }

 - (void) dealloc

 {

     LogTo(BLIPVerbose,@"DEALLOC %@",self);

     [_properties release];

     [_encodedBody release];

     [_mutableBody release];

     [_body release];

     [_connection release];

     [super dealloc];

 }

 - (NSString*) description

 {

     NSUInteger length = (_body.length ?: _mutableBody.length) ?: _encodedBody.length;

     NSMutableString *desc = [NSMutableString stringWithFormat: @"%@[#%u, %u bytes",

                              self.class,_number, length];

     if( _flags & kBLIP_Compressed ) {

         if( _encodedBody && _encodedBody.length != length )

             [desc appendFormat: @" (%u gzipped)", _encodedBody.length];

         else

             [desc appendString: @", gzipped"];

     }

     if( _flags & kBLIP_Urgent )

         [desc appendString: @", urgent"];

     if( _flags & kBLIP_NoReply )

         [desc appendString: @", noreply"];

     [desc appendString: @"]"];

     return desc;

 }

 - (NSString*) descriptionWithProperties

 {

     NSMutableString *desc = (NSMutableString*)self.description;

     [desc appendFormat: @" %@", self.properties.allProperties];

     return desc;

 }

 #pragma mark -

 #pragma mark PROPERTIES & METADATA:

 @synthesize connection=_connection, number=_number, isMine=_isMine, isMutable=_isMutable,

             _bytesWritten, sent=_sent, propertiesAvailable=_propertiesAvailable, complete=_complete;

 - (void) _setFlag: (BLIPMessageFlags)flag value: (BOOL)value

 {

     Assert(_isMine && _isMutable);

     if( value )

         _flags |= flag;

     else

         _flags &= ~flag;

 }

 - (BOOL) compressed                         {return (_flags & kBLIP_Compressed) != 0;}

 - (BOOL) urgent                             {return (_flags & kBLIP_Urgent) != 0;}

 - (void) setCompressed: (BOOL)compressed    {[self _setFlag: kBLIP_Compressed value: compressed];}

 - (void) setUrgent: (BOOL)high              {[self _setFlag: kBLIP_Urgent value: high];}

 - (NSData*) body

 {

     if( ! _body && _isMine )

         return [[_mutableBody copy] autorelease];

     else

         return _body;

 }

 - (void) setBody: (NSData*)body

 {

     Assert(_isMine && _isMutable);

     if( _mutableBody )

         [_mutableBody setData: body];

     else

         _mutableBody = [body mutableCopy];

 }

 - (void) _addToBody: (NSData*)data

 {

     if( data.length ) {

         if( _mutableBody )

             [_mutableBody appendData: data];

         else

             _mutableBody = [data mutableCopy];

         setObj(&_body,nil);

     }

 }

 - (void) addToBody: (NSData*)data

 {

     Assert(_isMine && _isMutable);

     [self _addToBody: data];

 }

 - (NSString*) bodyString

 {

     NSData *body = self.body;

     if( body )

         return [[[NSString alloc] initWithData: body encoding: NSUTF8StringEncoding] autorelease];

     else

         return nil;

 }

 - (void) setBodyString: (NSString*)string

 {

     self.body = [string dataUsingEncoding: NSUTF8StringEncoding];

     self.contentType = @"text/plain; charset=UTF-8";

 }

 - (BLIPProperties*) properties

 {

     return _properties;

 }

 - (BLIPMutableProperties*) mutableProperties

 {

     Assert(_isMine && _isMutable);

     return (BLIPMutableProperties*)_properties;

 }

 - (NSString*) valueOfProperty: (NSString*)property

 {

     return [_properties valueOfProperty: property];

 }

 - (void) setValue: (NSString*)value ofProperty: (NSString*)property

 {

     [self.mutableProperties setValue: value ofProperty: property];

 }

 - (NSString*) contentType               {return [_properties valueOfProperty: @"Content-Type"];}

 - (void) setContentType: (NSString*)t   {[self setValue: t ofProperty: @"Content-Type"];}

 - (NSString*) profile                   {return [_properties valueOfProperty: @"Profile"];}

 - (void) setProfile: (NSString*)p       {[self setValue: p ofProperty: @"Profile"];}

 #pragma mark -

 #pragma mark I/O:

 - (void) _encode

 {

     Assert(_isMine && _isMutable);

     _isMutable = NO;

     BLIPProperties *oldProps = _properties;

     _properties = [oldProps copy];

     [oldProps release];

     _encodedBody = [_properties.encodedData mutableCopy];

     Assert(_encodedBody.length>=2);

     NSData *body = _body ?: _mutableBody;

     NSUInteger length = body.length;

     if( length > 0 ) {

         if( self.compressed ) {

             body = [NSData gtm_dataByGzippingData: body compressionLevel: 5];

             LogTo(BLIPVerbose,@"Compressed %@ to %u bytes (%.0f%%)", self,body.length,

                   body.length*100.0/length);

         }

         [_encodedBody appendData: body];

     }

 }

 - (void) _assignedNumber: (UInt32)number

 {

     Assert(_number==0,@"%@ has already been sent",self);

     _number = number;

     _isMutable = NO;

 }

 - (BOOL) _writeFrameTo: (BLIPWriter*)writer maxSize: (UInt16)maxSize

 {

     Assert(_number!=0);

     Assert(_isMine);

     Assert(_encodedBody);

     if( _bytesWritten==0 )

         LogTo(BLIP,@"Now sending %@",self);

     ssize_t lengthToWrite = _encodedBody.length - _bytesWritten;

     if( lengthToWrite <= 0 && _bytesWritten > 0 )

         return NO; // done

     Assert(maxSize > sizeof(BLIPFrameHeader));

     maxSize -= sizeof(BLIPFrameHeader);

     UInt16 flags = _flags;

     if( lengthToWrite > maxSize ) {

         lengthToWrite = maxSize;

         flags |= kBLIP_MoreComing;

         LogTo(BLIPVerbose,@"%@ pushing frame, bytes %u-%u", self, _bytesWritten, _bytesWritten+lengthToWrite);

     } else {

         flags &= ~kBLIP_MoreComing;

         LogTo(BLIPVerbose,@"%@ pushing frame, bytes %u-%u (finished)", self, _bytesWritten, _bytesWritten+lengthToWrite);

     }

     // First write the frame header:

     BLIPFrameHeader header = {  EndianU32_NtoB(kBLIPFrameHeaderMagicNumber),

                                 EndianU32_NtoB(_number),

                                 EndianU16_NtoB(flags),

                                 EndianU16_NtoB(sizeof(BLIPFrameHeader) + lengthToWrite) };

     [writer writeData: [NSData dataWithBytes: &header length: sizeof(header)]];

     // Then write the body:

     if( lengthToWrite > 0 ) {

         [writer writeData: [NSData dataWithBytesNoCopy: (UInt8*)_encodedBody.bytes + _bytesWritten

                                               length: lengthToWrite

                                         freeWhenDone: NO]];

         _bytesWritten += lengthToWrite;

     }

     return (flags & kBLIP_MoreComing) != 0;

 }

 - (BOOL) _receivedFrameWithHeader: (const BLIPFrameHeader*)header body: (NSData*)body

 {

     Assert(!_isMine);

     AssertEq(header->number,_number);

     Assert(_flags & kBLIP_MoreComing);

     BLIPMessageType frameType = (header->flags & kBLIP_TypeMask), curType = (_flags & kBLIP_TypeMask);

     if( frameType != curType ) {

         Assert(curType==kBLIP_RPY && frameType==kBLIP_ERR && _mutableBody.length==0,

                @"Incoming frame's type %i doesn't match %@",frameType,self);

         _flags = (_flags & ~kBLIP_TypeMask) | frameType;

     }

     if( _encodedBody )

         [_encodedBody appendData: body];

     else

         _encodedBody = [body mutableCopy];

     LogTo(BLIPVerbose,@"%@ rcvd bytes %u-%u", self, _encodedBody.length-body.length, _encodedBody.length);

     if( ! _properties ) {

         // Try to extract the properties:

         ssize_t usedLength;

         setObj(&_properties, [BLIPProperties propertiesWithEncodedData: _encodedBody usedLength: &usedLength]);

         if( _properties ) {

             [_encodedBody replaceBytesInRange: NSMakeRange(0,usedLength)

                                     withBytes: NULL length: 0];

         } else if( usedLength < 0 )

             return NO;

         self.propertiesAvailable = YES;

     }

     if( ! (header->flags & kBLIP_MoreComing) ) {

         // After last frame, decode the data:

         _flags &= ~kBLIP_MoreComing;

         if( ! _properties )

             return NO;

         unsigned encodedLength = _encodedBody.length;

         if( self.compressed && encodedLength>0 ) {

             _body = [[NSData gtm_dataByInflatingData: _encodedBody] copy];

             if( ! _body )

                 return NO;

             LogTo(BLIPVerbose,@"Uncompressed %@ from %u bytes (%.1fx)", self, encodedLength,

                   _body.length/(float)encodedLength);

         } else {

             _body = [_encodedBody copy];

         }

         setObj(&_encodedBody,nil);

         self.propertiesAvailable = self.complete = YES;

     }

     return YES;

 }

 - (void) _connectionClosed

 {

     if( _isMine ) {

         _bytesWritten = 0;

         _flags |= kBLIP_MoreComing;

     }

 }

 @end

 /*

  Copyright (c) 2008, Jens Alfke <jens@mooseyard.com>. All rights reserved.

  Redistribution and use in source and binary forms, with or without modification, are permitted

  provided that the following conditions are met:

  * Redistributions of source code must retain the above copyright notice, this list of conditions

  and the following disclaimer.

  * Redistributions in binary form must reproduce the above copyright notice, this list of conditions

  and the following disclaimer in the documentation and/or other materials provided with the

  distribution.

  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR

  IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND 

  FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRI-

  BUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES

  (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR 

   PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 

  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF 

  THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

  */