duti.c

/* duti: set default handlers for document types based on a settings file. */

#include <ApplicationServices/ApplicationServices.h>
#include <CoreFoundation/CoreFoundation.h>

#include <sys/types.h>
#include <sys/stat.h>
#include <errno.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>

#include "handler.h"

extern char		*duti_version;
extern int		nroles;
int			verbose = 0;

struct roles		rtm[] = {
    { "none",   kLSRolesNone },
    { "viewer", kLSRolesViewer },
    { "editor", kLSRolesEditor },
    { "shell",  kLSRolesShell },
    { "all",    kLSRolesAll },
};

    int
main( int ac, char *av[] )
{
    struct stat		st;
    int			c, err = 0;
    int			set = 0;
    int			( *handler_f )( char * );
    char		*path = NULL;
    char		*p;

    extern int		optind;
    extern char		*optarg;

    while (( c = getopt( ac, av, "d:e:l:hsu:Vvx:" )) != -1 ) {
	switch ( c ) {
	case 'd':	/* show default handler for UTI */
	    return( uti_handler_show( optarg, 0 ));

	case 'e':	/* UTI declarations for extension */
		return( duti_utis_for_extension( optarg ));

	case 'h':	/* help */
	default:
	    err++;
	    break;

	case 'l':	/* list all handlers for UTI */
	    return( uti_handler_show( optarg, 1 ));

	case 's':	/* set handler */
	    set = 1;
	    break;

	case 'u':	/* UTI declarations */
		return( duti_utis( optarg ));

	case 'V':	/* version */
	    printf( "%s\n", duti_version );
	    exit( 0 );

	case 'v':	/* verbose */
	    verbose = 1;
	    break;

	case 'x':	/* info for extension */
	    return( duti_default_app_for_extension( optarg ));
	}
    }

    nroles = sizeof( rtm ) / sizeof( rtm[ 0 ] );

    switch (( ac - optind )) {
    case 0 :	/* read from stdin */
	if ( set ) {
	    err++;
	}
	break;

    case 1 :	/* read from file or directory */
	if ( set ) {
	    err++;
	}
	path = av[ optind ];
	break;

    case 2 :	/* set URI handler */
	if ( set ) {
	    return( duti_handler_set( av[ optind ],
		    av[ optind + 1 ], NULL ));
	}
	/* this fallthrough works because set == 0 */
	
    case 3 :	/* set UTI handler */
	if ( set ) {
	    return( duti_handler_set( av[ optind ],
		    av[ optind + 1 ], av[ optind + 2 ] ));
	}
	/* fallthrough to error */
    
    default :	/* error */
	err++;
	break;
    }

    if ( err ) {
	fprintf( stderr, "usage: %s [ -hvV ] [ -d uti ] [ -l uti ] "
			 "[ settings_path ]\n", av[ 0 ] );
	fprintf( stderr, "usage: %s -s bundle_id { uti | url_scheme } "
			 "[ role ]\n", av[ 0 ] );
	fprintf( stderr, "usage: %s -x extension\n", av[ 0 ] );
	exit( 1 );
    }

    /* by default, read from a FILE stream */
    handler_f = fsethandler;

    if ( path ) {
	if ( stat( path, &st ) != 0 ) {
	    fprintf( stderr, "stat %s: %s\n", path, strerror( errno ));
	    exit( 2 );
	}
	switch ( st.st_mode & S_IFMT ) {
	case S_IFDIR:	/* directory of settings files */
	    handler_f = dirsethandler;
	    break;

	case S_IFREG:	/* settings file or plist */
	    if (( p = strrchr( path, '.' )) != NULL ) {
		p++;
		if ( strcmp( p, "plist" ) == 0 ) {
		    handler_f = psethandler;
		}
	    }
	    break;

	default:
	    fprintf( stderr, "%s: not a supported settings path\n", path );
	    exit( 1 );
	}
    }

    return( handler_f( path ));
}