/*

 IMPORTANT:  This Apple software is supplied to you by Apple Computer, Inc. ("Apple") in

 consideration of your agreement to the following terms, and your use, installation, 

 modification or redistribution of this Apple software constitutes acceptance of these 

 terms.  If you do not agree with these terms, please do not use, install, modify or 

 redistribute this Apple software.

 In consideration of your agreement to abide by the following terms, and subject to these 

 terms, Apple grants you a personal, non-exclusive license, under Apple’s copyrights in 

 this original Apple software (the "Apple Software"), to use, reproduce, modify and 

 redistribute the Apple Software, with or without modifications, in source and/or binary 

 forms; provided that if you redistribute the Apple Software in its entirety and without 

 modifications, you must retain this notice and the following text and disclaimers in all 

 such redistributions of the Apple Software.  Neither the name, trademarks, service marks 

 or logos of Apple Computer, Inc. may be used to endorse or promote products derived from 

 the Apple Software without specific prior written permission from Apple. Except as expressly

 stated in this notice, no other rights or licenses, express or implied, are granted by Apple

 herein, including but not limited to any patent rights that may be infringed by your 

 derivative works or by other works in which the Apple Software may be incorporated.

 The Apple Software is provided by Apple on an "AS IS" basis.  APPLE MAKES NO WARRANTIES, 

 EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED WARRANTIES OF NON-INFRINGEMENT, 

 MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE, REGARDING THE APPLE SOFTWARE OR ITS 

 USE AND OPERATION ALONE OR IN COMBINATION WITH YOUR PRODUCTS.

 IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL OR CONSEQUENTIAL 

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

 OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) ARISING IN ANY WAY OUT OF THE USE, 

 REPRODUCTION, MODIFICATION AND/OR DISTRIBUTION OF THE APPLE SOFTWARE, HOWEVER CAUSED AND 

 WHETHER UNDER THEORY OF CONTRACT, TORT (INCLUDING NEGLIGENCE), STRICT LIABILITY OR 

 OTHERWISE, EVEN IF APPLE HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */

#import <WebKit/npapi.h>

#import <WebKit/npfunctions.h>

#import <WebKit/npruntime.h>

#import <Cocoa/Cocoa.h>

// Browser function table

static NPNetscapeFuncs* browser;

// Structure for per-instance storage

typedef struct PluginObject

{

    NPP npp;

    NPWindow window;

    bool pluginHasFocus;

    bool textFieldHasFocus;

    NSRect textFieldRect;

    NSRange selectedRange;

    NSTextStorage *textStorage;

    NSLayoutManager *layoutManager;

    NSTextContainer *textContainer;

} PluginObject;

NPError NPP_New(NPMIMEType pluginType, NPP instance, uint16 mode, int16 argc, char* argn[], char* argv[], NPSavedData* saved);

NPError NPP_Destroy(NPP instance, NPSavedData** save);

NPError NPP_SetWindow(NPP instance, NPWindow* window);

NPError NPP_NewStream(NPP instance, NPMIMEType type, NPStream* stream, NPBool seekable, uint16* stype);

NPError NPP_DestroyStream(NPP instance, NPStream* stream, NPReason reason);

int32 NPP_WriteReady(NPP instance, NPStream* stream);

int32 NPP_Write(NPP instance, NPStream* stream, int32 offset, int32 len, void* buffer);

void NPP_StreamAsFile(NPP instance, NPStream* stream, const char* fname);

void NPP_Print(NPP instance, NPPrint* platformPrint);

int16 NPP_HandleEvent(NPP instance, void* event);

void NPP_URLNotify(NPP instance, const char* URL, NPReason reason, void* notifyData);

NPError NPP_GetValue(NPP instance, NPPVariable variable, void* value);

NPError NPP_SetValue(NPP instance, NPNVariable variable, void* value);

#pragma export on

// Mach-o entry points

NPError NP_Initialize(NPNetscapeFuncs* browserFuncs);

NPError NP_GetEntryPoints(NPPluginFuncs* pluginFuncs);

void NP_Shutdown(void);

#pragma export off

NPError NP_Initialize(NPNetscapeFuncs* browserFuncs)

{

    browser = browserFuncs;

    return NPERR_NO_ERROR;

}

NPError NP_GetEntryPoints(NPPluginFuncs* pluginFuncs)

{

    pluginFuncs->version = 11;

    pluginFuncs->size = sizeof(pluginFuncs);

    pluginFuncs->newp = NPP_New;

    pluginFuncs->destroy = NPP_Destroy;

    pluginFuncs->setwindow = NPP_SetWindow;

    pluginFuncs->newstream = NPP_NewStream;

    pluginFuncs->destroystream = NPP_DestroyStream;

    pluginFuncs->asfile = NPP_StreamAsFile;

    pluginFuncs->writeready = NPP_WriteReady;

    pluginFuncs->write = (NPP_WriteProcPtr)NPP_Write;

    pluginFuncs->print = NPP_Print;

    pluginFuncs->event = NPP_HandleEvent;

    pluginFuncs->urlnotify = NPP_URLNotify;

    pluginFuncs->getvalue = NPP_GetValue;

    pluginFuncs->setvalue = NPP_SetValue;

    return NPERR_NO_ERROR;

}

