/*

 * Copyright (C) 2006, 2007 Apple Inc. All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 *

 * 1.  Redistributions of source code must retain the above copyright

 *     notice, this list of conditions and the following disclaimer. 

 * 2.  Redistributions in binary form must reproduce the above copyright

 *     notice, this list of conditions and the following disclaimer in the

 *     documentation and/or other materials provided with the distribution. 

 * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of

 *     its contributors may be used to endorse or promote products derived

 *     from this software without specific prior written permission. 

 *

 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY

 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE

 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY

 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES

 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;

 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND

 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */

#import "WebScriptDebugServer.h"

#import "WebScriptDebugServerPrivate.h"

#import "WebViewInternal.h"

#import <JavaScriptCore/Assertions.h>

NSString *WebScriptDebugServerProcessNameKey = @"WebScriptDebugServerProcessNameKey";

NSString *WebScriptDebugServerProcessBundleIdentifierKey = @"WebScriptDebugServerProcessBundleIdentifierKey";

NSString *WebScriptDebugServerProcessIdentifierKey = @"WebScriptDebugServerProcessIdentifierKey";

NSString *WebScriptDebugServerQueryNotification = @"WebScriptDebugServerQueryNotification";

NSString *WebScriptDebugServerQueryReplyNotification = @"WebScriptDebugServerQueryReplyNotification";

NSString *WebScriptDebugServerDidLoadNotification = @"WebScriptDebugServerDidLoadNotification";

NSString *WebScriptDebugServerWillUnloadNotification = @"WebScriptDebugServerWillUnloadNotification";

@implementation WebScriptDebugServer

static WebScriptDebugServer *sharedServer = nil;

static unsigned listenerCount = 0;

+ (WebScriptDebugServer *)sharedScriptDebugServer

{

    if (!sharedServer)

        sharedServer = [[WebScriptDebugServer alloc] init];

    return sharedServer;

}

+ (unsigned)listenerCount

{

    return listenerCount;

}

- (id)init

{

    self = [super init];

    NSProcessInfo *processInfo = [NSProcessInfo processInfo];

    serverName = [[NSString alloc] initWithFormat:@"WebScriptDebugServer-%@-%d", [processInfo processName], [processInfo processIdentifier]];

    serverConnection = [[NSConnection alloc] init];

    if ([serverConnection registerName:serverName]) {

        [serverConnection setRootObject:self];

        NSProcessInfo *processInfo = [NSProcessInfo processInfo];

        NSDictionary *info = [[NSDictionary alloc] initWithObjectsAndKeys:[processInfo processName], WebScriptDebugServerProcessNameKey,

            [[NSBundle mainBundle] bundleIdentifier], WebScriptDebugServerProcessBundleIdentifierKey,

            [NSNumber numberWithInt:[processInfo processIdentifier]], WebScriptDebugServerProcessIdentifierKey, nil];

        [[NSDistributedNotificationCenter defaultCenter] postNotificationName:WebScriptDebugServerDidLoadNotification object:serverName userInfo:info];

        [info release];

    } else {

        [serverConnection release];

        serverConnection = nil;

    }

    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(applicationTerminating:) name:NSApplicationWillTerminateNotification object:nil];

    [[NSDistributedNotificationCenter defaultCenter] addObserver:self selector:@selector(serverQuery:) name:WebScriptDebugServerQueryNotification object:nil];

    listeners = [[NSMutableSet alloc] init];

    return self;

}

- (void)dealloc

{

    // FIXME: Bad to do all this work in dealloc. What about under GC?

    ASSERT(listenerCount >= [listeners count]);

    listenerCount -= [listeners count];

    if (!listenerCount)

        [self detachScriptDebuggerFromAllWebViews];

    [[NSNotificationCenter defaultCenter] removeObserver:self name:NSApplicationWillTerminateNotification object:nil];

    [[NSDistributedNotificationCenter defaultCenter] removeObserver:self name:WebScriptDebugServerQueryNotification object:nil];

    [[NSDistributedNotificationCenter defaultCenter] postNotificationName:WebScriptDebugServerWillUnloadNotification object:serverName];

    [[NSNotificationCenter defaultCenter] removeObserver:self name:NSConnectionDidDieNotification object:nil];

    [serverConnection invalidate];

    [serverConnection release];

    [serverName release];

    [listeners release];

    [super dealloc];

}

- (void)applicationTerminating:(NSNotification *)notifiction

