probotic/src/irc.c

/* irc.c - IRC interface */

#define PREFIX_COMMAND_CHAR '!'

#define IRCMSG(msg) irc_cmd_msg(session, creds.channel, msg)

#define MAX(a,b) (a) > (b)

typedef struct
{
  char *  username;
  char *  password;
  char *  channel;
  char *  server;
  int     port;
} creds_t;

VARDECL creds_t creds =
{
  .username = NULL,
  .password = NULL,
  .channel  = NULL,
  .server   = NULL, /* localhost? */
#ifndef IRC_SSL_SUPPORT
  .port     = 6667
#else
  .port     = 6669
#endif /* !IRC_SSL_SUPPORT */
};

VARDECL char * ident_password = NULL;

VARDECL irc_session_t * session;
VARDECL irc_callbacks_t callbacks;

VARDECL char * current_username = NULL;

VARDECL char const * help_msg =
/* IRC_GREEN "!help           " IRC_STOP "   : This message\n" */
IRC_GREEN "!remind         " IRC_STOP "   : Dump current assignment\n"
IRC_GREEN "!reroll         " IRC_STOP "   : Rerolls assignment\n"
/* IRC_GREEN "!repo <link>" IRC_STOP "   : Sets project repository link\n" */
/* IRC_GREEN "!raw <sql>      " IRC_STOP "   : Execute raw SQL\n" */
/* IRC_GREEN "!dump           " IRC_STOP "   : List all possible projects\n" */
/* IRC_GREEN "!request        " IRC_STOP "   : Request personal project\n" */
IRC_GREEN "!remind         " IRC_STOP "   : Prints your assignment\n"
IRC_GREEN "!poll           " IRC_STOP "   : Start the democratic process (N message ...])\n"
IRC_GREEN "!vote           " IRC_STOP "   : Casts a vote (y/n [message ...])\n"
IRC_GREEN "!stop           " IRC_STOP "   : Stop the current polling\n"
IRC_BLUE  "!magic          " IRC_STOP "   : A random value at or below 100 or (N)\n";

DECL void parse_command(char const * cmd);
DECL size_t parse_secondary_username(char const * msg);

DECL char *
get_username(const char * origin)
{
  const char USERNAME_TERMINATOR = '!';
  int i = 0;
  char * r;
  while (origin[i] != USERNAME_TERMINATOR)
  { i++; }
  r = (char *) malloc(i + 1);
  if (r)
  {
    strncpy(r, origin, i);
    r[i] = '\00';
  }
  return r;
}

DECL void
ircmsg(char const * reciever, char const * fmt,
       ...)
{
  va_list args;
  char * fmtdmsg;
  char * swp;
  const char * delim = "\n";
  char * data;
  if(!strcmp(fmt, "") || fmt == NULL)
  { return; }

  va_start(args, fmt);
  if(vasprintf(&fmtdmsg, fmt, args) == -1)
  { exit(1); }

  puts(fmtdmsg);
  data = strtok(fmtdmsg, delim);
  do
  {
    swp = irc_color_convert_to_mirc(data);
    irc_cmd_msg(session, reciever, swp);
    free(swp);
  } while((data = strtok(NULL, delim), data));

  free(fmtdmsg);
  va_end(args);
}



DECL void
event_connect(irc_session_t * lsession,
              char const  * event,
              char const  * origin,
              char const ** params,
              unsigned int count)
{
  (void) event;
  (void) origin;
  (void) params;
  (void) count;
  /* msg ChanServ IDENTIFY? */
  irc_cmd_join(lsession, creds.channel, 0);
  if (ident_password)
  {
    ircmsg("NickServ", "IDENTIFY %s", ident_password);
    memset(ident_password, '\0', strlen(ident_password));
  }

#ifdef INITIAL_ASSIGNMENT_MESSAGE
  if(is_no_assignment(creds.channel))
  {
    ircmsg(creds.channel,
					 IRC_RED "No assignment for this channel. Finding a new..." IRC_STOP);
    random_assign(creds.channel);
  }
  ircmsg(creds.channel, remind(creds.channel));
#endif /* INITIAL_ASSIGNMENT_MESSAGE */
}

DECL void
event_channel(irc_session_t * lsession,
              char const  * event,
              char const  * origin,
              char const ** params,
              unsigned int count)
{
  /* char const * channel = params[0]; */
  char const * message = params[1];
  (void) lsession;
  (void) event;
  (void) origin;
  (void) message;
  (void) count;
  /* fetches username and logs message */
	if (!(current_username = get_username(origin)))
  { return; }
  printf("<%s> %s\n", current_username, message);
  /* Detects any respecified names */
  /* message += parse_secondary_username(message); */
  /* Parse commands */
  if (*message == PREFIX_COMMAND_CHAR &&
      message[1] != '\0')
  { parse_command(message+1); }
  free(current_username);
  current_username = NULL;
}

