// netfs_config.cpp

#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#include <SupportDefs.h>

#include "netfs_ioctl.h"

// usage
static const char* kUsage =
"Usage: netfs_config -h | --help\n"
"       netfs_config <mount point> -a <server name>\n"
"       netfs_config <mount point> -r <server name>\n"
"options:\n"
"  -a                - adds the supplied server\n"
"  -h, --help        - print this text\n"
"  -r                - removes the supplied server\n"
;

// print_usage
static
void
print_usage(bool error)
{
	fprintf((error ? stderr : stdout), kUsage);
}

// add_server
static
status_t
add_server(int fd, const char* serverName)
{
	netfs_ioctl_add_server params;
	if (strlen(serverName) >= sizeof(params.serverName))
		return B_BAD_VALUE;
	strcpy(params.serverName, serverName);
	if (ioctl(fd, NET_FS_IOCTL_ADD_SERVER, &params) < 0)
		return errno;
	return B_OK;
}

// remove_server
static
status_t
remove_server(int fd, const char* serverName)
{
	netfs_ioctl_remove_server params;
	if (strlen(serverName) >= sizeof(params.serverName))
		return B_BAD_VALUE;
	strcpy(params.serverName, serverName);
	if (ioctl(fd, NET_FS_IOCTL_REMOVE_SERVER, &params) < 0)
		return errno;
	return B_OK;
}

// main
int
main(int argc, char** argv)
{
	// parse the arguments
	if (argc < 2) {
		print_usage(true);
		return 1;
	}
	if (strcmp(argv[1], "--help") == 0 || strcmp(argv[1], "-h") == 0) {
		print_usage(false);
		return 0;
	}
	// add or remove
	if (argc != 4) {
		print_usage(true);
		return 1;
	}
	const char* mountPoint = argv[1];
	bool add = false;
	if (strcmp(argv[2], "-a") == 0) {
		add = true;
	} else if (strcmp(argv[2], "-r") == 0) {
		add = false;
	} else {
		print_usage(true);
		return 1;
	}
	const char* serverName = argv[3];
	// open the mount point
	int fd = open(mountPoint, O_RDONLY);
	if (fd < 0) {
		fprintf(stderr, "Opening `%s' failed: %s\n", mountPoint,
			strerror(errno));
		return 1;
	}
	// do the ioctl
	status_t error = B_OK;
	if (add)
		error = add_server(fd, serverName);
	else
		error = remove_server(fd, serverName);
	if (error != B_OK)
		fprintf(stderr, "Operation failed: %s\n", strerror(error));
	// clean up
	close(fd);
	return (error == B_OK ? 0 : 1);
}