rtbuf/rtbuf/cli.c

• Show log

  Commit

• Hash : f6bf2d89
  Author : Thomas de Grivel
  Date : 2020-02-06T14:32:48
  

  print error on function not found
  

Download

• cli.c

• /*
   * Copyright 2018 Thomas de Grivel <thoxdg@gmail.com> +33614550127
   *
   * Permission to use, copy, modify, and distribute this software for any
   * purpose with or without fee is hereby granted, provided that the above
   * copyright notice and this permission notice appear in all copies.
   *
   * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
   * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
   * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
   * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
   * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
   * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
   */
  
  #include <unistd.h>
  #include <stdlib.h>
  #include <stdio.h>
  #include <string.h>
  #include <readline/readline.h>
  #include <readline/history.h>
  
  char * readline (const char *prompt);
  
  #include "cli.h"
  
  void cli_init (s_cli *cli)
  {
    if (isatty(0)) {
      cli->prompt = "> ";
      using_history();
    }
    else
      cli->prompt = 0;
    cli->functions = 0;
  }
  
  void cli_prompt (s_cli *cli, const char *prompt)
  {
    if (isatty(0))
      cli->prompt = prompt;
    else
      cli->prompt = 0;
  }
  
  void cli_functions (s_cli *cli, const s_cli_function *functions)
  {
    cli->functions = functions;
  }
  
  int cli_scan (s_cli *cli)
  {
    unsigned i = 0;
    char *line = cli->line;
    char *arg = cli->arg;
    const char **argv = cli->argv;
    cli->argc = 0;
    *argv = arg;
    while (1) {
      *arg = ((++i == CLI_SIZE) ? 0 : *line);
      if (*arg == 0) {
        if (*argv != arg) {
          argv++;
          cli->argc++;
        }
        *argv = 0;
        return cli->argc;
      }
      else if (*arg == ' ') {
        *arg = 0;
        if (*argv != arg) {
          argv++;
          cli->argc++;
        }
        *argv = arg + 1;
      }
      arg++;
      line++;
    }
  }
  
  void trim_newline (char *str)
  {
    unsigned int i = 0;
    while (str[i])
      i++;
    if (i > 0 && str[i - 1] == '\n')
      str[i - 1] = 0;
  }
  
  int cli_read_file_line (s_cli *cli, FILE *fp)
  {
    if (fgets(cli->line, CLI_SIZE, fp)) {
      trim_newline(cli->line);
      fputs(cli->prompt, stdout);
      puts(cli->line);
      return 0;
    }
    return -1;
  }
  
  int cli_readline (s_cli *cli)
  {
    if (cli->prompt) {
      char *line = readline(cli->prompt);
      if (line == 0)
        return -1;
      strncpy(cli->line, line, CLI_SIZE - 1);
      cli->line[CLI_SIZE - 1] = 0;
      free(line);
      if (*cli->line)
        add_history(cli->line);
      return 0;
    }
    return cli_read_file_line(cli, stdin);
  }
  
  int cli_read_file (s_cli *cli, FILE *fp)
  {
    cli->argc = -1;
    cli->f = 0;
    if (cli_read_file_line(cli, fp))
      return -1;
    cli_scan(cli);
    if (0 < cli->argc)
      cli->f = cli_find_function(cli, cli->argv[0], cli->argc - 1);
    return cli->argc;
  }
  
  int cli_read (s_cli *cli)
  {
    cli->argc = -1;
    cli->f = 0;
    if (cli_readline(cli))
      return -1;
    cli_scan(cli);
    if (0 < cli->argc)
      cli->f = cli_find_function(cli, cli->argv[0], cli->argc - 1);
    return cli->argc;
  }
  
  f_cli cli_find_function (s_cli *cli, const char *name, int arity)
  {
    const s_cli_function *function = cli->functions;
    if (function == 0)
      return 0;
    while (1) {
      if (function->name == 0)
        return 0;
      if (strcmp(function->name, name) == 0 &&
          (function->arity < 0 || arity < 0 || function->arity == arity))
        return function->f;
      function++;
    }
  }
  
  int cli_eval (s_cli *cli)
  {
    if (0 < cli->argc) {
      if (cli->f)
        return cli->f(cli->argc - 1, cli->argv);
      fprintf(stderr, "rtbuf: function not found: %s\n", cli->argv[0]);
    }
    return -1;
  }
  

Download