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

  This file is part of OpenGroupware.org.

  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.
*/

#include <OGoFoundation/OGoComponent.h>

@class NSData, NSArray, NSMutableArray, NSDictionary, EOQualifier;

@interface LSWPersonAdvancedSearch : OGoComponent
{
@private
  NSString            *maxSearchCount;
  NSArray             *extendedAttributeKeys;
  NSString            *currentTeleType;  // non-retained
  NSMutableDictionary *person;           // LSGenericSearchRecord
  id                  item;              // non-retained
  BOOL                hasSearched;
  NSData              *formletterData;
  EOQualifier         *qualifier;
  NSString            *qualifierOperator;

  NSString            *udKey; // userDefaultKey
  NSString            *searchTitle;
  NSString            *saveTitle;
  BOOL                showTab;
}

@end

#include "common.h"
#include <NGExtensions/NSString+Ext.h>
#include <OGoContacts/SkyAddressConverterDataSource.h>
#include <OGoContacts/SkyPersonDataSource.h>

@interface LSWPersonAdvancedSearch(PrivateMethodes)
- (NSArray *)extendedPersonAttributeKeys;
- (void)_createQualifier;
- (NSMutableDictionary *)_createSearchInfo;
- (void)_setSearchFields:(NSDictionary *)_fields;
- (void)setMaxSearchCount:(NSString *)_maxSearchCount;
- (NSString *)maxSearchCount;
- (id)search;
@end

@implementation LSWPersonAdvancedSearch

static NSArray  *SkyPublicExtendedPersonAttributes = nil;
static NSString *KeywordSeparator = @", ";

+ (void)initialize {
  NSUserDefaults *ud = [NSUserDefaults standardUserDefaults];
  
  SkyPublicExtendedPersonAttributes = 
    [[ud arrayForKey:@"SkyPublicExtendedPersonAttributes"] copy];
}

- (NSArray *)_extractKeysFromAttributes:(NSArray *)_attrs {
  // TODO: looks like we could just use -valueForKey:@"key"?
  NSMutableArray *keys;
  int i, cnt;

  cnt  = [_attrs count];
  keys = [NSMutableArray arrayWithCapacity:cnt];
  for (i = 0; i < cnt; i++) {
    [keys addObject:
            [(NSDictionary *)[_attrs objectAtIndex:i] objectForKey:@"key"]];
  }
  return [keys sortedArrayUsingSelector:@selector(compare:)];
}

- (id)init {
  if ((self = [super init])) {
    
    self->extendedAttributeKeys = 
      [[self _extractKeysFromAttributes:SkyPublicExtendedPersonAttributes]
             copy];
    
    self->person = [[NSMutableDictionary alloc] initWithCapacity:16];
  }
  return self;
}

- (void)dealloc {
  [self->qualifierOperator     release];
  [self->maxSearchCount        release];
  [self->extendedAttributeKeys release];
  [self->person                release];
  [self->formletterData        release];
  [self->qualifier             release];
  [self->udKey                 release];
  [self->searchTitle           release];
  [self->saveTitle             release];
  [super dealloc];
}

/* accessors */

- (void)setPerson:(NSMutableDictionary *)_person {
  ASSIGN(self->person, _person);
}
- (NSMutableDictionary *)person {
  return self->person;
}

// TODO: dup to LSWEnterpriseAdvancedSearch, this is a hack used in Logic
- (void)setKeywordsAsArray:(NSArray *)_a {
  [[self person] takeValue:[_a componentsJoinedByString:KeywordSeparator] 
		 forKey:@"keywords"];
}
- (NSArray *)keywordsAsArray {
  NSString *s;
  
  if (![(s = [[self person] valueForKey:@"keywords"]) isNotNull])
    return nil;
  return [s componentsSeparatedByString:KeywordSeparator];
}

- (void)setQualifier:(EOQualifier *)_qualifier {
}

