Added a CYOutput object between std::ostream and the actual mechanism.

[cycript.git] / Console.cpp

diff --git a/Console.cpp b/Console.cpp
index f83c3a25118cf8b8764a77f4d0fbd9dba1d20f9a..6ed69e66dce574b5dcf16d79617720bd8aa7ae05 100644 (file)
--- a/Console.cpp
+++ b/Console.cpp
@@ -276,7 +276,8 @@ static void Console(int socket) {
                 code = command;
             else {
                 std::ostringstream str;
-                driver.source_->Show(str);
+                CYOutput out(str);
+                driver.source_->Show(out);
                 code = str.str();
             }
         }
@@ -431,7 +432,8 @@ int main(int argc, char *argv[]) {
                 Run(socket, start, end - start, stdout);
             else {
                 std::ostringstream str;
-                driver.source_->Show(str);
+                CYOutput out(str);
+                driver.source_->Show(out);
                 std::string code(str.str());
                 if (compile)
                     std::cout << code;