import eog, gtk, os

class HelloWorldPlugin(eog.Plugin):

    raw_file_extensions = [ 'cr2', '3fr', 'ari', 'arw', 'srf', 'sr2', 'bay',
        'crw', 'cr2', 'cap', 'iiq', 'eip', 'dcs', 'dcr', 'drf', 'k25', 'kdc',
        'dng', 'erf', 'fff', 'mef', 'mos', 'mrw', 'nef', 'nrw', 'orf', 'ptx',
        'pef', 'pxn', 'r3d', 'raf', 'raw', 'rw2', 'rwl', 'rwz', 'x3f' ]

    def __init__( self ):
        eog.Plugin.__init__( self )

    def activate( self, window ):
        self._window = window

        # insert menu item
        manager = window.get_ui_manager()
        self._action_group = gtk.ActionGroup( "EogRawToolsPluginActions" )
        self._action_group.add_actions( [ ( "EogPluginRunKeepRaw", None,
            _( "Keep Raw Image" ), "K",
            _( "Keep accompanying raw image" ),
            self.on_keep_raw_image ) ] )
        manager.insert_action_group( self._action_group, -1 )
        self._ui_id = manager.add_ui_from_string( """
            <ui>
                <menubar name="MainMenu">
                    <menu name="ToolsMenu" action="Tools">
                        <separator />
                        <menuitem name="EogPluginRawTools"
                            action="EogPluginRunKeepRaw" />
                        <separator />
                    </menu>
                </menubar>
                <popup name="ViewPopup">
                    <separator />
                    <menuitem action="EogPluginRunKeepRaw" />
                    <separator />
                </popup>
            </ui>
        """ )

        # connect to selection change
        thumb_view = window.get_thumb_view()
        self._on_selection_change_id = thumb_view.connect_after( \
            "selection_changed", self.on_selection_change, window )

        # init ui state
        self.update_action_group_sensitivity( window )

    def deactivate( self, window ):

        # remove menu items
        manager = window.get_ui_manager()
        manager.remove_ui( self._ui_id )
        manager.remove_action_group( self._action_group )
        manager.ensure_update()

        # disconnect handlers
        thumb_view = window.get_thumb_view()
        thumb_view.disconnect( self._on_selection_change_id )

    def on_selection_change( self, view, data ):
        self.update_action_group_sensitivity( data )

    def update_action_group_sensitivity( self, window ):

        # do we have just the one selected image? we can't handle multiple
        # images because EogThumbView.get_selected_images() doesn't work.
        thumb_view = window.get_thumb_view()
        self._action_group.set_sensitive( \
            thumb_view.get_n_selected() == 1 )

    def on_keep_raw_image( self, action ):

        # do we have just the one selected image? we can't handle multiple
        # images because EogThumbView.get_selected_images() doesn't work.
        thumb_view = self._window.get_thumb_view()
        if( thumb_view.get_n_selected() == 1 ):

            # get the image
            image = thumb_view.get_first_selected_image()
            if( image != None ):
                self.keep_raw_image( image )

    def keep_raw_image( self, image ):

        fname = image.get_file().get_path()

        # strip the file extension off, ready to search for raw files
        base_fname = image.get_file().get_path()
        pos = base_fname.rfind( '.' )
        if( pos != -1 ):
            base_fname = base_fname[ 0 : pos ]
        
        # check for stupidity
        if( base_fname[ -1 : None ] == '/' ):
            dialog = gtk.MessageDialog( None, gtk.DIALOG_MODAL, gtk.MESSAGE_INFO,
                gtk.BUTTONS_CLOSE, "Image is invalid!" )
            dialog.run()
            dialog.destroy()
            return

        # path the raw images will be moved to
        raw_path = os.path.dirname( fname ) + '/raw';

        # loop through valid raw file extensions
        found = False;
        for ext in self.raw_file_extensions + \
             map( lambda x: x.upper(), self.raw_file_extensions ):

            # calculate the name of the raw file and where we would move it to
            src_fname = base_fname + '.' + ext
            dst_fname = raw_path + '/' + os.path.basename( src_fname )

            # look for existing files
            if( os.path.isfile( src_fname ) ):
                found = True;

                # create the raw directory, if it doesn't exist
                if( not os.path.isdir( raw_path ) ):
                    os.mkdir( raw_path );

                # move the raw file in to the raw directory
                os.rename( src_fname, dst_fname )

            # check for already-moved raw files
            elif( os.path.isfile( dst_fname ) ):
                found = True;

        # check we found the raw file
        if( not found ):
            dialog = gtk.MessageDialog( None, gtk.DIALOG_MODAL, gtk.MESSAGE_INFO,
                gtk.BUTTONS_CLOSE, "Raw file not found!" )
            dialog.format_secondary_text( "This image doesn't appear to have an accompanying raw file." )
            dialog.run()
            dialog.destroy()
            return