- (EOQualifier *)qualifier {
  return self->qualifier;
}

- (NSString *)currentTeleTypeLabel {
  NSString *s;
  s = [self->currentTeleType stringByAppendingString:@"_search"];
  return [[self labels] valueForKey:s];
}

- (NSArray *)currentValues {
  return self->extendedAttributeKeys;
}

- (NSArray *)currentLabel {
  NSString *label = [[self labels] valueForKey:self->item];
  return (label == nil) ? self->item : label;
}

- (BOOL)hasSearched {
  return self->hasSearched;
}
- (void)setHasSearched:(BOOL)_searched {
  self->hasSearched = _searched;
}

- (void)setItem:(id)_item {
  self->item = _item;
}
- (id)item {
  return self->item;
}

- (void)setCurrentTeleType:(NSString *)_teleType {
  self->currentTeleType = _teleType;
}
- (NSString *)currentTeleType {
  return self->currentTeleType;
}

- (NSString *)maxSearchCount {
  return self->maxSearchCount;
}
- (void)setMaxSearchCount:(NSString *)_maxSearchCount {
  ASSIGN(self->maxSearchCount, _maxSearchCount);
}

/* actions */

- (NSMutableDictionary *)_createSearchInfo {
  NSMutableDictionary *dict;
  NSArray             *keys;
  int                 i, cnt;

  keys = [self->person allKeys];
  cnt  = [keys count];
  dict = [NSMutableDictionary dictionaryWithCapacity:cnt+1];
  
  for (i = 0; i < cnt; i++) {
    NSArray  *components;
    NSString *key;
    id       value;
    
    key        = [keys objectAtIndex:i];
    value      = [self->person objectForKey:key];
    
    if (![value isNotNull] || [value length] == 0)
      continue;

    components = [key componentsSeparatedByString:@"#"];
    key        = [components componentsJoinedByString:@"."];

    [dict setObject:value forKey:key];
  }

  return dict;
}

- (void)_setSearchFields:(NSDictionary *)_fields {
  NSArray             *keys;
  int                 i, cnt;

  keys = [_fields allKeys];
  cnt  = [keys count];

  // clear form
  [self->person removeAllObjects];
  [self takeValue:nil forKey:@"companyValueAttribute"];
  [self takeValue:nil forKey:@"companyValueValue"];
  
  for (i = 0; i < cnt; i++) {
    NSArray  *components;
    NSString *key;
    id       value;
    
    key        = [keys objectAtIndex:i];
    value      = [_fields objectForKey:key];
    
    components = [key componentsSeparatedByString:@"."];
    key        = [components componentsJoinedByString:@"#"];

    if ([key isEqualToString:@"companyValueValue"]) 
      [self takeValue:value forKey:@"companyValueValue"];
    else if ([key isEqualToString:@"companyValueAttribute"]) 
      [self takeValue:value forKey:@"companyValueAttribute"];
    
    else if ([value isNotNull] && [value length] != 0)
      [self->person setObject:value forKey:key];
  }
}

- (void)setQualifierOperator:(NSString *)_op {
  ASSIGNCOPY(self->qualifierOperator, _op);
}
- (NSString *)qualifierOperator {
  return [self->qualifierOperator isNotNull] 
    ? self->qualifierOperator : @"AND";
}