void NP_Shutdown(void)

{

}

NPError NPP_New(NPMIMEType pluginType, NPP instance, uint16 mode, int16 argc, char* argn[], char* argv[], NPSavedData* saved)

{

    // Create per-instance storage

    PluginObject* obj = (PluginObject*)malloc(sizeof(PluginObject));

    bzero(obj, sizeof(PluginObject));

    obj->npp = instance;

    instance->pdata = obj;

    // Ask the browser if it supports the CoreGraphics drawing model

    NPBool supportsCoreGraphics;

    if (browser->getvalue(instance, NPNVsupportsCoreGraphicsBool, &supportsCoreGraphics) != NPERR_NO_ERROR)

        supportsCoreGraphics = FALSE;

    if (!supportsCoreGraphics)

        return NPERR_INCOMPATIBLE_VERSION_ERROR;

    // If the browser supports the CoreGraphics drawing model, enable it.

    browser->setvalue(instance, NPPVpluginDrawingModel, (void*)NPDrawingModelCoreGraphics);

    // If the browser supports the Cocoa event model, enable it.

    NPBool supportsCocoa;

    if (browser->getvalue(instance, NPNVsupportsCocoaBool, &supportsCocoa) != NPERR_NO_ERROR)

        supportsCocoa = FALSE;

    if (!supportsCocoa)

        return NPERR_INCOMPATIBLE_VERSION_ERROR;

    browser->setvalue(instance, NPPVpluginEventModel, (void*)NPEventModelCocoa);

    obj->textFieldRect = NSMakeRect(10, 10, 200, 100);

    obj->textStorage = [[NSTextStorage alloc] initWithString:@""];

    obj->layoutManager = [[NSLayoutManager alloc] init];

    [obj->textStorage addLayoutManager:obj->layoutManager];

    obj->textContainer = [[NSTextContainer alloc] initWithContainerSize:obj->textFieldRect.size];

    [obj->layoutManager addTextContainer:obj->textContainer];

    obj->selectedRange.location = [obj->textStorage length];

    return NPERR_NO_ERROR;

}

NPError NPP_Destroy(NPP instance, NPSavedData** save)

{

    // Free per-instance storage

    PluginObject* obj = instance->pdata;

    [obj->textStorage release];

    [obj->layoutManager release];

    free(obj);

    return NPERR_NO_ERROR;

}

NPError NPP_SetWindow(NPP instance, NPWindow* window)

{

    PluginObject* obj = instance->pdata;

    obj->window = *window;

    return NPERR_NO_ERROR;

}

NPError NPP_NewStream(NPP instance, NPMIMEType type, NPStream* stream, NPBool seekable, uint16* stype)

{

    *stype = NP_ASFILEONLY;

    return NPERR_NO_ERROR;

}

NPError NPP_DestroyStream(NPP instance, NPStream* stream, NPReason reason)

{

    return NPERR_NO_ERROR;

}

int32 NPP_WriteReady(NPP instance, NPStream* stream)

{

    return 0;

}

int32 NPP_Write(NPP instance, NPStream* stream, int32 offset, int32 len, void* buffer)

{

    return 0;

}

void NPP_StreamAsFile(NPP instance, NPStream* stream, const char* fname)

{

}

void NPP_Print(NPP instance, NPPrint* platformPrint)

{

}

static void handleDraw(PluginObject* obj, NPCocoaEvent *event)

{

    NSGraphicsContext *oldContext = [[NSGraphicsContext currentContext] retain];

    NSGraphicsContext *context = [NSGraphicsContext graphicsContextWithGraphicsPort:event->data.draw.context

                                                                            flipped:YES];

    [NSGraphicsContext setCurrentContext:context];

    NSRect rect = NSMakeRect(0, 0, obj->window.width, obj->window.height);

    [[NSColor lightGrayColor] set];

    [NSBezierPath fillRect:rect];

    if (obj->pluginHasFocus) {

        [[NSColor blackColor] set];

        [NSBezierPath strokeRect:rect];

    }

    [[NSColor whiteColor] set];

    [NSBezierPath fillRect:obj->textFieldRect];

    // Draw the text

    NSRange glyphRange = [obj->layoutManager glyphRangeForTextContainer:obj->textContainer];

    if (glyphRange.length > 0) {

        [obj->layoutManager drawBackgroundForGlyphRange:glyphRange atPoint:obj->textFieldRect.origin];

        [obj->layoutManager drawGlyphsForGlyphRange:glyphRange atPoint:obj->textFieldRect.origin];

    }

    NSBezierPath *textInputBorder = [NSBezierPath bezierPathWithRect:obj->textFieldRect];

    [[NSColor blackColor] set];

    if (obj->pluginHasFocus && obj->textFieldHasFocus)

        [textInputBorder setLineWidth:2];

    else

        [textInputBorder setLineWidth:1];

    [textInputBorder stroke];

    if (obj->pluginHasFocus && obj->textFieldHasFocus) {

        NSUInteger rectCount;

        NSRect *rectArray = [obj->layoutManager rectArrayForCharacterRange:obj->selectedRange

                                            withinSelectedCharacterRange:obj->selectedRange

                                                        inTextContainer:obj->textContainer

                                                                rectCount:&rectCount];

        [[NSColor blackColor] set];

        for (unsigned i = 0; i < rectCount; i++) {

            NSRect rect = rectArray[i];

            rect.origin.x += obj->textFieldRect.origin.x;

            rect.origin.y += obj->textFieldRect.origin.y;

            [NSBezierPath strokeRect:rect];

        }        

    }

    [NSGraphicsContext setCurrentContext:oldContext];

}

