plugins/cd-recorder/rb-cd-recorder-plugin.c

   1 /* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*-
   2  *
   3  *  Copyright (C) 2006 William Jon McCann <mccann@jhu.edu>
   4  *
   5  * This program 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 2, or (at your option)
   8  * any later version.
   9  *
  10  * This program 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 this program; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301  USA.
  18  */
  19
  20 #define __EXTENSIONS__
  21
  22 #include "config.h"
  23
  24 #include <string.h> /* For strlen */
  25 #include <glib/gi18n-lib.h>
  26 #include <gmodule.h>
  27 #include <gtk/gtk.h>
  28 #include <glib.h>
  29 #include <glib-object.h>
  30
  31 #include "rb-plugin.h"
  32 #include "rb-debug.h"
  33 #include "rb-shell.h"
  34 #include "rb-source.h"
  35 #include "rb-playlist-source.h"
  36 #include "rb-dialog.h"
  37 #include "rb-file-helpers.h"
  38
  39 #include "rb-recorder.h"
  40 #include "rb-playlist-source-recorder.h"
  41
  42 #include <nautilus-burn-drive.h>
  43 #ifndef NAUTILUS_BURN_CHECK_VERSION
  44 #define NAUTILUS_BURN_CHECK_VERSION(a,b,c) FALSE
  45 #endif
  46
  47 #if NAUTILUS_BURN_CHECK_VERSION(2,15,3)
  48 #include <nautilus-burn.h>
  49 #endif
  50
  51 #define RB_TYPE_CD_RECORDER_PLUGIN              (rb_cd_recorder_plugin_get_type ())
  52 #define RB_CD_RECORDER_PLUGIN(o)                (G_TYPE_CHECK_INSTANCE_CAST ((o), RB_TYPE_CD_RECORDER_PLUGIN, RBCdRecorderPlugin))
  53 #define RB_CD_RECORDER_PLUGIN_CLASS(k)          (G_TYPE_CHECK_CLASS_CAST((k), RB_TYPE_CD_RECORDER_PLUGIN, RBCdRecorderPluginClass))
  54 #define RB_IS_CD_RECORDER_PLUGIN(o)             (G_TYPE_CHECK_INSTANCE_TYPE ((o), RB_TYPE_CD_RECORDER_PLUGIN))
  55 #define RB_IS_CD_RECORDER_PLUGIN_CLASS(k)       (G_TYPE_CHECK_CLASS_TYPE ((k), RB_TYPE_CD_RECORDER_PLUGIN))
  56 #define RB_CD_RECORDER_PLUGIN_GET_CLASS(o)      (G_TYPE_INSTANCE_GET_CLASS ((o), RB_TYPE_CD_RECORDER_PLUGIN, RBCdRecorderPluginClass))
  57
  58 typedef struct
  59 {
  60         RBPlugin        parent;
  61
  62         RBShell        *shell;
  63         GtkActionGroup *action_group;
  64         guint           ui_merge_id;
  65
  66         RBSource       *selected_source;
  67         guint           enabled : 1;
  68 } RBCdRecorderPlugin;
  69
  70 typedef struct
  71 {
  72         RBPluginClass parent_class;
  73 } RBCdRecorderPluginClass;
  74
  75 G_MODULE_EXPORT GType register_rb_plugin (GTypeModule *module);
  76 GType   rb_cd_recorder_plugin_get_type          (void) G_GNUC_CONST;
  77
  78 static void rb_cd_recorder_plugin_init (RBCdRecorderPlugin *plugin);
  79 static void rb_cd_recorder_plugin_finalize (GObject *object);
  80 static void impl_activate (RBPlugin *plugin, RBShell *shell);
  81 static void impl_deactivate (RBPlugin *plugin, RBShell *shell);
  82 static void cmd_burn_source (GtkAction          *action,
  83                              RBCdRecorderPlugin *pi);
  84
  85 static GtkActionEntry rb_cd_recorder_plugin_actions [] = {
  86         { "MusicPlaylistBurnPlaylist", GTK_STOCK_CDROM, N_("_Create Audio CD..."), NULL,
  87           N_("Create an audio CD from playlist"),
  88           G_CALLBACK (cmd_burn_source) }
  89 };
  90
  91 RB_PLUGIN_REGISTER(RBCdRecorderPlugin, rb_cd_recorder_plugin)
  92
  93 static void
  94 rb_cd_recorder_plugin_class_init (RBCdRecorderPluginClass *klass)
  95 {
  96         GObjectClass *object_class = G_OBJECT_CLASS (klass);
  97         RBPluginClass *plugin_class = RB_PLUGIN_CLASS (klass);
  98
  99         object_class->finalize = rb_cd_recorder_plugin_finalize;
 100
 101         plugin_class->activate = impl_activate;
 102         plugin_class->deactivate = impl_deactivate;
 103 }
 104
 105 static void
 106 rb_cd_recorder_plugin_init (RBCdRecorderPlugin *plugin)
 107 {
 108         rb_debug ("RBCdRecorderPlugin initializing");
 109
 110 #if NAUTILUS_BURN_CHECK_VERSION(2,15,3)
 111         nautilus_burn_init ();
 112 #endif
 113 }
 114
 115 static void
 116 rb_cd_recorder_plugin_finalize (GObject *object)
 117 {
 118 /*
 119         RBCdRecorderPlugin *plugin = RB_CD_RECORDER_PLUGIN (object);
 120 */
 121         rb_debug ("RBCdRecorderPlugin finalizing");
 122
 123 #if NAUTILUS_BURN_CHECK_VERSION(2,15,3)
 124         nautilus_burn_shutdown ();
 125 #endif
 126
 127         G_OBJECT_CLASS (rb_cd_recorder_plugin_parent_class)->finalize (object);
 128 }
 129
 130 static gboolean
 131 burn_source_iter_func (GtkTreeModel *model,
 132                        GtkTreeIter  *iter,
 133                        char        **uri,
 134                        char        **artist,
 135                        char        **title,
 136                        gulong       *duration)
 137 {
 138         RhythmDBEntry *entry;
 139
 140         gtk_tree_model_get (model, iter, 0, &entry, -1);
 141
 142         *uri = rhythmdb_entry_dup_string (entry, RHYTHMDB_PROP_LOCATION);
 143         *title = rhythmdb_entry_dup_string (entry, RHYTHMDB_PROP_TITLE);
 144         *artist = rhythmdb_entry_dup_string (entry, RHYTHMDB_PROP_ARTIST);
 145         *duration = rhythmdb_entry_get_ulong (entry, RHYTHMDB_PROP_DURATION);
 146
 147         return TRUE;
 148 }
 149
 150 static void
 151 source_burn (RBSource *source)
 152 {
 153         GtkWidget    *recorder;
 154         GtkWidget    *parent;
 155         char         *name;
 156         RBShell      *shell;
 157         gboolean      res;
 158         GError       *error;
 159         GtkTreeModel *model;
 160
 161         g_object_get (G_OBJECT (source), "query-model", &model, NULL);
 162
 163         /* don't burn if the source is empty */
 164         if (gtk_tree_model_iter_n_children (model, NULL) == 0) {
 165                 g_object_unref (model);
 166                 return;
 167         }
 168
 169         rb_debug ("burning source");
 170
 171         g_object_get (source, "name", &name, "shell", &shell, NULL);
 172
 173         parent = gtk_widget_get_toplevel (GTK_WIDGET (source));
 174         recorder = rb_playlist_source_recorder_new (parent,
 175                                                     shell,
 176                                                     name);
 177         g_object_unref (shell);
 178         g_free (name);
 179
 180         error = NULL;
 181         res = rb_playlist_source_recorder_add_from_model (RB_PLAYLIST_SOURCE_RECORDER (recorder),
 182                                                           model,
 183                                                           burn_source_iter_func,
 184                                                           &error);
 185         g_object_unref (model);
 186
 187         if (! res) {
 188                 rb_error_dialog (GTK_WINDOW (parent),
 189                                  _("Unable to create audio CD"),
 190                                  "%s", error->message);
 191                 g_error_free (error);
 192
 193                 gtk_widget_destroy (recorder);
 194
 195                 return;
 196         }
 197
 198         g_signal_connect (recorder,
 199                           "response",
 200                           G_CALLBACK (gtk_widget_destroy),
 201                           NULL);
 202
 203         gtk_widget_show (recorder);
 204 }
 205
 206 static void
 207 cmd_burn_source (GtkAction          *action,
 208                  RBCdRecorderPlugin *pi)
 209 {
 210         if (pi->selected_source != NULL) {
 211                 source_burn (pi->selected_source);
 212         }
 213 }
 214
 215 static void
 216 playlist_entries_changed (GtkTreeModel       *model,
 217                           RhythmDBEntry      *entry,
 218                           RBCdRecorderPlugin *pi)
 219 {
 220         int        num_tracks;
 221         GtkAction *action;
 222
 223         num_tracks = gtk_tree_model_iter_n_children (model, NULL);
 224
 225         action = gtk_action_group_get_action (pi->action_group, "MusicPlaylistBurnPlaylist");
 226         gtk_action_set_sensitive (action, (num_tracks > 0));
 227 }
 228
 229 static void
 230 playlist_row_inserted_cb (GtkTreeModel       *model,
 231                           GtkTreePath        *path,
 232                           GtkTreeIter        *iter,
 233                           RBCdRecorderPlugin *pi)
 234 {
 235         RhythmDBEntry *entry = rhythmdb_query_model_iter_to_entry (RHYTHMDB_QUERY_MODEL (model), iter);
 236
 237         playlist_entries_changed (model, entry, pi);
 238
 239         rhythmdb_entry_unref (entry);
 240 }
 241
 242 static void
 243 update_source (RBCdRecorderPlugin *pi,
 244                RBShell            *shell)
 245 {
 246         GtkAction *action;
 247         gboolean   playlist_active;
 248         RBSource  *selected_source;
 249
 250         if (pi->selected_source != NULL) {
 251                 RhythmDBQueryModel *model;
 252
 253                 g_object_get (G_OBJECT (pi->selected_source), "query-model", &model, NULL);
 254                 g_signal_handlers_disconnect_by_func (model, playlist_row_inserted_cb, pi);
 255                 g_signal_handlers_disconnect_by_func (model, playlist_entries_changed, pi);
 256                 g_object_unref (model);
 257         }
 258
 259         g_object_get (G_OBJECT (shell), "selected-source", &selected_source, NULL);
 260
 261         /* for now restrict to playlist sources */
 262         playlist_active = RB_IS_PLAYLIST_SOURCE (selected_source);
 263
 264         action = gtk_action_group_get_action (pi->action_group,
 265                                               "MusicPlaylistBurnPlaylist");
 266
 267         if (pi->enabled && playlist_active && rb_recorder_enabled ()) {
 268                 RhythmDBQueryModel *model;
 269
 270                 g_object_get (G_OBJECT (selected_source), "query-model", &model, NULL);
 271                 /* monitor for changes, to enable/disable the burn menu item */
 272                 g_signal_connect_object (G_OBJECT (model),
 273                                          "row_inserted",
 274                                          G_CALLBACK (playlist_row_inserted_cb),
 275                                          pi, 0);
 276                 g_signal_connect_object (G_OBJECT (model),
 277                                          "post-entry-delete",
 278                                          G_CALLBACK (playlist_entries_changed),
 279                                          pi, 0);
 280
 281                 playlist_entries_changed (GTK_TREE_MODEL (model), NULL, pi);
 282                 g_object_unref (model);
 283                 gtk_action_set_visible (action, TRUE);
 284         } else {
 285                 gtk_action_set_visible (action, FALSE);
 286         }
 287
 288
 289         if (pi->selected_source != NULL) {
 290                 g_object_unref (pi->selected_source);
 291         }
 292         pi->selected_source = selected_source;
 293 }
 294
 295 static void
 296 shell_selected_source_notify_cb (RBShell            *shell,
 297                                  GParamSpec         *param,
 298                                  RBCdRecorderPlugin *pi)
 299 {
 300         rb_debug ("RBCdRecorderPlugin selected source changed");
 301
 302         update_source (pi, shell);
 303 }
 304
 305 static const char *ui_paths [] = {
 306         "/MenuBar/MusicMenu/PlaylistMenu/MusicPlaylistPlaylistMenuPluginPlaceholder",
 307         "/ToolBar/ToolBarPluginPlaceholder",
 308         "/PlaylistSourcePopup/PlaylistSourcePopupPluginPlaceholder",
 309         "/AutoPlaylistSourcePopup/AutoPlaylistSourcePopupPluginPlaceholder",
 310         "/QueueSourcePopup/QueueSourcePopupPluginPlaceholder",
 311 };
 312
 313 static void
 314 impl_activate (RBPlugin *plugin,
 315                RBShell  *shell)
 316 {
 317         RBCdRecorderPlugin *pi = RB_CD_RECORDER_PLUGIN (plugin);
 318         GtkUIManager       *uimanager = NULL;
 319         int                 i;
 320
 321         pi->enabled = TRUE;
 322
 323         rb_debug ("RBCdRecorderPlugin activating");
 324
 325         pi->shell = shell;
 326
 327         g_object_get (G_OBJECT (shell),
 328                       "ui-manager", &uimanager,
 329                       NULL);
 330
 331         g_signal_connect_object (G_OBJECT (shell),
 332                                  "notify::selected-source",
 333                                  G_CALLBACK (shell_selected_source_notify_cb),
 334                                  pi, 0);
 335
 336         /* add UI */
 337         pi->action_group = gtk_action_group_new ("CdRecorderActions");
 338         gtk_action_group_set_translation_domain (pi->action_group,
 339                                                  GETTEXT_PACKAGE);
 340         gtk_action_group_add_actions (pi->action_group,
 341                                       rb_cd_recorder_plugin_actions, G_N_ELEMENTS (rb_cd_recorder_plugin_actions),
 342                                       pi);
 343         gtk_ui_manager_insert_action_group (uimanager, pi->action_group, 0);
 344
 345
 346         pi->ui_merge_id = gtk_ui_manager_new_merge_id (uimanager);
 347         for (i = 0; i < G_N_ELEMENTS (ui_paths); i++) {
 348                 gtk_ui_manager_add_ui (uimanager,
 349                                        pi->ui_merge_id,
 350                                        ui_paths [i],
 351                                        "MusicPlaylistBurnPlaylistMenu",
 352                                        "MusicPlaylistBurnPlaylist",
 353                                        GTK_UI_MANAGER_AUTO,
 354                                        FALSE);
 355         }
 356
 357         update_source (pi, shell);
 358 }
 359
 360 static void
 361 impl_deactivate (RBPlugin *plugin,
 362                  RBShell  *shell)
 363 {
 364         RBCdRecorderPlugin *pi = RB_CD_RECORDER_PLUGIN (plugin);
 365         GtkUIManager       *uimanager = NULL;
 366
 367         pi->enabled = FALSE;
 368
 369         rb_debug ("RBCdRecorderPlugin deactivating");
 370
 371         update_source (pi, shell);
 372
 373         if (pi->selected_source) {
 374                 pi->selected_source = NULL;
 375         }
 376
 377         g_signal_handlers_disconnect_by_func (shell, shell_selected_source_notify_cb, pi);
 378
 379         g_object_get (G_OBJECT (shell),
 380                       "ui-manager", &uimanager,
 381                       NULL);
 382
 383         gtk_ui_manager_remove_ui (uimanager, pi->ui_merge_id);
 384         gtk_ui_manager_remove_action_group (uimanager, pi->action_group);
 385
 386         g_object_unref (G_OBJECT (uimanager));
 387 }
 388