/* Implementation of class GSStoryboardTransform
   Copyright (C) 2020 Free Software Foundation, Inc.
   
   By: Gregory John Casamento
   Date: Sat 04 Jul 2020 03:48:15 PM EDT

   This file is part of the GNUstep Library.
   
   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.1 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., 51 Franklin Street, Fifth Floor,
   Boston, MA 02110 USA.
*/

#import <Foundation/NSData.h>
#import <Foundation/NSDictionary.h>
#import <Foundation/NSString.h>
#import <Foundation/NSMapTable.h>
#import <Foundation/NSXMLDocument.h>
#import <Foundation/NSXMLNode.h>
#import <Foundation/NSXMLElement.h>
#import <Foundation/NSUUID.h>
#import <Foundation/NSBundle.h>
#import <Foundation/NSKeyedArchiver.h>
#import <Foundation/NSException.h>

#import "AppKit/NSSeguePerforming.h"
#import "AppKit/NSStoryboard.h"
#import "AppKit/NSStoryboardSegue.h"
#import "AppKit/NSNibDeclarations.h"
#import "AppKit/NSViewController.h"
#import "AppKit/NSWindowController.h"

#import "GSStoryboardTransform.h"
#import "GSFastEnumeration.h"

#define APPLICATION @"application"

@interface NSStoryboardSegue (__private__)
- (void) _setDestinationController: (id)controller; 
- (void) _setSourceController: (id)controller;
@end

@interface NSStoryboardSegue (__StoryboardPrivate__)
// Private to this class...
- (void) _setKind: (NSString *)k;
- (void) _setRelationship: (NSString *)r;
- (NSString *) _kind;
- (NSString *) _relationship;
@end

// this needs to be set on segues
@implementation NSStoryboardSegue (__StoryboardPrivate__)
- (void) _setKind: (NSString *)k
{
  ASSIGN(_kind, k);
}

- (void) _setRelationship: (NSString *)r
{
  ASSIGN(_relationship, r);
}

- (NSString *) _kind
{
  return _kind;
}

- (NSString *) _relationship
{
  return _relationship;
}
@end

@implementation NSStoryboardSeguePerformAction
- (id) target
{
  return _target;
}

- (void) setTarget: (id)target
{
  ASSIGN(_target, target);
}

- (SEL) action
{
  return _action;
}

- (void) setAction: (SEL)action
{
  _action = action;
}

- (NSString *) selector
{
  return NSStringFromSelector(_action);
}

- (void) setSelector: (NSString *)s
{
  _action = NSSelectorFromString(s);
}

- (id) sender
{
  return _sender;
}

- (void) setSender: (id)sender
{
  ASSIGN(_sender, sender);
}

- (NSString *) identifier
{
  return _identifier;
}

- (void) setIdentifier: (NSString *)identifier
{
  ASSIGN(_identifier, identifier);
}

- (NSString *) kind
{
  return _kind;
}

- (void) setKind: (NSString *)kind
{
  ASSIGN(_kind, kind);
}

- (NSStoryboard *) storyboard
{
  return _storyboard;
}

- (void) setStoryboard: (NSStoryboard *)storyboard
{
  ASSIGN(_storyboard, storyboard);
}

- (NSStoryboardSegue *) storyboardSegue
{
  return _storyboardSegue;
}

- (void) setStoryboardSegue: (NSStoryboardSegue *)ss
{
  ASSIGN(_storyboardSegue, ss);
}

- (id) nibInstantiate
{
  return self;
}

- (void) dealloc
{
  RELEASE(_storyboard);
  RELEASE(_kind);
  RELEASE(_identifier);
  RELEASE(_sender);
  RELEASE(_storyboardSegue);
  [super dealloc];
}

- (IBAction) doAction: (id)sender
{
  if (_sender != nil)
    {
      [_sender performSegueWithIdentifier: _identifier
                                   sender: _sender];
    }
  else // This is a special case where there is no source controller and we don't ask "should"
    {
      id destCon = nil;
      if ([[_storyboardSegue destinationController] isKindOfClass: [NSViewController class]] ||
          [[_storyboardSegue destinationController] isKindOfClass: [NSWindowController class]])
        {
          destCon = [_storyboardSegue destinationController];
        }
      else
        {
          NSString *destId = [_storyboardSegue destinationController];
          destCon = [_storyboard instantiateControllerWithIdentifier: destId]; 
        }
      [_storyboardSegue _setSourceController: nil]; 
      [_storyboardSegue _setDestinationController: destCon];  // replace with actual controller...
      [_storyboardSegue perform];
    }
}

