/**
    STEnvironment
    Scripting environment
  
    Copyright (c) 2002 Free Software Foundation
  
    Written by: Stefan Urbanek <urbanek@host.sk>
    Date: 2000
   
    This file is part of the StepTalk project.
 
    This library 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 of the License, or (at your option) any later version.

    This library 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 this library; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    <title>STEnvironment class reference</title>
 
   */

#import <StepTalk/STEnvironment.h>

#import <StepTalk/STClassInfo.h>
#import <StepTalk/STEnvironmentDescription.h>
#import <StepTalk/STExterns.h>
#import <StepTalk/STFunctions.h>
#import <StepTalk/STModule.h>
#import <StepTalk/STObjectReference.h>
#import <StepTalk/STUndefinedObject.h>
#import <StepTalk/STScripting.h>

#import <Foundation/NSArray.h>
#import <Foundation/NSDebug.h>
#import <Foundation/NSDictionary.h>
#import <Foundation/NSException.h>
#import <Foundation/NSSet.h>
#import <Foundation/NSString.h>
#import <Foundation/NSInvocation.h>

@interface STEnvironment(STPrivateMethods)
- (STClassInfo *)findClassInfoForObject:(id)anObject;
@end

@implementation STEnvironment
/**
   Creates and initialises scripting environment using standard description.
 */
+ (STEnvironment *)defaultScriptingEnvironment
{
    NSString *name;

    name = [STEnvironmentDescription defaultEnvironmentDescriptionName];
    
    return [self environmentWithDescriptionName:name];
}

/**
   Creates and initialises scripting environment using description with name
   <var>descName</var>.
 */
+ environmentWithDescriptionName:(NSString *)descName
{
    return AUTORELEASE([[self alloc] initWithDescriptionName:descName]);
}

/**
   Creates and initialises scripting environment using description
   from dictionary <var>dict</var>.
 */
+ environmentWithDescriptionFromDictionary:(NSDictionary *)dict
{
    return AUTORELEASE([[self alloc] initWithDescriptionFromDictionary:dict]);

}

/**
   Creates and initialises scripting environment using description
   from file at <var>path</var>.
 */
+ environmentWithDescriptionFromFile:(NSString *)path
{
    return AUTORELEASE([[self alloc] initWithDescriptionFromFile:path]);

}

- init
{
    defaultPool = [[NSMutableDictionary alloc] init];
/* FIXME: */
//    [defaultPool setObject:STNil forKey:@"nil"];
    infoCache = [[NSMutableDictionary alloc] init];
    
    return [super init];
}

/**
   Initialises scripting environment using description with name
   <var>descName</var>.
 */
- initWithDescriptionName:(NSString *)descName
{
    [self init];
    
    description = [STEnvironmentDescription descriptionWithName:descName];
    classes = [description classes];

    RETAIN(description);

    return self;
}

/**
   Initialises scripting environment using description from dictionary
   <var>dict</var>.
 */
- initWithDescriptionFromDictionary:(NSDictionary *)dict
{
    [self init];
    
    description = [STEnvironmentDescription descriptionFromDictionary:dict];
    classes = [description classes];

    RETAIN(description);

    return self;
}

/**
   Initialises scripting environment using description 
   from file at <var>path</var>
 */
- initWithDescriptionFromFile:(NSString *)path
{
    [self init];
    
    description = [STEnvironmentDescription descriptionFromFile:path];
    classes = [description classes];
    
    RETAIN(description);

    return self;
}

- (void)dealloc
{
    RELEASE(defaultPool);
    RELEASE(pools);
    RELEASE(description);

    RELEASE(infoCache);

    [super dealloc];
}

/**
   Enable or disable full scripting. When full scripting is enabled, 
   you may send any message to any object.
 */
- (void)setFullScriptingEnabled:(BOOL)flag
{
    fullScripting = flag;
}

/**
   Returns YES if full scripting is enabled.
 */
- (BOOL)fullScriptingEnabled
{
    return fullScripting;
}

