1/*
2 * Copyright (C) 2009 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#ifndef JSStringBuilder_h
27#define JSStringBuilder_h
28
29#include "ExceptionHelpers.h"
30#include "JSString.h"
31#include "UStringConcatenate.h"
32#include "Vector.h"
33
34namespace JSC {
35
36class JSStringBuilder {
37public:
38    JSStringBuilder()
39        : m_okay(true)
40    {
41    }
42
43    void append(const UChar u)
44    {
45        m_okay &= buffer.tryAppend(&u, 1);
46    }
47
48    void append(const char* str)
49    {
50        append(str, strlen(str));
51    }
52
53    void append(const char* str, size_t len)
54    {
55        m_okay &= buffer.tryReserveCapacity(buffer.size() + len);
56        for (size_t i = 0; i < len; i++) {
57            UChar u = static_cast<unsigned char>(str[i]);
58            m_okay &= buffer.tryAppend(&u, 1);
59        }
60    }
61
62    void append(const UChar* str, size_t len)
63    {
64        m_okay &= buffer.tryAppend(str, len);
65    }
66
67    void append(const UString& str)
68    {
69        m_okay &= buffer.tryAppend(str.characters(), str.length());
70    }
71
72    JSValue build(ExecState* exec)
73    {
74        if (!m_okay)
75            return throwOutOfMemoryError(exec);
76        buffer.shrinkToFit();
77        if (!buffer.data())
78            return throwOutOfMemoryError(exec);
79        return jsString(exec, UString::adopt(buffer));
80    }
81
82protected:
83    Vector<UChar, 64> buffer;
84    bool m_okay;
85};
86
87template<typename StringType1, typename StringType2>
88inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2)
89{
90    PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2);
91    if (!result)
92        return throwOutOfMemoryError(exec);
93    return jsNontrivialString(exec, result);
94}
95
96template<typename StringType1, typename StringType2, typename StringType3>
97inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3)
98{
99    PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3);
100    if (!result)
101        return throwOutOfMemoryError(exec);
102    return jsNontrivialString(exec, result);
103}
104
105template<typename StringType1, typename StringType2, typename StringType3, typename StringType4>
106inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3, StringType4 string4)
107{
108    PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3, string4);
109    if (!result)
110        return throwOutOfMemoryError(exec);
111    return jsNontrivialString(exec, result);
112}
113
114template<typename StringType1, typename StringType2, typename StringType3, typename StringType4, typename StringType5>
115inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3, StringType4 string4, StringType5 string5)
116{
117    PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3, string4, string5);
118    if (!result)
119        return throwOutOfMemoryError(exec);
120    return jsNontrivialString(exec, result);
121}
122
123template<typename StringType1, typename StringType2, typename StringType3, typename StringType4, typename StringType5, typename StringType6>
124inline JSValue jsMakeNontrivialString(ExecState* exec, StringType1 string1, StringType2 string2, StringType3 string3, StringType4 string4, StringType5 string5, StringType6 string6)
125{
126    PassRefPtr<StringImpl> result = WTF::tryMakeString(string1, string2, string3, string4, string5, string6);
127    if (!result)
128        return throwOutOfMemoryError(exec);
129    return jsNontrivialString(exec, result);
130}
131
132}
133
134#endif
135