- (id) copyWithZone: (NSZone *)z
{
  NSStoryboardSeguePerformAction *pa = [[NSStoryboardSeguePerformAction allocWithZone: z] init];
  [pa setTarget: _target];
  [pa setSelector: [self selector]];
  [pa setSender: _sender];
  [pa setIdentifier: _identifier];
  [pa setStoryboardSegue: _storyboardSegue];
  [pa setStoryboard: _storyboard];
  return pa;
}

- (instancetype) initWithCoder: (NSCoder *)coder
{
  self = [super init];
  if ([coder allowsKeyedCoding])
    {
      if ([coder containsValueForKey: @"NSTarget"])
        {
          [self setTarget: [coder decodeObjectForKey: @"NSTarget"]];
        }
      if ([coder containsValueForKey: @"NSSelector"])
        {
          [self setSelector: [coder decodeObjectForKey: @"NSSelector"]];
        }
      if ([coder containsValueForKey: @"NSSender"])
        {
          [self setSender: [coder decodeObjectForKey: @"NSSender"]];
        }
      if ([coder containsValueForKey: @"NSIdentifier"])
        {
          [self setIdentifier: [coder decodeObjectForKey: @"NSIdentifier"]];
        }
      if ([coder containsValueForKey: @"NSKind"])
        {
          [self setKind: [coder decodeObjectForKey: @"NSKind"]];
        }
    }
  return self;
}

- (void) encodeWithCoder: (NSCoder *)coder
{
  // this is never encoded directly...
}
@end

@implementation NSControllerPlaceholder

- (NSString *) storyboardName
{
  return _storyboardName;
}

- (void) setStoryboardName: (NSString *)name
{
  ASSIGNCOPY(_storyboardName, name);
}

- (id) copyWithZone: (NSZone *)z
{
  NSControllerPlaceholder *c = [[NSControllerPlaceholder allocWithZone: z] init];
  [c setStoryboardName: _storyboardName];
  return c;
}

- (instancetype) initWithCoder: (NSCoder *)coder
{
  self = [super init];
  if ([coder allowsKeyedCoding])
    {
      if ([coder containsValueForKey: @"NSStoryboardName"])
        {
          [self setStoryboardName: [coder decodeObjectForKey: @"NSStoryboardName"]];
        }
    }
  return self;
}

- (void) encodeWithCoder: (NSCoder *)coder
{
  // this is never encoded directly...
}

- (id) instantiate
{
  NSStoryboard *sb = [NSStoryboard storyboardWithName: _storyboardName
                                               bundle: [NSBundle mainBundle]];
  return [sb instantiateInitialController];
}

@end

@implementation GSStoryboardTransform

- (instancetype) initWithData: (NSData *)data
{
  self = [super init];
  if (self != nil)
    {
      NSXMLDocument *xml = [[NSXMLDocument alloc] initWithData: data
                                                       options: 0
                                                         error: NULL];
      
      _scenesMap = [[NSMutableDictionary alloc] initWithCapacity: 10];
      _controllerMap = [[NSMutableDictionary alloc] initWithCapacity: 10];
      _identifierToSegueMap = [[NSMutableDictionary alloc] initWithCapacity: 10];
      
      [self processStoryboard: xml];
      RELEASE(xml);
    }
  return self;
}

- (void) dealloc
{
  RELEASE(_initialViewControllerId);
  RELEASE(_applicationSceneId);
  RELEASE(_scenesMap);
  RELEASE(_controllerMap);
  RELEASE(_identifierToSegueMap);
  [super dealloc];
}

- (NSString *) initialViewControllerId
{
  return _initialViewControllerId;
}

- (NSString *) applicationSceneId
{
  return _applicationSceneId;
}

- (NSDictionary *) scenesMap
{
  return _scenesMap;
}

- (NSDictionary *) controllerMap
{
  return _controllerMap;
}

- (NSMapTable *) segueMapForIdentifier: (NSString *)identifier
{
  return [_identifierToSegueMap objectForKey: identifier];
}

- (NSDictionary *) identifierToSegueMap
{
  return _identifierToSegueMap;
}

