FShGetLine.cpp

Go to the documentation of this file.

/*

$Header$

*/

#include <ctype.h>

#include "Error.h"
#include "FSh.h"
#include "Core.h"
#include "NamedObject.h"
#include "Set.h"

#include "FShGetLine.h"

// Public methods.

FShGetLine::FShGetLine(FSh *shell) : GetLine()
{
  int res;

  _forth = FORTH();

  if (!(_shell = shell)) {
    failure(ERROR(MSG_NULL_POINTER));
    return;
  }

  if (!(_connection = shell->_connection)) {
    failure(ERROR(MSG_NULL_POINTER));
    return;
  }

  if (FAILED(res = _connection->writeStr(_forth->output()))) {
    failure(ERROR_BACKTRACE(res));
    return;
  }

  if (FAILED(res = _connection->writeStr(_forth->welcome()))) {
    failure(ERROR_BACKTRACE(res));
    return;
  }
  if (FAILED(res = setPrompt(CSTRING(_forth->prompt())))) {
    failure(ERROR_BACKTRACE(res));
    return;
  }
}

FShGetLine::~FShGetLine()
{
}

// Protected methods.

int FShGetLine::onTab(char *buf, const int, int *cursorLoc, int *i)
{
  if (*cursorLoc == 0 || strlen(buf) == 0)
    return OK;

  String matchedWords;
  unsigned int matchedWordsNum = 0;
  bool quotation = false;
  String completedPart;
  StringSet words(_forth->words());
  int res;

  words += NamedObject::names();

  int j;
  unsigned int wordsPerLine, len;
  char *wordPart = buf;

  j = *cursorLoc - 1;
  while (j >= 0) {
    if (buf[j] == ' ' || buf[j] == '\t' || buf[j] == '"') {
      if (buf[j] == '"')
        quotation = true;
      wordPart = buf + j + 1;
      break;
    }
    --j;
  }
  if ((len = strlen(wordPart)) == 0)
    return OK;

  StringSet::const_iterator k = words.begin();
  StringSet::const_iterator e = words.end();

  wordsPerLine = 0;
  while (k != e) {
    if (strncasecmp(wordPart, CSTRING(*k), len) == 0) {
      if (wordsPerLine > 0)
        matchedWords += '\t';
      matchedWords += *k;
      completedPart.intersection(*k, true);
      ++matchedWordsNum;
      if (++wordsPerLine >= LIMIT_FORTH_WORDS_PER_LINE) {
        wordsPerLine = 0;
        matchedWords.append(String::eol());
      }
    }
    ++k;
  }

  if (matchedWordsNum == 0) {
    int actuallyWritten;

    if (FAILED(res = _connection->write(&FSH_GETLINE_BEEP, sizeof(FSH_GETLINE_BEEP), actuallyWritten)))
      return ERROR_BACKTRACE(res);

    return OK;
  }

  if (matchedWordsNum == 1) {
    if (!quotation)
      matchedWords += ' ';
    strncpy(wordPart, CSTRING(matchedWords), GL_LIMIT_BUFFER_SIZE - 1);
    *i = wordPart - buf;
    *cursorLoc += matchedWords.length();
  } else {
      if (FAILED(res = _connection->writeStr(String::eol() + matchedWords + (iscntrl(matchedWords[matchedWords.length() - 1]) ? "" : String::eol()) + prompt() + buf)))
        return ERROR_BACKTRACE(res);
      if (!completedPart.empty()) {
        strncpy(wordPart, CSTRING(completedPart), GL_LIMIT_BUFFER_SIZE - 1);
        *i = wordPart - buf;
        *cursorLoc += completedPart.length();
      }
    }

  return OK;
}

int FShGetLine::onOutput(const char *buf)
{
  int res;

  if (strlen(buf) == 0)
    if (FAILED(res = CORE()->kill(_shell)))
      return ERROR_BACKTRACE(res);

  if (FAILED(res = _forth->eval(buf)))
    ERROR_BACKTRACE(res);

  if (FAILED(res = setPrompt(CSTRING(_forth->prompt()))))
    return ERROR_BACKTRACE(res);

  String out(_forth->output());
  if (!out.empty())
    if (FAILED(res = _connection->writeStr(String::eol() + out)))
      return ERROR_BACKTRACE(res);

  return OK;
}

int FShGetLine::onPutc(const char c)
{
  int actuallyWritten;
  int res;

  if (FAILED(res = _connection->write(&c, sizeof(char), actuallyWritten)))
    return ERROR_BACKTRACE(res);
  if (c == '\n') {
    char ch = '\r'; // RAW mode needs '\r', does not hurt.
    if (FAILED(res = _connection->write(&ch, sizeof(char), actuallyWritten)))
      return ERROR_BACKTRACE(res);
  }

  return OK;
}

int FShGetLine::onPuts(const char *buf)
{
  int actuallyWritten;
  int res;

  if (buf)
    if (FAILED(res = _connection->write(buf, strlen(buf), actuallyWritten)))
      return ERROR_BACKTRACE(res);

  return OK;
}

// Caclulates length of colored text. Pay attention to non printable characters,
// so according to ANSI terminal definitions all sequences between
// ESC and 'm' treat as non-printable.
unsigned int FShGetLine::onStrlen(const char *str)
{
  return String::colorLength(str);
}

---