/*
 * global_commands.c
 * Copyright (C) 2002-2003 Ushodaya Enterprises Limited
 * Author: Dan Dennedy <dan@dennedy.org>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/poll.h>
#include <fcntl.h>
#include <unistd.h>
#include <dirent.h>
#include <pthread.h>

#include "dvunit.h"
#include "global_commands.h"
#include "raw1394util.h"
#include <libavc1394/rom1394.h>
#include "log.h"

static dv_unit g_units[MAX_UNITS];


/** Return the dv_unit given a numeric index.
*/

dv_unit dv1394d_get_unit( int n )
{
	if (n < MAX_UNITS)
		return g_units[n];
	else
		return NULL;
}

/** Destroy the dv_unit given its numeric index.
*/

void dv1394d_delete_unit( int n )
{
	if (n < MAX_UNITS)
	{
		dv_unit unit = dv1394d_get_unit(n);
		if (unit != NULL)
		{
			dv_unit_close( unit );
			g_units[ n ] = NULL;
			dv1394d_log( LOG_NOTICE, "Deleted unit U%d.", n ); 
		}
	}
}

/** Destroy all allocated units on the server.
*/

void dv1394d_delete_all_units( void )
{
	int i;
	for (i = 0; i < MAX_UNITS; i++)
		if ( dv1394d_get_unit(i) != NULL )
		{
			dv_unit_close( dv1394d_get_unit(i) );
			dv1394d_log( LOG_NOTICE, "Deleted unit U%d.", i ); 
		}
}

/** Add a DV virtual vtr to the server.
*/
response_codes dv1394d_add_unit( command_argument cmd_arg )
{
	int i;
	int channel = -1;
	char *guid_str = (char*) cmd_arg->argument;
	octlet_t guid;
	uint32_t guid_hi;
	uint32_t guid_lo;
	
	sscanf( guid_str, "%08x%08x", &guid_hi, &guid_lo );
	guid = (octlet_t)guid_hi << 32 | (octlet_t) guid_lo;

	if ( dv_tokeniser_count( cmd_arg->tokeniser ) == 3 )
		channel = atoi( dv_tokeniser_get_string( cmd_arg->tokeniser, 2 ) );

	/* make sure unit does not already exit */
	for (i = 0; i < MAX_UNITS; i++)
	{
		if (g_units[i] != NULL)
			if ( dv_unit_get_guid( g_units[i] ) == guid )
			{
				dv_response_printf( cmd_arg->response, 1024, "a unit already exists for that node\n\n" );
				return RESPONSE_ERROR;
			}
	}
	
	for (i = 0; i < MAX_UNITS; i++)
	{
		if (g_units[i] == NULL)
		{
		
			g_units[ i ] = dv_unit_init( guid, channel );
			if ( g_units[ i ] == NULL )
			{
				dv_response_printf( cmd_arg->response, 1024, "failed to allocate unit\n" );
				return RESPONSE_ERROR;
			}
			g_units[ i ]->unit = i;
			dv_unit_set_notifier( g_units[ i ], dv_parser_get_notifier( cmd_arg->parser ), cmd_arg->root_dir );

			dv1394d_log( LOG_NOTICE, "added unit %d to send to node %d over channel %d", 
				i, dv_unit_get_nodeid( g_units[i] ), dv_unit_get_channel( g_units[i] ) );
			dv_response_printf( cmd_arg->response, 10, "U%1d\n\n", i );
			return RESPONSE_SUCCESS_N;
		}
	}
	
	dv_response_printf( cmd_arg->response, 1024, "no more units can be created\n\n" );

	return RESPONSE_ERROR;
}


/** List all AV/C nodes on the bus.
*/
response_codes dv1394d_list_nodes( command_argument cmd_arg )
{
	response_codes error = RESPONSE_SUCCESS_N;
	raw1394handle_t handle;
	int i, j;
	char line[1024];
	octlet_t guid;
	rom1394_directory dir;

	for ( j = 0; j < raw1394_get_num_ports(); j++ )
	{
		handle = raw1394_open(j);
		for ( i = 0; i < raw1394_get_nodecount(handle); ++i )
		{
			rom1394_get_directory( handle, i, &dir);
			if ( (rom1394_get_node_type(&dir) == ROM1394_NODE_TYPE_AVC) )
			{
				guid = rom1394_get_guid(handle, i);
				if (dir.label != NULL)
				{
					snprintf( line, 1023, "%02d %08x%08x \"%s\"\n", i, 
						(quadlet_t) (guid>>32), (quadlet_t) (guid & 0xffffffff), dir.label );
				} else {
					snprintf( line, 1023, "%02d %08x%08x \"Unlabeled Node %d\"\n", i, 
						(quadlet_t) (guid>>32), (quadlet_t) (guid & 0xffffffff), i );
				}
				dv_response_write( cmd_arg->response, line, strlen(line) );
				rom1394_free_directory( &dir);
			}
		}
		raw1394_close( handle );
	}
	dv_response_write( cmd_arg->response, "\n", 1 );
	return error;
}


