client.c

#if HAVE_CONFIG_H
#include "config.h"
#endif

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <err.h>
#include <fcntl.h>
#include <getopt.h>
#include <signal.h>
#include <sysexits.h>
#include <syslog.h>
#include <unistd.h>

#include "array.h"
#include "vink.h"
#include "vink-internal.h"

static int print_version;
static int print_help;

struct
{
  ARRAY_MEMBERS (const char *);
} recipients;

static struct option long_options[] =
{
    { "recipient",      required_argument, 0, 'r' },
    { "version",        no_argument, &print_version, 1 },
    { "help",           no_argument, &print_help,    1 },
    { 0, 0, 0, 0 }
};

static void
client_message (struct vink_xmpp_state *state, struct vink_message *message)
{
  fprintf (stderr, "From: %s\nTo: %s\nContent-Length: %zu\n\n%s\n",
           message->from, message->to, strlen (message->body), message->body);

  vink_message_free (message);
}

static void
client_idle (struct vink_xmpp_state *state)
{
  /* We were only supposed to send one message, so we can safely terminate the stream now */
  if (ARRAY_COUNT (&recipients))
    vink_xmpp_end_stream (state);
}

static void
read_to_buffer (int fd, struct VINK_buffer* buf)
{
  char buffer[1024];
  int result;

  ARRAY_INIT (buf);

  while (0 != (result = read (fd, buffer, sizeof (buffer))))
    {
      if (result < 0)
        err (EX_OSERR, "Read error");

      ARRAY_ADD_SEVERAL (buf, buffer, result);
    }
}

int
main (int argc, char **argv)
{
  char *config_path;
  struct vink_client* cl;
  struct vink_xmpp_callbacks callbacks;
  struct VINK_buffer message;
  int i;

  while ((i = getopt_long (argc, argv, "", long_options, 0)) != -1)
    {
      switch (i)
        {
        case 0:

          break;

        case 'r':

          ARRAY_ADD (&recipients, optarg);

          break;

        case '?':

          fprintf (stderr, "Try `%s --help' for more information.\n", argv[0]);

          return EXIT_FAILURE;
        }
    }

  if (print_help)
    {
      printf ("Usage: %s [OPTION]...\n"
              "\n"
              "      --recipient=JID        send a message to the given recipient.  This\n"
              "                             option can be given multiple times to send\n"
              "                             copies to multiple addresses\n"
              "      --help     display this help and exit\n"
              "      --version  display version information\n"
              "\n"
              "Report bugs to <morten@rashbox.org>\n", argv[0]);

      return EXIT_SUCCESS;
    }

  if (print_version)
    {
      fprintf (stdout, "%s\n", PACKAGE_STRING);

      return EXIT_SUCCESS;
    }

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

  if (ARRAY_COUNT (&recipients))
    read_to_buffer (0, &message);

  if (!(config_path = getenv ("HOME")))
    errx (EXIT_FAILURE, "HOME environment variable is not set");

  if (-1 == asprintf (&config_path, "%s/.config/vink/vink.conf", config_path))
    err (EXIT_FAILURE, "asprintf failed");

  openlog ("cvink", LOG_PID | LOG_PERROR, LOG_USER);

  if (-1 == vink_init (config_path, VINK_CLIENT, VINK_API_VERSION))
    errx (EXIT_FAILURE, "vink_init failed: %s", vink_last_error ());

  free (config_path);

  if (0 == (cl = vink_client_alloc ()))
    errx (EXIT_FAILURE, "vink_client_alloc failed: %s", vink_last_error ());

  if (-1 == vink_client_connect (cl, vink_config ("domain"), VINK_XMPP))
    errx (EXIT_FAILURE, "vink_client_connect failed: %s", vink_last_error ());

  callbacks.message = client_message;
  callbacks.queue_empty = client_idle;

  vink_xmpp_set_callbacks (vink_client_state (cl), &callbacks);

  if (!ARRAY_COUNT (&recipients))
    {
      if (-1 == vink_xmpp_set_presence (vink_client_state (cl), VINK_PRESENT))
        errx (EXIT_FAILURE, "Failed to set presence: %s", vink_last_error ());
    }
  else
    {
      size_t recipient_index;
      char* escaped_message;

      escaped_message = vink_xml_escape (&ARRAY_GET (&message, 0),
                                         ARRAY_COUNT (&message));

      for (recipient_index = 0; recipient_index < ARRAY_COUNT (&recipients); ++recipient_index)
        {
          if (-1 == vink_xmpp_send_message (vink_client_state (cl),
                                            ARRAY_GET (&recipients, recipient_index),
                                            escaped_message))
            {
              fprintf (stderr, "Warning: Failed to send message to '%s': %s\n",
                       ARRAY_GET (&recipients, recipient_index),
                       vink_last_error ());
            }
        }

      free (escaped_message);
    }

  if (-1 == vink_client_run (cl))
    errx (EXIT_FAILURE, "vink_client_run failed: %s", vink_last_error ());

  return EXIT_SUCCESS;
}