Application.mm

   1 #define _GNU_SOURCE
   2
   3 #include <substrate.h>
   4 #include "cycript.hpp"
   5
   6 #include <cstdio>
   7 #include <sstream>
   8
   9 #include <setjmp.h>
  10
  11 #include <readline/readline.h>
  12 #include <readline/history.h>
  13
  14 #include <sys/mman.h>
  15
  16 #include <errno.h>
  17 #include <unistd.h>
  18
  19 #include <sys/types.h>
  20 #include <sys/stat.h>
  21 #include <fcntl.h>
  22
  23 #include "Cycript.tab.hh"
  24
  25 static jmp_buf ctrlc_;
  26
  27 static void sigint(int) {
  28     longjmp(ctrlc_, 1);
  29 }
  30
  31 static JSStringRef Result_;
  32
  33 void Run(const char *code, FILE *fout) { _pooled
  34     JSStringRef script(JSStringCreateWithUTF8CString(code));
  35
  36     JSContextRef context(CYGetJSContext());
  37
  38     JSValueRef exception(NULL);
  39     JSValueRef result(JSEvaluateScript(context, script, NULL, NULL, 0, &exception));
  40     JSStringRelease(script);
  41
  42     if (exception != NULL) { error:
  43         result = exception;
  44         exception = NULL;
  45     }
  46
  47     if (!JSValueIsUndefined(context, result)) {
  48         CYPool pool;
  49         const char *json;
  50
  51         json = CYPoolJSONString(pool, context, result, &exception);
  52         if (exception != NULL)
  53             goto error;
  54
  55         CYSetProperty(context, CYGetGlobalObject(context), Result_, result);
  56
  57         if (fout != NULL) {
  58             fputs(json, fout);
  59             fputs("\n", fout);
  60             fflush(fout);
  61         }
  62     }
  63 }
  64
  65 static void Console() {
  66     bool bypass(false);
  67     bool debug(false);
  68
  69     FILE *fout(stdout);
  70
  71     rl_bind_key('\t', rl_insert);
  72
  73     struct sigaction action;
  74     sigemptyset(&action.sa_mask);
  75     action.sa_handler = &sigint;
  76     action.sa_flags = 0;
  77     sigaction(SIGINT, &action, NULL);
  78
  79     restart: for (;;) {
  80         std::string command;
  81         std::vector<std::string> lines;
  82
  83         bool extra(false);
  84         const char *prompt("cy# ");
  85
  86         if (setjmp(ctrlc_) != 0) {
  87             fputs("\n", fout);
  88             fflush(fout);
  89             goto restart;
  90         }
  91
  92       read:
  93         char *line(readline(prompt));
  94         if (line == NULL)
  95             break;
  96
  97         if (!extra) {
  98             extra = true;
  99             if (line[0] == '\\') {
 100                 std::string data(line + 1);
 101                 if (data == "bypass") {
 102                     bypass = !bypass;
 103                     fprintf(fout, "bypass == %s\n", bypass ? "true" : "false");
 104                     fflush(fout);
 105                 } else if (data == "debug") {
 106                     debug = !debug;
 107                     fprintf(fout, "debug == %s\n", debug ? "true" : "false");
 108                     fflush(fout);
 109                 }
 110                 add_history(line);
 111                 goto restart;
 112             }
 113         }
 114
 115         lines.push_back(line);
 116         command += line;
 117         free(line);
 118
 119         std::string code;
 120
 121         if (bypass)
 122             code = command;
 123         else {
 124             CYDriver driver("");
 125             cy::parser parser(driver);
 126
 127             driver.data_ = command.c_str();
 128             driver.size_ = command.size();
 129
 130             if (parser.parse() != 0 || !driver.errors_.empty()) {
 131                 for (CYDriver::Errors::const_iterator i(driver.errors_.begin()); i != driver.errors_.end(); ++i) {
 132                     cy::position begin(i->location_.begin);
 133                     if (begin.line != lines.size() || begin.column - 1 != lines.back().size()) {
 134                         std::cerr << i->message_ << std::endl;
 135                         add_history(command.c_str());
 136                         goto restart;
 137                     }
 138                 }
 139
 140                 driver.errors_.clear();
 141
 142                 command += '\n';
 143                 prompt = "cy> ";
 144                 goto read;
 145             }
 146
 147             if (driver.source_ == NULL)
 148                 goto restart;
 149
 150             std::ostringstream str;
 151             driver.source_->Show(str);
 152             code = str.str();
 153         }
 154
 155         add_history(command.c_str());
 156
 157         if (debug)
 158             std::cout << code << std::endl;
 159
 160         Run(code.c_str(), fout);
 161     }
 162
 163     fputs("\n", fout);
 164     fflush(fout);
 165 }
 166
 167 static void *Map(const char *path, size_t *psize) {
 168     int fd;
 169     _syscall(fd = open(path, O_RDONLY));
 170
 171     struct stat stat;
 172     _syscall(fstat(fd, &stat));
 173     size_t size(stat.st_size);
 174
 175     *psize = size;
 176
 177     void *base;
 178     _syscall(base = mmap(NULL, size, PROT_READ, MAP_SHARED, fd, 0));
 179
 180     _syscall(close(fd));
 181     return base;
 182 }
 183
 184 int main(int argc, const char *argv[]) {
 185     const char *script;
 186
 187     if (argc == 1)
 188         script = NULL;
 189     else {
 190         CYSetArgs(argc - 1, argv + 1);
 191         script = argv[1];
 192     }
 193
 194     Result_ = CYCopyJSString("_");
 195
 196     if (script == NULL || strcmp(script, "-") == 0)
 197         Console();
 198     else {
 199         CYDriver driver(script);
 200         cy::parser parser(driver);
 201
 202         size_t size;
 203         char *start(reinterpret_cast<char *>(Map(script, &size)));
 204         char *end(start + size);
 205
 206         if (size >= 2 && start[0] == '#' && start[1] == '!') {
 207             start += 2;
 208             while (start != end && *start++ != '\n');
 209         }
 210
 211         driver.data_ = start;
 212         driver.size_ = end - start;
 213
 214         if (parser.parse() != 0 || !driver.errors_.empty()) {
 215             for (CYDriver::Errors::const_iterator i(driver.errors_.begin()); i != driver.errors_.end(); ++i)
 216                 std::cerr << i->location_.begin << ": " << i->message_ << std::endl;
 217         } else if (driver.source_ != NULL) {
 218             std::ostringstream str;
 219             driver.source_->Show(str);
 220             std::string code(str.str());
 221             std::cout << code << std::endl;
 222             Run(code.c_str(), stdout);
 223         }
 224     }
 225
 226     return 0;
 227 }