# Copyright (C) 2009  Lars Wirzenius <liw@liw.fi>
# 
# 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 3 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, see <http://www.gnu.org/licenses/>.


# This is necessary for running under Python 2.5, which we need to
# do on Debian, for now.
from __future__ import with_statement


import gtk

import dimbola


class PhotoTags(dimbola.Plugin):

    '''Show/edit selected photo's tags.'''

    def __init__(self, mwc):
        self.mwc = mwc
        mwc.connect('setup-widgets', self.setup_widgets)
        
    def setup_widgets(self, mwc):
        self.textview = mwc.widgets['tags_textview']
        self.popup = mwc.widgets['tags_textview_popup']
        self.remove_menuitem = mwc.widgets['remove_photo_tag_menuitem']
        self.taglist = dimbola.Taglist(self.textview, self.popup, 
                                       self.prepare_popup)
        self.photoid = None
        
    def enable(self):
        self.enable_signal(self.mwc.grid.model, 'selection-changed', 
                           self.selection_changed)
        self.enable_signal(self.mwc, 'tagtree-changed', 
                           lambda *args: self.refresh_tags())
        self.mwc.add_to_sidebar('right_sidebar', 'photo_tags_expander')

    def disable(self):
        self.disable_signals()
        self.mwc.remove_from_sidebar('right_sidebar', 'photo_tags_expander')
    
    def refresh_tags(self):
        with self.mwc.db:
            tags = []
            for tagid in self.mwc.db.get_tagids(self.photoid):
                tagname = self.mwc.db.get_tagname(tagid)
                if tagname is not None:
                    tags.append((tagid, tagname))
        tags.sort()
        self.taglist.set_tags(self.photoid, tags)

    def selection_changed(self, model):
        if self.mwc.db:
            if len(model.selected) == 1:
                self.photoid = model.selected[0]
                self.refresh_tags()
            else:
                self.taglist.clear()
                self.photoid = None
        else:
            self.photoid = None
    
    def tags_textview_is_sensitive(self):
        return len(self.mwc.grid.model.selected) == 1
        
    def prepare_popup(self):
        self.remove_menuitem.set_sensitive(
            self.taglist.buf.get_has_selection())
        
    def on_tags_textview_button_press_event(self, widget, event):
        return self.taglist.button_press_event(widget, event)
            
    def on_remove_photo_tag_menuitem_activate(self, *args):
        photoid, tagid = self.taglist.selected_tag()
        if photoid is not None:
            with self.mwc.db:
                self.mwc.db.remove_tagid(photoid, tagid)
            self.refresh_tags()

    def on_tags_textview_drag_motion(self, *args):
        return self.taglist.drag_motion(*args)

    def on_tags_textview_drag_leave(self, *args):
        return self.taglist.drag_leave(*args)

    def on_tags_textview_drag_data_received(self, *args):
        if self.photoid is None:
            return

        tagids = self.taglist.drag_data_received(*args)
        with self.mwc.db:
            for tagid in tagids:
                self.mwc.db.add_tagid(self.photoid, tagid)
        self.refresh_tags()