# 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 os

import gobject
import gtk

import dimbola
        

class FolderList(dimbola.Plugin):

    '''Show list of folders with imported photos in left sidebar.'''

    ID_COL = 0
    NAME_COL = 1
        
    def __init__(self, mwc):
        self.mwc = mwc
        mwc.connect('setup-widgets', self.setup_widgets)
        self.tagids = set()
        self.store = gtk.TreeStore(gobject.TYPE_INT, gobject.TYPE_STRING)
        
        mwc.new_hook('folder-selection-changed', gobject.TYPE_NONE,
                     (gobject.TYPE_PYOBJECT, gobject.TYPE_PYOBJECT))
        
    def enable(self):
        self.mwc.add_to_sidebar('left_sidebar', 'folderlist_expander',
                                weight=dimbola.MIN_WEIGHT,
                                expand=True, fill=True)
        self.enable_signal(self.mwc, 'db-changed', self.db_changed)
        
    def disable(self):
        self.mwc.remove_from_sidebar('left_sidebar', 'folderlist_expander')
        self.disable_signals()
    
    def setup_widgets(self, mwc):
        cr = gtk.CellRendererText()
        col = gtk.TreeViewColumn()
        col.pack_start(cr)
        col.add_attribute(cr, 'text', self.NAME_COL)
        self.treeview = self.mwc.widgets['folders_treeview']
        self.treeview.append_column(col)
        self.treeview.set_model(self.store)

        sel = self.treeview.get_selection()
        sel.set_mode(gtk.SELECTION_MULTIPLE)
        sel.connect('changed', self.selection_changed)

    def db_changed(self, mwc):
        self.refresh_store()
        
    def refresh_store(self):
        tb = dimbola.TreeBuilder()
        fakes = dict()
        with self.mwc.db:
            photoids = self.mwc.db.find_photoids()
            for photoid in photoids:
                folderid, b, c, d = \
                    self.mwc.db.get_basic_photo_metadata(photoid)
                foldername = self.mwc.db.get_folder_name(folderid)
                parentname = os.path.dirname(foldername)
                parentid = self.mwc.db.find_folder(parentname)
                if parentid is None:
                    parentid = self.make_fake_parent(fakes, parentname)
                basename = os.path.basename(foldername)
                tb.add(folderid, basename, basename, parentid)
        for fakename in fakes:
            parentname = os.path.dirname(fakename)
            parentid = fakes[parentname]
            fakeid = fakes[fakename]
            if parentid == fakeid:
                parentid = None
            basename = os.path.basename(fakename) or os.sep
            tb.add(fakeid, basename, basename, parentid)
        tb.done()
        self.store.clear()
        self.populate_treemodel(tb.tree)
        self.treeview.expand_all()

    def make_fake_parent(self, fakes, parentname):
        if parentname not in fakes:
            fakes[parentname] = -len(fakes) - 1
            self.make_fake_parent(fakes, os.path.dirname(parentname))
        return fakes[parentname]

    def populate_treemodel(self, nodes, parent_iter=None):
        for node in nodes:
            folderid, foldername, children = node
            it = self.store.append(parent_iter, (folderid, foldername))
            self.populate_treemodel(children, parent_iter=it)

    def photos_in_selected_folders(self):
        '''Return photoids for all currently selected folders.'''
        sel = self.treeview.get_selection()
        model, paths = sel.get_selected_rows()
        folderids = list()
        photoids = list()
        with self.mwc.db:
            for path in paths:
                it = self.store.get_iter(path)
                folderid = self.store.get_value(it, self.ID_COL)
                photoids += self.mwc.db.find_photoids_in_folder(folderid)
                folderids.append(folderid)
        return folderids, photoids

    def selection_changed(self, *args):
        folderids, photoids = self.photos_in_selected_folders()
        self.mwc.emit('folder-selection-changed', folderids, photoids)