/**
   Enable or disable creation of unknown objects. Normally you get nil if you
   request for non-existant object. If <var>flag</var> is YES
   then by requesting non-existant object, name for that object is created
   and it is set no STNil.
 */
-(void)setCreatesUnknownObjects:(BOOL)flag
{
    createsUnknownObjects = flag;
}

/**
   Returns YES if unknown objects are being created.
 */
-(BOOL)createsUnknownObjects
{
    return createsUnknownObjects;
}

- (void)addAllClasses
{
    [self addNamedObjectsFromDictionary:STAllObjectiveCClasses()];
}
/**
    Add classes specified by the names in the <var>names</var> array.
*/
- (void)addClassesWithNames:(NSArray *)names
{
    [self addNamedObjectsFromDictionary:STClassDictionaryFromNames(names)];
}

/**
   Load StepTalk module with the name <var>moduleName</var>. Module extension
   '.stmodule' is optional. Modules are stored in the Library/StepTalk/Modules
   directory.
 */

- (void) loadModule:(NSString *)moduleName
{
    STModule *module = [STModule moduleWithName:moduleName];
    
    [self addNamedObjectsFromDictionary:[module namedObjects]];
    [self addClassesWithNames:[module providedClasses]];
}

- (NSMutableDictionary *)defaultObjectPool
{
    return defaultPool;
}

/* ----------------------------------------------------------------------- 
   Object pools
   ----------------------------------------------------------------------- */

/**
    Register object <var>anObject</var> with name <var>objName</var>.
 */
- (void)addObject:(id)anObject
         withName:(NSString *)objName
{
    NSLog(@"In STEnvironment: addObject:withName: used. Use setObject:forName:");
    [defaultPool setObject:anObject forKey:objName];
}

/**
    Register object <var>anObject</var> with name <var>objName</var>.
 */

- (void)setObject:(id)anObject
          forName:(NSString *)objName
{
    if(anObject)
    {
        [defaultPool setObject:anObject forKey:objName];
    }
    else
    {
        [defaultPool setObject:STNil forKey:objName];
    }
}

- (NSMutableDictionary *) poolWithName:(NSString *)poolName
{
    NSMutableDictionary *pool = [pools objectForKey:poolName];
    
    if(!defaultPool)
    {
        defaultPool = [[NSMutableDictionary alloc] init];
    }

    if(!pool)
    {
        [NSException  raise:STGenericException
                     format:@"Undefined pool with name '%@'", poolName];


        pool = defaultPool;
    }

    return pool;
}


/**
    Register object <var>anObject</var> with name <var>objName</var> in pool
    <var>poolName</var>.
 */
- (void)addObject:(id)anObject
         withName:(NSString *)objName 
             pool:(NSString *)poolName
{
    
    [[self poolWithName:poolName] setObject:anObject forKey:objName];
}

/**
    Remove object named <var>objName</var>.
  */
- (void)removeObjectWithName:(NSString *)objName
{
    [defaultPool removeObjectForKey:objName];
}

/**
    Remove object named <var>objName</var> form pool <var>poolName</var>
  */
- (void)removeObjectWithName:(NSString *)objName
                        pool:(NSString *)poolName
{
    [[self poolWithName:poolName] removeObjectForKey:objName];
}

/**
    
  */
- (void)addNamedObjectsFromDictionary:(NSDictionary *)dict
{
    [defaultPool addEntriesFromDictionary:dict];
}


- (void)addNamedObjectsFromDictionary:(NSDictionary *)dict
                                 pool:(NSString *)poolName
{
    [[self poolWithName:poolName] addEntriesFromDictionary:dict];

}

- (id)objectWithName:(NSString *)objName
{
    id object;
    
    object = [defaultPool objectForKey:objName];
        
    return object;
}

- (id)objectWithName:(NSString *)objName 
                pool:(NSString *)poolName

{
    id object;
    
    object = [[self poolWithName:poolName] objectForKey:objName];
    
    return object;
}

