probotic/src/main.c

/* main.c */

#define VERSION_STRING "1"

/* Parses the format username[:password]@server[:port]  */

#define GENCOPY(dst,src,l)                      \
  do                                            \
  {                                             \
    dst = malloc(sep + 1);                      \
    if (dst)                                    \
    {                                           \
      strncpy(dst,src,l);                       \
      dst[len] = '\0';                          \
    }                                           \
    else                                        \
    { return 1; }                               \
  } while (0)

DECL int
parse_url(char const * url)
{
  size_t len = strlen(url);
  size_t sep = 0, ls = 0, rs;
  while (++sep < len && url[sep] != '@');
  while (++ls  < len && url[ls] != ':')
  {
    if (ls >= sep)
    { ls = 0; break; }
  }
  rs = sep;
  while (++rs < len && url[rs] != ':');
  if (rs == len)
  { rs = 0; }
  GENCOPY(creds.username, url, ls ? ls : sep);
  if (ls)
  {
    GENCOPY(creds.password, url + ls + 1, sep - ls - 1);
  }
  if (rs)
  {
    GENCOPY(creds.server, url + sep + 1, rs - sep - 1);
    creds.port = atoi(url + rs + 1);
  }
  else
  {
    GENCOPY(creds.server, url + sep + 1, len - sep - 1);
  }
  return 0;
}

#undef GENCOPY

DECL void
rope(void)
{
  if (session)
  { irc_destroy_session(session); }
  api_rope();
}

DECL int
init(void)
{
  srand(time(NULL));
  if(api_init())
  { ERR(DB_ERROR, "Error initializing database."); }
  memset(&callbacks, 0, sizeof(callbacks));
  callbacks.event_connect = event_connect;
  callbacks.event_channel = event_channel;
  session = irc_create_session(&callbacks);
  if (!session)
  {
    ERRMSG("Error creating IRC session");
    goto fail;
  }
  atexit(rope);
  assert(creds.username != NULL);
  assert(creds.server != NULL);
  if (irc_connect(session,
                  creds.server, creds.port, creds.password,
                  creds.username, creds.username, creds.username))
  {
      fprintf(stderr, "IRC ERROR: %s\n", irc_strerror(irc_errno(session)));
      exit(1);
  }
  FULL_FREE(creds.password);
  return 0;
fail:
  FULL_FREE(creds.password);
  return 1;
}

DECL int
loop(void)
{
  /* We should figure out how the failure happens so we can tell the user that. */
  if (irc_run(session) != 0)
  { ERR(1, "Error running IRC session\nPossible issue: bad URL,"
           " no network connection, bad port, refused connection."); }
  return 0;
}

DECL void
help(void)
{
  ERRMSG(PROGN ": usage\n"
         "-channel  CHANNEL  - Sets the target channel\n"
         "-url      URL      - Sets the target URL\n"
         "-db       DBFILE   - Sets the database file (default: probotic_data.sqlite)\n"
         "-identify PASSWORD - Identifies against NickServ\n"
         "-version           - Prints Version information\n"
         "-help              - Prints this information\n"
         "\nUse format username[:password]@server[:port], port defaults to 6667.\n");
}

DECL void
version(void)
{
  ERRMSG(PROGN ": " VERSION_STRING);
}

int
main (int argc,
      char ** argv)
{
  if (argc > 1)
  {
    char * arg;
    while (++argv, --argc)
    {
      arg = *argv;
      if (*arg == '-')
      {
        ++arg;
        if (strcmp(arg, "version") == 0)
        {
          version();
          return 1;
        }
        else if (strcmp(arg, "help") == 0)
        { goto help; }
        if (argc < 2)
        { goto help; }
        if (strcmp(arg, "db") == 0)
        { db = argv[1]; }
        else if (strcmp(arg, "url") == 0)
        { parse_url(argv[1]); }
        else if (strcmp(arg, "channel") == 0)
        { free(creds.channel); creds.channel = strdup(argv[1]); }
        else if (strcmp(arg, "identify") == 0)
        { ident_password = argv[1]; }
        else
        {
          ERR(1,"Unknown command provided");
        }
        ++argv; --argc;
      }
      else
      { goto help; }
    }
  }
  atexit(creds_free);

  fprintf(stderr, "Connecting to %s:%s@%s:%d\nChannel: %s\n",
          creds.username, creds.password,
          creds.server, creds.port, creds.channel);

  if (init())
  { return 1; }

  return loop();
help:
  help();
  return 1;
}