Logo Search packages:      
Sourcecode: f-spot version File versions  Download package

SmugMugExport.cs

/*
 * SmugMugExport.cs
 *
 * Authors:
 *   Thomas Van Machelen <thomas.vanmachelen@gmail.com>
 *
 * Based on PicasaWebExport code from Stephane Delcroix.
 *
 * Copyright (C) 2006 Thomas Van Machelen
 */

using System;
using System.Net;
using System.IO;
using System.Text;
using System.Threading;
using System.Collections;
using System.Collections.Specialized;
using System.Web;
using Mono.Unix;
using Gtk;

using FSpot;
using FSpot.Filters;
using FSpot.Widgets;
using Hyena;
using FSpot.UI.Dialog;

using Gnome.Keyring;
using SmugMugNet;

namespace FSpotSmugMugExport {
      public class SmugMugAccount {
            private string username;
            private string password;
            private SmugMugApi smugmug_proxy;

            public SmugMugAccount (string username, string password)
            {
                  this.username = username;
                  this.password = password;
            }

            public SmugMugApi Connect ()
            {
                  Log.Debug ("SmugMug.Connect() " + username);
                  SmugMugApi proxy = new SmugMugApi (username, password);
                  ServicePointManager.CertificatePolicy = new NoCheckCertificatePolicy ();
                  proxy.Login ();

                  this.smugmug_proxy = proxy;
                  return proxy;
            }

            private void MarkChanged()
            {
                  smugmug_proxy = null;
            }

            public bool Connected {
                  get {
                        return (smugmug_proxy != null && smugmug_proxy.Connected);
                  }
            }

            public string Username {
                  get {
                        return username;
                  }
                  set {
                        if (username != value) {
                              username = value;
                              MarkChanged ();
                        }
                  }
            }

            public string Password {
                  get {
                        return password;
                  }
                  set {
                        if (password != value) {
                              password = value;
                              MarkChanged ();
                        }
                  }
            }

            public SmugMugApi SmugMug {
                  get {
                        return smugmug_proxy;
                  }
            }
      }


      public class SmugMugAccountManager
      {
            private static SmugMugAccountManager instance;
            private const string keyring_item_name = "SmugMug Account";
            ArrayList accounts;

            public delegate void AccountListChangedHandler (SmugMugAccountManager manager, SmugMugAccount changed_account);
            public event AccountListChangedHandler AccountListChanged;

            public static SmugMugAccountManager GetInstance ()
            {
                  if (instance == null) {
                        instance = new SmugMugAccountManager ();
                  }

                  return instance;
            }

            private SmugMugAccountManager ()
            {
                  accounts = new ArrayList ();
                  ReadAccounts ();
            }

            public void MarkChanged ()
            {
                  MarkChanged (true, null);
            }

            public void MarkChanged (bool write, SmugMugAccount changed_account)
            {
                  if (write)
                        WriteAccounts ();

                  if (AccountListChanged != null)
                        AccountListChanged (this, changed_account);
            }

            public ArrayList GetAccounts ()
            {
                  return accounts;
            }

            public void AddAccount (SmugMugAccount account)
            {
                  AddAccount (account, true);
            }

            public void AddAccount (SmugMugAccount account, bool write)
            {
                  accounts.Add (account);
                  MarkChanged (write, account);
            }

            public void RemoveAccount (SmugMugAccount account)
            {
                  string keyring = Ring.GetDefaultKeyring();
                  Hashtable request_attributes = new Hashtable();
                  request_attributes["name"] = keyring_item_name;
                  request_attributes["username"] = account.Username;
                  try {
                        foreach(ItemData result in Ring.Find(ItemType.GenericSecret, request_attributes)) {
                              Ring.DeleteItem(keyring, result.ItemID);
                        }
                  } catch (Exception e) {
                        Log.Exception (e);
                  }
                  accounts.Remove (account);
                  MarkChanged ();
            }

            public void WriteAccounts ()
            {
                  string keyring = Ring.GetDefaultKeyring();
                  foreach (SmugMugAccount account in accounts) {
                        Hashtable update_request_attributes = new Hashtable();
                        update_request_attributes["name"] = keyring_item_name;
                        update_request_attributes["username"] = account.Username;

                        Ring.CreateItem(keyring, ItemType.GenericSecret, keyring_item_name, update_request_attributes, account.Password, true);
                  }
            }

