/*
  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 <LSFoundation/LSDBObjectBaseCommand.h>

@class NSArray;

@interface LSProjectsToCompanyAssignmentCommand : LSDBObjectBaseCommand
{
@private 
  NSArray *projects;
  NSArray *removedProjects;
}

@end

#import "common.h"

@implementation LSProjectsToCompanyAssignmentCommand

- (void)dealloc {
  [self->projects        release];
  [self->removedProjects release];
  [super dealloc];
}

/* command methods */

- (BOOL)_object:(id)_object isInList:(NSArray *)_list {
  NSEnumerator *listEnum;
  id           pkey;
  id           listObject;

  listEnum = [_list objectEnumerator];
  pkey     = [_object valueForKey:@"projectId"];
  
  while ((listObject = [listEnum nextObject])) {
    id opkey = [listObject valueForKey:@"projectId"];

    if ([pkey isEqual:opkey]) return YES;
  }
  return NO;
}

- (void)_removeOldAssignmentsInContext:(id)_context {
  NSEnumerator *pEnum;
  id           p;

  pEnum = [self->removedProjects objectEnumerator];
  while ((p = [pEnum nextObject]) != nil) {
    NSEnumerator *asEnum;
    NSArray      *assignments;
    id           as; 
    
    assignments = [p valueForKey:@"companyAssignments"];
    asEnum = [assignments objectEnumerator];
    
    while ((as = [asEnum nextObject]) != nil) {
      if ([[[self object] valueForKey:@"companyId"]
                  isEqual:[as valueForKey:@"companyId"]]) {
        LSRunCommandV(_context,        @"projectcompanyassignment",  @"delete",
                      @"object",       as,
                      @"reallyDelete", [NSNumber numberWithBool:YES], nil);
      }
    }
  }
}

- (void)_saveAssignmentsInContext:(id)_context {
  NSArray      *oldAssignments;
  NSEnumerator *pEnum          = nil;
  id           p               = nil; 
  id           obj             = [self object];

  LSRunCommandV(_context, [[self entityName] lowercaseString],
                @"get-project-assignments",
                @"object",      obj,
                @"relationKey", @"projectAssignments", nil);

  oldAssignments = [obj valueForKey:@"projectAssignments"];

  //[self logWithFormat:@"oldAssignments: %@", oldAssignments];

  pEnum = [self->projects objectEnumerator];

  while ((p = [pEnum nextObject]) != nil) {
    if (![self _object:p isInList:oldAssignments]) {
      LSRunCommandV(_context,     @"projectcompanyassignment",  @"new",
                    @"companyId", [obj valueForKey:@"companyId"],
                    @"projectId", [p valueForKey:@"projectId"],
                    @"hasAccess", [NSNumber numberWithBool:NO], nil);
    }
  }
}

- (void)_executeInContext:(id)_context {
  if (self->removedProjects) {
    [self _removeOldAssignmentsInContext:_context];
  }
  if (self->projects) {
    [self _saveAssignmentsInContext:_context];
  }
}

/* accessors */

- (void)setProjects:(NSArray *)_projects {
  ASSIGN(self->projects, _projects);
}
- (NSArray *)projects {
  return self->projects;
}

- (void)setRemovedProjects:(NSArray *)_removedProjects {
  ASSIGN(self->removedProjects, _removedProjects);
}
- (NSArray *)removedProjects {
  return self->removedProjects;
}

/* key/value coding */

- (void)takeValue:(id)_value forKey:(NSString *)_key {
  if ([_key isEqualToString:@"projects"]) {
    [self setProjects:_value];
    return;
  }
  if ([_key isEqualToString:@"removedProjects"]) {
    [self setRemovedProjects:_value];
    return;
  }
  [super takeValue:_value forKey:_key];
}

- (id)valueForKey:(NSString *)_key {
  if ([_key isEqualToString:@"removedProjects"])
    return [self removedProjects];
  return [super valueForKey:_key];
}

@end /* LSProjectsToCompanyAssignmentCommand */