toxcore/testing/toxic/chat.c

/*
 * Toxic -- Tox Curses Client
 */

#include <curses.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include <ctype.h>
#include <time.h>

#include "../../core/Messenger.h"
#include "../../core/network.h"

#include "windows.h"

typedef struct {
  int friendnum;

  char line[256];
  size_t pos;

  WINDOW* history;
  WINDOW* linewin;

} ChatContext;

extern int w_active;
extern void del_window(ToxWindow *w, int f_num);
extern void fix_name(uint8_t* name);
void print_help(ChatContext* self);
void execute(ToxWindow* self, ChatContext* ctx, char* cmd);

static void chat_onMessage(ToxWindow* self, int num, uint8_t* msg, uint16_t len) {
  ChatContext* ctx = (ChatContext*) self->x;
  uint8_t nick[MAX_NAME_LENGTH] = {0};
  
  time_t now;
  time(&now);
  struct tm * timeinfo;
  timeinfo = localtime(&now);

  if(ctx->friendnum != num)
    return;

  getname(num, (uint8_t*) &nick);

  msg[len-1] = '\0';
  nick[MAX_NAME_LENGTH-1] = '\0';

  fix_name(msg);
  fix_name(nick);

  wattron(ctx->history, COLOR_PAIR(2));
  wprintw(ctx->history, "[%02d:%02d:%02d] ", timeinfo->tm_hour, timeinfo->tm_min, timeinfo->tm_sec);
  wattroff(ctx->history, COLOR_PAIR(2));
  wattron(ctx->history, COLOR_PAIR(4));
  wprintw(ctx->history, "%s: ", nick);
  wattroff(ctx->history, COLOR_PAIR(4));
  wprintw(ctx->history, "%s\n", msg);

  self->blink = true;
  beep();
  flash();
}

static void chat_onNickChange(ToxWindow* self, int num, uint8_t* nick, uint16_t len) {
  ChatContext* ctx = (ChatContext*) self->x;

  if(ctx->friendnum != num)
    return;

  nick[len-1] = '\0';
  fix_name(nick);

  snprintf(self->title, sizeof(self->title), "[%s (%d)]", nick, num);

  wattron(ctx->history, COLOR_PAIR(3));
  wprintw(ctx->history, " * Your partner changed nick to '%s'\n", nick);
  wattroff(ctx->history, COLOR_PAIR(3));
}

static void chat_onStatusChange(ToxWindow* self, int num, uint8_t* status, uint16_t len) {

}

/* check that the string has one non-space character */
int string_is_empty(char *string)
{
  int rc = 0;
  char *copy = strdup(string);

  rc = ((strtok(copy, " ") == NULL) ? 1:0);
  free(copy);

  return rc;
}

static void chat_onKey(ToxWindow* self, int key) {
  ChatContext* ctx = (ChatContext*) self->x;

  time_t now;
  time(&now);
  struct tm * timeinfo;
  timeinfo = localtime(&now);

  /* PRINTABLE characters: Add to line */
  if(isprint(key)) {
    if(ctx->pos != sizeof(ctx->line)-1) {
      ctx->line[ctx->pos++] = key;
      ctx->line[ctx->pos] = '\0';
    }
  }

  /* RETURN key: Execute command or print line */
  else if(key == '\n') {
    if (ctx->line[0] == '/')
      execute(self, ctx, ctx->line);
    else {
      if(!string_is_empty(ctx->line)) {
        /* make sure the string has at least non-space character */
        wattron(ctx->history, COLOR_PAIR(2));
        wprintw(ctx->history, "[%02d:%02d:%02d] ", timeinfo->tm_hour, timeinfo->tm_min, timeinfo->tm_sec);
        wattroff(ctx->history, COLOR_PAIR(2));
        wattron(ctx->history, COLOR_PAIR(1));
        wprintw(ctx->history, "you: ", ctx->line);
        wattroff(ctx->history, COLOR_PAIR(1));
        wprintw(ctx->history, "%s\n", ctx->line);
      }
      if(m_sendmessage(ctx->friendnum, (uint8_t*) ctx->line, strlen(ctx->line)+1) == 0) {
        wattron(ctx->history, COLOR_PAIR(3));
        wprintw(ctx->history, " * Failed to send message.\n");
        wattroff(ctx->history, COLOR_PAIR(3));
      }
    }
    ctx->line[0] = '\0';
    ctx->pos = 0;
  }

  /* BACKSPACE key: Remove one character from line */
  else if(key == 0x107 || key == 0x8 || key == 0x7f) {
    if(ctx->pos != 0) {
      ctx->line[--ctx->pos] = '\0';
    }
  }
}