            private void ReadAccounts ()
            {

                  Hashtable request_attributes = new Hashtable();
                  request_attributes["name"] = keyring_item_name;
                  try {
                        foreach(ItemData result in Ring.Find(ItemType.GenericSecret, request_attributes)) {
                              if(!result.Attributes.ContainsKey("name") || !result.Attributes.ContainsKey("username") ||
                                    (result.Attributes["name"] as string) != keyring_item_name)
                                    continue;

                              string username = (string)result.Attributes["username"];
                              string password = result.Secret;

                              if (username == null || username == String.Empty || password == null || password == String.Empty)
                                    throw new ApplicationException ("Invalid username/password in keyring");

                              SmugMugAccount account = new SmugMugAccount(username, password);
                              if (account != null)
                                    AddAccount (account, false);

                        }
                  } catch (Exception e) {
                        Log.Exception (e);
                  }

                  MarkChanged ();
            }
      }

      public class SmugMugAccountDialog {
            public SmugMugAccountDialog (Gtk.Window parent) : this (parent, null) {
                  Dialog.Response += HandleAddResponse;
                  add_button.Sensitive = false;
            }

            public SmugMugAccountDialog (Gtk.Window parent, SmugMugAccount account)
            {
                  xml = new Glade.XML (null, "SmugMugExport.glade", dialog_name, "f-spot");
                  xml.Autoconnect (this);

                  Dialog.Modal = false;
                  Dialog.TransientFor = parent;
                  Dialog.DefaultResponse = Gtk.ResponseType.Ok;

                  this.account = account;

                  password_entry.ActivatesDefault = true;
                  username_entry.ActivatesDefault = true;

                  if (account != null) {
                        password_entry.Text = account.Password;
                        username_entry.Text = account.Username;
                        add_button.Label = Gtk.Stock.Ok;
                        Dialog.Response += HandleEditResponse;
                  }

                  if (remove_button != null)
                        remove_button.Visible = account != null;

                  this.Dialog.Show ();

                  password_entry.Changed += HandleChanged;
                  username_entry.Changed += HandleChanged;
                  HandleChanged (null, null);
            }

            private void HandleChanged (object sender, System.EventArgs args)
            {
                  password = password_entry.Text;
                  username = username_entry.Text;

                  add_button.Sensitive = !(password == String.Empty || username == String.Empty);
            }

            [GLib.ConnectBefore]
            protected void HandleAddResponse (object sender, Gtk.ResponseArgs args)
            {
                  if (args.ResponseId == Gtk.ResponseType.Ok) {
                        SmugMugAccount account = new SmugMugAccount (username, password);
                        SmugMugAccountManager.GetInstance ().AddAccount (account);
                  }
                  Dialog.Destroy ();
            }

            protected void HandleEditResponse (object sender, Gtk.ResponseArgs args)
            {
                  if (args.ResponseId == Gtk.ResponseType.Ok) {
                        account.Username = username;
                        account.Password = password;
                        SmugMugAccountManager.GetInstance ().MarkChanged (true, account);
                  } else if (args.ResponseId == Gtk.ResponseType.Reject) {
                        // NOTE we are using Reject to signal the remove action.
                        SmugMugAccountManager.GetInstance ().RemoveAccount (account);
                  }
                  Dialog.Destroy ();
            }

            private Gtk.Dialog Dialog {
                  get {
                        if (dialog == null)
                              dialog = (Gtk.Dialog) xml.GetWidget (dialog_name);

                        return dialog;
                  }
            }

            private SmugMugAccount account;
            private string password;
            private string username;
            private string dialog_name = "smugmug_add_dialog";
            private Glade.XML xml;

            // widgets
            [Glade.Widget] Gtk.Dialog dialog;
            [Glade.Widget] Gtk.Entry password_entry;
            [Glade.Widget] Gtk.Entry username_entry;

            [Glade.Widget] Gtk.Button add_button;
            [Glade.Widget] Gtk.Button remove_button;
      }

      public class SmugMugAddAlbum {
            //[Glade.Widget] Gtk.OptionMenu album_optionmenu;

