kc3-lang/kc3/ic3/buf_linenoise.c

• Show log

  Commit

• Hash : 3d8d7728
  Author : Thomas de Grivel
  Date : 2023-03-23T17:19:58
  

  licence

Download

• ic3/buf_linenoise.c

• /* c3
   * Copyright 2022,2023 kmx.io <contact@kmx.io>
   *
   * Permission is hereby granted to use this software granted the above
   * copyright notice and this permission paragraph are included in all
   * copies and substantial portions of this software.
   *
   * THIS SOFTWARE IS PROVIDED "AS-IS" WITHOUT ANY GUARANTEE OF
   * PURPOSE AND PERFORMANCE. IN NO EVENT WHATSOEVER SHALL THE
   * AUTHOR BE CONSIDERED LIABLE FOR THE USE AND PERFORMANCE OF
   * THIS SOFTWARE.
   */
  #include <assert.h>
  #include <err.h>
  #include <stdio.h>
  #include <stdlib.h>
  #include <string.h>
  #include <unistd.h>
  #include "../libc3/c3.h"
  #include "../linenoise/linenoise.h"
  
  typedef struct buf_linenoise {
    s_buf       buf;
    bool        eof;
    const char *prompt;
  } s_buf_linenoise;
  
  sw buf_linenoise_refill_fgets (s_buf *buf);
  sw buf_linenoise_refill_linenoise (s_buf *buf);
  
  void buf_linenoise_close (s_buf *buf)
  {
    assert(buf);
    buf->refill = NULL;
    free(buf->user_ptr);
    buf->user_ptr = NULL;
    if (isatty(STDIN_FILENO))
      puts("");
  }
  
  s_buf * buf_linenoise_open_r (s_buf *buf, const char *prompt)
  {
    s_buf_linenoise *buf_linenoise;
    assert(buf);
    buf_linenoise = malloc(sizeof(s_buf_linenoise));
    if (! buf_linenoise)
      errx(1, "buf_linenoise_open_r: out of memory");
    buf_init_1(&buf_linenoise->buf, "");
    buf_linenoise->eof = false;
    if (isatty(STDIN_FILENO))
      buf->refill = buf_linenoise_refill_linenoise;
    else
      buf->refill = buf_linenoise_refill_fgets;
    buf_linenoise->prompt = prompt;
    buf->user_ptr = buf_linenoise;
    return buf;
  }
  
  sw buf_linenoise_refill_fgets (s_buf *buf)
  {
    sw c = 0;
    uw result = 0;
    assert(buf);
    assert(buf->user_ptr);
    if (buf->rpos > buf->wpos ||
        buf->wpos > buf->size)
      return -1;
    if (feof(stdin))
      return -1;
    while (buf->wpos < buf->size &&
           c != '\n' &&
           (c = getchar()) != EOF &&
           c <= 255) {
      buf_write_u8(buf, c);
      result++;
    }      
    return result;
  }
  
  sw buf_linenoise_refill_linenoise (s_buf *buf)
  {
    s_buf_linenoise *buf_linenoise;
    uw buf_linenoise_len;
    uw r;
    uw size;
    assert(buf);
    assert(buf->user_ptr);
    if (buf->rpos > buf->wpos ||
        buf->wpos > buf->size)
      return -1;
    size = buf->size - buf->wpos;
    if (size == 0)
      return 0;
    buf_linenoise = buf->user_ptr;
    if (buf_linenoise->eof)
      return -1;
    buf_linenoise_len = buf_linenoise->buf.wpos - buf_linenoise->buf.rpos;
    if (buf_linenoise_len == 0) {
      free(buf_linenoise->buf.ptr.p);
      if (! (buf_linenoise->buf.ptr.p = linenoise(buf_linenoise->prompt))) {
        buf_linenoise->eof = true;
        buf_linenoise->buf.rpos =
          buf_linenoise->buf.wpos =
          buf_linenoise->buf.size = 0;
        return -1;
      }
      linenoiseHistoryAdd(buf_linenoise->buf.ptr.p);
      buf_linenoise_len = strlen(buf_linenoise->buf.ptr.p);
      buf_linenoise->buf.ptr.ps8[buf_linenoise_len++] = '\n';
      buf_linenoise->buf.size = buf_linenoise->buf.wpos = buf_linenoise_len;
      buf_linenoise->buf.rpos = 0;
    }
    if (buf_linenoise_len < size)
      size = buf_linenoise_len;
    if ((r = buf_xfer(buf, &buf_linenoise->buf, size)) != size)
      return -1;
    return size;
  }
  

Download