- (void)_createQualifier {
  // DUP: LSWEnterpriseAdvancedSearch
  NSMutableDictionary *dict;
  NSArray             *keys;
  int                 i, cnt;
  NSMutableArray      *qualifiers;
  
  dict = [self _createSearchInfo];
  
  if ([self valueForKey:@"companyValueAttribute"] &&
      [self valueForKey:@"companyValueValue"]) {
    [dict setObject:[self valueForKey:@"companyValueValue"]
          forKey:[self valueForKey:@"companyValueAttribute"]];
  }

  qualifiers = [[NSMutableArray alloc] initWithCapacity:4];
  keys       = [dict allKeys];
  
  for (i = 0, cnt = [keys count]; i < cnt; i++) {
    EOQualifier *q;
    NSString *key;
    NSString *value;
    
    key   = [keys objectAtIndex:i];
    value = [[dict objectForKey:key] stringValue];
    
    /* special keywords processing */
    
    if ([key isEqualToString:@"keywords"]) {
      if ([value rangeOfString:KeywordSeparator].length > 0) {
        NSArray  *keywords;
        unsigned j, jcnt;
        
        keywords = [value componentsSeparatedByString:KeywordSeparator];
        for (j = 0, jcnt = [keywords count]; j < jcnt; j++) {
          // Note: datasource only supports like with a suffix-star
          q = [[EOKeyValueQualifier alloc]
                initWithKey:@"keywords"
                operatorSelector:EOQualifierOperatorLike
                value:
                  [[keywords objectAtIndex:j] stringByAppendingString:@"*"]];
          if (q != nil) [qualifiers addObject:q];
          [q release]; q = nil;
        }
        continue;
      }
    }
    
    /* create LIKE qualifiers for fields */
    
    if ([value rangeOfString:@"%"].length > 0) 
      value = [value stringByReplacingString:@"%" withString:@"*"];
    if (![(NSString *)value hasSuffix:@"*"])
      value = [value stringByAppendingString:@"*"];
    
    q = [[EOKeyValueQualifier alloc]
          initWithKey:key
          operatorSelector:EOQualifierOperatorLike
          value:value];
    if (q != nil) [qualifiers addObject:q];
    [q release]; q = nil;
  }
  
  [self->qualifier release]; self->qualifier = nil;
  
  self->qualifier = [[self qualifierOperator] isEqualToString:@"OR"]
    ? [[EOOrQualifier alloc]  initWithQualifierArray:qualifiers]
    : [[EOAndQualifier alloc] initWithQualifierArray:qualifiers];
  [qualifiers release]; qualifiers = nil;
}

- (id)search {
  /* creates qualifier an asks LSWPersons page to display results */
  [self _createQualifier];
  [self->formletterData release]; self->formletterData = nil;
  return [self performParentAction:@"advancedSearch"];
}

- (NSString *)defaultFormLetterKind {
  return [[[self session] userDefaults] objectForKey:@"formletter_kind"];
}

- (id)formletter {
  SkyAddressConverterDataSource *ds;
  SkyPersonDataSource           *sds;
  EOFetchSpecification          *fspec;
  NSDictionary                  *hints;
  NSString                      *kind;
  LSCommandContext              *ctx;
  
  [self _createQualifier];
  
  kind  = [self defaultFormLetterKind];
  fspec = [[EOFetchSpecification alloc] init];
  hints = [[NSDictionary alloc] initWithObjectsAndKeys:kind, @"kind", nil];
  ctx   = [(OGoSession *)[self session] commandContext];
  
  sds   =
    [(SkyPersonDataSource *)[SkyPersonDataSource alloc] initWithContext:ctx];
  ds    = [[SkyAddressConverterDataSource alloc] 
	    initWithDataSource:sds context:ctx labels:[self labels]];
  [sds release]; sds = nil;
  
  [fspec setQualifier:self->qualifier];
  [fspec setFetchLimit:[self->maxSearchCount intValue]];
  [fspec setHints:hints];
  [ds setFetchSpecification:fspec];
  [fspec release]; fspec = nil;
  [hints release]; hints = nil;
  
  /* fetch */
  
  [self->formletterData release]; self->formletterData = nil;
  self->formletterData = [[[ds fetchObjects] lastObject] retain];
  [ds release]; ds = nil;
  return nil;
}

- (id)clearForm {
  [self->person removeAllObjects];
  [self _createQualifier];
  return nil;
}

- (BOOL)hasFormletter {
  return (self->formletterData != nil) ? YES : NO;
}