            [Glade.Widget] Gtk.Dialog dialog;
            [Glade.Widget] Gtk.Entry title_entry;
            [Glade.Widget] Gtk.CheckButton public_check;
            [Glade.Widget] Gtk.ComboBox category_combo;

            [Glade.Widget] Gtk.Button add_button;

            private string dialog_name = "smugmug_add_album_dialog";
            private Glade.XML xml;
            private SmugMugExport export;
            private SmugMugApi smugmug;
            private string title;
            private ListStore category_store;

            public SmugMugAddAlbum (SmugMugExport export, SmugMugApi smugmug)
            {
                  xml = new Glade.XML (null, "SmugMugExport.glade", dialog_name, "f-spot");
                  xml.Autoconnect (this);

                  this.export = export;
                  this.smugmug = smugmug;

                  this.category_store = new ListStore (typeof(int), typeof(string));
                  CellRendererText display_cell = new CellRendererText();
                  category_combo.PackStart (display_cell, true);
                  category_combo.SetCellDataFunc (display_cell, new CellLayoutDataFunc (CategoryDataFunc));
                  this.category_combo.Model = category_store;
                  PopulateCategoryCombo ();

                  Dialog.Response += HandleAddResponse;

                  title_entry.Changed += HandleChanged;
                  HandleChanged (null, null);
            }

            private void HandleChanged (object sender, EventArgs args)
            {
                  title = title_entry.Text;

                  if (title == String.Empty)
                        add_button.Sensitive = false;
                  else
                        add_button.Sensitive = true;
            }

            [GLib.ConnectBefore]
            protected void HandleAddResponse (object sender, Gtk.ResponseArgs args)
            {
                  if (args.ResponseId == Gtk.ResponseType.Ok) {
                        smugmug.CreateAlbum (title, CurrentCategoryId, public_check.Active);
                        export.HandleAlbumAdded (title);
                  }
                  Dialog.Destroy ();
            }

            void CategoryDataFunc (CellLayout layout, CellRenderer renderer, TreeModel model, TreeIter iter)
            {
                  string name = (string)model.GetValue (iter, 1);
                  (renderer as CellRendererText).Text = name;
            }

            protected void PopulateCategoryCombo ()
            {
                  SmugMugNet.Category[] categories = smugmug.GetCategories ();

                  foreach (SmugMugNet.Category category in categories) {
                        category_store.AppendValues (category.CategoryID, category.Title);
                  }

                  category_combo.Active = 0;

                  category_combo.ShowAll ();
            }

            protected int CurrentCategoryId
            {
                  get {
                        TreeIter current;
                        category_combo.GetActiveIter (out current);
                        return (int)category_combo.Model.GetValue (current, 0);
                  }
            }

            private Gtk.Dialog Dialog {
                  get {
                        if (dialog == null)
                              dialog = (Gtk.Dialog) xml.GetWidget (dialog_name);

                        return dialog;
                  }
            }
      }


      public class SmugMugExport : FSpot.Extensions.IExporter {
            public SmugMugExport ()
            {
            }
            public void Run (IBrowsableCollection selection)
            {
                  xml = new Glade.XML (null, "SmugMugExport.glade", dialog_name, "f-spot");
                  xml.Autoconnect (this);

                  this.items = selection.Items;
                  album_button.Sensitive = false;
                  FSpot.Widgets.IconView view = new FSpot.Widgets.IconView (selection);
                  view.DisplayDates = false;
                  view.DisplayTags = false;

                  Dialog.Modal = false;
                  Dialog.TransientFor = null;
                  Dialog.Close += HandleCloseEvent;

                  thumb_scrolledwindow.Add (view);
                  view.Show ();
                  Dialog.Show ();

                  SmugMugAccountManager manager = SmugMugAccountManager.GetInstance ();
                  manager.AccountListChanged += PopulateSmugMugOptionMenu;
                  PopulateSmugMugOptionMenu (manager, null);

                  if (edit_button != null)
                        edit_button.Clicked += HandleEditGallery;

                  rh = new Gtk.ResponseHandler (HandleResponse);
                  Dialog.Response += HandleResponse;
                  connect = true;
                  HandleSizeActive (null, null);
                  Connect ();

                  scale_check.Toggled += HandleScaleCheckToggled;

                  LoadPreference (SCALE_KEY);
                  LoadPreference (SIZE_KEY);
                  LoadPreference (ROTATE_KEY);
                  LoadPreference (BROWSER_KEY);
            }