- (NSXMLElement *) createCustomObjectWithId: (NSString *)ident
                                   userLabel: (NSString *)userLabel
                                 customClass: (NSString *)className
{
  NSXMLElement *customObject =
    [[NSXMLElement alloc] initWithName: @"customObject"];
  NSXMLNode *idValue =
    [NSXMLNode attributeWithName: @"id"
                     stringValue: ident];
  NSXMLNode *usrLabel =
    [NSXMLNode attributeWithName: @"userLabel"
                     stringValue: userLabel];
  NSXMLNode *customCls =
    [NSXMLNode attributeWithName: @"customClass"
                     stringValue: className];
  
  [customObject addAttribute: idValue];
  [customObject addAttribute: usrLabel];
  [customObject addAttribute: customCls];

  AUTORELEASE(customObject);
  
  return customObject;
}

- (NSData *) dataForIdentifier: (NSString *)identifier
{
  NSString *sceneId = [_controllerMap objectForKey: identifier];
  NSXMLDocument *xml = [_scenesMap objectForKey: sceneId];
  return [xml XMLData];
}

- (NSData *) dataForApplicationScene
{
  return [self dataForIdentifier: APPLICATION];
}

- (void) processChild: (NSXMLElement *)child
              withDoc: (NSXMLElement *)doc
          withAppNode: (NSXMLNode *)appNode
              sceneId: (NSString *)sceneId
{
  NSString *customClassString = nil;
  if (appNode != nil)
    {
      NSXMLElement *objects = (NSXMLElement *)[appNode parent];
      NSArray *appConsArr = [appNode nodesForXPath: @"connections" error: NULL];
      NSXMLNode *appCons = [appConsArr objectAtIndex: 0];
      if (appCons != nil)
        {
          [appCons detach];
        }
      
      NSArray *appChildren = [appNode children];
      NSEnumerator *ace = [appChildren objectEnumerator];
      NSXMLElement *ae = nil;
      
      // Assign application scene...
      ASSIGN(_applicationSceneId, sceneId);
      [_controllerMap setObject: _applicationSceneId
                         forKey: APPLICATION];
      
      // Move all application children to objects...
      while ((ae = [ace nextObject]) != nil)
        {
          [ae detach];
          [objects addChild: ae];
        }
      
      // Remove the appNode
      [appNode detach];
      
      // Add it to the document
      [objects detach];
      [doc addChild: objects];
      
      // create a customObject entry for NSApplication reference...
      NSXMLNode *appCustomClass = (NSXMLNode *)[(NSXMLElement *)appNode
                                                   attributeForName: @"customClass"];
      customClassString = ([appCustomClass stringValue] == nil) ?
        @"NSApplication" : [appCustomClass stringValue];
      NSXMLElement *customObject = nil;
      
      customObject = 
        [self createCustomObjectWithId: @"-3"
                             userLabel: @"Application"
                           customClass: @"NSObject"];
      [child insertChild: customObject
                 atIndex: 0];
      customObject = 
        [self createCustomObjectWithId: @"-1"
                             userLabel: @"First Responder"
                           customClass: @"FirstResponder"];
      [child insertChild: customObject
                 atIndex: 0];
      customObject =
        [self createCustomObjectWithId: @"-2"
                             userLabel: @"File's Owner"
                           customClass: customClassString];
      if (appCons != nil)
        {
          [customObject addChild: appCons];
        }
      [child insertChild: customObject
                 atIndex: 0]; 
    }
  else
    {
      NSXMLElement *customObject = nil;
      
      customObject = 
        [self createCustomObjectWithId: @"-3"
                             userLabel: @"Application"
                           customClass: @"NSObject"];
      [child insertChild: customObject
                 atIndex: 0];
      customObject = 
        [self createCustomObjectWithId: @"-1"
                             userLabel: @"First Responder"
                           customClass: @"FirstResponder"];
      [child insertChild: customObject
                 atIndex: 0];
      customObject = 
        [self createCustomObjectWithId: @"-2"
                             userLabel: @"File's Owner"
                           customClass: customClassString];
      [child insertChild: customObject
                 atIndex: 0];
      
      [child detach];
      [doc addChild: child];
    }
}

