1/*
2 * Copyright (C) 2012 Google 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 are
6 * met:
7 *
8 *     * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 *     * Redistributions in binary form must reproduce the above
11 * copyright notice, this list of conditions and the following disclaimer
12 * in the documentation and/or other materials provided with the
13 * distribution.
14 *     * Neither the name of Google Inc. nor the names of its
15 * contributors may be used to endorse or promote products derived from
16 * this software without specific prior written permission.
17 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30
31#include "config.h"
32#include "SurroundingText.h"
33
34#include "Document.h"
35#include "TextIterator.h"
36#include "VisibleSelection.h"
37#include "VisibleUnits.h"
38
39namespace WebCore {
40
41SurroundingText::SurroundingText(const VisiblePosition& visiblePosition, unsigned maxLength)
42    : m_positionOffsetInContent(0)
43{
44    if (visiblePosition.isNull())
45        return;
46
47    const unsigned halfMaxLength = maxLength / 2;
48    CharacterIterator forwardIterator(makeRange(visiblePosition, endOfDocument(visiblePosition)).get(), TextIteratorStopsOnFormControls);
49    if (!forwardIterator.atEnd())
50        forwardIterator.advance(maxLength - halfMaxLength);
51
52    Position position = visiblePosition.deepEquivalent().parentAnchoredEquivalent();
53    Document* document = position.document();
54    RefPtr<Range> forwardRange = forwardIterator.range();
55    if (!forwardRange || !Range::create(document, position, forwardRange->startPosition())->text().length()) {
56        ASSERT(forwardRange);
57        return;
58    }
59
60    BackwardsCharacterIterator backwardsIterator(makeRange(startOfDocument(visiblePosition), visiblePosition).get(), TextIteratorStopsOnFormControls);
61    if (!backwardsIterator.atEnd())
62        backwardsIterator.advance(halfMaxLength);
63
64    RefPtr<Range> backwardsRange = backwardsIterator.range();
65    if (!backwardsRange) {
66        ASSERT(backwardsRange);
67        return;
68    }
69
70    m_positionOffsetInContent = Range::create(document, backwardsRange->endPosition(), position)->text().length();
71    m_contentRange = Range::create(document, backwardsRange->endPosition(), forwardRange->startPosition());
72    ASSERT(m_contentRange);
73}
74
75PassRefPtr<Range> SurroundingText::rangeFromContentOffsets(unsigned startOffsetInContent, unsigned endOffsetInContent)
76{
77    if (startOffsetInContent >= endOffsetInContent || endOffsetInContent > content().length())
78        return 0;
79
80    CharacterIterator iterator(m_contentRange.get());
81
82    ASSERT(!iterator.atEnd());
83    iterator.advance(startOffsetInContent);
84
85    ASSERT(iterator.range());
86    Position start = iterator.range()->startPosition();
87
88    ASSERT(!iterator.atEnd());
89    iterator.advance(endOffsetInContent - startOffsetInContent);
90
91    ASSERT(iterator.range());
92    Position end = iterator.range()->startPosition();
93
94    return Range::create(start.document(), start, end);
95}
96
97String SurroundingText::content() const
98{
99    if (m_contentRange)
100        return m_contentRange->text();
101    return String();
102}
103
104unsigned SurroundingText::positionOffsetInContent() const
105{
106    return m_positionOffsetInContent;
107}
108
109} // namespace WebCore
110