            Gtk.ResponseHandler rh;

            private bool scale;
            private int size;
            private bool browser;
            private bool rotate;
//          private bool meta;
            private bool connect = false;

            private long approx_size = 0;
            private long sent_bytes = 0;

            IBrowsableItem [] items;
            int photo_index;
            ThreadProgressDialog progress_dialog;

            ArrayList accounts;
            private SmugMugAccount account;
            private Album album;

            private string dialog_name = "smugmug_export_dialog";
            private Glade.XML xml;

            // Dialogs
            private SmugMugAccountDialog gallery_add;
            private SmugMugAddAlbum album_add;

            // Widgets
            [Glade.Widget] Gtk.Dialog dialog;
            [Glade.Widget] Gtk.OptionMenu gallery_optionmenu;
            [Glade.Widget] Gtk.OptionMenu album_optionmenu;

            [Glade.Widget] Gtk.CheckButton browser_check;
            [Glade.Widget] Gtk.CheckButton scale_check;
            [Glade.Widget] Gtk.CheckButton rotate_check;

            [Glade.Widget] Gtk.SpinButton size_spin;

            [Glade.Widget] Gtk.Button album_button;
            [Glade.Widget] Gtk.Button edit_button;

            [Glade.Widget] Gtk.Button export_button;

            [Glade.Widget] Gtk.ScrolledWindow thumb_scrolledwindow;

            System.Threading.Thread command_thread;

            public const string EXPORT_SERVICE = "smugmug/";
            public const string SCALE_KEY = Preferences.APP_FSPOT_EXPORT + EXPORT_SERVICE + "scale";
            public const string SIZE_KEY = Preferences.APP_FSPOT_EXPORT + EXPORT_SERVICE + "size";
            public const string ROTATE_KEY = Preferences.APP_FSPOT_EXPORT + EXPORT_SERVICE + "rotate";
            public const string BROWSER_KEY = Preferences.APP_FSPOT_EXPORT + EXPORT_SERVICE + "browser";

            private void HandleResponse (object sender, Gtk.ResponseArgs args)
            {
                  if (args.ResponseId != Gtk.ResponseType.Ok) {
                        Dialog.Destroy ();
                        return;
                  }

                  if (scale_check != null) {
                        scale = scale_check.Active;
                        size = size_spin.ValueAsInt;
                  } else
                        scale = false;

                  browser = browser_check.Active;
                  rotate = rotate_check.Active;
//                meta = meta_check.Active;

                  if (account != null) {
                        //System.Console.WriteLine ("history = {0}", album_optionmenu.History);
                        album = (Album) account.SmugMug.GetAlbums() [Math.Max (0, album_optionmenu.History)];
                        photo_index = 0;

                        Dialog.Destroy ();

                        command_thread = new System.Threading.Thread (new System.Threading.ThreadStart (this.Upload));
                        command_thread.Name = Mono.Unix.Catalog.GetString ("Uploading Pictures");

                        progress_dialog = new ThreadProgressDialog (command_thread, items.Length);
                        progress_dialog.Start ();

                        // Save these settings for next time
                        Preferences.Set (SCALE_KEY, scale);
                        Preferences.Set (SIZE_KEY, size);
                        Preferences.Set (ROTATE_KEY, rotate);
                        Preferences.Set (BROWSER_KEY, browser);
                  }
            }

            public void HandleSizeActive (object sender, EventArgs args)
            {
                  size_spin.Sensitive = scale_check.Active;
            }