/* 'abc' SINGLE
   'def ' SINGLE
   'ghi jkl' MULTI */

DECL int
has_arg(char const * cmd)
{
  char const * start = cmd;
  while (isalnum(*cmd))
  {
    if (*cmd == '\0')
    { break; }
    ++cmd;
  }
  while (*cmd != '\0')
  {
    if (!isspace(*cmd))
    { return cmd - start; }
    ++cmd;
  }
  return 0;
}

DECL void
parse_command(char const * cmd)
{
  static int vote_count;
  size_t i = 0;
  char * msgswp = NULL;
  /* size_t len = strlen(cmd); */
  if (!(i = has_arg(cmd)))
  {
		printf("NARG Handling '%s'\n", cmd);
    /* NO ARGUMENTS */
    if (strncmp(cmd, "remind", 6) == 0)
    {
      msgswp = remind(current_username);
      ircmsg(creds.channel, "%s: %s", current_username, msgswp);
    }
    else if (strncmp(cmd, "help", 4) == 0)
    { ircmsg(creds.channel, help_msg); }
    else if (strncmp(cmd, "magic", 5) == 0)
    { ircmsg(creds.channel, "%s: " IRC_YELLOW "%d" IRC_STOP, current_username, (rand() % 100) + 1); }
#ifndef NO_VULN_COMMANDS
    else if (strncmp(cmd, "dump", 4) == 0)
    {
      ircmsg(creds.channel, "%s: All projects:", current_username);
      msgswp = dump();
      ircmsg(creds.channel, msgswp);
    }
#endif /* !NO_VULN_COMMANDS */
    else if (strncmp(cmd, "reroll", 6) == 0)
    {
      purge_assignments(current_username);
      random_assign(current_username);
      ircmsg(creds.channel, "%s: %s", current_username, remind(current_username));
    }
    else if (strncmp(cmd, "stop", 4) == 0)
    {
      if (vote_count)
      {
        ircmsg(creds.channel, "poll: stop");
        vote_count = 0;
      }
    }
  }
  else /* HAS ARGUMENTS */
  {
		printf("ARG Handling '%s'\n", cmd);
    static int yes, no;
    char const * const arg = cmd + i;
    /* fprintf(stderr, "argoff: %p; i: %ld; arg: %sEOA\n", cmd + i + 1, i, arg); */
#ifndef NO_VULN_COMMANDS
    if (strncmp(cmd, "raw", 3) == 0)
    {
      printf("RAW\n");
      ircmsg(creds.channel, "%s: Executing SQL `%s'.", current_username, arg);
      msgswp = raw(arg);
      ircmsg(creds.channel, msgswp);
    } else
#endif /* !NO_VULN_COMMANDS */
#if 0
    if (strncmp(cmd, "repo", 4) == 0)
    {
      /* ircmsg(creds.channel, "%s: Setting project repository...", current_username); */
      set_repo(creds.channel, arg);
      msgswp = remind(creds.channel);
      ircmsg(creds.channel, "%s: %s", current_username, msgswp);
    }
#endif /* 0 */
    if (strncmp(cmd, "magic", 5) == 0)
    { ircmsg(creds.channel, "%s: " IRC_YELLOW "%d" IRC_STOP, current_username, (rand() % atoi(arg)) + 1); }
		else if (strncmp(cmd, "echo", 3) == 0)
		{ ircmsg(creds.channel, "%s", arg); }
    else if (strncmp(cmd, "poll", 4) == 0)
    {
      yes = no = 0;
      vote_count = atoi(arg);
      if (!vote_count)
      { ircmsg(creds.channel, "!poll NUMBER_OF_VOTES What your voting on ..."); }
      else
      { ircmsg(creds.channel, "poll: start"); }
    }
    else if (strncmp(cmd, "vote", 4) == 0)
    {
      /* fprintf(stderr, "y%d n%d vote_count %d\n", yes, no, vote_count); */
      if (vote_count)
      {
        switch (*arg)
        {
        case 'Y': case 'y': ++yes; break;
        case 'N': case 'n': ++no;  break;
        default: ircmsg(creds.channel, "Unknown: '%c', use y/n", *arg); goto stop;
        }
        if (--vote_count)
        { ircmsg(creds.channel, "Votes remaining: %d", vote_count); }
        else
        { ircmsg(creds.channel, "poll: %s", MAX(yes,no) ? "PASSED" : "REJECTED"); }
      }
    }
  }
stop:
  free(msgswp);
}

/* mutates current_username and returns positional offset to the first character that isn't related to the uname */
DECL size_t
parse_secondary_username(char const * msg)
{
  int ret = 0;
#define SECONDARY_NAMES_BOT "cnile"
  if (strcmp(current_username,SECONDARY_NAMES_BOT) == 0)
  if (*msg == '<')
  { while (msg[++ret] != '\0' ||
           msg[ret] != '>'); }
  fprintf(stderr, "msg[ret] = %c", msg[ret]);
  return ret;
}