- (NSString *) controllerIdWithDocument: (NSXMLDocument *)document
{
  NSString *controllerId = nil;
  NSArray *windowControllers = [document nodesForXPath: @"//windowController" error: NULL];
  NSArray *viewControllers = [document nodesForXPath: @"//viewController" error: NULL];
  NSArray *controllerPlaceholders = [document nodesForXPath: @"//controllerPlaceholder" error: NULL];
  
  if ([windowControllers count] > 0)
    {
      NSXMLElement *ce = [windowControllers objectAtIndex: 0];
      NSXMLNode *attr = [ce attributeForName: @"id"];
      controllerId = [attr stringValue];
      
      NSEnumerator *windowControllerEnum = [windowControllers objectEnumerator];
      NSXMLElement *o = nil;
      while ((o = [windowControllerEnum nextObject]) != nil)
        {
          NSXMLElement *objects = (NSXMLElement *)[o parent];
          NSArray *windows = [o nodesForXPath: @"//window" error: NULL];
          NSEnumerator *windowEn = [windows objectEnumerator];
          NSXMLNode *w = nil;
          while ((w = [windowEn nextObject]) != nil)
            {
              [w detach];
              [objects addChild: w];
            }
        }
    }
  
  if ([viewControllers count] > 0)
    {
      NSXMLElement *ce = [viewControllers objectAtIndex: 0];
      NSXMLNode *attr = [ce attributeForName: @"id"];
      controllerId = [attr stringValue];
    }
  
  if ([controllerPlaceholders count] > 0)
    {
      NSXMLElement *ce = [controllerPlaceholders objectAtIndex: 0];
      NSXMLNode *attr = [ce attributeForName: @"id"];
      controllerId = [attr stringValue];
    }
  
  return controllerId;
}

- (void) processCustomElement: (NSXMLElement *)coel
                  forDocument: (NSXMLDocument *)document
{
  NSXMLNode *attr = [coel attributeForName: @"sceneMemberID"];
  if ([[attr stringValue] isEqualToString: @"firstResponder"])
    {
      NSXMLNode *customClassAttr = [coel attributeForName: @"customClass"];
      NSXMLNode *idAttr = [coel attributeForName: @"id"];
      NSString *originalId = [idAttr stringValue];
      
      [idAttr setStringValue: @"-1"]; // set to first responder id
      [customClassAttr setStringValue: @"FirstResponder"];
      
      // Actions
      NSArray *cons = [document nodesForXPath: @"//action" error: NULL];
      FOR_IN(NSXMLElement*, celem, cons) 
        {
          NSXMLNode *targetAttr = [celem attributeForName: @"target"];
          NSString *val = [targetAttr stringValue];
          if ([val isEqualToString: originalId])
            {
              [targetAttr setStringValue: @"-1"];
            }
        }
      END_FOR_IN(cons);
      
      // Outlets
      cons = [document nodesForXPath: @"//outlet" error: NULL];
      FOR_IN(NSXMLElement*, celem, cons)
        {
          NSXMLNode *attr = [celem attributeForName: @"destination"];
          NSString *val = [attr stringValue];
          if ([val isEqualToString: originalId])
            {
              [attr setStringValue: @"-1"];
            }
        }
      END_FOR_IN(cons);
    }
}

- (void) processStoryboard: (NSXMLDocument *)xml
{
  NSArray *docNodes = [xml nodesForXPath: @"document" error: NULL];

  if ([docNodes count] > 0)
    {
      NSXMLElement *docNode = [docNodes objectAtIndex: 0];
      NSArray *array = [docNode nodesForXPath: @"//scene" error: NULL];
      
      // Set initial view controller...
      ASSIGN(_initialViewControllerId, [[docNode attributeForName: @"initialViewController"] stringValue]);             
      FOR_IN(NSXMLElement*, e, array) 
        {
          NSXMLElement *doc = [[NSXMLElement alloc] initWithName: @"document"];
          NSArray *children = [e children];
          NSXMLDocument *document = nil;
          NSString *sceneId = [[e attributeForName: @"sceneID"] stringValue]; 
          NSString *controllerId = nil;

          // Move children...
          FOR_IN(NSXMLElement*, child, children)
            {
              if ([[child name] isEqualToString: @"point"] == YES)
                continue; // go on if it's a point element, we don't use that in the app...
              
              NSArray *subnodes = [child nodesForXPath: @"//application" error: NULL];
              NSXMLNode *appNode = [subnodes objectAtIndex: 0];
              [self processChild: child
                         withDoc: doc
                     withAppNode: appNode
                         sceneId: sceneId];
              
              // fix other custom objects
              document = [[NSXMLDocument alloc] initWithRootElement: doc]; // put it into the document, so we can use Xpath.
              controllerId = [self controllerIdWithDocument: document];
              controllerId = (controllerId != nil) ? controllerId : APPLICATION;
              RELEASE(doc);

              NSArray *customObjects = [document nodesForXPath: @"//objects/customObject" error: NULL];
              FOR_IN(NSXMLElement*, coel, customObjects)
                {
                  [self processCustomElement: coel
                                 forDocument: document];
                }
              END_FOR_IN(customObjects);

              // Create document...
              [_scenesMap setObject: document
                             forKey: sceneId];
              
              // Map controllerId's to scenes...
              if (controllerId != nil)
                {
                  [_controllerMap setObject: sceneId
                                     forKey: controllerId];

                  [self processSegues: document
                      forControllerId: controllerId];
                }
              RELEASE(document);
            }
          END_FOR_IN(children);
        }
      END_FOR_IN(array);
    }
  else
    {
      [NSException raise: NSInternalInconsistencyException
                  format: @"No document element found in storyboard file"];
    }
}