{

    [[NSDistributedNotificationCenter defaultCenter] postNotificationName:WebScriptDebugServerWillUnloadNotification object:serverName];

}

- (void)attachScriptDebuggerToAllWebViews

{

    [WebView _makeAllWebViewsPerformSelector:@selector(_attachScriptDebuggerToAllFrames)];

}

- (void)detachScriptDebuggerFromAllWebViews

{

    [WebView _makeAllWebViewsPerformSelector:@selector(_detachScriptDebuggerFromAllFrames)];

}

- (void)serverQuery:(NSNotification *)notification

{

    NSProcessInfo *processInfo = [NSProcessInfo processInfo];

    NSDictionary *info = [[NSDictionary alloc] initWithObjectsAndKeys:[processInfo processName], WebScriptDebugServerProcessNameKey,

        [[NSBundle mainBundle] bundleIdentifier], WebScriptDebugServerProcessBundleIdentifierKey,

        [NSNumber numberWithInt:[processInfo processIdentifier]], WebScriptDebugServerProcessIdentifierKey, nil];

    [[NSDistributedNotificationCenter defaultCenter] postNotificationName:WebScriptDebugServerQueryReplyNotification object:serverName userInfo:info];

    [info release];

}

- (void)listenerConnectionDidDie:(NSNotification *)notification

{

    NSConnection *connection = [notification object];

    NSMutableSet *listenersToRemove = [[NSMutableSet alloc] initWithCapacity:[listeners count]];

    NSEnumerator *enumerator = [listeners objectEnumerator];

    NSDistantObject *listener = nil;

    [[NSNotificationCenter defaultCenter] removeObserver:self name:NSConnectionDidDieNotification object:connection];

    while ((listener = [enumerator nextObject]))

        if ([[listener connectionForProxy] isEqualTo:connection])

            [listenersToRemove addObject:listener];

    ASSERT(listenerCount >= [listenersToRemove count]);

    listenerCount -= [listenersToRemove count];

    [listeners minusSet:listenersToRemove];

    [listenersToRemove release];

    if (!listenerCount)

        [self detachScriptDebuggerFromAllWebViews];

}

- (oneway void)addListener:(id<WebScriptDebugListener>)listener

{

    // can't use isKindOfClass: here because that will send over the wire and not check the proxy object

    if (!listener || [listener class] != [NSDistantObject class] || ![listener conformsToProtocol:@protocol(WebScriptDebugListener)])

        return;

    listenerCount++;

    [listeners addObject:listener];

    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(listenerConnectionDidDie:) name:NSConnectionDidDieNotification object:[(NSDistantObject *)listener connectionForProxy]];

    if (listenerCount == 1)

        [self attachScriptDebuggerToAllWebViews];

}

- (oneway void)removeListener:(id<WebScriptDebugListener>)listener

{

    if (!listener || ![listeners containsObject:listener])

        return;

    ASSERT(listenerCount >= 1);

    listenerCount--;

    [[NSNotificationCenter defaultCenter] removeObserver:self name:NSConnectionDidDieNotification object:[(NSDistantObject *)listener connectionForProxy]];

    [listeners removeObject:listener];

    if (!listenerCount)

        [self detachScriptDebuggerFromAllWebViews];

}

- (oneway void)step

{

    step = YES;

    paused = NO;

}

- (oneway void)pause

{

    paused = YES;

    step = NO;

}

- (oneway void)resume

{

    paused = NO;

    step = NO;

}

- (oneway BOOL)isPaused

{

    return paused;

}

- (void)suspendProcessIfPaused

{

    // this method will suspend this process when called during the dubugging callbacks

    // we need to do this to implement breakpoints and pausing of JavaScript

    while (paused)

        [[NSRunLoop currentRunLoop] runMode:NSConnectionReplyMode beforeDate:[NSDate distantFuture]];

    if (step) {

        step = NO;

        paused = YES;

    }

}

- (void)webView:(WebView *)webView didLoadMainResourceForDataSource:(WebDataSource *)dataSource

{

    if (![listeners count] || inCallback)

        return;

    inCallback = YES;

    NSEnumerator *enumerator = [listeners objectEnumerator];

    NSDistantObject <WebScriptDebugListener> *listener = nil;

    while ((listener = [enumerator nextObject])) {

        if ([[listener connectionForProxy] isValid])

            [listener webView:webView didLoadMainResourceForDataSource:dataSource];

    }

    inCallback = NO;

}

