class WebcastsBrowser(gtk.VBox, SignalContainer):
    
    def __init__(self):
        gtk.VBox.__init__(self)
        SignalContainer.__init__(self)

        # Init categorys.
        self.get_categorys()
        
        # load data.
        self.__load_webcast_query()
        
        # Init webcastbar.
        self.__init_webcastbar()
        
        # Init iconview.
        self.metro_view = self.get_icon_view()
        self.metro_view_sw = self.metro_view.get_scrolled_window()
        
        self.page_box = gtk.VBox()
        self.page_box.add(self.metro_view_sw)
        
        
        # webcasts view
        self.webcast_view = self.get_webcast_view()
        self.webcast_view_sw = self.webcast_view.get_scrolled_window()
        
        # collect view
        self.collected_db_file = get_config_file("favorite_webcasts.db")
        self.collected_view = self.get_webcast_view()
        self.collected_view_sw = self.collected_view.get_scrolled_window()
        
        # Init collect view.
        
        # init listview page.
        self.init_listview_page()
        Dispatcher.connect("change-webcast", self.on_dispatcher_change_webcast)
        WebcastDB.connect("changed", self.on_db_update_songs)
        
        body_paned = HPaned(handle_color=app_theme.get_color("panedHandler"), enable_drag=True)
        body_paned.add1(self.webcastbar)
        body_paned.add2(self.page_box)
        self.add(body_paned)
        
    def get_categorys(self):    
        lang = utils.get_system_lang()
        if lang.startswith("zh"):
            self.__categorys = ["region", "genre", "composite"]
            self.current_category = "region"            
        else:    
            self.__categorys = ["region_en", "genre_en"]
            self.current_category = "region_en"            
            
        self.__category_gettexts = {
            "region" : "地域分类",
            "genre"  : "流派分类",
            "region_en" : _("Region"),
            "genre_en"  : _("Genre"),
            "composite"  : "综合分类",
            "finance" : "财经",
            "sports"  : "体育",
            "music"   : "音乐",
            "news"    : "新闻",
            "network" : "网络",
            }    
        
    def __load_webcast_query(self):    
        self.__db_query = WebcastQuery()        
        if WebcastDB.isloaded():
            self.__on_db_loaded()
        else:    
            WebcastDB.connect("loaded", lambda obj: self.__on_db_loaded())
            
    def __on_db_loaded(self):        
        self.autoconnect(self.__db_query, "added", self.__on_added_songs)
        self.autoconnect(self.__db_query, "removed", self.__on_removed_songs)
        self.autoconnect(self.__db_query, "update-songs", self.__on_update_songs)
        self.autoconnect(self.__db_query, "full-update", self.__on_full_update)
        self.__db_query.set_query()
        
    def __on_added_songs(self, db_query, songs):    
        self.reload_flag = True
    
    def __on_removed_songs(self, db_query, songs):
        self.reload_flag = True
    
    def __on_update_songs(self, db_query, songs):
        self.reload_flag = True
    
    def __on_full_update(self, db_query):
        self.load_view_data()
        self.load_collect_data()
        
    def __init_webcastbar(self):    
        self.webcastbar = TreeView(enable_drag_drop=False, enable_multiple_select=False)
        self.webcastbar.connect("single-click-item", self.on_webcastbar_single_click_item)
        items = []
        for category in self.__categorys:
            items.append(CategoryTreeItem(self.__category_gettexts[category], category=category))
        items.append(CollectTreeItem(_("Favorites")))    
        self.webcastbar.add_items(items)
        self.webcastbar.select_items([self.webcastbar.visible_items[0]])
        self.webcastbar.set_size_request(121, -1)
        self.webcastbar.draw_mask = self.on_webcastbar_draw_mask        
        
    def on_webcastbar_draw_mask(self, cr, x, y, w, h):    
        draw_alpha_mask(cr, x, y, w, h ,"layoutRight")
        
    def on_webcastbar_single_click_item(self, widget, item, column, x, y):    
        if hasattr(item, "collect_flag"):
            switch_tab(self.page_box, self.collected_view_sw)
        else:    
            widget = self.page_box.get_children()[0]
            if widget != self.metro_view_sw:
                switch_tab(self.page_box, self.metro_view_sw)            
            
            if self.current_category != item.category:
                self.current_category = item.category
                self.load_view_data()
        
    def init_listview_page(self):    
        self.listview_page = gtk.VBox()
        self.text_prompt = TextPrompt("Default")
        prompt_align = set_widget_gravity(self.text_prompt, 
                                          paddings=(10, 10, 0, 0))
        prompt_box = gtk.HBox()
        back_button = BackButton()
        back_button.connect("clicked", self.on_backbutton_clicked)
        back_button_align = set_widget_gravity(back_button, gravity=(0.5, 0.5, 0, 0),
                                               paddings=(0, 0, 10, 5))
        prompt_box.pack_start(back_button_align, False, True)
        prompt_box.pack_start(prompt_align, False, False)
        
        self.listview_page.pack_start(prompt_box, False, True)
        self.listview_page.pack_start(self.webcast_view_sw, True, True)
        
    def on_backbutton_clicked(self, widget):    
        switch_tab(self.page_box, self.metro_view_sw)
        
    def switch_to_listview(self, category, title):    
        self.text_prompt.set_text(title)
        self.webcast_view.clear()
        
        songs = self.__db_query.get_songs(category, title)
        self.webcast_view.add_webcasts(songs)
        switch_tab(self.page_box, self.listview_page)
        
        
    def load_view_data(self):    
        self.metro_view.clear()                    
        if self.current_category == "composite":
            child_datas = self.__db_query.get_composite_categorys()
            gettext_datas = []
            for child in child_datas:
                gettext_datas.append((self.__category_gettexts[child], child))
            self.metro_view.add_composite_items(gettext_datas)    
                
        else:    
            child_datas = self.__db_query.get_info(self.current_category)[0]
            child_datas.sort(key=lambda x: locale.strxfrm(x))
            self.metro_view.add_webcast_items(child_datas)            
            
    def load_collect_data(self):        
        try:
            collected_objs = utils.load_db(self.collected_db_file)
        except:    
            collected_objs = None

        if collected_objs:    
            songs = [ WebcastDB.get_song(uri) for uri in collected_objs]
        else:    
            songs = None
            
        if songs:
            songs = [ song for song in songs if song.get("collected", False)]            
            if songs:
                self.collected_view.add_webcasts(songs)
        
    def get_icon_view(self):
        icon_view = WebcastIconView()
        icon_view.connect("single-click-item", self.on_iconview_single_click_item)
        icon_view.connect("double-click-item", self.on_iconview_single_click_item)
        return icon_view
    
    
    def on_iconview_single_click_item(self, widget, item, x, y):
        if item.is_composited:
            category = item.category
        else:    
            category = self.current_category
            
        title = item.title
        if item.is_composited:
            self.switch_to_listview(category, title)
        else:    
            self.switch_to_listview(category, title)
        
    def get_webcast_view(self):    
        webcast_view = MultiDragWebcastView()
        webcast_view.keymap.update({"BackSpace" : lambda : self.on_backbutton_clicked(None)})        
        return webcast_view

    
    def on_dispatcher_change_webcast(self, widget, song):
        item = self.collected_view.get_webcast_item(song)
        if item and not song.get("collected", False):
            self.collected_view.delete_items([item])
        elif song.get("collected", False):    
            self.collected_view.add_webcasts([song])
        
    
    def on_db_update_songs(self, db, infos):
        for song, tags, new_tags in infos:
            if new_tags.has_key("collected"):
                item = self.collected_view.get_webcast_item(song)
                if item:
                    if not new_tags.get("collected", False):
                        self.collected_view.delete_items([item])
                else:    
                    if new_tags.get("collected", False):
                        self.collected_view.add_webcasts([song])
        
    def save(self):
        songs = self.collected_view.get_webcasts()
        uris = [ song.get("uri") for song in songs if song.get("uri")]
        utils.save_db(uris, self.collected_db_file)
