DO NOT SUBMIT

Source/WebKit/chromium/src/mac/WebTextHelper.mm

+/*
+ * Copyright (C) 2011 Google 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:
+ * 
+ *     * Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *     * 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.
+ *     * Neither the name of Google Inc. 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 THE COPYRIGHT HOLDERS AND 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 THE COPYRIGHT
+ * OWNER 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.
+ */
+
+#include "config.h"
+#include "WebFrameImpl.h"
+#include "WebTextHelper.h"
+
+#include "ColorMac.h"
+#include "Document.h"
+#include "Element.h"
+#include "Frame.h"
+#include "FrameView.h"
+#include "HTMLElement.h"
+#include "Node.h"
+#include "Range.h"
+#include "TextIterator.h"
+#include "WebRect.h"
+#include "WebString.h"
+
+#import <Cocoa/Cocoa.h>
+
+using namespace WebCore;
+
+namespace WebKit {
+
+// Helper functions.
+
+WebFrameImpl* getWebFrameImpl(WebFrame* frame)
+{
+    return static_cast<WebFrameImpl*>(frame);
+}
+
+Element* rangeScope(WebFrame* webFrame)
+{
+    Frame* frame = getWebFrameImpl(webFrame)->frame();
+    Element* selectionRoot = frame->selection()->rootEditableElement();
+    return selectionRoot ? selectionRoot : frame->document()->documentElement();
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+WebTextHelper::WebTextHelper(WebFrame* frame)
+    : m_frame(frame)
+{
+}
+
+// This function is copied from /WebKit/mac/Misc/WebNSAttributedStringExtras.mm.
+WebKit::WebString WebTextHelper::substringInRange(uint location, uint length)
+{
+    Frame* frame = getWebFrameImpl(m_frame)->frame();
+    if (frame->view()->needsLayout())
+        frame->view()->layout();
+
+    RefPtr<Range> range(TextIterator::rangeFromLocationAndLength(rangeScope(m_frame), location, length));
+    if (!range)
+        return WebKit::WebString();
+
+    NSMutableAttributedString* string = [[NSMutableAttributedString alloc] init];
+    NSMutableDictionary* attrs = [NSMutableDictionary dictionary];
+
+    unsigned position = 0;
+    for (TextIterator it(range.get()); !it.atEnd() && [string length] < length; it.advance()) {
+        unsigned numCharacters = it.length();
+        if (!numCharacters)
+            continue;
+
+        ExceptionCode exception = 0;
+        Node* container = it.range()->startContainer(exception);
+        RenderObject* renderer = container->renderer();
+        ASSERT(renderer);
+        if (!renderer)
+            continue;
+
+        RenderStyle* style = renderer->style();
+        NSFont* font = style->font().primaryFont()->getNSFont();
+        // If the platform font can't be loaded, it's likely that the site is
+        // using a web font. For now, just use the default font instead.
+        // TODO(rsesek): Change the font activation flags to allow other processes
+        // to use the font.
+        if (!font)
+          font = [NSFont systemFontOfSize:style->font().size()];
+        [attrs setObject:font forKey:NSFontAttributeName];
+
+        if (style->visitedDependentColor(CSSPropertyColor).alpha())
+            [attrs setObject:nsColor(style->visitedDependentColor(CSSPropertyColor)) forKey:NSForegroundColorAttributeName];
+        else
+            [attrs removeObjectForKey:NSForegroundColorAttributeName];
+        if (style->visitedDependentColor(CSSPropertyBackgroundColor).alpha())
+            [attrs setObject:nsColor(style->visitedDependentColor(CSSPropertyBackgroundColor)) forKey:NSBackgroundColorAttributeName];
+        else
+            [attrs removeObjectForKey:NSBackgroundColorAttributeName];
+
+        NSString* substring =
+            [[[NSString alloc] initWithCharactersNoCopy:const_cast<UChar*>(it.characters())
+                                                 length:it.length()
+                                           freeWhenDone:NO] autorelease];
+        [string replaceCharactersInRange:NSMakeRange(position, 0)
+                              withString:substring];
+        [string setAttributes:attrs range:NSMakeRange(position, numCharacters)];
+        position += numCharacters;
+    }
+    NSData* archivedData([NSArchiver archivedDataWithRootObject:string]);
+    if (!archivedData)
+        return WebString();
+    return WebKit::WebString(static_cast<const WebUChar*>([archivedData bytes]),
+        [archivedData length]);
+}
+
+} // namespace WebKit