IMDb plugin: add movie action to open the IMDb web page for a given movie
[cinaest] / src / plugins / imdb-plugin.vala
1 /* This file is part of Cinaest.
2  *
3  * Copyright (C) 2009 Philipp Zabel
4  *
5  * Cinaest is free software: you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation, either version 3 of the License, or
8  * (at your option) any later version.
9  *
10  * Cinaest is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License
16  * along with Cinaest. If not, see <http://www.gnu.org/licenses/>.
17  */
18
19 using Gtk;
20 using Hildon;
21
22 class IMDbPlugin : Plugin {
23         private dynamic DBus.Object server;
24         List<MovieSource> sources;
25         private weak Osso.Context osso_context;
26         private Gtk.Window _window;
27
28         public override void hello (Gtk.Window window, Osso.Context context) {
29                 string filename = Path.build_filename (Environment.get_user_cache_dir(),
30                                                        "cinaest", "imdb.db", null);
31                 stdout.printf ("IMDb Plugin Loaded. Cache Db: %s\n", filename);
32
33                 if (!FileUtils.test (filename, FileTest.EXISTS)) {
34                         var note = new Note.confirmation (window, "There is no local copy of the Internet Movie Database (IMDb). Download it now?\n\nThis can be started later from the Settings dialog.");
35
36                         var response = note.run ();
37                         note.destroy ();
38
39                         if (response == ResponseType.OK) {
40                                 download_imdb (window);
41                         }
42                 }
43
44                 var source = new IMDBSource ();
45
46                 sources = new List<MovieSource> ();
47                 sources.append (source);
48
49                 osso_context = context;
50                 _window = window;
51
52                 // FIXME - this forces the inclusion of config.h
53                 (void) Config.GETTEXT_PACKAGE;
54         }
55
56         private void download_imdb (Gtk.Window window) {
57                 var picker = new PickerDialog (window);
58                 var selector = new TouchSelector.text ();
59                 string[] mirrors;
60
61                 try {
62                         var conn = DBus.Bus.get (DBus.BusType.SESSION);
63                         server = conn.get_object (IMDbDownloader.DBUS_SERVICE,
64                                                   IMDbDownloader.DBUS_OBJECT,
65                                                   IMDbDownloader.DBUS_IFACE);
66
67                         mirrors = server.get_mirrors ();
68                 } catch (DBus.Error e) {
69                         warning ("Failed connect to IMDb downloader: %s", e.message);
70                         return;
71                 }
72
73                 foreach (string mirror in mirrors) {
74                         selector.append_text (mirror);
75                 }
76
77                 picker.set_title ("Please select a source mirror");
78                 picker.set_selector (selector);
79
80                 int res = picker.run();
81                 string mirror = selector.get_current_text ();
82                 picker.destroy();
83
84                 if (res == ResponseType.OK) {
85                         var download = new IMDbDownloadDialog (window);
86                         download.run (server, mirror);
87                         download.destroy ();
88                 }
89         }
90
91         public override unowned List<MovieSource> get_sources () {
92                 return sources;
93         }
94
95         public override List<MovieAction> get_actions (Movie movie) {
96                 List<MovieAction> list = null;
97
98                 if (movie.year > 0)
99                         list.append (new MovieAction (_("IMDb page"), on_visit_imdb, movie));
100                 else
101                         list.append (new MovieAction (_("Lookup on IMDb"), on_visit_imdb, movie));
102
103                 return list;
104         }
105
106         private void on_visit_imdb (Movie movie) {
107                 var url = "http://www.imdb.com/find?s=tt&q=" + movie.title.replace(" ", "+");
108                 if (movie.year > 0)
109                         url += "+(%d)".printf (movie.year);
110
111                 var status = osso_context.rpc_run_with_defaults ("osso_browser", "open_new_window", null, 's', url, 'b', false);
112                 if (status != Osso.Status.OK) {
113                         var banner = (Banner) Hildon.Banner.show_information_with_markup (_window, null, "Failed to open browser.");
114                         banner.set_timeout (1500);
115                 }
116         }
117
118         public override void settings_dialog (Gtk.Window window) {
119                 var dialog = new Gtk.Dialog ();
120                 dialog.set_transient_for (window);
121                 dialog.set_title (_("IMDb plugin settings"));
122
123                 string filename = Path.build_filename (Environment.get_user_cache_dir(),
124                                                        "cinaest", "imdb.db", null);
125                 var file = File.new_for_path (filename);
126                 var info = file.query_info (FILE_ATTRIBUTE_TIME_MODIFIED, FileQueryInfoFlags.NONE, null);
127                 TimeVal tv;
128                 info.get_modification_time (out tv);
129
130                 var date = Date ();
131                 date.set_time_val (tv);
132                 char[] s = new char[64];
133                 date.strftime (s, "%x");
134
135                 var download = new Hildon.Button.with_text (SizeType.FINGER_HEIGHT, ButtonArrangement.VERTICAL, _("Download"), "Last update: " + (string) s);
136
137                 VBox content = (VBox) dialog.get_content_area ();
138                 content.pack_start (download, true, true, 0);
139
140                 download.clicked.connect (() => {
141                         download_imdb (window);
142                 });
143
144                 dialog.show_all ();
145                 dialog.run ();
146                 dialog.destroy ();
147         }
148
149         public override unowned string get_name () {
150                 return "IMDb";
151         }
152 }
153
154 class IMDBSource : MovieSource {
155         MovieSource.ReceiveMovieFunction _get_callback;
156         public override async void get_movies (MovieFilter filter, MovieSource.ReceiveMovieFunction callback, int limit, Cancellable? cancellable) {
157                 // IMDb has too many movies
158                 if (filter.title == "")
159                         return;
160                 var sqlite = new IMDbSqlite (Path.build_filename (Environment.get_user_cache_dir (),
161                                              "cinaest", "imdb.db", null));
162
163                 _get_callback = callback;
164                 yield sqlite.query (filter, receive_movie, cancellable);
165         }
166
167         private void receive_movie (string title, int year, int rating, int genres) {
168                 Movie movie = new Movie ();
169                 movie.title = title;
170                 movie.year = year;
171                 movie.rating = rating;
172                 movie.genres.field = genres;
173                 // TODO - depending on settings, this could be something else, like director info or runtime
174                 movie.secondary = movie.genres.to_string ();
175                 _get_callback (movie);
176         }
177
178         public override void add_movie (Movie movie) {
179         }
180
181         public override unowned string get_name () {
182                 return "IMDb";
183         }
184
185         public override unowned string get_description () {
186                 return "Movies on IMDb";
187         }
188 }
189
190 [ModuleInit]
191 public Type register_plugin () {
192         // types are registered automatically
193         return typeof (IMDbPlugin);
194 }