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