/** List units already added to server.
*/
response_codes dv1394d_list_units( command_argument cmd_arg )
{
	response_codes error = RESPONSE_SUCCESS_N;
	char line[1024];
	int i;
	
	for (i = 0; i < MAX_UNITS; i++)
	{
		if (dv1394d_get_unit(i) != NULL)
		{
			snprintf( line, 1023, "U%d %02d %08x%08x %d\n", i, dv_unit_get_nodeid(g_units[i]),
			(quadlet_t) (dv_unit_get_guid(g_units[i]) >> 32), 
			(quadlet_t) (dv_unit_get_guid(g_units[i]) & 0xffffffff),
			!dv_unit_is_offline( g_units[i] ) );
			dv_response_write( cmd_arg->response, line, strlen(line) );
		}
	}
	dv_response_write( cmd_arg->response, "\n", 1 );

	return error;
}

static int
filter_files( const struct dirent *de )
{
	if ( de->d_name[ 0 ] != '.' )
		return 1;
	else
		return 0;
}

/** List clips in a directory.
*/
response_codes dv1394d_list_clips( command_argument cmd_arg )
{
	response_codes error = RESPONSE_BAD_FILE;
	const char *dir_name = (const char*) cmd_arg->argument;
	DIR *dir;
	char fullname[1024];
	struct dirent **de = NULL;
	int i, n;
	
	snprintf( fullname, 1023, "%s%s", cmd_arg->root_dir, dir_name );
	dir = opendir( fullname );
	if (dir != NULL)
	{
		struct stat info;
		error = RESPONSE_SUCCESS_N;
		n = scandir( fullname, &de, filter_files, alphasort );
		for (i = 0; i < n; i++ )
		{
			snprintf( fullname, 1023, "%s%s/%s", cmd_arg->root_dir, dir_name, de[i]->d_name );
			if ( stat( fullname, &info ) == 0 && S_ISDIR( info.st_mode ) )
				dv_response_printf( cmd_arg->response, 1024, "\"%s/\"\n", de[i]->d_name );
		}
		for (i = 0; i < n; i++ )
		{
			snprintf( fullname, 1023, "%s%s/%s", cmd_arg->root_dir, dir_name, de[i]->d_name );
			if ( lstat( fullname, &info ) == 0 && 
				 ( S_ISREG( info.st_mode ) || ( strstr( fullname, ".clip" ) && info.st_mode | S_IXUSR ) ) )
				dv_response_printf( cmd_arg->response, 1024, "\"%s\" %llu\n", de[i]->d_name, (unsigned long long) info.st_size );
			free( de[ i ] );
		}
		free( de );
		closedir( dir );
		dv_response_write( cmd_arg->response, "\n", 1 );
	}

	return error;
}

/** Set a server configuration property.
*/

response_codes dv1394d_set_global_property( command_argument cmd_arg )
{
	char *key = (char*) cmd_arg->argument;
	char *value = NULL;

	value = strchr( key, '=' );
	if (value == NULL)
		return RESPONSE_OUT_OF_RANGE;
	*value = 0;
	value++;
	dv1394d_log( LOG_DEBUG, "SET %s = %s", key, value );

	if ( strncasecmp( key, "root", 1024) == 0 )
	{
		int len = strlen(value);
		int i;
		
		/* stop all units and unload clips */
		for (i = 0; i < MAX_UNITS; i++)
		{
			if (g_units[i] != NULL)
				dv_unit_terminate( g_units[i] );
		}

		/* set the property */
		strncpy( cmd_arg->root_dir, value, 1023 );

		/* add a trailing slash if needed */
		if ( cmd_arg->root_dir[ len - 1 ] != '/')
		{
			cmd_arg->root_dir[ len ] = '/';
			cmd_arg->root_dir[ len + 1 ] = '\0';
		}
	}
	else
		return RESPONSE_OUT_OF_RANGE;
	
	return RESPONSE_SUCCESS;
}

/** Get a server configuration property.
*/