            private void Upload ()
            {
                  sent_bytes = 0;
                  approx_size = 0;

                  System.Uri album_uri = null;

                  Log.Debug ("Starting Upload to Smugmug, album " + album.Title + " - " + album.AlbumID);

                  FilterSet filters = new FilterSet ();
                  filters.Add (new JpegFilter ());

                  if (scale)
                        filters.Add (new ResizeFilter ((uint)size));

                  if (rotate)
                        filters.Add (new OrientationFilter ());

                  while (photo_index < items.Length) {
                        try {
                              IBrowsableItem item = items[photo_index];

                              FileInfo file_info;
                              Log.Debug ("uploading " + photo_index);

                              progress_dialog.Message = String.Format (Catalog.GetString ("Uploading picture \"{0}\" ({1} of {2})"),
                                                             item.Name, photo_index+1, items.Length);
                              progress_dialog.ProgressText = string.Empty;
                              progress_dialog.Fraction = ((photo_index) / (double) items.Length);
                              photo_index++;

                              FilterRequest request = new FilterRequest (item.DefaultVersion.Uri);

                              filters.Convert (request);

                              file_info = new FileInfo (request.Current.LocalPath);

                              if (approx_size == 0) //first image
                                    approx_size = file_info.Length * items.Length;
                              else
                                    approx_size = sent_bytes * items.Length / (photo_index - 1);

                              int image_id = account.SmugMug.Upload (request.Current.LocalPath, album.AlbumID);
                              if (App.Instance.Database != null && item is Photo && image_id >= 0)
                                    App.Instance.Database.Exports.Create ((item as Photo).Id,
                                                            (item as Photo).DefaultVersionId,
                                                            ExportStore.SmugMugExportType,
                                                            account.SmugMug.GetAlbumUrl (image_id).ToString ());

                              sent_bytes += file_info.Length;

                              if (album_uri == null)
                                    album_uri = account.SmugMug.GetAlbumUrl (image_id);
                        } catch (System.Exception e) {
                              progress_dialog.Message = String.Format (Mono.Unix.Catalog.GetString ("Error Uploading To Gallery: {0}"),
                                                             e.Message);
                              progress_dialog.ProgressText = Mono.Unix.Catalog.GetString ("Error");
                              Log.DebugException (e);

                              if (progress_dialog.PerformRetrySkip ()) {
                                    photo_index--;
                                    if (photo_index == 0)
                                          approx_size = 0;
                              }
                        }
                  }

                  progress_dialog.Message = Catalog.GetString ("Done Sending Photos");
                  progress_dialog.Fraction = 1.0;
                  progress_dialog.ProgressText = Mono.Unix.Catalog.GetString ("Upload Complete");
                  progress_dialog.ButtonLabel = Gtk.Stock.Ok;

                  if (browser && album_uri != null) {
                        GtkBeans.Global.ShowUri (Dialog.Screen, album_uri.ToString ());
                  }
            }

            private void HandleScaleCheckToggled (object o, EventArgs e)
            {
                  rotate_check.Sensitive = !scale_check.Active;
            }

            private void PopulateSmugMugOptionMenu (SmugMugAccountManager manager, SmugMugAccount changed_account)
            {
                  Gtk.Menu menu = new Gtk.Menu ();
                  this.account = changed_account;
                  int pos = -1;

                  accounts = manager.GetAccounts ();
                  if (accounts == null || accounts.Count == 0) {
                        Gtk.MenuItem item = new Gtk.MenuItem (Mono.Unix.Catalog.GetString ("(No Gallery)"));
                        menu.Append (item);
                        gallery_optionmenu.Sensitive = false;
                        edit_button.Sensitive = false;
                  } else {
                        int i = 0;
                        foreach (SmugMugAccount account in accounts) {
                              if (account == changed_account)
                                    pos = i;

                              Gtk.MenuItem item = new Gtk.MenuItem (account.Username);
                              menu.Append (item);
                              i++;
                        }
                        gallery_optionmenu.Sensitive = true;
                        edit_button.Sensitive = true;
                  }

                  menu.ShowAll ();
                  gallery_optionmenu.Menu = menu;
                  gallery_optionmenu.SetHistory ((uint)pos);
            }

            private void Connect ()
            {
                  Connect (null);
            }

            private void Connect (SmugMugAccount selected)
            {
                  Connect (selected, null);
            }

            private void Connect (SmugMugAccount selected, string text)
            {
                  try {
                        if (accounts.Count != 0 && connect) {
                              if (selected == null)
                                    account = (SmugMugAccount) accounts [gallery_optionmenu.History];
                              else
                                    account = selected;

                              if (!account.Connected)
                                    account.Connect ();

                              PopulateAlbumOptionMenu (account.SmugMug);
                        }
                  } catch (System.Exception) {
                        Log.Warning ("Can not connect to SmugMug. Bad username? Password? Network connection?");
                        //System.Console.WriteLine ("{0}",ex);
                        if (selected != null)
                              account = selected;

                        PopulateAlbumOptionMenu (account.SmugMug);

                        album_button.Sensitive = false;

                        new SmugMugAccountDialog (this.Dialog, account);
                  }
            }

