#include <stdio.h>
#include <ctype.h>
#include <hesiod.h>

/*
 * Make a hesiod cluster query
 * for the machine you are on
 * and produce a set of environment variable
 * assignments for the C shell or the Bourne shell,
 * depending on the '-b' flag
 *
 * If any stdio errors, truncate standard output to 0
 * and return an exit status.
 */

main(argc, argv)
char *argv[];
{
	register char **hp;
	int bourneshell = 0;
	char myself[80];

	if (argc < 3 || argc > 4) {
		fprintf(stderr, "usage: getcluster [-b] hostname version\n");
		exit(-1);
	}
	if (argc == 4 && strcmp(argv[1], "-b") == 0) {
		bourneshell++;
		argv++;
	}
	
	hp = hes_resolve(argv[1], "cluster");
	if (hp != NULL)
		shellenv(hp, bourneshell, argv[2]);
	return 0;
}

shellenv(hp, bourneshell, version)
char **hp;
int bourneshell;
char *version;
{
	char var[80], val[80], vers[80], **hp_save=hp;
	int specific = 0;

	while (specific < 2) {
		if (bourneshell) {
			while(*hp) {
				vers[0] = '\0';
				sscanf(*hp++, "%s %s %s", var, val, vers);
				if ((specific && !vers[0]) ||
				    (!specific && vers[0]) ||
				    (vers[0] != '\0' && strcmp(vers, version)))
					continue;
				upper(var);
				printf("%s=%s ; export %s\n", var, val, var);
			}
		} else
			while(*hp) {
				vers[0] = '\0';
				sscanf(*hp++, "%s %s %s", var, val, vers);
				if ((specific && !vers[0]) ||
				    (!specific && vers[0]) ||
				    (vers[0] != '\0' && strcmp(vers, version)))
					continue;
				upper(var);
				printf("setenv %s %s\n", var, val);
			}
		specific++;
		hp = hp_save;
	}
	if (ferror(stdout)) {
		ftruncate(fileno(stdout), 0L);
		exit(-1);
	}
}

upper(v)
register char *v;
{
	while(*v) {
		*v = toupper(*v);
		v++;
	}
}