/* aurblup - AUR blacklist updater
 *
 * Small utility to update the AUR package blacklist. Can be used in a cronjob.
 * Check the "README" file for details.
 */

#include <alpm.h>
#include <getopt.h>
#include <mysql.h>
#include <stdio.h>
#include <string.h>

#include "config.h"

#define alpm_die(...) die(__VA_ARGS__, alpm_strerror(alpm_errno(handle)));
#define mysql_die(...) die(__VA_ARGS__, mysql_error(c));

static void die(const char *, ...);
static alpm_list_t *pkglist_append(alpm_list_t *, const char *);
static alpm_list_t *blacklist_get_pkglist();
static void blacklist_add(const char *);
static void blacklist_remove(const char *);
static void blacklist_sync(alpm_list_t *, alpm_list_t *);
static alpm_list_t *dblist_get_pkglist(alpm_list_t *);
static alpm_list_t *dblist_create(void);
static int parse_options(int, char **);
static void init(void);
static void cleanup(void);

static char *mysql_host = "localhost";
static char *mysql_socket = NULL;
static char *mysql_user = "aur";
static char *mysql_passwd = "aur";
static char *mysql_db = "AUR";

static MYSQL *c;

static alpm_handle_t *handle;

static void
die(const char *format, ...)
{
  va_list arg;

  va_start(arg, format);
  fprintf(stderr, "aurblup: ");
  vfprintf(stderr, format, arg);
  va_end(arg);

  cleanup();
  exit(1);
}

static alpm_list_t *
pkglist_append(alpm_list_t *pkglist, const char *pkgname)
{
  int len = strcspn(pkgname, "<=>");
  if (!len)
    len = strlen(pkgname);

  char *s = malloc(len + 1);

  strncpy(s, pkgname, len);
  s[len] = '\0';

  if (alpm_list_find_str(pkglist, s))
    free(s);
  else
    pkglist = alpm_list_add(pkglist, s);

  return pkglist;
}

static alpm_list_t *
blacklist_get_pkglist()
{
  MYSQL_RES *res;
  MYSQL_ROW row;
  alpm_list_t *pkglist = NULL;

  if (mysql_query(c, "SELECT Name FROM PackageBlacklist"))
    mysql_die("failed to read blacklist from MySQL database: %s\n");

  if (!(res = mysql_store_result(c)))
    mysql_die("failed to store MySQL result: %s\n");

  while ((row = mysql_fetch_row(res)))
    pkglist = pkglist_append(pkglist, row[0]);

  mysql_free_result(res);

  return pkglist;
}

static void
blacklist_add(const char *name)
{
  char *esc = malloc(strlen(name) * 2 + 1);
  char query[1024];

  mysql_real_escape_string(c, esc, name, strlen(name));
  snprintf(query, 1024, "INSERT INTO PackageBlacklist (Name) "
                        "VALUES ('%s')", esc);
  free(esc);

  if (mysql_query(c, query))
    mysql_die("failed to query MySQL database (\"%s\"): %s\n", query);
}

static void
blacklist_remove(const char *name)
{
  char *esc = malloc(strlen(name) * 2 + 1);
  char query[1024];

  mysql_real_escape_string(c, esc, name, strlen(name));
  snprintf(query, 1024, "DELETE FROM PackageBlacklist WHERE Name = '%s'", esc);
  free(esc);

  if (mysql_query(c, query))
    mysql_die("failed to query MySQL database (\"%s\"): %s\n", query);
}

static void
blacklist_sync(alpm_list_t *pkgs_cur, alpm_list_t *pkgs_new)
{
  alpm_list_t *pkgs_add, *pkgs_rem, *p;

  pkgs_add = alpm_list_diff(pkgs_new, pkgs_cur, (alpm_list_fn_cmp)strcmp);
  pkgs_rem = alpm_list_diff(pkgs_cur, pkgs_new, (alpm_list_fn_cmp)strcmp);

  if (mysql_query(c, "START TRANSACTION"))
    mysql_die("failed to start MySQL transaction: %s\n");

  for (p = pkgs_add; p; p = alpm_list_next(p))
    blacklist_add(p->data);

  for (p = pkgs_rem; p; p = alpm_list_next(p))
    blacklist_remove(p->data);

  if (mysql_query(c, "COMMIT"))
    mysql_die("failed to commit MySQL transaction: %s\n");

  alpm_list_free(pkgs_add);
  alpm_list_free(pkgs_rem);
}

