/*
  Copyright (C) 2000-2003 SKYRIX Software AG

  This file is part of OGo

  OGo is free software; you can redistribute it and/or modify it under
  the terms of the GNU Lesser General Public License as published by the
  Free Software Foundation; either version 2, or (at your option) any
  later version.

  OGo is distributed in the hope that it will be useful, but WITHOUT ANY
  WARRANTY; without even the implied warranty of MERCHANTABILITY or
  FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  License for more details.

  You should have received a copy of the GNU Lesser General Public
  License along with OGo; see the file COPYING.  If not, write to the
  Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  02111-1307, USA.
*/
// $Id$

#include "common.h"
#include "LSGenericSearchRecord.h"
#include "LSExtendedSearch.h"
#include <GDLAccess/EOSQLQualifier.h>

@implementation LSExtendedSearch

+ (void)initialize {
  NSAssert2([super version] == 1,
            @"invalid superclass (%@) version %i !",
            NSStringFromClass([self superclass]), [super version]);
}

- (id)init {
  if ((self = [super init])) {
    self->operator = @"AND";
  }
  return self;
} 

- (id)initWithSearchRecord:(LSGenericSearchRecord *)_searchRecord
  andRelatedRecords:(NSArray *)_relatedRecords
{
  if ((self = [self init])) {
    self->searchRecord   = RETAIN(_searchRecord);
    self->relatedRecords = RETAIN(_relatedRecords);
  }
  return self;
}

#if !LIB_FOUNDATION_BOEHM_GC
- (void)dealloc {
  RELEASE(self->operator);
  RELEASE(self->searchRecord);
  RELEASE(self->relatedRecords);
  [super dealloc];
}
#endif

- (NSString *)_qualifierFormatForRecord:(LSGenericSearchRecord *)_record {
  NSMutableString *format;
  NSEnumerator    *keys;
  id              key;

  format = [NSMutableString stringWithCapacity:32];
  keys   = [[_record searchDict] keyEnumerator];
  
  [self setComparator:[_record comparator]];
  
  while ((key = [keys nextObject])) {
    EOAttribute *attr;
    id          value;
    NSString    *extType;

    attr    = [[_record entity]     attributeNamed:key];
    value   = [[_record searchDict] valueForKey:key];
    extType = [[attr externalType]  lowercaseString];

    NSAssert2(attr, @"missing attribute for key %@ in entity %@",
              key, [_record entity]);
    
    value = [value stringValue];
    
    if ([value length] > 0) {
      if ([format length] > 0) {
        [format appendString:@" "];
        [format appendString:self->operator];
        [format appendString:@" "];
      }
      if ([[attr valueClassName] isEqualToString:@"NSNumber"] ||
          [[attr valueClassName] isEqualToString:@"NSCalendarDate"]) {
        [format appendString:
                [self _formatForNumberAttribute:attr andValue:value]];
      }
      else if ([[attr valueClassName] isEqualToString:@"NSString"] &&
                 [extType hasSuffix:@"text"]) {
        [format appendString:[self _formatForTextAttribute:attr andValue:value]];
      }
      else if ([[attr valueClassName] isEqualToString:@"NSString"] &&
                 ![extType hasSuffix:@"text"]) {
        [format appendString:
                [self _formatForStringAttribute:attr andValue:value]];
      }
    }
  }
  return format;
}

- (NSString *)_qualifierFormat {
  NSMutableString *format   = nil;
  NSEnumerator    *listEnum = nil;
  id              record    = nil;
  NSString        *tmp      = nil;
  NSString        *startQ   = nil;
  int             startQLength = 0;

  startQ       = @"( ";
  startQLength = [startQ length];
  
  format = [NSMutableString stringWithCapacity:64];

  [format appendString:startQ];
  [format appendString:[self _qualifierFormatForRecord:self->searchRecord]];

  listEnum = [self->relatedRecords objectEnumerator];
  
  while ((record = [listEnum nextObject])) {
    tmp  = [self _qualifierFormatForRecord:record];
    if ([tmp length] > 0) {
      if ([format length] > startQLength) {
        [format appendString:@" "];
        [format appendString:self->operator];
        [format appendString:@" "];
      }
      [format appendString:tmp];
    }
  }
  if ([format length] == startQLength)
    [format appendString:@"1=2"];
  [format appendString:@" )"];
  return format;
}

- (EOSQLQualifier *)qualifier {
  EOSQLQualifier *qualifier = nil;

  qualifier = [[EOSQLQualifier alloc] initWithEntity:[self entity]
                                      qualifierFormat:[self _qualifierFormat]];
  [qualifier setUsesDistinct:YES];

  return AUTORELEASE(qualifier);
}

- (EOEntity *)entity {
  return [self->searchRecord entity];
}

// accessors

- (void)setOperator:(NSString *)_operator {
  NSMutableString *string = [[NSMutableString allocWithZone:[self zone]]
                                       initWithString:@" "];
  [string appendString:_operator];
  [string appendString:@" "];
  if (self->operator != nil) {
    RELEASE(self->operator); self->operator = nil;
  }
  self->operator = string;
}
- (NSString *)operator {
  return self->operator;
}

- (void)setSearchRecord:(LSGenericSearchRecord *)_searchRecord {
  ASSIGN(searchRecord, _searchRecord);
}
- (LSGenericSearchRecord *)searchRecord {
  return self->searchRecord;
}

- (void)setRelatedRecords:(NSArray *)_relatedRecords {
  ASSIGN(self->relatedRecords, _relatedRecords);
}
- (NSArray *)relatedRecords {
  return self->relatedRecords;
}

@end