jim-interactive.c

   1 #include <errno.h>
   2 #include <string.h>
   3
   4 #include "jimautoconf.h"
   5 #include <jim.h>
   6
   7 #ifdef USE_LINENOISE
   8 #include <unistd.h>
   9 #include "linenoise.h"
  10 #else
  11 #define MAX_LINE_LEN 512
  12 #endif
  13
  14 /**
  15  * Returns an allocated line, or NULL if EOF.
  16  */
  17 char *Jim_HistoryGetline(const char *prompt)
  18 {
  19 #ifdef USE_LINENOISE
  20     return linenoise(prompt);
  21 #else
  22     int len;
  23     char *line = malloc(MAX_LINE_LEN);
  24
  25     fputs(prompt, stdout);
  26     fflush(stdout);
  27
  28     if (fgets(line, MAX_LINE_LEN, stdin) == NULL) {
  29         free(line);
  30         return NULL;
  31     }
  32     len = strlen(line);
  33     if (len && line[len - 1] == '\n') {
  34         line[len - 1] = '\0';
  35     }
  36     return line;
  37 #endif
  38 }
  39
  40 void Jim_HistoryLoad(const char *filename)
  41 {
  42 #ifdef USE_LINENOISE
  43     linenoiseHistoryLoad(filename);
  44 #endif
  45 }
  46
  47 void Jim_HistoryAdd(const char *line)
  48 {
  49 #ifdef USE_LINENOISE
  50     linenoiseHistoryAdd(line);
  51 #endif
  52 }
  53
  54 void Jim_HistorySave(const char *filename)
  55 {
  56 #ifdef USE_LINENOISE
  57     linenoiseHistorySave(filename);
  58 #endif
  59 }
  60
  61 void Jim_HistoryShow(void)
  62 {
  63 #ifdef USE_LINENOISE
  64     /* built-in history command */
  65     int i;
  66     int len;
  67     char **history = linenoiseHistory(&len);
  68     for (i = 0; i < len; i++) {
  69         printf("%4d %s\n", i + 1, history[i]);
  70     }
  71 #endif
  72 }
  73
  74 int Jim_InteractivePrompt(Jim_Interp *interp)
  75 {
  76     int retcode = JIM_OK;
  77     char *history_file = NULL;
  78 #ifdef USE_LINENOISE
  79     const char *home;
  80
  81     home = getenv("HOME");
  82     if (home && isatty(STDIN_FILENO)) {
  83         int history_len = strlen(home) + sizeof("/.jim_history");
  84         history_file = Jim_Alloc(history_len);
  85         snprintf(history_file, history_len, "%s/.jim_history", home);
  86         Jim_HistoryLoad(history_file);
  87     }
  88 #endif
  89
  90     printf("Welcome to Jim version %d.%d\n",
  91         JIM_VERSION / 100, JIM_VERSION % 100);
  92     Jim_SetVariableStrWithStr(interp, JIM_INTERACTIVE, "1");
  93
  94     while (1) {
  95         Jim_Obj *scriptObjPtr;
  96         const char *result;
  97         int reslen;
  98         char prompt[20];
  99
 100         if (retcode != JIM_OK) {
 101             const char *retcodestr = Jim_ReturnCode(retcode);
 102
 103             if (*retcodestr == '?') {
 104                 snprintf(prompt, sizeof(prompt) - 3, "[%d] . ", retcode);
 105             }
 106             else {
 107                 snprintf(prompt, sizeof(prompt) - 3, "[%s] . ", retcodestr);
 108             }
 109         }
 110         else {
 111             strcpy(prompt, ". ");
 112         }
 113
 114         scriptObjPtr = Jim_NewStringObj(interp, "", 0);
 115         Jim_IncrRefCount(scriptObjPtr);
 116         while (1) {
 117             char state;
 118             char *line;
 119
 120             line = Jim_HistoryGetline(prompt);
 121             if (line == NULL) {
 122                 if (errno == EINTR) {
 123                     continue;
 124                 }
 125                 Jim_DecrRefCount(interp, scriptObjPtr);
 126                 retcode = JIM_OK;
 127                 goto out;
 128             }
 129             if (Jim_Length(scriptObjPtr) != 0) {
 130                 /* Line continuation */
 131                 Jim_AppendString(interp, scriptObjPtr, "\n", 1);
 132             }
 133             Jim_AppendString(interp, scriptObjPtr, line, -1);
 134             free(line);
 135             if (Jim_ScriptIsComplete(interp, scriptObjPtr, &state))
 136                 break;
 137
 138             snprintf(prompt, sizeof(prompt), "%c> ", state);
 139         }
 140 #ifdef USE_LINENOISE
 141         if (strcmp(Jim_String(scriptObjPtr), "h") == 0) {
 142             /* built-in history command */
 143             Jim_HistoryShow();
 144             Jim_DecrRefCount(interp, scriptObjPtr);
 145             continue;
 146         }
 147
 148         Jim_HistoryAdd(Jim_String(scriptObjPtr));
 149         if (history_file) {
 150             Jim_HistorySave(history_file);
 151         }
 152 #endif
 153         retcode = Jim_EvalObj(interp, scriptObjPtr);
 154         Jim_DecrRefCount(interp, scriptObjPtr);
 155
 156         if (retcode == JIM_EXIT) {
 157             break;
 158         }
 159         if (retcode == JIM_ERR) {
 160             Jim_MakeErrorMessage(interp);
 161         }
 162         result = Jim_GetString(Jim_GetResult(interp), &reslen);
 163         if (reslen) {
 164             printf("%s\n", result);
 165         }
 166     }
 167   out:
 168     Jim_Free(history_file);
 169     return retcode;
 170 }