- (NSString *) findControllerIdInDocument: (NSXMLDocument *)xmlIn
                              withObjects: (NSXMLElement *)objects
{
  NSString *src = nil;
  NSArray *controllers = [objects nodesForXPath: @"windowController"
                                          error: NULL];
  if ([controllers count] > 0)
    {
      NSXMLElement *controller = (NSXMLElement *)[controllers objectAtIndex: 0];
      NSXMLNode *idAttr = [controller attributeForName: @"id"];
      src = [idAttr stringValue];
    }
  else
    {
      controllers = [objects nodesForXPath: @"viewController"
                                     error: NULL];
      if ([controllers count] > 0)
        {
          NSXMLElement *controller = (NSXMLElement *)[controllers objectAtIndex: 0];
          NSXMLNode *idAttr = [controller attributeForName: @"id"];
          src = [idAttr stringValue];
        }
    }
  return src;
}

- (BOOL) isProcessedDocument: (NSXMLDocument *)xmlIn
                        uuid: (NSMutableString *)uuid
{
  NSArray *docArray = [xmlIn nodesForXPath: @"document" error: NULL];
  if ([docArray count] > 0)
    {
      NSXMLElement *docElem = (NSXMLElement *)[docArray objectAtIndex: 0];
      NSXMLNode *a = [docElem attributeForName: @"uuid"];
      NSString *value = [a stringValue];
      if (value != nil)
        {
          [uuid appendString: value];
          return YES;
        }
      else
        {
          NSString *uuidString = [[NSUUID UUID] UUIDString];
          NSXMLNode *new_uuid_attr = [NSXMLNode attributeWithName: @"uuid"
                                                      stringValue: uuidString];
          [docElem addAttribute: new_uuid_attr];
          [uuid appendString: uuidString];
        }
    }
  return NO;
}

- (NSXMLElement *) createStoryboardProxyElementWithSelector: (NSString *)selector
                                                     target: (NSString *)dst
                                            segueIdentifier: (NSString *)ident
                                                     sender: (NSString *)src
                                                       kind: (NSString *)kind
{
  NSXMLElement *sbproxy = [NSXMLElement elementWithName: @"storyboardSeguePerformAction"];

  NSXMLNode *pselector
    = [NSXMLNode attributeWithName: @"selector"
                       stringValue: selector];
  NSXMLNode *ptarget
    = [NSXMLNode attributeWithName: @"target"
                       stringValue: dst];
  NSString *pident_value = [[NSUUID UUID] UUIDString];
  NSXMLNode *pident
    = [NSXMLNode attributeWithName: @"id"
                       stringValue: pident_value];
  NSXMLNode *psegueIdent
    = [NSXMLNode attributeWithName: @"identifier"
                       stringValue: ident];
  NSXMLNode *psender
    = [NSXMLNode attributeWithName: @"sender"
                       stringValue: src];
  NSXMLNode *pkind
    = [NSXMLNode attributeWithName: @"kind"
                       stringValue: kind];
  
  [sbproxy addAttribute: pselector];
  [sbproxy addAttribute: ptarget];
  [sbproxy addAttribute: pident];
  [sbproxy addAttribute: psegueIdent];
  [sbproxy addAttribute: psender];
  [sbproxy addAttribute: pkind];

  return sbproxy;
}

