/* MDKQueryManager.m
 *  
 * Copyright (C) 2006 Free Software Foundation, Inc.
 *
 * Author: Enrico Sersale <enrico@dtedu.net>
 * Date: October 2006
 *
 * This file is part of the GNUstep GWorkspace application
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 * 
 * This program 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 General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02111 USA.
 */

#include <AppKit/AppKit.h>
#include "MDKQueryManager.h"
#include "MDKFSFilter.h"
#include "FSNode.h"
#include "config.h"

#define GWDebugLog(format, args...) \
  do { if (GW_DEBUG_LOG) \
    NSLog(format , ## args); } while (0)

static MDKQueryManager *queryManager = nil;


@protocol	GMDSClientProtocol

- (BOOL)queryResults:(NSData *)results;

- (oneway void)endOfQueryWithNumber:(NSNumber *)qnum;

@end


@protocol	GMDSProtocol

- (oneway void)registerClient:(id)remote;

- (oneway void)unregisterClient:(id)remote;

- (oneway void)performQuery:(NSDictionary *)queryInfo;

@end


@implementation MDKQueryManager

+ (MDKQueryManager *)queryManager
{
  if (queryManager == nil) {
    queryManager = [MDKQueryManager new];
  }
  return queryManager;
}

- (void)dealloc
{  
  [dnc removeObserver: self];
  [nc removeObserver: self];
  RELEASE (queries);
  RELEASE (liveQueries);
  
  [super dealloc];
}

- (id)init
{
  self = [super init];
  
  if (self) {
    queries = [NSMutableArray new];
    liveQueries = [NSMutableArray new];
  
    tableNumber = 0L;
    queryNumber = 0L;
    gmds = nil;
    nc = [NSNotificationCenter defaultCenter];
    dnc = [NSDistributedNotificationCenter defaultCenter];

    [dnc addObserver: self
            selector: @selector(metadataDidUpdate:)
	              name: @"GWMetadataDidUpdateNotification"
	            object: nil];
  }
  
  return self;
}

- (BOOL)startQuery:(MDKQuery *)query
{
  if ([query isRoot] == NO) {
    [NSException raise: NSInvalidArgumentException
	              format: @"\"%@\" is not the root query.", [query description]];
  }
  
  if ([queries containsObject: query]) { 
    [NSException raise: NSInvalidArgumentException
	              format: @"\"%@\" is already started.", [query description]];
  }
  
  [self connectGMDs];
  
  if (gmds) {
    unsigned count = [queries count];
    unsigned i;

    for (i = 0; i < count; i++) {
      MDKQuery *q = [queries objectAtIndex: i];
      
      if (([q isGathering] == NO) && [q isStopped]) {
        [queries removeObjectAtIndex: i];
        i--;
        count--;
      }
    }
    
    NS_DURING
	    {
        if ([query isClosed] == NO) {
          [query closeSubqueries];
        }
        if ([query isBuilt] == NO) {
          [query buildQuery];
        }
	    }
    NS_HANDLER
	    {
        NSLog(@"%@", localException); 
        return NO;
	    }
    NS_ENDHANDLER
        
    [queries insertObject: query atIndex: 0];
    
    if ([queries count] == 1) {
      [query setStarted];
      [gmds performQuery: [query sqlDescription]];
    }
      
  } else {
    [NSException raise: NSInternalInconsistencyException
	              format: @"The query manager is unable to contact the gmds daemon."];  
  }
  
  return YES;
}

- (BOOL)queryResults:(NSData *)results
{
  CREATE_AUTORELEASE_POOL(arp);
  NSDictionary *dict = [NSUnarchiver unarchiveObjectWithData: results];
  NSNumber *qnum = [dict objectForKey: @"qnumber"];
  MDKQuery *query = [self queryWithNumber: qnum];
  BOOL resok = NO;
  
  if (query && ([query isStopped] == NO)) {
    [query appendResults: [dict objectForKey: @"lines"]];
    resok = YES;
  }

  RELEASE (arp);

  return resok;
}

- (oneway void)endOfQueryWithNumber:(NSNumber *)qnum
{
  MDKQuery *query = [self queryWithNumber: qnum];
    
  if (query) {
    if ([query isUpdating]) {    
      GWDebugLog(@"REMOVING UPDATING QUERY %i", [queries count]);
    } else {    
      GWDebugLog(@"REMOVING SIMPLE QUERY %i", [queries count]);
    }

    if ([query isUpdating]) {
      [query updatingDone];
    }  
    [query gatheringDone];
    [queries removeObject: query];
  }

  query = [self nextQuery];

  if (query && ([query isGathering] == NO)) {
    if ([query isStopped] == NO) {
      if ([query isUpdating] == NO) {    
        [query setStarted];
        [gmds performQuery: [query sqlDescription]];
      } else {
        [query updatingStarted];
        
        GWDebugLog(@"PERFORMING UPDATE (2) %i", [queries count]);
        
        [gmds performQuery: [query sqlUpdatesDescription]];
      }
    } else {
      [queries removeObject: query];
    }
  }
}

- (MDKQuery *)queryWithNumber:(NSNumber *)qnum
{
  unsigned i;

  for (i = 0; i < [queries count]; i++) {
    MDKQuery *query = [queries objectAtIndex: i];
    
    if ([[query queryNumber] isEqual: qnum]) {
      return query;
    }
  }

  return nil;
}

- (MDKQuery *)nextQuery
{
  return [queries lastObject];
}

- (unsigned long)tableNumber
{
  return tableNumber++;
}

- (unsigned long)queryNumber
{
  return queryNumber++;  
}

- (void)connectGMDs
{
  if (gmds == nil) {
    gmds = [NSConnection rootProxyForConnectionWithRegisteredName: @"gmds" 
                                                             host: @""];

    if (gmds == nil) {
	    NSString *cmd;
      int i;
    
      cmd = [[NSSearchPathForDirectoriesInDomains(
                GSToolsDirectory, NSSystemDomainMask, YES) objectAtIndex: 0]
                                      stringByAppendingPathComponent: @"gmds"];    
                    
      [NSTask launchedTaskWithLaunchPath: cmd arguments: nil];
   
      for (i = 0; i < 40; i++) {
	      [[NSRunLoop currentRunLoop] runUntilDate:
		                     [NSDate dateWithTimeIntervalSinceNow: 0.1]];

        gmds = [NSConnection rootProxyForConnectionWithRegisteredName: @"gmds" 
                                                                 host: @""];                  
        if (gmds) {
          break;
        }
      }
    }
    
    if (gmds) {
      RETAIN (gmds);
      [gmds setProtocolForProxy: @protocol(GMDSProtocol)];
    
	    [[NSNotificationCenter defaultCenter] addObserver: self
	                   selector: @selector(gmdsConnectionDidDie:)
		                     name: NSConnectionDidDieNotification
		                   object: [gmds connectionForProxy]];

      [gmds registerClient: self];                              
      NSLog(@"gmds connected!");     
                       
    } else {
      NSLog(@"unable to contact gmds.");  
    }
  }  
}

- (void)gmdsConnectionDidDie:(NSNotification *)notif
{
  [nc removeObserver: self
	                    name: NSConnectionDidDieNotification
	                  object: [notif object]];
  DESTROY (gmds);
  NSLog(@"gmds connection died!");  
  [[NSRunLoop currentRunLoop] runUntilDate:
		                    [NSDate dateWithTimeIntervalSinceNow: 1.0]];
  [self connectGMDs];
}

@end


@implementation MDKQueryManager (updates)

- (void)startUpdateForQuery:(MDKQuery *)query
{
  if ([liveQueries containsObject: query] == NO) {
    [liveQueries insertObject: query atIndex: 0];
  }
}

- (void)metadataDidUpdate:(NSNotification *)notif
{
  CREATE_AUTORELEASE_POOL(arp);
  NSArray *removed = [[notif userInfo] objectForKey: @"removed"];
  unsigned count = [liveQueries count];
  unsigned i;

  for (i = 0; i < count; i++) {
    MDKQuery *query = [liveQueries objectAtIndex: i];
    
    if ([query updatesEnabled]) {
      [query removePaths: removed];
    
      if ([queries containsObject: query] == NO) {
        [queries addObject: query];
    
        GWDebugLog(@"INSERTING UPDATING QUERY %i", [queries count]);
      }
      
    } else {
      [liveQueries removeObjectAtIndex: i];
      count--;
      i--;
    }
  }
  
  if (count && (count == [queries count])) {  
    MDKQuery *query = [queries lastObject];
    
    GWDebugLog(@"PERFORMING UPDATE (1) %i", [queries count]);
    
    [query updatingStarted];
    [gmds performQuery: [query sqlUpdatesDescription]];
  }

  RELEASE (arp);
}

@end


@implementation MDKQueryManager (results_filtering)

static NSArray *imageExtensions(void)
{
  static NSMutableArray *extensions = nil;

  if (extensions == nil) {
    extensions = [NSMutableArray new]; 
    
    [extensions addObjectsFromArray: [NSImage imageFileTypes]];
    [extensions addObject: @"xpm"];
    [extensions addObject: @"xbm"];
    
    [extensions makeImmutableCopyOnFail: NO];
  }

  return extensions;
}

static NSArray *movieExtensions(void)
{
  static NSArray *extensions = nil;

  if (extensions == nil) {
    extensions = [[NSArray alloc] initWithObjects: @"avi", @"mpg", @"mpeg",
                          @"mov", @"divx", @"m1v", @"m2p", @"m2v", @"moov", 
                          @"mp4", @"mpv", @"ogm", @"qt", @"rm", @"swf", 
                          @"vob", @"wmv", nil];
  }

  return extensions;
}

static NSArray *musicExtensions(void)
{
  static NSArray *extensions = nil;

  if (extensions == nil) {
    extensions = [[NSArray alloc] initWithObjects: @"aac", @"ac3", @"aif",
                          @"aiff", @"mpa", @"mp1", @"mp2", @"mp3", @"ogg", 
                          @"omf", @"ram", @"wav", @"wma", nil];
  }

  return extensions;
}

static NSArray *sourceExtensions(void)
{
  static NSArray *extensions = nil;

  if (extensions == nil) {
    extensions = [[NSArray alloc] initWithObjects: @"asm", @"c", @"class", 
                          @"cpp", @"cxx", @"cc", @"c++", @"h", @"hpp", 
                          @"hxx", @"java", @"jar", @"m", @"mm", @"pl", 
                          @"py", @"y", @"yxx", nil];
  }

  return extensions;
}

- (NSString *)categoryNameForNode:(FSNode *)node
{
  NSString *category = nil;
    
  if ([node isApplication]) {
    category = @"applications";
  } else if ([node isDirectory] && ([node isPackage] == NO)) {
    category = @"folders";
  } else {
    NSString *ext = [[[node path] pathExtension] lowercaseString];
    
    if (ext && [ext length]) {
      if ([ext isEqual: @"pdf"]) {
        category = @"pdfdocs";
      } else if ([sourceExtensions() containsObject: ext]) {
        category = @"sources";      
      } else if ([imageExtensions() containsObject: ext]) {
        category = @"images";      
      } else if ([movieExtensions() containsObject: ext]) {
        category = @"movies";      
      } else if ([musicExtensions() containsObject: ext]) {
        category = @"music";      
      }
    }
  }
  
  if (category == nil) {
    if ([node application]) {
      category = @"documents";  
    } else {
      category = @"plainfiles";
    }
  }
  
  return category;
}

- (BOOL)filterNode:(FSNode *)node
     withFSFilters:(NSArray *)filters
{
  int i;
  
  for (i = 0; i < [filters count]; i++) {
    if ([[filters objectAtIndex: i] filterNode: node] == NO) {
      return NO;
    }
  }
    
  return YES;
}

@end