Tools/WebKitTestRunner/StringFunctions.h

   1 /*
   2  * Copyright (C) 2010 Apple Inc. All rights reserved.
   3  * Copyright (C) 2010 University of Szeged. All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
  15  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  16  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  17  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
  18  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  19  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  20  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  21  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  22  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  23  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
  24  * THE POSSIBILITY OF SUCH DAMAGE.
  25  */
  26
  27 #ifndef StringFunctions_h
  28 #define StringFunctions_h
  29
  30 #include <JavaScriptCore/JSRetainPtr.h>
  31 #include <JavaScriptCore/JavaScript.h>
  32 #include <sstream>
  33 #include <string>
  34 #include <WebKit2/WKRetainPtr.h>
  35 #include <WebKit2/WKString.h>
  36 #include <WebKit2/WKStringPrivate.h>
  37 #include <WebKit2/WKURL.h>
  38 #include <wtf/OwnArrayPtr.h>
  39 #include <wtf/PassOwnArrayPtr.h>
  40 #include <wtf/Platform.h>
  41 #include <wtf/text/WTFString.h>
  42 #include <wtf/Vector.h>
  43
  44 namespace WTR {
  45
  46 // Conversion functions
  47
  48 inline WKRetainPtr<WKStringRef> toWK(JSStringRef string)
  49 {
  50     return adoptWK(WKStringCreateWithJSString(string));
  51 }
  52
  53 inline WKRetainPtr<WKStringRef> toWK(JSRetainPtr<JSStringRef> string)
  54 {
  55     return toWK(string.get());
  56 }
  57
  58 inline JSRetainPtr<JSStringRef> toJS(WKStringRef string)
  59 {
  60     return JSRetainPtr<JSStringRef>(Adopt, WKStringCopyJSString(string));
  61 }
  62
  63 inline JSRetainPtr<JSStringRef> toJS(const WKRetainPtr<WKStringRef>& string)
  64 {
  65     return toJS(string.get());
  66 }
  67
  68 inline std::string toSTD(WKStringRef string)
  69 {
  70     size_t bufferSize = WKStringGetMaximumUTF8CStringSize(string);
  71     OwnArrayPtr<char> buffer = adoptArrayPtr(new char[bufferSize]);
  72     size_t stringLength = WKStringGetUTF8CString(string, buffer.get(), bufferSize);
  73     return std::string(buffer.get(), stringLength - 1);
  74 }
  75
  76 inline std::string toSTD(const WKRetainPtr<WKStringRef>& string)
  77 {
  78     return toSTD(string.get());
  79 }
  80
  81 inline WTF::String toWTFString(WKStringRef string)
  82 {
  83     size_t bufferSize = WKStringGetMaximumUTF8CStringSize(string);
  84     OwnArrayPtr<char> buffer = adoptArrayPtr(new char[bufferSize]);
  85     size_t stringLength = WKStringGetUTF8CString(string, buffer.get(), bufferSize);
  86     return WTF::String::fromUTF8WithLatin1Fallback(buffer.get(), stringLength - 1);
  87 }
  88
  89 inline WTF::String toWTFString(const WKRetainPtr<WKStringRef>& string)
  90 {
  91     return toWTFString(string.get());
  92 }
  93
  94 } // namespace WTR
  95
  96 #endif // StringFunctions_h