            private void HandleAccountSelected (object sender, System.EventArgs args)
            {
                  Connect ();
            }

            public void HandleAlbumAdded (string title) {
                  SmugMugAccount account = (SmugMugAccount) accounts [gallery_optionmenu.History];
                  PopulateAlbumOptionMenu (account.SmugMug);

                  // make the newly created album selected
                  Album[] albums = account.SmugMug.GetAlbums();
                  for (int i=0; i < albums.Length; i++) {
                        if (((Album)albums[i]).Title == title) {
                              album_optionmenu.SetHistory((uint)i);
                        }
                  }
            }

            private void PopulateAlbumOptionMenu (SmugMugApi smugmug)
            {
                  Album[] albums = null;
                  if (smugmug != null) {
                        try {
                              albums = smugmug.GetAlbums();
                        } catch (Exception) {
                              Log.Debug ("Can't get the albums");
                              smugmug = null;
                        }
                  }

                  Gtk.Menu menu = new Gtk.Menu ();

                  bool disconnected = smugmug == null || !account.Connected || albums == null;

                  if (disconnected || albums.Length == 0) {
                        string msg = disconnected ? Mono.Unix.Catalog.GetString ("(Not Connected)")
                              : Mono.Unix.Catalog.GetString ("(No Albums)");

                        Gtk.MenuItem item = new Gtk.MenuItem (msg);
                        menu.Append (item);

                        export_button.Sensitive = false;
                        album_optionmenu.Sensitive = false;
                        album_button.Sensitive = false;
                  } else {
                        foreach (Album album in albums) {
                              System.Text.StringBuilder label_builder = new System.Text.StringBuilder ();

                              label_builder.Append (album.Title);

                              Gtk.MenuItem item = new Gtk.MenuItem (label_builder.ToString ());
                              ((Gtk.Label)item.Child).UseUnderline = false;
                              menu.Append (item);
                        }

                        export_button.Sensitive = items.Length > 0;
                        album_optionmenu.Sensitive = true;
                        album_button.Sensitive = true;
                  }

                  menu.ShowAll ();
                  album_optionmenu.Menu = menu;
            }

            public void HandleAddGallery (object sender, System.EventArgs args)
            {
                  gallery_add = new SmugMugAccountDialog (this.Dialog);
            }

            public void HandleEditGallery (object sender, System.EventArgs args)
            {
                  gallery_add = new SmugMugAccountDialog (this.Dialog, account);
            }

            public void HandleAddAlbum (object sender, System.EventArgs args)
            {
                  if (account == null)
                        throw new Exception (Catalog.GetString ("No account selected"));

                  album_add = new SmugMugAddAlbum (this, account.SmugMug);
            }

            void LoadPreference (string key)
            {
                  switch (key) {
                  case SCALE_KEY:
                        if (scale_check.Active != Preferences.Get<bool> (key)) {
                              scale_check.Active = Preferences.Get<bool> (key);
                              rotate_check.Sensitive = !Preferences.Get<bool> (key);
                        }
                        break;

                  case SIZE_KEY:
                        size_spin.Value = (double) Preferences.Get<int> (key);
                        break;

                  case BROWSER_KEY:
                        if (browser_check.Active != Preferences.Get<bool> (key))
                              browser_check.Active = Preferences.Get<bool> (key);
                        break;

                  case ROTATE_KEY:
                        if (rotate_check.Active != Preferences.Get<bool> (key))
                              rotate_check.Active = Preferences.Get<bool> (key);
                        break;
                  }
            }

            protected void HandleCloseEvent (object sender, System.EventArgs args)
            {
                  account.SmugMug.Logout ();
            }

            private Gtk.Dialog Dialog {
                  get {
                        if (dialog == null)
                              dialog = (Gtk.Dialog) xml.GetWidget (dialog_name);

                        return dialog;
                  }
            }
      }
}

Generated by  Doxygen 1.6.0   Back to index