- (NSMapTable *) processConnections: (NSArray *)connectionsArray
                        withObjects: (NSXMLElement *)objects
                       controllerId: (NSString *)src
{
  NSMapTable *mapTable = [NSMapTable strongToWeakObjectsMapTable];
  
  FOR_IN (NSXMLElement*, connObj, connectionsArray)
    {
      NSXMLElement *connections = (NSXMLElement *)connObj;
      NSArray *children = [connections children]; // there should be only one per set.
      
      FOR_IN (id, obj, children)
        {
          if ([[obj name] isEqualToString: @"segue"])
            {
              // get the information from the segue.
              id segue_parent_parent = [[obj parent] parent];
              id segue_parent = [obj parent];
              NSString *segue_parent_parent_name = [segue_parent_parent name];
              NSXMLNode *attr = [obj attributeForName: @"destination"];
              NSString *dst = [attr stringValue];
              attr = [obj attributeForName: @"kind"];
              NSString *kind =  [attr stringValue];
              attr = [obj attributeForName: @"relationship"];
              NSString *rel = [attr stringValue];
              [obj detach]; // segue can't be in the archive since it doesn't conform to NSCoding
              attr = [obj attributeForName: @"id"];
              NSString *uid = [attr stringValue];
              attr = [obj attributeForName: @"identifier"];
              NSString *ident = [attr stringValue];
              if (ident == nil)
                {
                  ident = [[NSUUID UUID] UUIDString];
                }
              
              // Create proxy object to invoke methods on the window controller
              NSXMLElement *sbproxy =  [self createStoryboardProxyElementWithSelector: @"doAction:"
                                                                               target: dst
                                                                      segueIdentifier: ident
                                                                               sender: src
                                                                                 kind: kind];
              
              NSUInteger count = [[objects children] count];
              [objects insertChild: sbproxy
                           atIndex: count - 1];
              
              // add action to parent ONLY if it is NOT a controller..
              if (![segue_parent_parent_name isEqualToString: @"windowController"] &&
                  ![segue_parent_parent_name isEqualToString: @"viewController"])
                {              
                  // Create action...
                  NSXMLElement *action = [NSXMLElement elementWithName: @"action"];
                  NSXMLNode *selector
                    = [NSXMLNode attributeWithName: @"selector"
                                       stringValue: @"doAction:"];
                  NSXMLNode *target
                    = [NSXMLNode attributeWithName: @"target"
                                       stringValue: [[sbproxy attributeForName: @"id"] stringValue]];
                  NSXMLNode *controller_ident
                    = [NSXMLNode attributeWithName: @"id"
                                       stringValue: uid]; 
                  [action addAttribute: selector];
                  [action addAttribute: target];
                  [action addAttribute: controller_ident];
                  [segue_parent addChild: action];
                }
              
              // Create the segue...
              NSStoryboardSegue *ss = [[NSStoryboardSegue alloc] initWithIdentifier: ident
                                                                             source: src
                                                                        destination: dst];
              [ss _setKind: kind];
              [ss _setRelationship: rel];
              
              // Add to maptable...
              [mapTable setObject: ss
                           forKey: ident];
              
            } // only process segue objects...
        } // iterate over objects in each set of connections
      END_FOR_IN(children);          
    } // iterate over connection objs
  END_FOR_IN(connectionsArray);

  return mapTable;
}

- (void) processSegues: (NSXMLDocument *)xml
       forControllerId: (NSString *)identifier
{
  NSMutableString *uuidString = [NSMutableString string];
  BOOL processed = [self isProcessedDocument: xml
                                        uuid: uuidString];
  if (processed)
    {
      return; // don't processed the document again
    }
  else // Get the controller... there is only one per scene.
    {
      NSArray *array = [xml nodesForXPath: @"//objects[1]"
                                    error: NULL];
      NSXMLElement *objects = [array objectAtIndex: 0]; // get the "objects" section
      NSString *src = [self findControllerIdInDocument: xml
                                           withObjects: objects];
      NSArray *connectionsArray = [xml nodesForXPath: @"//connections"
                                               error: NULL];
      NSMapTable *mapTable = [self processConnections: connectionsArray
                                          withObjects: objects
                                         controllerId: src];           
      [_identifierToSegueMap setObject: mapTable
                                forKey: identifier];                  
    }
}

@end