test.c

#include <stdlib.h> // getenv
#include <stdio.h>
#include <string.h>
#include <sxupdate/api.h>

static char *fgets_no_trailing_white(char * restrict str, int size, FILE * restrict stream) {
  char *s = fgets(str, size, stream);
  while(s && *s && strchr(" \n\r\t\f", s[strlen(s)-1]))
    s[strlen(s)-1] = '\0';
  return s && *s ? s : NULL;
}
static void interaction_handler(sxupdate_t handle, enum sxupdate_step step,
                                void (*resume)(sxupdate_t, enum sxupdate_action)) {
  switch(step) {
  case sxupdate_step_have_newer_version:
    fprintf(stderr, "A newer version of this software is available. Download and update now (Y)?\n");
    {
      char answer[200];
      if(fgets_no_trailing_white(answer, sizeof(answer), stdin) && *answer && strchr("Yy", *answer))
        resume(handle, sxupdate_action_proceed);
      else {
        fprintf(stderr, "Update cancelled\n");
        resume(handle, sxupdate_action_abort);
      }
    }
    break;
  case sxupdate_step_already_up_to_date:
    fprintf(stderr, "This version is already up-to-date\n");
    // intentional fall-through
  default:
    resume(handle, sxupdate_action_none);
    break;
  }
}

struct sxupdate_semantic_version get_version() {
  struct sxupdate_semantic_version v = {};
  v.major = 1;
  v.minor = 1;
  v.patch = 1;
  v.prerelease = "alpha";
  return v;
}

int main(int argc, const char *argv[]) {
  char url[1024];
  const char *envvar;
#define var_from_env(n,v) ((envvar = getenv(n)) && strncpy(v, envvar, sizeof(v)))

  if(!(var_from_env("SXUPDATE_URL", url))) {
    fprintf(stderr, "(set SXUPDATE_URL to skip) Enter URL to fetch appcast.json file from (should begin with 'https://' or 'file://'\n e.g. https://raw.githubusercontent.com/liquidaty/sxupdate/main/test/example1/appcast.json):\n");
    if(!fgets_no_trailing_white(url, sizeof(url), stdin))
      return 1;
  }

  char installer_arg[FILENAME_MAX];
  if(!var_from_env("SXUPDATE_INSTALLER_ARGUMENT", installer_arg)) {
    fprintf(stderr, "(set SXUPDATE_INSTALLER_ARGUMENT to skip) Enter the name, if any, of any additional argument to pass to the installer e.g. /D=C:\\Program Files\\NSIS\n");
    if(!fgets_no_trailing_white(installer_arg, sizeof(installer_arg), stdin))
      *installer_arg = '\0';
  }

  char pem_path[FILENAME_MAX];
  if(!var_from_env("SXUPDATE_PEMFILE", pem_path)) {
    fprintf(stderr, "(set SXUPDATE_PEMFILE to skip) Enter the path to the public key pem file used to check the installation file, or blank to skip signature checking\n");
    if(!fgets_no_trailing_white(pem_path, sizeof(pem_path), stdin))
      *pem_path = '\0';
  }

  char have_file = 0;
  size_t len = strlen(url);
  if(len < 9 || memcmp(url, "https://", 8)) {
    if(len > 8 && !memcmp(url, "file://", 7))
      have_file = 1;
    else {
      fprintf(stderr, "Invalid URL: %s\n", url);
      return 1;
    }
  }

  char have_custom_header = 0;
  char header_name[128];
  char header_value[512];
  if(!have_file) {
    char got_value = 0;
    if(var_from_env("SXUPDATE_HEADERNAME", header_name))
      got_value = *header_name;
    else {
      fprintf(stderr, "(set SXUPDATE_HEADERNAME to skip) Enter custom header name (e.g. 'Authorization') to send with request, if any\n");
      got_value = !!fgets_no_trailing_white(header_name, sizeof(header_name), stdin);
    }
    if(got_value) {
      if(!var_from_env("SXUPDATE_HEADERVALUE", header_value)) {
        fprintf(stderr, "Enter custom header value (e.g. 'token xxxxxyyyyy')\n");
        if(!fgets_no_trailing_white(header_value, sizeof(header_value), stdin))
          *header_value = '\0';
      }
      if(!*header_value) {
        fprintf(stderr, "No header value provided!\n");
        return 1;
      }
      have_custom_header = 1;
    }
  }

  sxupdate_t sxu = sxupdate_new();
  if(sxu) {
    int err = 0;
    sxupdate_set_verbosity(sxu, 5);

    if(*pem_path == '\0')
      sxupdate_set_public_key(sxu, NULL);
    else {
      if(sxupdate_set_public_key_from_file(sxu, pem_path) != sxupdate_status_ok)
        err = 1;
    }

    if(*installer_arg)
      sxupdate_add_installer_arg(sxu, installer_arg);

    if(!err) {
      sxupdate_set_current_version(sxu, get_version);
      sxupdate_set_interaction_handler(sxu, interaction_handler);
      sxupdate_set_url(sxu, url);
      if(have_custom_header)
        sxupdate_add_header(sxu, header_name, header_value);
      if(sxupdate_execute(sxu)) {
        char *err_msg = sxupdate_err_msg(sxu);
        fprintf(stderr, "Error: %s\n", err_msg ? err_msg : "Unknown");
        free(err_msg);
      }
    }
    sxupdate_delete(sxu);
  }
}