conf.c

#include "conf.h"

#include <errno.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>

#include <glib.h>
#include <gtk/gtk.h>

#define streq(a,b) (strcmp(a,b) == 0)

ConfigEntry *configs = NULL;
static char *profile = NULL;

enum {
  SYMBOL_TRUE = G_TOKEN_LAST + 1,
  SYMBOL_FALSE,
};

static int conf_from_file(const char *path)
{
  int fd = open(path, O_RDONLY);
  if(!fd) {
    fprintf(stderr, "Cannot open configuration file %s: %s\n", path, strerror(errno));
    return 0;
  }

  GScanner *scanner = g_scanner_new(NULL);
  /* Don't skip linefeeds */
  scanner->config->cset_skip_characters = " \t";
  /* Identifier needs to include * and - */
  scanner->config->cset_identifier_first = G_CSET_A_2_Z G_CSET_a_2_z "_*";
  scanner->config->cset_identifier_nth   = G_CSET_A_2_Z G_CSET_a_2_z "_-*";

  g_scanner_scope_add_symbol(scanner, 0, "true",  GINT_TO_POINTER(SYMBOL_TRUE));
  g_scanner_scope_add_symbol(scanner, 0, "false", GINT_TO_POINTER(SYMBOL_FALSE));

  scanner->config->symbol_2_token = TRUE;

  scanner->input_name = g_strdup(path);
  g_scanner_input_file(scanner, fd);

  int matching_profile = 1;

  GTokenType t;
  while((t = g_scanner_get_next_token(scanner)) != G_TOKEN_EOF) {
    if(t == '\n') // Skip linefeeds here
      continue;

    if(t == G_TOKEN_IDENTIFIER) {
      if(!matching_profile) {
        // Skip tokens until linefeed
        while(g_scanner_get_next_token(scanner) != '\n')
          ;
        continue;
      }

      char *name = scanner->value.v_identifier;

      ConfigEntry *cfg = NULL;
      for(ConfigEntry *p = configs; p; p = p->next) {
        if(strcmp(name, p->longname) != 0)
          continue;
        cfg = p;
        break;
      }

      if(!cfg) {
        g_scanner_error(scanner, "\"%s\" is not a recognised setting name", name);
        goto abort;
      }

      int key;

      if(cfg->is_parametric) {
        if(g_scanner_get_next_token(scanner) != ':') {
          g_scanner_error(scanner, "Expected ':'");
          goto abort;
        }

        if(g_scanner_get_next_token(scanner) != G_TOKEN_INT) {
          g_scanner_error(scanner, "Expected integer parameter index");
          goto abort;
        }

        key = scanner->value.v_int;
      }

      if(g_scanner_get_next_token(scanner) != G_TOKEN_EQUAL_SIGN) {
        g_scanner_error(scanner, "Expected '='");
        goto abort;
      }

      t = g_scanner_get_next_token(scanner);
      switch(cfg->type) {
        case CONF_TYPE_STRING:
          if(t != G_TOKEN_STRING) {
            g_scanner_error(scanner, "Expected \"%s\" to take a string value", cfg->longname);
            goto abort;
          }
          if(cfg->is_parametric) {
            ConfigValue value = { .s = scanner->value.v_string };
            (*cfg->apply)(key, value);
          }
          else {
            if(cfg->from_file.s)
              g_free(cfg->from_file.s);
            cfg->from_file.s = g_strdup(scanner->value.v_string);
            cfg->var_set_from_file = TRUE;
          }
          break;
        case CONF_TYPE_INT:
          if(t != G_TOKEN_INT) {
            g_scanner_error(scanner, "Expected \"%s\" to take an integer value", cfg->longname);
            goto abort;
          }
          cfg->from_file.i = scanner->value.v_int;
          cfg->var_set_from_file = TRUE;
          break;
        case CONF_TYPE_DOUBLE:
          if(t == G_TOKEN_INT) {
            t = G_TOKEN_FLOAT;
            scanner->value.v_float = scanner->value.v_int;
          }
          if(t != G_TOKEN_FLOAT) {
            g_scanner_error(scanner, "Expected \"%s\" to take a float value", cfg->longname);
            goto abort;
          }
          cfg->from_file.d = scanner->value.v_float;
          cfg->var_set_from_file = TRUE;
          break;
        case CONF_TYPE_BOOL:
          if(t != (GTokenType)SYMBOL_TRUE && t != (GTokenType)SYMBOL_FALSE) {
            g_scanner_error(scanner, "Expected \"%s\" to take a boolean value", cfg->longname);
            goto abort;
          }
          cfg->from_file.i = (t == (GTokenType)SYMBOL_TRUE);
          cfg->var_set_from_file = TRUE;
          break;
      }

      if(g_scanner_get_next_token(scanner) != '\n') {
        g_scanner_error(scanner, "Expected EOL");
        goto abort;
      }
    }
    else if(t == '[') {
      t = g_scanner_get_next_token(scanner);
      if(t != G_TOKEN_IDENTIFIER ||
         strcmp(scanner->value.v_identifier, "Profile")) {
        g_scanner_error(scanner, "Expected 'Profile'");
        goto abort;
      }

      if(g_scanner_get_next_token(scanner) != G_TOKEN_IDENTIFIER) {
        g_scanner_error(scanner, "Expected profile name");
        goto abort;
      }

      matching_profile = profile && g_pattern_match_simple(scanner->value.v_identifier, profile);

      if(g_scanner_get_next_token(scanner) != ']') {
        g_scanner_error(scanner, "Expected ']'");
        goto abort;
      }

      if(g_scanner_get_next_token(scanner) != '\n') {
        g_scanner_error(scanner, "Expected EOL");
        goto abort;
      }
    }
    else {
      g_scanner_error(scanner, "Expected a setting name");
      goto abort;
    }
  }

  g_scanner_destroy(scanner);
  close(fd);

  /* Config parse successful - copy to vars */

  for(ConfigEntry *p = configs; p; p = p->next) {
    if(p->var_set || !p->var_set_from_file)
      continue;

    switch(p->type) {
      case CONF_TYPE_STRING:
        *(char**)p->var = p->from_file.s; break;
      case CONF_TYPE_INT:
        *(int*)p->var = p->from_file.i; break;
      case CONF_TYPE_DOUBLE:
        *(double*)p->var = p->from_file.d; break;
      case CONF_TYPE_BOOL:
        *(int*)p->var = p->from_file.i; break;
    }
    p->var_set = TRUE;
  }

  return 1;

abort:
  g_scanner_destroy(scanner);
  close(fd);

  return 0;
}