static alpm_list_t *
dblist_get_pkglist(alpm_list_t *dblist)
{
  alpm_list_t *d, *p, *q;
  alpm_list_t *pkglist = NULL;

  for (d = dblist; d; d = alpm_list_next(d)) {
    alpm_db_t *db = d->data;

    if (alpm_trans_init(handle, 0))
      alpm_die("failed to initialize ALPM transaction: %s\n");
    if (alpm_db_update(0, db) < 0)
      alpm_die("failed to update ALPM database: %s\n");
    if (alpm_trans_release(handle))
      alpm_die("failed to release ALPM transaction: %s\n");

    for (p = alpm_db_get_pkgcache(db); p; p = alpm_list_next(p)) {
      alpm_pkg_t *pkg = p->data;

      pkglist = pkglist_append(pkglist, alpm_pkg_get_name(pkg));

      for (q = alpm_pkg_get_provides(pkg); q; q = alpm_list_next(q)) {
        alpm_depend_t *provide = q->data;
        pkglist = pkglist_append(pkglist, provide->name);
      }

      for (q = alpm_pkg_get_replaces(pkg); q; q = alpm_list_next(q)) {
        alpm_depend_t *replace = q->data;
        pkglist = pkglist_append(pkglist, replace->name);
      }
    }
  }

  return pkglist;
}

static alpm_list_t *
dblist_create(void)
{
  alpm_list_t *d;
  alpm_list_t *dblist = NULL;
  int i;

  for (i = 0; i < sizeof(alpm_repos) / sizeof(char *); i++) {
    if (!alpm_register_syncdb(handle, alpm_repos[i], 0))
      alpm_die("failed to register sync db \"%s\": %s\n", alpm_repos[i]);
  }

  if (!(dblist = alpm_get_syncdbs(handle)))
    alpm_die("failed to get sync DBs: %s\n");

  for (d = dblist; d; d = alpm_list_next(d)) {
    alpm_db_t *db = d->data;

    char server[1024];
    snprintf(server, 1024, ALPM_MIRROR, alpm_db_get_name(db));

    if (alpm_db_add_server(db, server))
      alpm_die("failed to set server \"%s\": %s\n", server);
  }

  return dblist;
}

static int parse_options(int argc, char **argv)
{
  int opt;

  static const struct option opts[] = {
    { "mysql-host",   required_argument, 0, 'h' },
    { "mysql-socket", required_argument, 0, 'S' },
    { "mysql-user",   required_argument, 0, 'u' },
    { "mysql-passwd", required_argument, 0, 'p' },
    { "mysql-db",     required_argument, 0, 'D' },
    { 0, 0, 0, 0 }
  };

  while((opt = getopt_long(argc, argv, "h:S:u:p:D:", opts, NULL)) != -1) {
    switch(opt) {
      case 'h':
        mysql_host = optarg;
        break;;
      case 'S':
        mysql_socket = optarg;
        break;;
      case 'u':
        mysql_user = optarg;
        break;;
      case 'p':
        mysql_passwd = optarg;
        break;;
      case 'D':
        mysql_db = optarg;
        break;;
      default:
        return 0;
    }
  }

  return 1;
}

static void
init(void)
{
  enum _alpm_errno_t alpm_err;
  if (mysql_library_init(0, NULL, NULL))
    mysql_die("could not initialize MySQL library: %s\n");
  if (!(c = mysql_init(NULL)))
    mysql_die("failed to setup MySQL client: %s\n");
  if (!mysql_real_connect(c, mysql_host, mysql_user, mysql_passwd,
                          mysql_db, 0, mysql_socket, 0))
    mysql_die("failed to initiate MySQL connection to %s: %s\n", mysql_host);

  if ((handle = alpm_initialize("/", ALPM_DBPATH, &alpm_err)) == NULL)
    die("failed to initialize ALPM: %s\n", alpm_strerror(alpm_err));
}

static void
cleanup(void)
{
  alpm_release(handle);
  mysql_close(c);
  mysql_library_end();
}

int main(int argc, char *argv[])
{
  alpm_list_t *pkgs_cur, *pkgs_new;

  if (!parse_options(argc, argv))
    return 1;

  init();

  pkgs_cur = blacklist_get_pkglist();
  pkgs_new = dblist_get_pkglist(dblist_create());
  blacklist_sync(pkgs_cur, pkgs_new);
  FREELIST(pkgs_new);
  FREELIST(pkgs_cur);

  cleanup();

  return 0;
}