1 files changed, 70 insertions, 7 deletions
diff --git a/src/backend-direct.c b/src/backend-direct.c
index 41a4053..dd381fb 100644
--- a/src/backend-direct.c
+++ b/src/backend-direct.c
@@ -126,6 +126,16 @@ struct gamma_state {
 	unsigned crtcs_selected : 1;
 
 	/**
+	 * Whether the program has connected to the site
+	 */
+	unsigned connected : 1;
+
+	/**
+	 * The libgamma constant for the adjustment method
+	 */
+	int method;
+
+	/**
 	 * Selected parition (if `partition_selected`)
 	 */
 	size_t selected_partition;
@@ -142,10 +152,18 @@ struct gamma_state {
 
 	/**
 	 * Indices of selected CRTCs
+	 * 
+	 * Deallocated by when no longer needed
 	 */
 	size_t *selected_crtcs;
 
 	/**
+	 * Name of site to connect to, `NULL` if not selected
+	 * or once connected to the site
+	 */
+	char *site_name;
+
+	/**
 	 * State for selected paritions
 	 */
 	struct partition_state *partitions;
@@ -180,11 +198,16 @@ direct_create(struct gamma_state **state_out, int method, const char *method_nam
 	state = *state_out = ecalloc(1, sizeof(**state_out));
 	state->selected_crtcs = NULL;
 	state->partitions = NULL;
+	state->site_name = NULL;
+	state->method = method;
 	state->multiple_sites = caps.multiple_sites;
 	state->multiple_partitions = caps.multiple_partitions;
 	state->multiple_crtcs = caps.multiple_crtcs;
 	state->partitions_are_graphics_cards = caps.partitions_are_graphics_cards;
-	
+
+	if (state->multiple_sites)
+		return 0;
+
 	err = libgamma_site_initialise(&state->site, method, NULL);
 	if (err) {
 		weprintf("libgamma_site_initialise %s NULL: %s", libgamma_const_of_method(method), libgamma_strerror(err));
@@ -208,16 +231,19 @@ direct_print_help(FILE *f, int method)
 		return;
 	}
 
+	if (caps.multiple_sites)
+		fprintf(f, "  display=%s  %s\n", _("NAME  "), _("Display server instance to apply adjustments to"));
+
 	if (caps.multiple_partitions && caps.partitions_are_graphics_cards)
 		/* TRANSLATORS: "N" represents "ordinal"; right-pad with spaces to preserve display width */
-		fprintf(f, "  card=%s   %s\n", _("N     "), _("Graphics card to apply adjustments to"));
+		fprintf(f, "  card=%s     %s\n", _("N     "), _("Graphics card to apply adjustments to"));
 	else if (caps.multiple_partitions)
-		fprintf(f, "  screen=%s %s\n", _("N     "), _("X screen to apply adjustments to"));
+		fprintf(f, "  screen=%s   %s\n", _("N     "), _("X screen to apply adjustments to"));
 
 	if (caps.multiple_crtcs)
-		fprintf(f, "  crtc=%s   %s\n", _("N     "), _("List of comma-separated CRTCs to apply adjustments to"));
+		fprintf(f, "  crtc=%s     %s\n", _("N     "), _("List of comma-separated CRTCs to apply adjustments to"));
 
-	if (caps.multiple_partitions || caps.multiple_crtcs)
+	if (caps.multiple_sites || caps.multiple_partitions || caps.multiple_crtcs)
 		fprintf(f, "\n");
 }
 
@@ -225,7 +251,9 @@ direct_print_help(FILE *f, int method)
 int
 direct_set_option(struct gamma_state *state, const char *key, const char *value)
 {
-	if (state->multiple_partitions && !strcasecmp(key, state->partitions_are_graphics_cards ? "card" : "screen")) {
+	if (state->multiple_sites && !strcasecmp(key, "display")) {
+		return direct_set_site(state, key, value);
+	} else if (state->multiple_partitions && !strcasecmp(key, state->partitions_are_graphics_cards ? "card" : "screen")) {
 		return direct_set_partitions(state, key, value);
 	} else if (state->multiple_crtcs && !strcasecmp(key, "crtc")) {
 		return direct_set_crtcs(state, key, value);
@@ -240,6 +268,18 @@ direct_set_option(struct gamma_state *state, const char *key, const char *value)
 
 
 int
+direct_set_site(struct gamma_state *state, const char *key, const char *value)
+{
+	if (state->site_name) {
+		weprintf(_("`%s' option specified multiple times, using last selection."), key);
+		free(state->site_name);
+	}
+	state->site_name = estrdup(value);
+	return 0;
+}
+
+
+int
 direct_set_partitions(struct gamma_state *state, const char *key, const char *value)
 {
 	const char *end;
@@ -310,6 +350,25 @@ direct_start(struct gamma_state *state)
 	size_t i, j, k, num, part;
 	int err;
 
+	/* Connect to display server */
+	if (state->multiple_sites) {
+		if (state->site_name && !*state->site_name) {
+			free(state->site_name);
+			state->site_name = NULL;
+		}
+		err = libgamma_site_initialise(&state->site, state->method, state->site_name);
+		if (err) {
+			weprintf("libgamma_site_initialise %s %s: %s",
+			         libgamma_const_of_method(state->method),
+			         state->site_name ? state->site_name : "NULL",
+			         libgamma_strerror(err));
+			return -1;
+		}
+		state->connected = 1;
+		free(state->site_name);
+		state->site_name = NULL;
+	}
+
 	/* Allocate partition states */
 	if (state->partition_selected) {
 		state->partitions = ecalloc(1, sizeof(*state->partitions));
@@ -505,6 +564,8 @@ direct_start(struct gamma_state *state)
 		return -1;
 	}
 
+	free(state->selected_crtcs);
+	state->selected_crtcs = NULL;
 	return 0;
 }
 
@@ -646,6 +707,8 @@ direct_free(struct gamma_state *state)
 		free(state->partitions);
 	}
 	free(state->selected_crtcs);
-	libgamma_site_destroy(&state->site);
+	if (state->connected)
+		libgamma_site_destroy(&state->site);
+	free(state->site_name);
 	free(state);
 }