cli.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>

#include "kamby.h"

struct KaNode *builtin_puts(struct KaNode *node, struct KaNode **env) {
  while (node) {
    switch (node->type) {
      case KA_NUM:
        printf("%lld ", node->num);
        break;
      case KA_STR:
        printf("%s ", node->str);
        break;
      default:;
    }
    node = node->next;
  }
  printf("\b\n");
  return calloc(1, KANODE_SIZE);
}

char ident[256];
struct KaNode *builtin_tree(struct KaNode *node, struct KaNode **env) {
  if (!node) node = *env;
  while (node) {
    switch (node->type) {
      case KA_EXPR:
        printf("%sEXPR %s:\n", ident, node->key);
        strcat(ident, "..");
        builtin_tree(node->chld, env);
        strcpy(ident, ident + 2);
        break;
      case KA_BLCK:
        printf("%sBLCK %s:\n", ident, node->key);
        strcat(ident, "..");
        builtin_tree(node->chld, env);
        strcpy(ident, ident + 2);
        break;
      case KA_LIST:
        printf("%sLIST %s:\n", ident, node->key);
        strcat(ident, "..");
        builtin_tree(node->chld, env);
        strcpy(ident, ident + 2);
        break;
      case KA_NUM:
        printf("%sNUM %s: %lld\n", ident, node->key, node->num);
        break;
      case KA_STR:
        printf("%sSTR %s: %s\n", ident, node->key, node->str);
        break;
      case KA_IDF:
        printf("%sIDF %s: %s\n", ident, node->key, node->str);
        break;
      default:
        printf("%sNONE %s: %s\n", ident, node->key, node->str);
    }
    node = node->next && node->next->type ? node->next : NULL;
  }
  return calloc(1, KANODE_SIZE);
}

int main(int argc, char **argv) {
  struct KaNode *env = ka_init();
  struct KaNode *pos = calloc(1, KANODE_SIZE);

  ka_def(ka_link(ka_idf("puts"), ka_fn(builtin_puts), 0), &env);
  ka_def(ka_link(ka_idf("tree"), ka_fn(builtin_tree), 0), &env);

  if (argc == 1) {
    char input[1024];
    while (1) {
      printf("kamby> ");
      fflush(stdout);
      fgets(input, 1024, stdin);
      if (input[0] == '\n') continue;
      else if (strcmp(input, "exit\n") == 0) break;
      free(pos);
      pos = calloc(1, KANODE_SIZE);
      ka_eval(ka_parser(input, &pos), &env);
      input[0] = '\0';
    }
  } else if (argc == 2) {
    FILE *file = fopen(argv[1], "r");
    fseek(file, 0, SEEK_END);
    long size = ftell(file);
    rewind(file);
    char *text = calloc(1, size + 1);
    fread(text, size, 1, file);
    struct KaNode *ast = ka_parser(text, &pos);
    ka_eval(ast, &env);
    fclose(file);
    free(text);
  }

  free(pos);
  free(env);

  return 0;
}