Movie list menu: store the movie source

[cinaest] / src / movie-list-menu.vala

diff --git a/src/movie-list-menu.vala b/src/movie-list-menu.vala
index ac9e9f6..daea9e4 100644 (file)
--- a/src/movie-list-menu.vala
+++ b/src/movie-list-menu.vala
@@ -20,7 +20,7 @@ using Gtk;
 using Hildon;
 
 public class MovieListMenu : AppMenu {
-       public TreeSortable sortable;
+       public MovieListStore store;
        public MovieFilter filter;
        private weak MovieListWindow movie_list_window;
        private Hildon.Button filter_year;
@@ -53,15 +53,15 @@ public class MovieListMenu : AppMenu {
                // Connect signals
                sort_by_title.toggled.connect (button => {
                        if (button.get_active ())
-                               sortable.set_sort_column_id (MovieListStore.Columns.TITLE, Gtk.SortType.ASCENDING);
+                               store.set_sort_column_id (MovieListStore.Columns.TITLE, Gtk.SortType.ASCENDING);
                });
                sort_by_year.toggled.connect (button => {
                        if (button.get_active ())
-                               sortable.set_sort_column_id (MovieListStore.Columns.YEAR, Gtk.SortType.DESCENDING);
+                               store.set_sort_column_id (MovieListStore.Columns.YEAR, Gtk.SortType.DESCENDING);
                });
                sort_by_rating.toggled.connect (button => {
                        if (button.get_active ())
-                               sortable.set_sort_column_id (MovieListStore.Columns.RATING, Gtk.SortType.DESCENDING);
+                               store.set_sort_column_id (MovieListStore.Columns.RATING, Gtk.SortType.DESCENDING);
                });
 
                add_filter (sort_by_title);
@@ -101,9 +101,11 @@ public class MovieListMenu : AppMenu {
                show_all ();
        }
 
+       private MovieSource source_;
        public MovieSource source {
                set {
-                       if (SourceFlags.EDITABLE in value.get_flags ()) {
+                       source_ = value;
+                       if (source_ != null && SourceFlags.EDITABLE in source_.get_flags ()) {
                                delete_movies.show ();
                                import_movies.show ();
                        } else {