static void invalidatePlugin(PluginObject* obj)

{

    NPRect rect;

    rect.left = 0;

    rect.top = 0;

    rect.right = obj->window.width;

    rect.bottom = obj->window.height;

    browser->invalidaterect(obj->npp, &rect);    

}

static void handleFocusChanged(NPCocoaEvent* cocoaEvent, PluginObject* obj)

{

    obj->pluginHasFocus = cocoaEvent->data.focus.hasFocus;

    invalidatePlugin(obj);

}

static void handleMouseMoved(NPCocoaEvent* cocoaEvent, PluginObject* obj)

{

    NSPoint point = NSMakePoint(cocoaEvent->data.mouse.pluginX, cocoaEvent->data.mouse.pluginY);

    if (NSPointInRect(point, obj->textFieldRect))

        [[NSCursor IBeamCursor] set];

    else

        [[NSCursor arrowCursor] set];

}

static void handleMouseDown(NPCocoaEvent* cocoaEvent, PluginObject* obj) 

{

    NSPoint point = NSMakePoint(cocoaEvent->data.mouse.pluginX, cocoaEvent->data.mouse.pluginY);

    obj->textFieldHasFocus = NSPointInRect(point, obj->textFieldRect);

    invalidatePlugin(obj);

}

static int16_t handleTextFieldKeyDown(NPCocoaEvent* event, PluginObject* obj)

{

    NSString *string = (NSString *)event->data.key.charactersIgnoringModifiers;

    unichar c = [string length] > 0 ? [string characterAtIndex:0] : 0;

    switch (c) {

        case NSLeftArrowFunctionKey:

            if (obj->selectedRange.location > 0) {

                obj->selectedRange.location--;

                invalidatePlugin(obj);

            }

            return 1;

        case NSRightArrowFunctionKey:

            if (obj->selectedRange.location < [obj->textStorage length]) {

                obj->selectedRange.location++;  

                invalidatePlugin(obj);

            }

            return 1;

        default:

            // Return 0 and let the text input system handle it.

            return 0;

    }

}

static int16_t handleTextInput(NPCocoaEvent* event, PluginObject* obj)

{

    NSString *string = (NSString *)event->data.text.text;

    NSRange range = obj->selectedRange;

    [obj->textStorage replaceCharactersInRange:range withString:string];

    obj->selectedRange.location = range.location + [string length];

    obj->selectedRange.length = 0;

    invalidatePlugin(obj);

    return 1;

}

int16 NPP_HandleEvent(NPP instance, void* event)

{

    PluginObject* obj = instance->pdata;

    NPCocoaEvent* cocoaEvent = event;

    switch (cocoaEvent->type) {

        case NPCocoaEventDrawRect:

            handleDraw(obj, cocoaEvent);

            return 1;

        case NPCocoaEventFocusChanged:

            handleFocusChanged(cocoaEvent, obj);

            return 1;

        case NPCocoaEventMouseMoved:

            handleMouseMoved(cocoaEvent, obj);

            return 1;

        case NPCocoaEventMouseDown:

            handleMouseDown(cocoaEvent, obj);

            return 1;

        case NPCocoaEventKeyDown:

            // If the text field has focus we ignore the event, causing it

            // to be sent to the input manager.

            if (obj->textFieldHasFocus)

                return handleTextFieldKeyDown(cocoaEvent, obj);

            else

                return 1;

        case NPCocoaEventTextInput:

            return handleTextInput(cocoaEvent, obj);

            return 1;

    }

    return 0;

}

void NPP_URLNotify(NPP instance, const char* url, NPReason reason, void* notifyData)

{

}

NPError NPP_GetValue(NPP instance, NPPVariable variable, void* value)

{

    return NPERR_GENERIC_ERROR;

}

NPError NPP_SetValue(NPP instance, NPNVariable variable, void* value)

{

    return NPERR_GENERIC_ERROR;

}