/*

 * Copyright (C) 2004, 2005, 2006 Apple Computer, Inc.  All rights reserved.

 * Copyright (C) 2005, 2006 Alexey Proskuryakov (ap@nypop.com)

 * Copyright (C) 2006 David Smith (catfish.man@gmail.com)

 *

 * 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.

 *

 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``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 COMPUTER, INC. OR

 * 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 "config.h"

#import "WebCoreFrameBridge.h"

#import "AXObjectCache.h"

#import "CSSHelper.h"

#import "Cache.h"

#import "ClipboardMac.h"

#import "ColorMac.h"

#import "DOMImplementation.h"

#import "DOMInternal.h"

#import "DOMWindow.h"

#import "DeleteSelectionCommand.h"

#import "DocLoader.h"

#import "DocumentFragment.h"

#import "DocumentLoader.h"

#import "DocumentType.h"

#import "Editor.h"

#import "EditorClient.h"

#import "EventHandler.h"

#import "FloatRect.h"

#import "FontData.h"

#import "FormDataStreamMac.h"

#import "Frame.h"

#import "FrameLoader.h"

#import "FrameLoaderClient.h"

#import "FrameTree.h"

#import "FrameView.h"

#import "GraphicsContext.h"

#import "HTMLDocument.h"

#import "HTMLFormElement.h"

#import "HTMLInputElement.h"

#import "HTMLNames.h"

#import "HitTestResult.h"

#import "Image.h"

#import "LoaderNSURLExtras.h"

#import "MoveSelectionCommand.h"

#import "Page.h"

#import "PlatformMouseEvent.h"

#import "PlatformScreen.h"

#import "PlugInInfoStore.h"

#import "RenderImage.h"

#import "RenderPart.h"

#import "RenderTreeAsText.h"

#import "RenderView.h"

#import "RenderWidget.h"

#import "ReplaceSelectionCommand.h"

#import "ResourceRequest.h"

#import "SelectionController.h"

#import "SmartReplace.h"

#import "SubresourceLoader.h"

#import "SystemTime.h"

#import "Text.h"

#import "TextEncoding.h"

#import "TextIterator.h"

#import "TextResourceDecoder.h"

#import "TypingCommand.h"

#import "WebCoreSystemInterface.h"

#import "WebCoreViewFactory.h"

#import "XMLTokenizer.h"

#import "htmlediting.h"

#import "kjs_proxy.h"

#import "kjs_window.h"

#import "markup.h"

#import "visible_units.h"

#import <OpenScripting/ASRegistry.h>

#import <JavaScriptCore/array_instance.h>

#import <JavaScriptCore/date_object.h>

#import <JavaScriptCore/runtime_root.h>

#import <wtf/RetainPtr.h>

@class NSView;

using namespace std;

using namespace WebCore;

using namespace HTMLNames;

using KJS::ArrayInstance;

using KJS::BooleanType;

using KJS::DateInstance;

using KJS::ExecState;

using KJS::GetterSetterType;

using KJS::JSImmediate;

using KJS::JSLock;

using KJS::JSObject;

using KJS::JSValue;

using KJS::NullType;

using KJS::NumberType;

using KJS::ObjectType;

using KJS::SavedBuiltins;

using KJS::SavedProperties;

using KJS::StringType;

using KJS::UndefinedType;

using KJS::UnspecifiedType;

using KJS::Window;

using KJS::Bindings::RootObject;

static PassRefPtr<RootObject> createRootObject(void* nativeHandle)

{

    NSView *view = (NSView *)nativeHandle;

    WebCoreFrameBridge *bridge = [[WebCoreViewFactory sharedFactory] bridgeForView:view];

    if (!bridge)

        return 0;

    Frame* frame = [bridge _frame];

    return frame->createRootObject(nativeHandle, frame->scriptProxy()->interpreter());

}

static pthread_t mainThread = 0;

static void updateRenderingForBindings(ExecState* exec, JSObject* rootObject)

{

    if (pthread_self() != mainThread)

        return;

    if (!rootObject)

        return;

    Window* window = static_cast<Window*>(rootObject);

    if (!window)

        return;

    if (Frame* frame = window->impl()->frame())

        if (Document* doc = frame->document())

            doc->updateRendering();

}

static NSAppleEventDescriptor* aeDescFromJSValue(ExecState* exec, JSValue* jsValue)

{

    NSAppleEventDescriptor* aeDesc = 0;

    switch (jsValue->type()) {

        case BooleanType:

            aeDesc = [NSAppleEventDescriptor descriptorWithBoolean:jsValue->getBoolean()];

            break;

        case StringType:

            aeDesc = [NSAppleEventDescriptor descriptorWithString:String(jsValue->getString())];

            break;

        case NumberType: {

            double value = jsValue->getNumber();

            int intValue = (int)value;

            if (value == intValue)

                aeDesc = [NSAppleEventDescriptor descriptorWithDescriptorType:typeSInt32 bytes:&intValue length:sizeof(intValue)];

            else

                aeDesc = [NSAppleEventDescriptor descriptorWithDescriptorType:typeIEEE64BitFloatingPoint bytes:&value length:sizeof(value)];

            break;

        }

        case ObjectType: {

            JSObject* object = jsValue->getObject();

            if (object->inherits(&DateInstance::info)) {

                DateInstance* date = static_cast<DateInstance*>(object);

                double ms = 0;

                int tzOffset = 0;

                if (date->getTime(ms, tzOffset)) {

                    CFAbsoluteTime utcSeconds = ms / 1000 - kCFAbsoluteTimeIntervalSince1970;

                    LongDateTime ldt;

                    if (noErr == UCConvertCFAbsoluteTimeToLongDateTime(utcSeconds, &ldt))

                        aeDesc = [NSAppleEventDescriptor descriptorWithDescriptorType:typeLongDateTime bytes:&ldt length:sizeof(ldt)];

                }

            }

            else if (object->inherits(&ArrayInstance::info)) {

                static HashSet<JSObject*> visitedElems;

                if (!visitedElems.contains(object)) {

                    visitedElems.add(object);

                    ArrayInstance* array = static_cast<ArrayInstance*>(object);

                    aeDesc = [NSAppleEventDescriptor listDescriptor];

                    unsigned numItems = array->getLength();

                    for (unsigned i = 0; i < numItems; ++i)

                        [aeDesc insertDescriptor:aeDescFromJSValue(exec, array->getItem(i)) atIndex:0];

                    visitedElems.remove(object);

                }

            }

            if (!aeDesc) {

                JSValue* primitive = object->toPrimitive(exec);

                if (exec->hadException()) {

                    exec->clearException();

                    return [NSAppleEventDescriptor nullDescriptor];

                }

                return aeDescFromJSValue(exec, primitive);

            }

            break;

        }

        case UndefinedType:

            aeDesc = [NSAppleEventDescriptor descriptorWithTypeCode:cMissingValue];

            break;

        default:

            LOG_ERROR("Unknown JavaScript type: %d", jsValue->type());

            // no break;

        case UnspecifiedType:

        case NullType:

        case GetterSetterType:

            aeDesc = [NSAppleEventDescriptor nullDescriptor];

            break;

    }

    return aeDesc;

}

@implementation WebCoreFrameBridge

static inline WebCoreFrameBridge *bridge(Frame *frame)

{

    if (!frame)

        return nil;

    return frame->bridge();

}

- (NSString *)domain

{

    Document *doc = m_frame->document();

    if (doc)

        return doc->domain();

    return nil;

}

+ (WebCoreFrameBridge *)bridgeForDOMDocument:(DOMDocument *)document

{

    return bridge([document _document]->frame());

}

- (id)init

{

    static bool initializedKJS;

    if (!initializedKJS) {

        initializedKJS = true;

        mainThread = pthread_self();

        RootObject::setCreateRootObject(createRootObject);

        KJS::Bindings::Instance::setDidExecuteFunction(updateRenderingForBindings);

    }

    if (!(self = [super init]))

        return nil;

    _shouldCreateRenderers = YES;

    return self;

}

- (void)dealloc

{

    ASSERT(_closed);

    [super dealloc];

}

- (void)finalize

{

    ASSERT(_closed);

    [super finalize];

}

- (void)close

{

    [self clearFrame];

    _closed = YES;

}

- (void)addData:(NSData *)data

{

    Document *doc = m_frame->document();

    // Document may be nil if the part is about to redirect

    // as a result of JS executing during load, i.e. one frame

    // changing another's location before the frame's document

    // has been created. 

    if (doc) {

        doc->setShouldCreateRenderers([self shouldCreateRenderers]);

        m_frame->loader()->addData((const char *)[data bytes], [data length]);

    }

}

- (BOOL)scrollOverflowInDirection:(WebScrollDirection)direction granularity:(WebScrollGranularity)granularity

{

    if (!m_frame)

        return NO;

    return m_frame->eventHandler()->scrollOverflow((ScrollDirection)direction, (ScrollGranularity)granularity);

}

- (void)clearFrame

{

    m_frame = 0;

}

- (void)createFrameViewWithNSView:(NSView *)view marginWidth:(int)mw marginHeight:(int)mh

{

    // If we own the view, delete the old one - otherwise the render m_frame will take care of deleting the view.

    if (m_frame)

        m_frame->setView(0);

    FrameView* frameView = new FrameView(m_frame);

    m_frame->setView(frameView);

    frameView->deref();

    frameView->setView(view);

    if (mw >= 0)

        frameView->setMarginWidth(mw);

    if (mh >= 0)

        frameView->setMarginHeight(mh);

}

- (NSString *)_stringWithDocumentTypeStringAndMarkupString:(NSString *)markupString

{

    return m_frame->documentTypeString() + markupString;

}

- (NSArray *)nodesFromList:(Vector<Node*> *)nodesVector

{

    size_t size = nodesVector->size();

    NSMutableArray *nodes = [NSMutableArray arrayWithCapacity:size];

    for (size_t i = 0; i < size; ++i)

        [nodes addObject:[DOMNode _wrapNode:(*nodesVector)[i]]];

    return nodes;

}

- (NSString *)markupStringFromNode:(DOMNode *)node nodes:(NSArray **)nodes

{

    // FIXME: This is never "for interchange". Is that right? See the next method.

    Vector<Node*> nodeList;

    NSString *markupString = createMarkup([node _node], IncludeNode, nodes ? &nodeList : 0).getNSString();

    if (nodes)

        *nodes = [self nodesFromList:&nodeList];

    return [self _stringWithDocumentTypeStringAndMarkupString:markupString];

}

- (NSString *)markupStringFromRange:(DOMRange *)range nodes:(NSArray **)nodes

{

    // FIXME: This is always "for interchange". Is that right? See the previous method.

    Vector<Node*> nodeList;

    NSString *markupString = createMarkup([range _range], nodes ? &nodeList : 0, AnnotateForInterchange).getNSString();

    if (nodes)

        *nodes = [self nodesFromList:&nodeList];

    return [self _stringWithDocumentTypeStringAndMarkupString:markupString];

}

- (NSString *)selectedString

{

    String text = m_frame->selectedText();

    text.replace('\\', m_frame->backslashAsCurrencySymbol());

    return text;

}

- (NSString *)stringForRange:(DOMRange *)range

{

    // This will give a system malloc'd buffer that can be turned directly into an NSString

    unsigned length;

    UChar* buf = plainTextToMallocAllocatedBuffer([range _range], length);

    if (!buf)

        return [NSString string];

    UChar backslashAsCurrencySymbol = m_frame->backslashAsCurrencySymbol();

    if (backslashAsCurrencySymbol != '\\')

        for (unsigned n = 0; n < length; n++) 

            if (buf[n] == '\\')

                buf[n] = backslashAsCurrencySymbol;

    // Transfer buffer ownership to NSString

    return [[[NSString alloc] initWithCharactersNoCopy:buf length:length freeWhenDone:YES] autorelease];

}

- (void)reapplyStylesForDeviceType:(WebCoreDeviceType)deviceType

{

    if (m_frame->view())

        m_frame->view()->setMediaType(deviceType == WebCoreDeviceScreen ? "screen" : "print");

    Document *doc = m_frame->document();

    if (doc)

        doc->setPrinting(deviceType == WebCoreDevicePrinter);

    m_frame->reparseConfiguration();

}

- (void)forceLayoutAdjustingViewSize:(BOOL)flag

{

    m_frame->forceLayout(!flag);

    if (flag)

        m_frame->view()->adjustViewSize();

}

- (void)forceLayoutWithMinimumPageWidth:(float)minPageWidth maximumPageWidth:(float)maxPageWidth adjustingViewSize:(BOOL)flag

{

    m_frame->forceLayoutWithPageWidthRange(minPageWidth, maxPageWidth, flag);

}

- (void)sendResizeEvent

{

    m_frame->sendResizeEvent();

}

- (void)sendScrollEvent

{

    m_frame->sendScrollEvent();

}

- (void)drawRect:(NSRect)rect

{

    PlatformGraphicsContext* platformContext = static_cast<PlatformGraphicsContext*>([[NSGraphicsContext currentContext] graphicsPort]);

    ASSERT([[NSGraphicsContext currentContext] isFlipped]);

    GraphicsContext context(platformContext);

    m_frame->paint(&context, enclosingIntRect(rect));

}

// Used by pagination code called from AppKit when a standalone web page is printed.

- (NSArray*)computePageRectsWithPrintWidthScaleFactor:(float)printWidthScaleFactor printHeight:(float)printHeight

{

    NSMutableArray* pages = [NSMutableArray arrayWithCapacity:5];

    if (printWidthScaleFactor <= 0) {

        LOG_ERROR("printWidthScaleFactor has bad value %.2f", printWidthScaleFactor);

        return pages;

    }

    if (printHeight <= 0) {

        LOG_ERROR("printHeight has bad value %.2f", printHeight);

        return pages;

    }

    if (!m_frame || !m_frame->document() || !m_frame->view()) return pages;

    RenderView* root = static_cast<RenderView *>(m_frame->document()->renderer());

    if (!root) return pages;

    FrameView* view = m_frame->view();

    if (!view)

        return pages;

    NSView* documentView = view->getDocumentView();

    if (!documentView)

        return pages;

    float currPageHeight = printHeight;

    float docHeight = root->layer()->height();

    float docWidth = root->layer()->width();

    float printWidth = docWidth/printWidthScaleFactor;

    // We need to give the part the opportunity to adjust the page height at each step.

    for (float i = 0; i < docHeight; i += currPageHeight) {

        float proposedBottom = min(docHeight, i + printHeight);

        m_frame->adjustPageHeight(&proposedBottom, i, proposedBottom, i);

        currPageHeight = max(1.0f, proposedBottom - i);

        for (float j = 0; j < docWidth; j += printWidth) {

            NSValue* val = [NSValue valueWithRect: NSMakeRect(j, i, printWidth, currPageHeight)];

            [pages addObject: val];

        }

    }

    return pages;

}

// This is to support the case where a webview is embedded in the view that's being printed

- (void)adjustPageHeightNew:(float *)newBottom top:(float)oldTop bottom:(float)oldBottom limit:(float)bottomLimit

{

    m_frame->adjustPageHeight(newBottom, oldTop, oldBottom, bottomLimit);

}

- (NSObject *)copyRenderNode:(RenderObject *)node copier:(id <WebCoreRenderTreeCopier>)copier

{

    NSMutableArray *children = [[NSMutableArray alloc] init];

    for (RenderObject *child = node->firstChild(); child; child = child->nextSibling()) {

        [children addObject:[self copyRenderNode:child copier:copier]];

    }

    NSString *name = [[NSString alloc] initWithUTF8String:node->renderName()];

    RenderWidget* renderWidget = node->isWidget() ? static_cast<RenderWidget*>(node) : 0;

    Widget* widget = renderWidget ? renderWidget->widget() : 0;

    NSView *view = widget ? widget->getView() : nil;

    int nx, ny;

    node->absolutePosition(nx, ny);

    NSObject *copiedNode = [copier nodeWithName:name

                                       position:NSMakePoint(nx,ny)

                                           rect:NSMakeRect(node->xPos(), node->yPos(), node->width(), node->height())

                                           view:view

                                       children:children];

    [name release];

    [children release];

    return copiedNode;

}

- (NSObject *)copyRenderTree:(id <WebCoreRenderTreeCopier>)copier

{

    RenderObject *renderer = m_frame->renderer();

    if (!renderer) {

        return nil;

    }

    return [self copyRenderNode:renderer copier:copier];

}

- (void)installInFrame:(NSView *)view

{

    // If this isn't the main frame, it must have a render m_frame set, or it

    // won't ever get installed in the view hierarchy.

    ASSERT(m_frame == m_frame->page()->mainFrame() || m_frame->ownerElement());

    m_frame->view()->setView(view);

    // FIXME: frame tries to do this too, is it needed?

    if (m_frame->ownerRenderer()) {

        m_frame->ownerRenderer()->setWidget(m_frame->view());

        // Now the render part owns the view, so we don't any more.