Just to be safe.

[cycript.git] / Console.cpp

diff --git a/Console.cpp b/Console.cpp
index f65323ea7e6f0e4c385a1ec868cbdefddfe03906..8e1c0dae2ce32515504c747453fc9eabbf583809 100644 (file)
--- a/Console.cpp
+++ b/Console.cpp
@@ -39,13 +39,26 @@
 
 #include "cycript.hpp"
 
 
 #include "cycript.hpp"
 

+#ifdef CY_EXECUTE
+#include "JavaScript.hpp"
+#endif
+

 #include <cstdio>
 #include <sstream>
 
 #include <setjmp.h>
 
 #include <cstdio>
 #include <sstream>
 
 #include <setjmp.h>
 

+#ifdef HAVE_READLINE_H
+#include <readline.h>
+#else

 #include <readline/readline.h>
 #include <readline/readline.h>

+#endif
+
+#ifdef HAVE_HISTORY_H
+#include <history.h>
+#else

 #include <readline/history.h>
 #include <readline/history.h>

+#endif

 
 #include <sys/mman.h>
 
 
 #include <sys/mman.h>
 


@@ -108,10 +121,9 @@ void Setup(CYDriver &driver, cy::parser &parser) {
         driver.strict_ = true;
 }
 
         driver.strict_ = true;
 }
 

-void Setup(CYOutput &out, CYDriver &driver) {
+void Setup(CYOutput &out, CYDriver &driver, CYOptions &options) {

     out.pretty_ = pretty_;
     out.pretty_ = pretty_;

-
-    CYContext context(driver.pool_);
+    CYContext context(driver.pool_, options);

     driver.program_->Replace(context);
 }
 
     driver.program_->Replace(context);
 }
 


@@ -178,7 +190,7 @@ void Run(int client, std::string &code, FILE *fout = NULL, bool expand = false)
 
 int (*append_history$)(int, const char *);
 
 
 int (*append_history$)(int, const char *);
 

-static void Console(apr_pool_t *pool, int client) {
+static void Console(apr_pool_t *pool, int client, CYOptions &options) {

     passwd *passwd;
     if (const char *username = getenv("LOGNAME"))
         passwd = getpwnam(username);
     passwd *passwd;
     if (const char *username = getenv("LOGNAME"))
         passwd = getpwnam(username);


@@ -320,8 +332,8 @@ static void Console(apr_pool_t *pool, int client) {
                 code = command;
             else {
                 std::ostringstream str;
                 code = command;
             else {
                 std::ostringstream str;

-                CYOutput out(str);
-                Setup(out, driver);
+                CYOutput out(str, options);
+                Setup(out, driver, options);

                 out << *driver.program_;
                 code = str.str();
             }
                 out << *driver.program_;
                 code = str.str();
             }


@@ -369,6 +381,7 @@ void InjectLibrary(pid_t pid);
 int Main(int argc, char const * const argv[], char const * const envp[]) {
     bool tty(isatty(STDIN_FILENO));
     bool compile(false);
 int Main(int argc, char const * const argv[], char const * const envp[]) {
     bool tty(isatty(STDIN_FILENO));
     bool compile(false);

+    CYOptions options;

 
     append_history$ = reinterpret_cast<int (*)(int, const char *)>(dlsym(RTLD_DEFAULT, "append_history"));
 
 
     append_history$ = reinterpret_cast<int (*)(int, const char *)>(dlsym(RTLD_DEFAULT, "append_history"));
 


@@ -416,6 +429,8 @@ int Main(int argc, char const * const argv[], char const * const envp[]) {
 
             case 'g':
                 if (false);
 
             case 'g':
                 if (false);

+                else if (strcmp(arg, "rename") == 0)
+                    options.verbose_ = true;

 #if YYDEBUG
                 else if (strcmp(arg, "bison") == 0)
                     bison_ = true;
 #if YYDEBUG
                 else if (strcmp(arg, "bison") == 0)
                     bison_ = true;


@@ -562,7 +577,7 @@ int Main(int argc, char const * const argv[], char const * const envp[]) {
 #endif
 
     if (script == NULL && tty)
 #endif
 
     if (script == NULL && tty)

-        Console(pool, client);
+        Console(pool, client, options);

     else {
         CYDriver driver(script ?: "<stdin>");
         cy::parser parser(driver);
     else {
         CYDriver driver(script ?: "<stdin>");
         cy::parser parser(driver);


@@ -597,12 +612,13 @@ int Main(int argc, char const * const argv[], char const * const envp[]) {
             for (CYDriver::Errors::const_iterator i(driver.errors_.begin()); i != driver.errors_.end(); ++i)
                 std::cerr << i->location_.begin << ": " << i->message_ << std::endl;
         } else if (driver.program_ != NULL)
             for (CYDriver::Errors::const_iterator i(driver.errors_.begin()); i != driver.errors_.end(); ++i)
                 std::cerr << i->location_.begin << ": " << i->message_ << std::endl;
         } else if (driver.program_ != NULL)

-            if (client != -1)
-                Run(client, start, end - start, stdout);
-            else {
+            if (client != -1) {
+                std::string code(start, end-start);
+                Run(client, code, stdout);
+            } else {

                 std::ostringstream str;
                 std::ostringstream str;

-                CYOutput out(str);
-                Setup(out, driver);
+                CYOutput out(str, options);
+                Setup(out, driver, options);

                 out << *driver.program_;
                 std::string code(str.str());
                 if (compile)
                 out << *driver.program_;
                 std::string code(str.str());
                 if (compile)