void execute(ToxWindow* self, ChatContext* ctx, char* cmd)
{
  if (!strcmp(cmd, "/clear") || !strcmp(cmd, "/c")) {
    wclear(self->window);
    wclear(ctx->history);
  }
  else if (!strcmp(cmd, "/help") || !strcmp(cmd, "/h"))
    print_help(ctx);
  else if (!strcmp(cmd, "/quit") || !strcmp(cmd, "/exit") || !strcmp(cmd, "/q")) {
    endwin();
    exit(0);
  }
  else if (!strncmp(cmd, "/status ", strlen("/status "))) {
    char* msg;
    msg = strchr(cmd, ' ');
    if(msg == NULL) {
      wprintw(ctx->history, "Invalid syntax.\n");
      return;
    }
    msg++;
    m_set_userstatus(USERSTATUS_KIND_RETAIN, (uint8_t*) msg, strlen(msg)+1);
    wprintw(ctx->history, "Status set to: %s\n", msg);
  }
  else if (!strncmp(cmd, "/nick ", strlen("/nick "))) {
    char* nick;
    nick = strchr(cmd, ' ');
    if(nick == NULL) {
      wprintw(ctx->history, "Invalid syntax.\n");
      return;
    }
    nick++;
    setname((uint8_t*) nick, strlen(nick)+1);
    wprintw(ctx->history, "Nickname set to: %s\n", nick);
  }
  else if(!strcmp(cmd, "/myid")) {
    char id[32*2 + 1] = {0};
    int i;
    for (i = 0; i < 32; i++) {
      char xx[3];
      snprintf(xx, sizeof(xx), "%02x",  self_public_key[i] & 0xff);
      strcat(id, xx);
    }
    wprintw(ctx->history, "Your ID: %s\n", id);
  }
  else if (strcmp(ctx->line, "/close") == 0) {
    w_active = 0;    // Go to prompt screen
    int f_num = ctx->friendnum;
    delwin(ctx->linewin);
    del_window(self, f_num);
  }
  else
    wprintw(ctx->history, "Invalid command.\n");
}

static void chat_onDraw(ToxWindow* self) {
  curs_set(1);
  int x, y;
  ChatContext* ctx = (ChatContext*) self->x;

  getmaxyx(self->window, y, x);

  (void) x;
  if(y < 3)
    return;

  wclear(ctx->linewin);
  mvwhline(ctx->linewin, 0, 0, '_', COLS);
  mvwprintw(self->window, y-1, 0, "%s\n", ctx->line);

  wrefresh(self->window);
}

static void chat_onInit(ToxWindow* self) {
  int x, y;
  ChatContext* ctx = (ChatContext*) self->x;

  getmaxyx(self->window, y, x);

  ctx->history = subwin(self->window, y - 4, x, 0, 0);
  scrollok(ctx->history, 1);

  ctx->linewin = subwin(self->window, 2, x, y - 3, 0);
  print_help(ctx);
}

void print_help(ChatContext* self) {
  wattron(self->history, COLOR_PAIR(2) | A_BOLD);
  wprintw(self->history, "Commands:\n");
  wattroff(self->history, A_BOLD);
  
  wprintw(self->history, "      /status <message>          : Set your status\n");
  wprintw(self->history, "      /nick <nickname>           : Set your nickname\n");
  wprintw(self->history, "      /myid                      : Print your ID\n");
  wprintw(self->history, "      /clear                     : Clear the screen\n");
  wprintw(self->history, "      /close                     : Close the current chat window\n");
  wprintw(self->history, "      /quit or /exit             : Exit program\n");
  wprintw(self->history, "      /help                      : Print this message again\n\n");

  wattroff(self->history, COLOR_PAIR(2));
}

ToxWindow new_chat(int friendnum) {
  ToxWindow ret;

  memset(&ret, 0, sizeof(ret));

  ret.onKey = &chat_onKey;
  ret.onDraw = &chat_onDraw;
  ret.onInit = &chat_onInit;
  ret.onMessage = &chat_onMessage;
  ret.onNickChange = &chat_onNickChange;
  ret.onStatusChange = &chat_onStatusChange;

  uint8_t nick[MAX_NAME_LENGTH] = {0};
  getname(friendnum, (uint8_t*) &nick);
  fix_name(nick);

  snprintf(ret.title, sizeof(ret.title), "[%s (%d)]", nick, friendnum);

  ChatContext* x = calloc(1, sizeof(ChatContext));
  x->friendnum = friendnum;

  ret.x = (void*) x;
  return ret;
}