From b26b445ded92c4af4c163a687fba7e5a9d1100b1 Mon Sep 17 00:00:00 2001
From: Dan Dennedy <dan@dennedy.org>
Date: Sun, 17 Apr 2011 13:10:13 -0700
Subject: [PATCH] Add a few options and arguments to mvcp-client.

---
 src/mvcp-client/remote.c |  145 ++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 140 insertions(+), 5 deletions(-)

diff --git a/src/mvcp-client/remote.c b/src/mvcp-client/remote.c
index 3da69e6..5c225b4 100644
--- a/src/mvcp-client/remote.c
+++ b/src/mvcp-client/remote.c
@@ -21,6 +21,9 @@
 /* System header files */
 #include <stdio.h>
 #include <stdint.h>
+#include <string.h>
+#include <stdlib.h>
+#include <unistd.h>
 
 #include <mvcp/mvcp_remote.h>
 
@@ -28,10 +31,25 @@
 #include "client.h"
 #include "io.h"
 
+static void show_usage( char *program_name )
+{
+	printf(
+"Usage: %s [options] [mvcp-command* | file | -]\n"
+"Options:\n"
+"  -s host[:port]       Set server address and port (default 5250)\n"
+"  -push unit           Push MLT XML from stdin/pipe\n"
+"  -h                   Display usage help\n"
+"No arguments initiates interactive text client.\n"
+"The -s option is required to process MVCP commands whether in file, pipe, or command line.\n"
+"MVCP commands on the command line that contain spaces must be quoted.\n",
+		program_name
+	);
+}
+
 /** Connect to a remote server.
 */
 
-static mvcp_parser create_parser( )
+static mvcp_parser create_parser_prompt( )
 {
 	char server[ 132 ];
 	int port;
@@ -54,20 +72,137 @@ static mvcp_parser create_parser( )
 	return parser;
 }
 
+static mvcp_parser create_parser_arg( char *arg )
+{
+	int port = 5250;
+	char *portstr = strchr( arg, ':' );
+
+	if ( portstr )
+	{
+		port = atoi( &portstr[1] );
+		*portstr = 0;
+	}
+	printf( "Connecting to %s:%d\n", arg, port );
+	return mvcp_parser_init_remote( arg, port );
+}
+
+void report( mvcp_response response )
+{
+	int index = 0;
+	if ( response != NULL )
+		for ( index = 0; index < mvcp_response_count( response ); index ++ )
+			printf( "%s\n", mvcp_response_get_line( response, index ) );
+}
+
+static int parse_command_line( mvcp_parser parser, int argc, char **argv )
+{
+	mvcp_response response = mvcp_parser_connect( parser );
+	int i;
+	int interactive = 1;
+
+	if ( response )
+	{
+		mvcp_response_close( response );
+		response = NULL;
+		for ( i = 1; i < argc; i++ )
+		{
+			// If pipe run stdin
+			if ( !strcmp( argv[i], "-" ) )
+			{
+				response = mvcp_parser_run_file( parser, stdin );
+				report( response );
+				mvcp_response_close( response );
+				interactive = 0;
+				break;
+			}
+			// If '-push unit' option
+			else if ( !strcmp( argv[i], "-push" ) && ( i + 1 ) < argc )
+			{
+				char command[10];
+				char line[1024];
+				char *buffer = NULL;
+				size_t size = 0;
+
+				sprintf( command, "PUSH U%s", argv[++i] );
+				while ( fgets( line, 1024, stdin ) )
+				{
+					if ( !strcmp( line, "" ) )
+						break;
+					buffer = realloc( buffer, size + strlen( line ) );
+					strcat( buffer, line);
+					size += strlen( line ) + 1;
+				}
+				response = mvcp_parser_received( parser, command, buffer );
+				if ( buffer )
+					free( buffer );
+				interactive = 0;
+			}
+			// Skip -s
+			else if ( !strcmp( argv[i], "-s" ) && ( i + 1 ) < argc )
+			{
+				i++;
+			}
+			// If next arg is a readable file run it
+			else if ( argv[i][0] != '-' && !access( argv[i], R_OK ) )
+			{
+				response = mvcp_parser_run( parser, argv[i] );
+				interactive = 0;
+			}
+			// Otherwise execute command
+			else if ( argv[i][0] != '-' )
+			{
+				response = mvcp_parser_execute( parser, argv[i] );
+				interactive = 0;
+			}
+			if ( response )
+			{
+				report( response );
+				mvcp_response_close( response );
+				response = NULL;
+			}
+		}
+	}
+	else
+	{
+		printf( "Failed to connect\n" );
+	}
+	return interactive;
+}
+
 /** Main function.
 */
 
 int main( int argc, char **argv )
 {
-	mvcp_parser parser = create_parser( );
+	mvcp_parser parser = NULL;
+	int i;
+	int interactive = 1;
 
-	if ( parser != NULL )
+	for ( i = 1; i < argc; i++ )
+	{
+		if ( !strncmp( argv[i], "-h", 2 ) || !strncmp( argv[i], "--h", 3 ) )
+		{
+			show_usage( argv[0] );
+			return 0;
+		}
+		else if ( !strcmp( argv[i], "-s" ) && ( i + 1 ) < argc )
+		{
+			parser = create_parser_arg( argv[++i] );
+			if ( parser && argc > 3 )
+				interactive = parse_command_line( parser, argc, argv );
+		}
+	}
+	if ( !parser )
+	{
+		// Prompt for the server
+		parser = create_parser_prompt( );
+	}
+	if ( interactive && parser)
 	{
 		client demo = client_init( parser );
 		client_run( demo );
 		client_close( demo );
-		mvcp_parser_close( parser );
 	}
-
+	mvcp_parser_close( parser );
 	return 0;
 }
-- 
1.7.4.4