#include "../../include/all.h"
#include "./common.h"

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

int main(int argc, char *argv[]) {
  lm_ctx_update_list_t *list = NULL;
  int                   ret  = EXIT_FAILURE;
  lm_pkg_t              pkg, *old = NULL, *new = NULL;
  lm_ctx_t              ctx;

  if (argc != 3) {
    printf("usage: %s <pool name> <pool url>\n", argv[0]);
    return ret;
  }

  if (!mkdir_ifnot(ROOT_DIR)) {
    printf("failed to create root dir: %s\n", ROOT_DIR);
    return ret;
  }

  lm_ctx_init(&ctx);

  if (!lm_ctx_set_data(&ctx, DATA_DIR)) {
    printf("failed to set data dir: %s (%d)\n", lm_strerror(), lm_error());
    goto end;
  }

  if (!lm_ctx_set_temp(&ctx, TEMP_DIR)) {
    printf("failed to set temp dir: %s (%d)\n", lm_strerror(), lm_error());
    goto end;
  }

  if (!lm_ctx_set_root(&ctx, ROOT_DIR)) {
    printf("failed to set root dir: %s (%d)\n", lm_strerror(), lm_error());
    goto end;
  }

  if (lm_ctx_pool_add(&ctx, argv[1], argv[2]) == NULL) {
    printf("failed to add pool: %s (%d)\n", lm_strerror(), lm_error());
    goto end;
  }

  lm_ctx_ping(&ctx, NULL, NULL);
  lm_ctx_sync(&ctx, true, NULL, NULL);

  if ((list = lm_ctx_update_list(&ctx)) == NULL) {
    printf("failed to get update list: %s (%d)", lm_strerror(), lm_error());
    goto end;
  }

  if (list->count == 0) {
    printf("all packages are up-to-date\n");
    goto end;
  }

  while ((old = lm_ctx_update_list_next(list)) != NULL) {
    printf("updating %s (%s)...\n", old->name, old->version);

    if ((new = lm_ctx_update(&ctx, old)) == NULL) {
      printf("failed to update package: %s\n", lm_strerror());
      goto end;
    }

    if (!lm_ctx_download(&ctx, new, NULL, NULL)) {
      printf("failed to download new package: %s", lm_strerror());
      goto end;
    }

    if (!lm_ctx_remove(&ctx, old, NULL, NULL)) {
      printf("failed to remove old package: %s", lm_strerror());
      goto end;
    }

    if (!lm_ctx_install(&ctx, new, NULL, NULL)) {
      printf("failed to install new package: %s", lm_strerror());
      goto end;
    }
  }

  ret = EXIT_SUCCESS;

end:
  lm_ctx_update_list_free(list);
  lm_ctx_free(&ctx);
  return ret;
}