class WebcastsBrowser(gtk.VBox, SignalContainer):
    def __init__(self):
        gtk.VBox.__init__(self)
        SignalContainer.__init__(self)

        # Init categorys.
        self.get_categorys()

        # load data.
        self.__load_webcast_query()

        # Init webcastbar.
        self.__init_webcastbar()

        # Init iconview.
        self.metro_view = self.get_icon_view()
        self.metro_view_sw = self.metro_view.get_scrolled_window()

        self.page_box = gtk.VBox()
        self.page_box.add(self.metro_view_sw)

        # webcasts view
        self.webcast_view = self.get_webcast_view()
        self.webcast_view_sw = self.webcast_view.get_scrolled_window()

        # collect view
        self.collected_db_file = get_config_file("favorite_webcasts.db")
        self.collected_view = self.get_webcast_view()
        self.collected_view_sw = self.collected_view.get_scrolled_window()

        # Init collect view.

        # init listview page.
        self.init_listview_page()
        Dispatcher.connect("change-webcast", self.on_dispatcher_change_webcast)
        WebcastDB.connect("changed", self.on_db_update_songs)

        body_paned = HPaned(handle_color=app_theme.get_color("panedHandler"),
                            enable_drag=True)
        body_paned.add1(self.webcastbar)
        body_paned.add2(self.page_box)
        self.add(body_paned)

    def get_categorys(self):
        lang = utils.get_system_lang()
        if lang.startswith("zh"):
            self.__categorys = ["region", "genre", "composite"]
            self.current_category = "region"
        else:
            self.__categorys = ["region_en", "genre_en"]
            self.current_category = "region_en"

        self.__category_gettexts = {
            "region": "地域分类",
            "genre": "流派分类",
            "region_en": _("Region"),
            "genre_en": _("Genre"),
            "composite": "综合分类",
            "finance": "财经",
            "sports": "体育",
            "music": "音乐",
            "news": "新闻",
            "network": "网络",
        }

    def __load_webcast_query(self):
        self.__db_query = WebcastQuery()
        if WebcastDB.isloaded():
            self.__on_db_loaded()
        else:
            WebcastDB.connect("loaded", lambda obj: self.__on_db_loaded())

    def __on_db_loaded(self):
        self.autoconnect(self.__db_query, "added", self.__on_added_songs)
        self.autoconnect(self.__db_query, "removed", self.__on_removed_songs)
        self.autoconnect(self.__db_query, "update-songs",
                         self.__on_update_songs)
        self.autoconnect(self.__db_query, "full-update", self.__on_full_update)
        self.__db_query.set_query()

    def __on_added_songs(self, db_query, songs):
        self.reload_flag = True

    def __on_removed_songs(self, db_query, songs):
        self.reload_flag = True

    def __on_update_songs(self, db_query, songs):
        self.reload_flag = True

    def __on_full_update(self, db_query):
        self.load_view_data()
        self.load_collect_data()

    def __init_webcastbar(self):
        self.webcastbar = TreeView(enable_drag_drop=False,
                                   enable_multiple_select=False)
        self.webcastbar.connect("single-click-item",
                                self.on_webcastbar_single_click_item)
        items = []
        for category in self.__categorys:
            items.append(
                CategoryTreeItem(self.__category_gettexts[category],
                                 category=category))
        items.append(CollectTreeItem(_("Favorites")))
        self.webcastbar.add_items(items)
        self.webcastbar.select_items([self.webcastbar.visible_items[0]])
        self.webcastbar.set_size_request(121, -1)
        self.webcastbar.draw_mask = self.on_webcastbar_draw_mask

    def on_webcastbar_draw_mask(self, cr, x, y, w, h):
        draw_alpha_mask(cr, x, y, w, h, "layoutRight")

    def on_webcastbar_single_click_item(self, widget, item, column, x, y):
        if hasattr(item, "collect_flag"):
            switch_tab(self.page_box, self.collected_view_sw)
        else:
            widget = self.page_box.get_children()[0]
            if widget != self.metro_view_sw:
                switch_tab(self.page_box, self.metro_view_sw)

            if self.current_category != item.category:
                self.current_category = item.category
                self.load_view_data()

    def init_listview_page(self):
        self.listview_page = gtk.VBox()
        self.text_prompt = TextPrompt("Default")
        prompt_align = set_widget_gravity(self.text_prompt,
                                          paddings=(10, 10, 0, 0))
        prompt_box = gtk.HBox()
        back_button = BackButton()
        back_button.connect("clicked", self.on_backbutton_clicked)
        back_button_align = set_widget_gravity(back_button,
                                               gravity=(0.5, 0.5, 0, 0),
                                               paddings=(0, 0, 10, 5))
        prompt_box.pack_start(back_button_align, False, True)
        prompt_box.pack_start(prompt_align, False, False)

        self.listview_page.pack_start(prompt_box, False, True)
        self.listview_page.pack_start(self.webcast_view_sw, True, True)

    def on_backbutton_clicked(self, widget):
        switch_tab(self.page_box, self.metro_view_sw)

    def switch_to_listview(self, category, title):
        self.text_prompt.set_text(title)
        self.webcast_view.clear()

        songs = self.__db_query.get_songs(category, title)
        self.webcast_view.add_webcasts(songs)
        switch_tab(self.page_box, self.listview_page)

    def load_view_data(self):
        self.metro_view.clear()
        if self.current_category == "composite":
            child_datas = self.__db_query.get_composite_categorys()
            gettext_datas = []
            for child in child_datas:
                gettext_datas.append((self.__category_gettexts[child], child))
            self.metro_view.add_composite_items(gettext_datas)

        else:
            child_datas = self.__db_query.get_info(self.current_category)[0]
            child_datas.sort(key=lambda x: locale.strxfrm(x))
            self.metro_view.add_webcast_items(child_datas)

    def load_collect_data(self):
        try:
            collected_objs = utils.load_db(self.collected_db_file)
        except:
            collected_objs = None

        if collected_objs:
            songs = [WebcastDB.get_song(uri) for uri in collected_objs]
        else:
            songs = None

        if songs:
            songs = [song for song in songs if song.get("collected", False)]
            if songs:
                self.collected_view.add_webcasts(songs)

    def get_icon_view(self):
        icon_view = WebcastIconView()
        icon_view.connect("single-click-item",
                          self.on_iconview_single_click_item)
        icon_view.connect("double-click-item",
                          self.on_iconview_single_click_item)
        return icon_view

    def on_iconview_single_click_item(self, widget, item, x, y):
        if item.is_composited:
            category = item.category
        else:
            category = self.current_category

        title = item.title
        if item.is_composited:
            self.switch_to_listview(category, title)
        else:
            self.switch_to_listview(category, title)

    def get_webcast_view(self):
        webcast_view = MultiDragWebcastView()
        webcast_view.keymap.update(
            {"BackSpace": lambda: self.on_backbutton_clicked(None)})
        return webcast_view

    def on_dispatcher_change_webcast(self, widget, song):
        item = self.collected_view.get_webcast_item(song)
        if item and not song.get("collected", False):
            self.collected_view.delete_items([item])
        elif song.get("collected", False):
            self.collected_view.add_webcasts([song])

    def on_db_update_songs(self, db, infos):
        for song, tags, new_tags in infos:
            if new_tags.has_key("collected"):
                item = self.collected_view.get_webcast_item(song)
                if item:
                    if not new_tags.get("collected", False):
                        self.collected_view.delete_items([item])
                else:
                    if new_tags.get("collected", False):
                        self.collected_view.add_webcasts([song])

    def save(self):
        songs = self.collected_view.get_webcasts()
        uris = [song.get("uri") for song in songs if song.get("uri")]
        utils.save_db(uris, self.collected_db_file)
 def __load_webcast_query(self):    
     self.__db_query = WebcastQuery()        
     if WebcastDB.isloaded():
         self.__on_db_loaded()
     else:    
         WebcastDB.connect("loaded", lambda obj: self.__on_db_loaded())
 def __load_webcast_query(self):
     self.__db_query = WebcastQuery()
     if WebcastDB.isloaded():
         self.__on_db_loaded()
     else:
         WebcastDB.connect("loaded", lambda obj: self.__on_db_loaded())