- (void)webView:(WebView *)webView       didParseSource:(NSString *)source

                                         baseLineNumber:(NSUInteger)lineNumber

                                                fromURL:(NSURL *)url

                                               sourceId:(int)sid

                                            forWebFrame:(WebFrame *)webFrame

{

    if (![listeners count] || inCallback)

        return;

    inCallback = YES;

    NSEnumerator *enumerator = [listeners objectEnumerator];

    NSDistantObject <WebScriptDebugListener> *listener = nil;

    while ((listener = [enumerator nextObject])) {

        if ([[listener connectionForProxy] isValid])

            [listener webView:webView didParseSource:source baseLineNumber:lineNumber fromURL:url sourceId:sid forWebFrame:webFrame];

    }

    inCallback = NO;

}

- (void)webView:(WebView *)webView  failedToParseSource:(NSString *)source

                                         baseLineNumber:(NSUInteger)lineNumber

                                                fromURL:(NSURL *)url

                                              withError:(NSError *)error

                                            forWebFrame:(WebFrame *)webFrame

{

    if (![listeners count] || inCallback)

        return;

    inCallback = YES;

    NSEnumerator *enumerator = [listeners objectEnumerator];

    NSDistantObject <WebScriptDebugListener> *listener = nil;

    while ((listener = [enumerator nextObject])) {

        if ([[listener connectionForProxy] isValid])

            [listener webView:webView failedToParseSource:source baseLineNumber:lineNumber fromURL:url withError:error forWebFrame:webFrame];

    }

    inCallback = NO;

}

- (void)webView:(WebView *)webView    didEnterCallFrame:(WebScriptCallFrame *)frame

                                               sourceId:(int)sid

                                                   line:(int)lineno

                                            forWebFrame:(WebFrame *)webFrame

{

    if (![listeners count] || inCallback)

        return;

    inCallback = YES;

    NSEnumerator *enumerator = [listeners objectEnumerator];

    NSDistantObject <WebScriptDebugListener> *listener = nil;

    while ((listener = [enumerator nextObject])) {

        if ([[listener connectionForProxy] isValid])

            [listener webView:webView didEnterCallFrame:frame sourceId:sid line:lineno forWebFrame:webFrame];

    }

    [self suspendProcessIfPaused];

    inCallback = NO;

}

- (void)webView:(WebView *)webView willExecuteStatement:(WebScriptCallFrame *)frame

                                               sourceId:(int)sid

                                                   line:(int)lineno

                                            forWebFrame:(WebFrame *)webFrame

{

    if (![listeners count] || inCallback)

        return;

    inCallback = YES;

    NSEnumerator *enumerator = [listeners objectEnumerator];

    NSDistantObject <WebScriptDebugListener> *listener = nil;

    while ((listener = [enumerator nextObject])) {

        if ([[listener connectionForProxy] isValid])

            [listener webView:webView willExecuteStatement:frame sourceId:sid line:lineno forWebFrame:webFrame];

    }

    [self suspendProcessIfPaused];

    inCallback = NO;

}

- (void)webView:(WebView *)webView   willLeaveCallFrame:(WebScriptCallFrame *)frame

                                               sourceId:(int)sid

                                                   line:(int)lineno

                                            forWebFrame:(WebFrame *)webFrame

{

    if (![listeners count] || inCallback)

        return;

    inCallback = YES;

    NSEnumerator *enumerator = [listeners objectEnumerator];

    NSDistantObject <WebScriptDebugListener> *listener = nil;

    while ((listener = [enumerator nextObject])) {

        if ([[listener connectionForProxy] isValid])

            [listener webView:webView willLeaveCallFrame:frame sourceId:sid line:lineno forWebFrame:webFrame];

    }

    [self suspendProcessIfPaused];

    inCallback = NO;

}

- (void)webView:(WebView *)webView   exceptionWasRaised:(WebScriptCallFrame *)frame

                                               sourceId:(int)sid

                                                   line:(int)lineno

                                            forWebFrame:(WebFrame *)webFrame

{

    if (![listeners count] || inCallback)

        return;

    inCallback = YES;

    NSEnumerator *enumerator = [listeners objectEnumerator];

    NSDistantObject <WebScriptDebugListener> *listener = nil;

    while ((listener = [enumerator nextObject])) {

        if ([[listener connectionForProxy] isValid])

            [listener webView:webView exceptionWasRaised:frame sourceId:sid line:lineno forWebFrame:webFrame];

    }

    [self suspendProcessIfPaused];

    inCallback = NO;

}

@end