response_codes dv1394d_get_global_property( command_argument cmd_arg )
{
	char *key = (char*) cmd_arg->argument;

	if ( strncasecmp( key, "root", 1024) == 0 )
	{
		dv_response_write( cmd_arg->response, cmd_arg->root_dir, strlen(cmd_arg->root_dir) );
		return RESPONSE_SUCCESS_1;
	}
	else
		return RESPONSE_OUT_OF_RANGE;
	
	return RESPONSE_SUCCESS;
}

/** IEEE 1394 Bus Reset handler 

    This is included here for now due to all the unit management involved.
*/

static int reset_handler( raw1394handle_t h, unsigned int generation )
{
	int i, j, count, retry = 3;
	int port = (int) raw1394_get_userdata( h );
	
	raw1394_update_generation( h, generation );
	dv1394d_log( LOG_NOTICE, "bus reset on port %d", port );

	while ( retry-- > 0 ) 
	{
		raw1394handle_t handle = raw1394_open( port );
		count = raw1394_get_nodecount( handle );
		
		if ( count > 0 )
		{
			dv1394d_log( LOG_DEBUG, "bus reset, checking units" );
			
			/* suspend all units on this port */
			for ( j = MAX_UNITS; j > 0; j-- )
			{
				if ( g_units[ j-1 ] != NULL && dv_unit_get_port( g_units[ j-1 ] ) == port )
					dv_unit_suspend( g_units[ j-1 ] );
			}
			dv1394d_log( LOG_DEBUG, "All units are now stopped" );
			
			/* restore units with known guid, take others offline */
			for ( j = 0; j < MAX_UNITS; j++ )
			{
				if ( g_units[j] != NULL && 
					( dv_unit_get_port( g_units[ j ] ) == port || dv_unit_get_port( g_units[ j ] ) == -1 ) )
				{
					int found = 0;
					for ( i = 0; i < count; i++ )
					{
						octlet_t guid;
						dv1394d_log( LOG_DEBUG, "attempting to get guid for node %d", i );
						guid = rom1394_get_guid( handle, i );
						if ( guid == g_units[ j ]->guid )
						{
							dv1394d_log( LOG_NOTICE, "unit with GUID %08x%08x found", 
								(quadlet_t) (g_units[j]->guid>>32), (quadlet_t) (g_units[j]->guid & 0xffffffff));
							if ( dv_unit_is_offline( g_units[ j ] ) )
								dv_unit_online( g_units[ j ] );
							else
								dv_unit_restore( g_units[ j ] );
							found = 1;
							break;
						}
					}
					if ( found == 0 )
						dv_unit_offline( g_units[ j ] );
				}
			}
			dv1394d_log( LOG_DEBUG, "completed bus reset handler");
			raw1394_close( handle );
			return 0;
		}
		raw1394_close( handle );
	}
	dv1394d_log( LOG_CRIT, "raw1394 reported zero nodes on the bus!" );
	return 0;
}


/** One pthread per IEEE 1394 port
*/

static pthread_t raw1394service_thread[4];

/** One raw1394 handle for each pthread/port
*/

static raw1394handle_t raw1394service_handle[4];

/** The service thread that polls raw1394 for new events.
*/

static void* raw1394_service( void *arg )
{
	raw1394handle_t handle = (raw1394handle_t) arg;
	struct pollfd raw1394_poll;
	raw1394_poll.fd = raw1394_get_fd( handle );
	raw1394_poll.events = POLLIN;
	raw1394_poll.revents = 0;
	while ( 1 )
	{
		if ( poll( &raw1394_poll, 1, 200) > 0 )
		{
			if ( (raw1394_poll.revents & POLLIN) 
					|| (raw1394_poll.revents & POLLPRI) )
				raw1394_loop_iterate( handle );
		}
		pthread_testcancel();
	}
	
}


/** Start the raw1394 service threads for handling bus reset.

    One thread is launched per port on the system.
*/

void raw1394_start_service_threads( void )
{
	int port;
	for ( port = 0; port < raw1394_get_num_ports(); port++ )
	{
		raw1394service_handle[port] = raw1394_open( port );
		raw1394_set_bus_reset_handler( raw1394service_handle[port], reset_handler );
		pthread_create( &(raw1394service_thread[port]), NULL, raw1394_service, raw1394service_handle[port] );
	}
	for ( ; port < 4; port++ )
		raw1394service_handle[port] = NULL;
}

/** Shutdown all the raw1394 service threads.
*/

void raw1394_stop_service_threads( void )
{
	int i;
	for ( i = 0; i < 4; i++ )
	{
		if ( raw1394service_handle[i] != NULL )
		{
			pthread_cancel( raw1394service_thread[i] );
			pthread_join( raw1394service_thread[i], NULL );
			raw1394_close( raw1394service_handle[i] );
		}
	}
}