int conf_parse(int *argcp, char ***argvp)
{
  int n_entries = 0;
  for(ConfigEntry *p = configs; p; p = p->next)
    n_entries += (p->type == CONF_TYPE_BOOL) ? 2 : 1;

  GOptionEntry *option_entries = g_malloc0(sizeof(GOptionEntry) * (n_entries + 3));

  char *config_file = NULL;
  option_entries[0].long_name  = "config-file";
  option_entries[0].short_name = 0;
  option_entries[0].flags      = 0;
  option_entries[0].arg        = G_OPTION_ARG_FILENAME;
  option_entries[0].arg_data   = &config_file;
  option_entries[0].description     = "Path to config file";
  option_entries[0].arg_description = "PATH";

  option_entries[1].long_name  = "profile";
  option_entries[1].short_name = 'p';
  option_entries[1].flags      = 0;
  option_entries[1].arg        = G_OPTION_ARG_STRING;
  option_entries[1].arg_data   = &profile;
  option_entries[1].description     = "Profile name";
  option_entries[1].arg_description = "PROFILE";

  int i = 2;
  for(ConfigEntry *cfg = configs; cfg; cfg = cfg->next, i++) {
    if(cfg->is_parametric)
      continue;

    char *longname = g_strdup(cfg->longname);

    /* Convert foo_bar to foo-bar; easier on commandline */
    for(char *s = longname; s[0]; s++)
      if(s[0] == '_')
        s[0] = '-';

    option_entries[i].long_name  = longname;
    option_entries[i].short_name = cfg->shortname;
    option_entries[i].flags      = 0;
    option_entries[i].arg_data        = cfg->var;
    option_entries[i].description     = cfg->desc;
    option_entries[i].arg_description = cfg->argdesc;

    switch(cfg->type) {
      case CONF_TYPE_STRING:
        option_entries[i].arg = G_OPTION_ARG_STRING;
        break;
      case CONF_TYPE_INT:
        option_entries[i].arg = G_OPTION_ARG_INT;
        break;
      case CONF_TYPE_DOUBLE:
        option_entries[i].arg = G_OPTION_ARG_DOUBLE;
        break;
      case CONF_TYPE_BOOL:
        option_entries[i].arg = G_OPTION_ARG_NONE;
        i++;

        option_entries[i].long_name  = g_strdup_printf("no-%s", longname);
        option_entries[i].short_name = 0;
        option_entries[i].arg        = G_OPTION_ARG_NONE;
        option_entries[i].arg_data   = &cfg->var_set;
        option_entries[i].description = g_strdup_printf("Disable %s", cfg->desc);
        break;
    }
  }

  option_entries[i].long_name = NULL;

  GError *args_error = NULL;
  GOptionContext *args_context;

  args_context = g_option_context_new("commandline...");
  g_option_context_add_main_entries(args_context, option_entries, NULL);

  g_option_context_add_group(args_context, gtk_get_option_group(TRUE));

  /* Convert -e to -- so as to treat a -e option as a commandline vector
   * to run, and satisfy the x-terminal-emulator spec
   */
  for(i = 0; i < *argcp; i++)
    if(streq((*argvp)[i], "-e")) {
      (*argvp)[i] = "--";
      break;
    }

  if(!g_option_context_parse(args_context, argcp, argvp, &args_error)) {
    fprintf(stderr, "Option parsing failed: %s\n", args_error->message);
    return 0;
  }

  for(i = 2; i < n_entries + 1; i++)
    g_free((void*)option_entries[i].long_name);
  free(option_entries);

  /* g_option doesn't give us any way to tell if variables were set or not;
   * this is the best we can do
   */
  for(ConfigEntry *cfg = configs; cfg; cfg = cfg->next) {
    if(cfg->is_parametric)
      continue;

    switch(cfg->type) {
      case CONF_TYPE_STRING:
        cfg->var_set = *(void**)cfg->var != NULL;
        break;
      case CONF_TYPE_INT:
        cfg->var_set = *(int*)cfg->var != -1;
        break;
      case CONF_TYPE_DOUBLE:
        cfg->var_set = *(double*)cfg->var != -1.0;
        break;
      case CONF_TYPE_BOOL:
        if(*(int*)cfg->var)
          cfg->var_set = 1;
        break;
    }
  }

  if(config_file) {
    if(!conf_from_file(config_file))
      return 0;
  }
  else {
    config_file = g_strdup_printf("%s/pangoterm.cfg", g_get_user_config_dir());
    struct stat st;
    if(stat(config_file, &st) == 0)
      if(!conf_from_file(config_file))
        return 0;
    g_free(config_file);
  }

  for(ConfigEntry *cfg = configs; cfg; cfg = cfg->next) {
    if(cfg->is_parametric)
      continue;

    switch(cfg->type) {
      case CONF_TYPE_STRING:
        if(!cfg->var_set)
          *(char**)cfg->var = cfg->dflt.s;
        break;
      case CONF_TYPE_INT:
        if(!cfg->var_set)
          *(int*)cfg->var = cfg->dflt.i;
        break;
      case CONF_TYPE_DOUBLE:
        if(!cfg->var_set)
          *(double*)cfg->var = cfg->dflt.d;
        break;
      case CONF_TYPE_BOOL:
        if(!cfg->var_set)
          *(int*)cfg->var = cfg->dflt.i;
    }
  }

  return 1;
}