- (STObjectReference *)objectReferenceForObjectWithName:(NSString *)name
{
    STObjectReference *ref;
    
    ref = [STObjectReference alloc];
    [ref initWithObjectName:name
                       pool:defaultPool
                     create:createsUnknownObjects];

    return AUTORELEASE(ref);
}

- (STObjectReference *)objectReferenceForObjectWithName:(NSString *)name
                                                   pool:(NSString *)poolName
{
    STObjectReference *ref;

    ref = [STObjectReference alloc];
    [ref initWithObjectName:name
                       pool:[self poolWithName:poolName]
                     create:createsUnknownObjects];
                       
    return AUTORELEASE(ref);

}

- (void)removePool:(NSString *)poolName
{
    [pools removeObjectForKey:poolName];
}


/* FIXME: rewrite */
- (STClassInfo *)findClassInfoForObject:(id)anObject
{
    NSString    *origName;
    NSString    *name;
    STClassInfo *info = nil;
    Class        class;
    
    if(!anObject)
    {
        anObject = STNil;
    }

    if( [anObject isProxy] )
    {
        NSDebugLog(@"FIXME: receiver is a distant object");
        return nil;
    }

    class = [anObject classForScripting];

    if([anObject isClass])
    {
        origName = name = [[class className] 
                            stringByAppendingString:@" class"];

        NSDebugLLog(@"STSending",
                    @"Looking for class info '%@'...",
                    name);

        info = [infoCache objectForKey:name];
        
        if(info)
        {
            return info;
        }
        
        while( !(info = [classes objectForKey:name]) )
        {
            class = [[class superclass] classForScripting];
            if(!class)
            {
                break;
            }
            
            name = [[class className]
                            stringByAppendingString:@" class"];
            NSDebugLLog(@"STSending",
                        @"    ... %@?",name);
        }
    }
    else
    {
        origName = name = [class className];

        NSDebugLLog(@"STSending",
                    @"Looking for class info '%@' (instance)...",
                    name);

        info = [infoCache objectForKey:name];
        if(info)
        {
            return info;
        }

        while( !(info = [classes objectForKey:name]) )
        {
            class = [[class superclass] classForScripting];
            if(!class)
            {
                break;
            }
            
            name = [class className];
            NSDebugLLog(@"STSending",
                        @"    ... %@?",name);
        }
    }
    
    if(!info)
    {
        NSDebugLLog(@"STSending",
                    @"No class info '%@'",
                    name);
        return nil;
    }

    NSDebugLLog(@"STSending",
                @"Found class info '%@'",
                name);
                
    [infoCache setObject:info forKey:origName]; 
    return info;
}

- (NSString  *)translateSelector:(NSString *)aString forReceiver:(id)anObject
{
    STClassInfo *class;
    NSString    *selector;

    if( [anObject isProxy] )
    {
        NSDebugLog(@"Warning: receiver is a distant object (FIXME)");
     
        return aString;
    }

    class    = [self findClassInfoForObject:anObject];

    NSDebugLLog(@"STSending",
                @"Lookup selector '%@' class %@", aString, [class behaviourName]);

    selector = [class translationForSelector:aString];

    NSDebugLLog(@"STSending",
                @"Found selector '%@'",selector);

#ifdef DEBUG
    if(! [selector isEqualToString:aString])
    {
       NSDebugLLog(@"STSending",
                    @"using selector '%@' instead of '%@'",
                    selector,aString);
    }
#endif
    
    if(!selector && fullScripting )
    {
        NSDebugLLog(@"STSending",
                    @"using selector '%@' (full scriptig)",
                    aString);

        selector = AUTORELEASE([aString copy]);
    }

    if(!selector)
    {
        [NSException raise:STScriptingException
                     format:@"Receiver of type %@ denies selector '%@'",
                            [anObject className],aString];

        /* if exception is ignored, then try to use original selector  */
        selector = AUTORELEASE([aString copy]);
    }


    return selector;
}
@end