- (void)setFormletterData:(id)_data {
  ASSIGN(self->formletterData, _data);
}
- (id)formletterData {
  return self->formletterData;
}

/* SavedSearches */

- (void)setUserDefaultKey:(NSString *)_key {
  ASSIGN(self->udKey,_key);
}
- (NSString *)userDefaultKey {
  return self->udKey;
}

- (void)setSearchTitle:(NSString *)_title {
  ASSIGN(self->searchTitle,_title);
}
- (NSString *)searchTitle {
  return self->searchTitle;
}

- (void)setSaveTitle:(NSString *)_title {
  ASSIGN(self->saveTitle,_title);
}
- (NSString *)saveTitle {
  return self->saveTitle;
}

- (void)setShowTab:(BOOL)_flag {
  self->showTab = _flag;
}
- (BOOL)showTab {
  return self->showTab;  
}

- (NSArray *)savedSearches {
  return [[[[[self session] userDefaults] objectForKey:[self userDefaultKey]]
                   allKeys] sortedArrayUsingSelector:@selector(compare:)];
}
- (BOOL)hasSavedSearches {
  return [[self savedSearches] count] ? YES : NO;
}

- (id)saveSearch {
  NSUserDefaults      *ud;
  NSMutableDictionary *settings;
  NSMutableDictionary *info;
  EOQualifier  *qual;

  if (![self->saveTitle length]) {
    [[self parent] setErrorString:
                   [[self labels] valueForKey:@"error_setSearchTitle"]];
    return nil;
  }

  [self _createQualifier];
  info = [self _createSearchInfo];
  {
    id tmp;
    
    if ((tmp = [self valueForKey:@"companyValueAttribute"]))
      [info setObject:tmp forKey:@"companyValueAttribute"];
    if ((tmp = [self valueForKey:@"companyValueValue"]))
      [info setObject:tmp forKey:@"companyValueValue"];
  }
  qual = self->qualifier;

  info = [NSDictionary dictionaryWithObjectsAndKeys:
                       info,                  @"searchFields",
                       [qual description],    @"qualifier",
                       [self maxSearchCount], @"maxSearchCount",
                       [NSNumber numberWithBool:[self showTab]], @"showTab",
                       nil];

  ud       = [[self session] userDefaults];
  settings = [ud objectForKey:[self userDefaultKey]];
  if (settings == nil) {
    settings = [NSDictionary dictionaryWithObjectsAndKeys:
                             info, self->saveTitle, nil];
  }
  else {
    settings = [[settings mutableCopy] autorelease];
    [settings setObject:info forKey:self->saveTitle];
  }
  [ud setObject:settings forKey:[self userDefaultKey]];
  [ud synchronize];
  [self setSaveTitle:@""];
  [self setShowTab:NO];

  return nil;
}

- (id)saveAndSearch {
  [self saveSearch];
  return [self search];
}

- (id)loadSavedSearch {
  NSDictionary   *settings;
  NSDictionary   *info;
  NSUserDefaults *ud;

  if (![self->searchTitle length]) {
    [self _setSearchFields:nil];
    [self setSaveTitle:@""];
    [self setShowTab:NO];
    return nil;
  }

  ud       = [[self session] userDefaults];
  settings = [ud objectForKey:[self userDefaultKey]];
  info     = [settings objectForKey:self->searchTitle];

  if (info == nil) {
    NSString *s;

    s = [[self labels] valueForKey:@"error_cannotLoadSavedSearch"];
    [[self parent] setErrorString:s];
    return nil;
  }

  [self _setSearchFields:[info objectForKey:@"searchFields"]];
  [self setMaxSearchCount:[info objectForKey:@"maxSearchCount"]];
  [self setShowTab:[[info objectForKey:@"showTab"] boolValue]];
  [self setSaveTitle:self->searchTitle];

  return nil;
}

@end /* LSWPersonAdvancedSearch(SavedSearches) */