Move most of TextIterator off of live ranges
[WebKit-https.git] / Source / WebKitLegacy / mac / WebView / WebTextIterator.mm
1 /*
2  * Copyright (C) 2008-2020 Apple Inc. All Rights Reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #import "WebTextIterator.h"
27
28 #import "DOMNodeInternal.h"
29 #import "DOMRangeInternal.h"
30 #import "WebTypesInternal.h"
31 #import <JavaScriptCore/InitializeThreading.h>
32 #import <WebCore/Range.h>
33 #import <WebCore/TextIterator.h>
34 #import <wtf/MainThread.h>
35 #import <wtf/RunLoop.h>
36 #import <wtf/Vector.h>
37
38 @interface WebTextIteratorPrivate : NSObject {
39 @public
40     std::unique_ptr<WebCore::TextIterator> _textIterator;
41     Vector<unichar> _upconvertedText;
42 }
43 @end
44
45 @implementation WebTextIteratorPrivate
46
47 + (void)initialize
48 {
49 #if !PLATFORM(IOS_FAMILY)
50     JSC::initializeThreading();
51     RunLoop::initializeMainRunLoop();
52 #endif
53 }
54
55 @end
56
57 @implementation WebTextIterator
58
59 - (void)dealloc
60 {
61     [_private release];
62     [super dealloc];
63 }
64
65 - (id)initWithRange:(DOMRange *)range
66 {
67     self = [super init];
68     if (!self)
69         return self;
70     
71     _private = [[WebTextIteratorPrivate alloc] init];
72     if (!range)
73         return self;
74
75     _private->_textIterator = makeUnique<WebCore::TextIterator>(*core(range));
76     return self;
77 }
78
79 - (void)advance
80 {
81     if (_private->_textIterator)
82         _private->_textIterator->advance();
83     _private->_upconvertedText.shrink(0);
84 }
85
86 - (BOOL)atEnd
87 {
88     return _private->_textIterator && _private->_textIterator->atEnd();
89 }
90
91 - (DOMRange *)currentRange
92 {
93     if (!_private->_textIterator)
94         return nil;
95     auto& textIterator = *_private->_textIterator;
96     if (textIterator.atEnd())
97         return nil;
98     return kit(createLiveRange(textIterator.range()).ptr());
99 }
100
101 // FIXME: Consider deprecating this method and creating one that does not require copying 8-bit characters.
102 - (const unichar*)currentTextPointer
103 {
104     if (!_private->_textIterator)
105         return nullptr;
106     StringView text = _private->_textIterator->text();
107     unsigned length = text.length();
108     if (!length)
109         return nullptr;
110     if (!text.is8Bit())
111         return text.characters16();
112     if (_private->_upconvertedText.isEmpty())
113         _private->_upconvertedText.appendRange(text.characters8(), text.characters8() + length);
114     ASSERT(_private->_upconvertedText.size() == text.length());
115     return _private->_upconvertedText.data();
116 }
117
118 - (NSUInteger)currentTextLength
119 {
120     return _private->_textIterator ? _private->_textIterator->text().length() : 0;
121 }
122
123 @end
124
125 @implementation WebTextIterator (WebTextIteratorDeprecated)
126
127 - (DOMNode *)currentNode
128 {
129     return _private->_textIterator ? kit(_private->_textIterator->node()) : nil;
130 }
131
132 - (NSString *)currentText
133 {
134     return _private->_textIterator ? _private->_textIterator->text().createNSString().autorelease() : @"";
135 }
136
137 @end