Automatically generate FFI bridges using libclang.

[cycript.git] / Library.cpp

diff --git a/Library.cpp b/Library.cpp
index aa85e81cc595da8fa04ebebb1c8b90ce10b760a4..05cdde163c5b482b0681bcf70e764870765a9244 100644 (file)
--- a/Library.cpp
+++ b/Library.cpp
@@ -32,6 +32,7 @@
 
 #include <sys/mman.h>
 
+#include "Code.hpp"
 #include "ConvertUTF.h"
 #include "Driver.hpp"
 #include "Error.hpp"
@@ -103,15 +104,20 @@ bool CYGetOffset(const char *value, ssize_t &index) {
 }
 /* }}} */
 /* JavaScript *ify {{{ */
-void CYStringify(std::ostringstream &str, const char *data, size_t size) {
-    unsigned quot(0), apos(0);
-    for (const char *value(data), *end(data + size); value != end; ++value)
-        if (*value == '"')
-            ++quot;
-        else if (*value == '\'')
-            ++apos;
-
-    bool single(quot > apos);
+void CYStringify(std::ostringstream &str, const char *data, size_t size, bool c) {
+    bool single;
+    if (c)
+        single = false;
+    else {
+        unsigned quot(0), apos(0);
+        for (const char *value(data), *end(data + size); value != end; ++value)
+            if (*value == '"')
+                ++quot;
+            else if (*value == '\'')
+                ++apos;
+
+        single = quot > apos;
+    }
 
     str << (single ? '\'' : '"');
 
@@ -240,6 +246,11 @@ CYUTF8String CYPoolCode(CYPool &pool, std::streambuf &stream) {
     return $pool.strdup(str.str().c_str());
 }
 
+CYUTF8String CYPoolCode(CYPool &pool, CYUTF8String code) {
+    CYStream stream(code.data, code.data + code.size);
+    return CYPoolCode(pool, stream);
+}
+
 CYPool &CYGetGlobalPool() {
     static CYPool pool;
     return pool;