New check to enable/disable selective sync
[pithos-ms-client] / trunk / Pithos.Client.WPF / Shell / ShellViewModel.cs
1 #region
2 /* -----------------------------------------------------------------------
3  * <copyright file="ShellViewModel.cs" company="GRNet">
4  * 
5  * Copyright 2011-2012 GRNET S.A. All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or
8  * without modification, are permitted provided that the following
9  * conditions are met:
10  *
11  *   1. Redistributions of source code must retain the above
12  *      copyright notice, this list of conditions and the following
13  *      disclaimer.
14  *
15  *   2. Redistributions in binary form must reproduce the above
16  *      copyright notice, this list of conditions and the following
17  *      disclaimer in the documentation and/or other materials
18  *      provided with the distribution.
19  *
20  *
21  * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
22  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
23  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
24  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
25  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
26  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
27  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
28  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
29  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
31  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
32  * POSSIBILITY OF SUCH DAMAGE.
33  *
34  * The views and conclusions contained in the software and
35  * documentation are those of the authors and should not be
36  * interpreted as representing official policies, either expressed
37  * or implied, of GRNET S.A.
38  * </copyright>
39  * -----------------------------------------------------------------------
40  */
41 #endregion
42 using System.Collections.Concurrent;
43 using System.Diagnostics;
44 using System.Diagnostics.Contracts;
45 using System.IO;
46 using System.Net;
47 using System.Reflection;
48 using System.Runtime.InteropServices;
49 using System.ServiceModel;
50 using System.Threading.Tasks;
51 using System.Windows;
52 using System.Windows.Controls.Primitives;
53 using AppLimit.NetSparkle;
54 using Caliburn.Micro;
55 using Hardcodet.Wpf.TaskbarNotification;
56 using Pithos.Client.WPF.Configuration;
57 using Pithos.Client.WPF.FileProperties;
58 using Pithos.Client.WPF.Preferences;
59 using Pithos.Client.WPF.SelectiveSynch;
60 using Pithos.Client.WPF.Services;
61 using Pithos.Client.WPF.Shell;
62 using Pithos.Core;
63 using Pithos.Core.Agents;
64 using Pithos.Interfaces;
65 using System;
66 using System.Collections.Generic;
67 using System.Linq;
68 using Pithos.Network;
69 using StatusService = Pithos.Client.WPF.Services.StatusService;
70
71 namespace Pithos.Client.WPF {
72         using System.ComponentModel.Composition;
73
74         
75         ///<summary>
76         /// The "shell" of the Pithos application displays the taskbar  icon, menu and notifications.
77         /// The shell also hosts the status service called by shell extensions to retrieve file info
78         ///</summary>
79         ///<remarks>
80         /// It is a strange "shell" as its main visible element is an icon instead of a window
81         /// The shell subscribes to the following events:
82         /// * Notification:  Raised by components that want to notify the user. Usually displayed in a balloon
83         /// * SelectiveSynchChanges: Notifies that the user made changes to the selective synch folders for an account. Raised by the Selective Synch dialog. Located here because the monitors are here
84         /// * ShowFilePropertiesEvent: Raised when a shell command requests the display of the file/container properties dialog
85         ///</remarks>           
86         //TODO: CODE SMELL Why does the shell handle the SelectiveSynchChanges?
87     [Export(typeof(IShell)), Export(typeof(ShellViewModel))]
88         public class ShellViewModel : Screen, IStatusNotification, IShell,
89                 IHandle<Notification>, IHandle<SelectiveSynchChanges>, IHandle<ShowFilePropertiesEvent>
90         {
91
92                 //The Status Checker provides the current synch state
93                 //TODO: Could we remove the status checker and use events in its place?
94                 private readonly IStatusChecker _statusChecker;
95                 private readonly IEventAggregator _events;
96
97                 public PithosSettings Settings { get; private set; }
98
99
100                 private readonly ConcurrentDictionary<Uri, PithosMonitor> _monitors = new ConcurrentDictionary<Uri, PithosMonitor>();
101                 ///<summary>
102                 /// Dictionary of account monitors, keyed by account
103                 ///</summary>
104                 ///<remarks>
105                 /// One monitor class is created for each account. The Shell needs access to the monitors to execute start/stop/pause commands,
106                 /// retrieve account and boject info            
107                 ///</remarks>
108                 // TODO: Does the Shell REALLY need access to the monitors? Could we achieve the same results with a better design?
109                 // TODO: The monitors should be internal to Pithos.Core, even though exposing them makes coding of the Object and Container windows easier
110                 public ConcurrentDictionary<Uri, PithosMonitor> Monitors
111                 {
112                         get { return _monitors; }
113                 }
114
115
116                 ///<summary>
117                 /// The status service is used by Shell extensions to retrieve file status information
118                 ///</summary>
119                 //TODO: CODE SMELL! This is the shell! While hosting in the shell makes executing start/stop commands easier, it is still a smell
120                 private ServiceHost _statusService;
121
122                 //Logging in the Pithos client is provided by log4net
123         private static readonly log4net.ILog Log = log4net.LogManager.GetLogger(MethodBase.GetCurrentMethod().DeclaringType);
124
125             private readonly PollAgent _pollAgent;
126             private readonly NetworkAgent _networkAgent;
127
128
129             private MiniStatusViewModel _miniStatus;
130
131             [Import]
132         public MiniStatusViewModel MiniStatus
133             {
134                 get { return _miniStatus; }
135                 set
136                 {
137                     _miniStatus = value;
138                     _miniStatus.Shell = this;
139                     _miniStatus.Deactivated += (sender, arg) =>
140                                                    {
141                                                        _statusVisible = false;
142                                                    NotifyOfPropertyChange(()=>MiniStatusCaption);
143                                                    };
144                 }
145             }
146
147             ///<summary>
148                 /// The Shell depends on MEF to provide implementations for windowManager, events, the status checker service and the settings
149                 ///</summary>
150                 ///<remarks>
151                 /// The PithosSettings class encapsulates the app's settings to abstract their storage mechanism (App settings, a database or registry)
152                 ///</remarks>
153                 [ImportingConstructor]          
154                 public ShellViewModel(IWindowManager windowManager, IEventAggregator events, IStatusChecker statusChecker, PithosSettings settings,PollAgent pollAgent,NetworkAgent networkAgent)
155                 {
156                         try
157                         {
158
159                                 _windowManager = windowManager;
160                                 //CHECK: Caliburn doesn't need explicit command construction
161                                 //OpenPithosFolderCommand = new PithosCommand(OpenPithosFolder);
162                                 _statusChecker = statusChecker;
163                                 //The event subst
164                                 _events = events;
165                                 _events.Subscribe(this);
166
167                             _pollAgent = pollAgent;
168                             _networkAgent = networkAgent;
169                                 Settings = settings;
170
171                                 Proxy.SetFromSettings(settings);
172
173                 StatusMessage = Settings.Accounts.Count==0 
174                     ? "No Accounts added\r\nPlease add an account" 
175                     : "Starting";
176
177                                 _accounts.CollectionChanged += (sender, e) =>
178                                                                                                    {
179                                                                                                            NotifyOfPropertyChange(() => OpenFolderCaption);
180                                                                                                            NotifyOfPropertyChange(() => HasAccounts);
181                                                                                                    };
182
183                 SetVersionMessage();
184                         }
185                         catch (Exception exc)
186                         {
187                                 Log.Error("Error while starting the ShellViewModel",exc);
188                                 throw;
189                         }
190
191                 }
192
193             private void SetVersionMessage()
194             {
195                 Assembly assembly = Assembly.GetExecutingAssembly();
196                 var fileVersion = FileVersionInfo.GetVersionInfo(assembly.Location);
197                 VersionMessage = String.Format("Pithos+ {0}", fileVersion.FileVersion);
198             }
199
200         public void CurrentSyncStatus()
201         {
202             if (Accounts.Count == 0)
203             {
204                 ShowPreferences("AccountTab");
205             }
206             else
207             {
208                 if (!_statusVisible)
209                     _windowManager.ShowWindow(MiniStatus);
210                 else
211                 {
212                     if (MiniStatus.IsActive)
213                         MiniStatus.TryClose();
214                 }
215                 _statusVisible = !_statusVisible;
216
217                 NotifyOfPropertyChange(() => MiniStatusCaption);
218             }
219         }
220
221             protected override void OnActivate()
222                 {
223                         base.OnActivate();
224
225             InitializeSparkle();
226
227                 //Must delay opening the upgrade window
228             //to avoid Windows Messages sent by the TaskbarIcon
229             TaskEx.Delay(5000).ContinueWith(_=>
230                 Execute.OnUIThread(()=> _sparkle.StartLoop(true,Settings.UpdateForceCheck,Settings.UpdateCheckInterval)));
231
232
233                         StartMonitoring();                    
234                 }
235
236
237             private void OnCheckFinished(object sender, bool updaterequired)
238             {
239             
240             Log.InfoFormat("Upgrade check finished. Need Upgrade: {0}", updaterequired);
241             if (_manualUpgradeCheck)
242             {
243                 _manualUpgradeCheck = false;
244                 if (!updaterequired)
245                     //Sparkle raises events on a background thread
246                     Execute.OnUIThread(()=>
247                         ShowBalloonFor(new Notification{Title="Pithos+ is up to date",Message="You have the latest Pithos+ version. No update is required"}));
248             }
249             }
250
251             private void OnUpgradeDetected(object sender, UpdateDetectedEventArgs e)
252             {            
253                 Log.InfoFormat("Update detected {0}",e.LatestVersion);
254             }
255
256         public void CheckForUpgrade()
257         {
258             ShowBalloonFor(new Notification{Title="Checking for upgrades",Message="Contacting the server to retrieve the latest Pithos+ version."});
259             _sparkle.StopLoop();
260             _sparkle.updateDetected -= OnUpgradeDetected;
261             _sparkle.checkLoopFinished -= OnCheckFinished;
262             _sparkle.Dispose();
263
264             _manualUpgradeCheck = true;
265             InitializeSparkle();
266             _sparkle.StartLoop(true,true,Settings.UpdateCheckInterval);
267         }
268
269         private void InitializeSparkle()
270         {
271             _sparkle = new Sparkle(Settings.UpdateUrl);
272             _sparkle.updateDetected += OnUpgradeDetected;
273             _sparkle.checkLoopFinished += OnCheckFinished;
274             _sparkle.ShowDiagnosticWindow = Settings.UpdateDiagnostics;
275         }
276
277             private async void StartMonitoring()
278                 {
279                         try
280                         {
281                 if (Settings.IgnoreCertificateErrors)
282                 {
283                     ServicePointManager.ServerCertificateValidationCallback = (sender, certificate, chain, errors) => true;
284                 }
285                     
286                                 var accounts = Settings.Accounts.Select(MonitorAccount);
287                                 await TaskEx.WhenAll(accounts);
288                                 _statusService = StatusService.Start();
289
290                         }
291                         catch (AggregateException exc)
292                         {
293                                 exc.Handle(e =>
294                                 {
295                                         Log.Error("Error while starting monitoring", e);
296                                         return true;
297                                 });
298                                 throw;
299                         }
300                 }
301
302                 protected override void OnDeactivate(bool close)
303                 {
304                         base.OnDeactivate(close);
305                         if (close)
306                         {
307                                 StatusService.Stop(_statusService);
308                                 _statusService = null;
309                         }
310                 }
311
312                 public Task MonitorAccount(AccountSettings account)
313                 {
314                         return Task.Factory.StartNew(() =>
315                         {                                                
316                                 PithosMonitor monitor;
317                                 var accountName = account.AccountName;
318
319                                 if (Monitors.TryGetValue(account.AccountKey, out monitor))
320                                 {
321                                         //If the account is active
322                     if (account.IsActive)
323                     {
324                         //The Api Key may have changed throuth the Preferences dialog
325                         monitor.ApiKey = account.ApiKey;
326                                                 Debug.Assert(monitor.StatusNotification == this,"An existing monitor should already have a StatusNotification service object");
327                         monitor.RootPath = account.RootPath;
328                         //Start the monitor. It's OK to start an already started monitor,
329                         //it will just ignore the call                        
330                         StartMonitor(monitor).Wait();
331                     }
332                     else
333                     {
334                         //If the account is inactive
335                         //Stop and remove the monitor
336                         RemoveMonitor(account.ServerUrl,accountName);
337                     }
338                                         return;
339                                 }
340
341                                 
342                                 //Create a new monitor/ Can't use MEF here, it would return a single instance for all monitors
343                                 monitor = new PithosMonitor
344                                                           {
345                                                                   UserName = accountName,
346                                                                   ApiKey = account.ApiKey,                                  
347                                                                   StatusNotification = this,
348                                                                   RootPath = account.RootPath
349                                                           };
350                                 //PithosMonitor uses MEF so we need to resolve it
351                                 IoC.BuildUp(monitor);
352
353                                 monitor.AuthenticationUrl = account.ServerUrl;
354
355                                 Monitors[account.AccountKey] = monitor;
356
357                                 if (account.IsActive)
358                                 {
359                                         //Don't start a monitor if it doesn't have an account and ApiKey
360                                         if (String.IsNullOrWhiteSpace(monitor.UserName) ||
361                                                 String.IsNullOrWhiteSpace(monitor.ApiKey))
362                                                 return;
363                                         StartMonitor(monitor);
364                                 }
365                         });
366                 }
367
368
369                 protected override void OnViewLoaded(object view)
370                 {
371                         UpdateStatus();
372                         var window = (Window)view;            
373                         TaskEx.Delay(1000).ContinueWith(t => Execute.OnUIThread(window.Hide));
374                         base.OnViewLoaded(view);
375                 }
376
377
378                 #region Status Properties
379
380                 private string _statusMessage;
381                 public string StatusMessage
382                 {
383                         get { return _statusMessage; }
384                         set
385                         {
386                                 _statusMessage = value;
387                                 NotifyOfPropertyChange(() => StatusMessage);
388                 NotifyOfPropertyChange(() => TooltipMessage);
389                         }
390                 }
391
392         public string VersionMessage { get; set; }
393
394             public string TooltipMessage
395             {
396                 get
397                 {
398                     return String.Format("{0}\r\n{1}",VersionMessage,StatusMessage);
399                 }
400             }
401
402         public string TooltipMiniStatus
403         {
404             get
405             {
406                 return String.Format("{0}\r\n{1}", "Status Window", "Enable / Dissable the stuatus window");
407             }
408         }
409
410         public string ToggleStatusWindowMessage
411         {
412             get
413             {
414                 return String.Format("{0}" + Environment.NewLine + "{1} Toggle Mini Status");
415             }
416         }
417
418             private readonly ObservableConcurrentCollection<AccountInfo> _accounts = new ObservableConcurrentCollection<AccountInfo>();
419                 public ObservableConcurrentCollection<AccountInfo> Accounts
420                 {
421                         get { return _accounts; }
422                 }
423
424                 public bool HasAccounts
425                 {
426                         get { return _accounts.Count > 0; }
427                 }
428
429
430                 public string OpenFolderCaption
431                 {
432                         get
433                         {
434                                 return (_accounts.Count == 0)
435                                                 ? "No Accounts Defined"
436                                                 : "Open Pithos Folder";
437                         }
438                 }
439
440                 private string _pauseSyncCaption="Pause Synching";
441                 public string PauseSyncCaption
442                 {
443                         get { return _pauseSyncCaption; }
444                         set
445                         {
446                                 _pauseSyncCaption = value;
447                                 NotifyOfPropertyChange(() => PauseSyncCaption);
448                         }
449                 }
450
451                 private readonly ObservableConcurrentCollection<FileEntry> _recentFiles = new ObservableConcurrentCollection<FileEntry>();
452                 public ObservableConcurrentCollection<FileEntry> RecentFiles
453                 {
454                         get { return _recentFiles; }
455                 }
456
457
458                 private string _statusIcon="../Images/Pithos.ico";
459                 public string StatusIcon
460                 {
461                         get { return _statusIcon; }
462                         set
463                         {
464                                 //TODO: Ensure all status icons use the Pithos logo
465                                 _statusIcon = value;
466                                 NotifyOfPropertyChange(() => StatusIcon);
467                         }
468                 }
469
470                 #endregion
471
472                 #region Commands
473
474         public void CancelCurrentOperation()
475         {
476             _networkAgent.CancelCurrentOperation();
477         }
478
479         public void ShowPreferences()
480         {
481             ShowPreferences(null);
482         }
483
484                 public void ShowPreferences(string currentTab)
485                 {
486                         //Settings.Reload();
487                     var preferences = new PreferencesViewModel(_windowManager, _events, this, Settings,currentTab);
488                     _windowManager.ShowDialog(preferences);
489                         
490                 }
491
492                 public void AboutPithos()
493                 {
494                         var about = IoC.Get<AboutViewModel>();
495                     about.LatestVersion=_sparkle.LatestVersion;
496                         _windowManager.ShowWindow(about);
497                 }
498
499                 public void SendFeedback()
500                 {
501                         var feedBack =  IoC.Get<FeedbackViewModel>();
502                         _windowManager.ShowWindow(feedBack);
503                 }
504
505                 //public PithosCommand OpenPithosFolderCommand { get; private set; }
506
507                 public void OpenPithosFolder()
508                 {
509                         var account = Settings.Accounts.FirstOrDefault(acc => acc.IsActive);
510                         if (account == null)
511                                 return;
512                         Process.Start(account.RootPath);
513                 }
514
515                 public void OpenPithosFolder(AccountInfo account)
516                 {
517                         Process.Start(account.AccountPath);
518                 }
519
520                 
521
522                 public void GoToSite()
523                 {            
524                         var site = Properties.Settings.Default.ProductionServer;
525                         Process.Start(site);            
526                 }
527
528
529                 public void GoToSite(AccountInfo account)
530                 {
531                     var uri = account.SiteUri.Replace("http://","https://");            
532                     Process.Start(uri);
533                 }
534
535             private bool _statusVisible;
536
537             public string MiniStatusCaption
538             {
539                 get
540                 {
541                     return  _statusVisible ? "Hide Status Window" : "Show Status Window";
542                 }
543             }
544
545             public bool HasConflicts
546             {
547             get { return true; }
548             }
549         public void ShowConflicts()
550         {
551             _windowManager.ShowWindow(IoC.Get<ConflictsViewModel>());            
552         }
553
554             /// <summary>
555         /// Open an explorer window to the target path's directory
556         /// and select the file
557         /// </summary>
558         /// <param name="entry"></param>
559         public void GoToFile(FileEntry entry)
560         {
561             var fullPath = entry.FullPath;
562             if (!File.Exists(fullPath) && !Directory.Exists(fullPath))
563                 return;
564             Process.Start("explorer.exe","/select, " + fullPath);
565         }
566
567         public void OpenLogPath()
568         {
569             var pithosDataPath = PithosSettings.PithosDataPath;
570
571             Process.Start(pithosDataPath);
572         }
573         
574         public void ShowFileProperties()
575                 {
576                         var account = Settings.Accounts.First(acc => acc.IsActive);            
577                         var dir = new DirectoryInfo(account.RootPath + @"\pithos");
578                         var files=dir.GetFiles();
579                         var r=new Random();
580                         var idx=r.Next(0, files.Length);
581                         ShowFileProperties(files[idx].FullName);            
582                 }
583
584                 public void ShowFileProperties(string filePath)
585                 {
586                         if (String.IsNullOrWhiteSpace(filePath))
587                                 throw new ArgumentNullException("filePath");
588                         if (!File.Exists(filePath) && !Directory.Exists(filePath))
589                                 throw new ArgumentException(String.Format("Non existent file {0}",filePath),"filePath");
590                         Contract.EndContractBlock();
591
592                         var pair=(from monitor in  Monitors
593                                                            where filePath.StartsWith(monitor.Value.RootPath, StringComparison.InvariantCultureIgnoreCase)
594                                                                    select monitor).FirstOrDefault();
595                         var accountMonitor = pair.Value;
596
597                         if (accountMonitor == null)
598                                 return;
599
600                         var infoTask=Task.Factory.StartNew(()=>accountMonitor.GetObjectInfo(filePath));
601
602                         
603
604                         var fileProperties = new FilePropertiesViewModel(this, infoTask,filePath);
605                         _windowManager.ShowWindow(fileProperties);
606                 } 
607                 
608                 public void ShowContainerProperties()
609                 {
610                         var account = Settings.Accounts.First(acc => acc.IsActive);            
611                         var dir = new DirectoryInfo(account.RootPath);
612                         var fullName = (from folder in dir.EnumerateDirectories()
613                                                         where (folder.Attributes & FileAttributes.Hidden) == 0
614                                                         select folder.FullName).First();
615                         ShowContainerProperties(fullName);            
616                 }
617
618                 public void ShowContainerProperties(string filePath)
619                 {
620                         if (String.IsNullOrWhiteSpace(filePath))
621                                 throw new ArgumentNullException("filePath");
622                         if (!Directory.Exists(filePath))
623                                 throw new ArgumentException(String.Format("Non existent file {0}",filePath),"filePath");
624                         Contract.EndContractBlock();
625
626                         var pair=(from monitor in  Monitors
627                                                            where filePath.StartsWith(monitor.Value.RootPath, StringComparison.InvariantCultureIgnoreCase)
628                                                                    select monitor).FirstOrDefault();
629                         var accountMonitor = pair.Value;            
630                         var info = accountMonitor.GetContainerInfo(filePath);
631
632                         
633
634                         var containerProperties = new ContainerPropertiesViewModel(this, info,filePath);
635                         _windowManager.ShowWindow(containerProperties);
636                 }
637
638                 public void SynchNow()
639                 {
640                         _pollAgent.SynchNow();
641                 }
642
643                 public ObjectInfo RefreshObjectInfo(ObjectInfo currentInfo)
644                 {
645                         if (currentInfo==null)
646                                 throw new ArgumentNullException("currentInfo");
647                         Contract.EndContractBlock();                
648             var monitor = Monitors[currentInfo.AccountKey];
649                         var newInfo=monitor.CloudClient.GetObjectInfo(currentInfo.Account, currentInfo.Container, currentInfo.Name);
650                         return newInfo;
651                 }
652
653                 public ContainerInfo RefreshContainerInfo(ContainerInfo container)
654                 {
655                         if (container == null)
656                                 throw new ArgumentNullException("container");
657                         Contract.EndContractBlock();
658
659                         var monitor = Monitors[container.AccountKey];
660                         var newInfo = monitor.CloudClient.GetContainerInfo(container.Account, container.Name);
661                         return newInfo;
662                 }
663
664             private bool _isPaused;
665             public bool IsPaused
666             {
667                 get { return _isPaused; }
668                 set
669                 {
670                     _isPaused = value;
671                 PauseSyncCaption = IsPaused ? "Resume syncing" : "Pause syncing";
672                 var iconKey = IsPaused ? "TraySyncPaused" : "TrayInSynch";
673                 StatusIcon = String.Format(@"../Images/{0}.ico", iconKey);
674
675                 NotifyOfPropertyChange(() => IsPaused);
676                 }
677             }
678
679             public void ToggleSynching()
680                 {
681                         IsPaused=!IsPaused;
682                         foreach (var monitor in Monitors.Values)
683                         {
684                             monitor.Pause = IsPaused ;
685                         }
686             _pollAgent.Pause = IsPaused;
687             _networkAgent.Pause = IsPaused;
688
689
690                 }
691
692         public void ExitPithos()
693         {
694             try
695             {
696
697                 foreach (var monitor in Monitors.Select(pair => pair.Value))
698                 {
699                     monitor.Stop();
700                 }
701
702                 var view = GetView() as Window;
703                 if (view != null)
704                     view.Close();
705             }
706             catch (Exception exc)
707             {
708                 Log.Info("Exception while exiting", exc);                
709             }
710             finally
711             {
712                 Application.Current.Shutdown();
713             }
714         }
715
716             #endregion
717
718
719                 private readonly Dictionary<PithosStatus, StatusInfo> _iconNames = new List<StatusInfo>
720                         {
721                                 new StatusInfo(PithosStatus.InSynch, "All files up to date", "TrayInSynch"),
722                                 new StatusInfo(PithosStatus.PollSyncing, "Polling Files", "TraySynching"),
723                 new StatusInfo(PithosStatus.LocalSyncing, "Syncing Files", "TraySynching"),
724                                 new StatusInfo(PithosStatus.SyncPaused, "Sync Paused", "TraySyncPaused")
725                         }.ToDictionary(s => s.Status);
726
727                 readonly IWindowManager _windowManager;
728                 
729         //private int _syncCount=0;
730
731
732         private PithosStatus _pithosStatus = PithosStatus.Disconnected;
733
734         public void SetPithosStatus(PithosStatus status)
735         {
736             if (_pithosStatus == PithosStatus.LocalSyncing && status == PithosStatus.PollComplete)
737                 return;
738             if (_pithosStatus == PithosStatus.PollSyncing && status == PithosStatus.LocalComplete)
739                 return;
740             if (status == PithosStatus.LocalComplete || status == PithosStatus.PollComplete)
741                 _pithosStatus = PithosStatus.InSynch;
742             else
743                 _pithosStatus = status;
744             UpdateStatus();
745         }
746
747         public void SetPithosStatus(PithosStatus status,string message)
748         {
749             StatusMessage = message;
750             SetPithosStatus(status);
751         }
752
753           /*  public Notifier GetNotifier(Notification startNotification, Notification endNotification)
754             {
755                 return new Notifier(this, startNotification, endNotification);
756             }*/
757
758             public Notifier GetNotifier(string startMessage, string endMessage, params object[] args)
759             {
760                 return new Notifier(this, 
761                 new StatusNotification(String.Format(startMessage,args)), 
762                 new StatusNotification(String.Format(endMessage,args)));
763             }
764
765
766             ///<summary>
767                 /// Updates the visual status indicators of the application depending on status changes, e.g. icon, stat                
768                 ///</summary>
769                 public void UpdateStatus()
770                 {
771
772                         if (_iconNames.ContainsKey(_pithosStatus))
773                         {
774                                 var info = _iconNames[_pithosStatus];
775                                 StatusIcon = String.Format(@"../Images/{0}.ico", info.IconName);
776                         }
777
778             if (_pithosStatus == PithosStatus.InSynch)
779                 StatusMessage = "All files up to date";
780                 }
781
782
783            
784                 private Task StartMonitor(PithosMonitor monitor,int retries=0)
785                 {
786                         return Task.Factory.StartNew(() =>
787                         {
788                                 using (log4net.ThreadContext.Stacks["Monitor"].Push("Start"))
789                                 {
790                                         try
791                                         {
792                                                 Log.InfoFormat("Start Monitoring {0}", monitor.UserName);
793
794                                                 monitor.Start();
795                                         }
796                                         catch (WebException exc)
797                                         {
798                                                 if (AbandonRetry(monitor, retries))
799                                                         return;
800
801                                                 HttpStatusCode statusCode =HttpStatusCode.OK;
802                                                 var response = exc.Response as HttpWebResponse;
803                                                 if(response!=null)
804                                                         statusCode = response.StatusCode;
805
806                                                 switch (statusCode)
807                                                 {
808                                                         case HttpStatusCode.Unauthorized:
809                                                                 var message = String.Format("API Key Expired for {0}. Starting Renewal",
810                                                                                                                         monitor.UserName);
811                                                                 Log.Error(message, exc);
812                                 var account = Settings.Accounts.Find(acc => acc.AccountKey == new Uri(new Uri(monitor.AuthenticationUrl), monitor.UserName));                                
813                                                         account.IsExpired = true;
814                                 Notify(new ExpirationNotification(account));
815                                                                 //TryAuthorize(monitor.UserName, retries).Wait();
816                                                                 break;
817                                                         case HttpStatusCode.ProxyAuthenticationRequired:
818                                                                 TryAuthenticateProxy(monitor,retries);
819                                                                 break;
820                                                         default:
821                                                                 TryLater(monitor, exc, retries);
822                                                                 break;
823                                                 }
824                                         }
825                                         catch (Exception exc)
826                                         {
827                                                 if (AbandonRetry(monitor, retries)) 
828                                                         return;
829
830                                                 TryLater(monitor,exc,retries);
831                                         }
832                                 }
833                         });
834                 }
835
836                 private void TryAuthenticateProxy(PithosMonitor monitor,int retries)
837                 {
838                         Execute.OnUIThread(() =>
839                                                                    {                                       
840                                                                            var proxyAccount = IoC.Get<ProxyAccountViewModel>();
841                                                                                 proxyAccount.Settings = Settings;
842                                                                            if (true != _windowManager.ShowDialog(proxyAccount)) 
843                                                                                    return;
844                                                                            StartMonitor(monitor, retries);
845                                                                            NotifyOfPropertyChange(() => Accounts);
846                                                                    });
847                 }
848
849                 private bool AbandonRetry(PithosMonitor monitor, int retries)
850                 {
851                         if (retries > 1)
852                         {
853                                 var message = String.Format("Monitoring of account {0} has failed too many times. Will not retry",
854                                                                                         monitor.UserName);
855                                 _events.Publish(new Notification
856                                                                         {Title = "Account monitoring failed", Message = message, Level = TraceLevel.Error});
857                                 return true;
858                         }
859                         return false;
860                 }
861
862
863             private void TryLater(PithosMonitor monitor, Exception exc,int retries)
864                 {
865                         var message = String.Format("An exception occured. Can't start monitoring\nWill retry in 10 seconds");
866                         Task.Factory.StartNewDelayed(10000, () => StartMonitor(monitor,retries+1));
867                         _events.Publish(new Notification
868                                                                 {Title = "Error", Message = message, Level = TraceLevel.Error});
869                         Log.Error(message, exc);
870                 }
871
872
873                 public void NotifyChange(string status, TraceLevel level=TraceLevel.Info)
874                 {
875                         StatusMessage = status;
876                         
877                         _events.Publish(new Notification { Title = "Pithos+", Message = status, Level = level });
878                 }
879
880                 public void NotifyChangedFile(string filePath)
881                 {
882             if (RecentFiles.Any(e => e.FullPath == filePath))
883                 return;
884             
885                         IProducerConsumerCollection<FileEntry> files=RecentFiles;
886                         FileEntry popped;
887                         while (files.Count > 5)
888                                 files.TryTake(out popped);
889             var entry = new FileEntry { FullPath = filePath };
890                         files.TryAdd(entry);
891                 }
892
893                 public void NotifyAccount(AccountInfo account)
894                 {
895                         if (account== null)
896                                 return;
897                         //TODO: What happens to an existing account whose Token has changed?
898                         account.SiteUri= String.Format("{0}/ui/?token={1}&user={2}",
899                                 account.SiteUri, Uri.EscapeDataString(account.Token),
900                                 Uri.EscapeDataString(account.UserName));
901
902                         if (!Accounts.Any(item => item.UserName == account.UserName && item.SiteUri == account.SiteUri))
903                                 Accounts.TryAdd(account);
904
905                 }
906
907                 public void NotifyConflicts(IEnumerable<FileSystemInfo> conflictFiles, string message)
908                 {
909                         if (conflictFiles == null)
910                                 return;
911                     //Convert to list to avoid multiple iterations
912             var files = conflictFiles.ToList();
913                         if (files.Count==0)
914                                 return;
915
916                         UpdateStatus();
917                         //TODO: Create a more specific message. For now, just show a warning
918                         NotifyForFiles(files,message,TraceLevel.Warning);
919
920                 }
921
922                 public void NotifyForFiles(IEnumerable<FileSystemInfo> files, string message,TraceLevel level=TraceLevel.Info)
923                 {
924                         if (files == null)
925                                 return;
926                         if (!files.Any())
927                                 return;
928
929                         StatusMessage = message;
930
931                         _events.Publish(new Notification { Title = "Pithos+", Message = message, Level = level});
932                 }
933
934                 public void Notify(Notification notification)
935                 {
936                         _events.Publish(notification);
937                 }
938
939
940                 public void RemoveMonitor(string serverUrl,string accountName)
941                 {
942                         if (String.IsNullOrWhiteSpace(accountName))
943                                 return;
944
945                         var accountInfo=_accounts.FirstOrDefault(account => account.UserName == accountName && account.StorageUri.ToString().StartsWith(serverUrl));
946             if (accountInfo != null)
947             {
948                 _accounts.TryRemove(accountInfo);
949                 _pollAgent.RemoveAccount(accountInfo);
950             }
951
952             var accountKey = new Uri(new Uri(serverUrl),accountName);
953                     PithosMonitor monitor;
954                         if (Monitors.TryRemove(accountKey, out monitor))
955                         {
956                                 monitor.Stop();
957                 //TODO: Also remove any pending actions for this account
958                 //from the network queue                
959                         }
960                 }
961
962                 public void RefreshOverlays()
963                 {
964                         foreach (var pair in Monitors)
965                         {
966                                 var monitor = pair.Value;
967
968                                 var path = monitor.RootPath;
969
970                                 if (String.IsNullOrWhiteSpace(path))
971                                         continue;
972
973                                 if (!Directory.Exists(path) && !File.Exists(path))
974                                         continue;
975
976                                 IntPtr pathPointer = Marshal.StringToCoTaskMemAuto(path);
977
978                                 try
979                                 {
980                                         NativeMethods.SHChangeNotify(HChangeNotifyEventID.SHCNE_UPDATEITEM,
981                                                                                                  HChangeNotifyFlags.SHCNF_PATHW | HChangeNotifyFlags.SHCNF_FLUSHNOWAIT,
982                                                                                                  pathPointer, IntPtr.Zero);
983                                 }
984                                 finally
985                                 {
986                                         Marshal.FreeHGlobal(pathPointer);
987                                 }
988                         }
989                 }
990
991                 #region Event Handlers
992                 
993                 public void Handle(SelectiveSynchChanges message)
994                 {
995                     TaskEx.Run(() =>
996                     {
997                         PithosMonitor monitor;
998                         if (Monitors.TryGetValue(message.Account.AccountKey, out monitor))
999                         {
1000                             monitor.SetSelectivePaths(message.Uris, message.Added, message.Removed);
1001                         }
1002
1003                         var account = Accounts.FirstOrDefault(acc => acc.AccountKey == message.Account.AccountKey);
1004                         if (account != null)
1005                         {
1006                             this._pollAgent.SetSelectivePaths(account, message.Added, message.Removed);
1007                         }
1008                     });
1009
1010                 }
1011
1012
1013                 private bool _pollStarted;
1014             private Sparkle _sparkle;
1015             private bool _manualUpgradeCheck;
1016
1017             //SMELL: Doing so much work for notifications in the shell is wrong
1018                 //The notifications should be moved to their own view/viewmodel pair
1019                 //and different templates should be used for different message types
1020                 //This will also allow the addition of extra functionality, eg. actions
1021                 //
1022                 public void Handle(Notification notification)
1023                 {
1024                         UpdateStatus();
1025
1026                         if (!Settings.ShowDesktopNotifications)
1027                                 return;
1028
1029                         if (notification is PollNotification)
1030                         {
1031                                 _pollStarted = true;
1032                                 return;
1033                         }
1034                         if (notification is CloudNotification)
1035                         {
1036                                 if (!_pollStarted) 
1037                                         return;
1038                                 _pollStarted= false;
1039                                 notification.Title = "Pithos+";
1040                                 notification.Message = "Start Synchronisation";
1041                         }
1042
1043                     var deleteNotification = notification as CloudDeleteNotification;
1044             if (deleteNotification != null)
1045             {
1046                 StatusMessage = String.Format("Deleted {0}", deleteNotification.Data.Name);
1047                 return;
1048             }
1049
1050                     var progress = notification as ProgressNotification;
1051                     
1052                     
1053             if (progress != null)
1054                     {
1055                         StatusMessage = String.Format("{0} {1:p2} of {2} - {3}",                                                      
1056                                               progress.Action,
1057                                                       progress.Block/(double)progress.TotalBlocks,
1058                                                       progress.FileSize.ToByteSize(),
1059                                                       progress.FileName);
1060                         return;
1061                     }
1062
1063                     var info = notification as StatusNotification;
1064             if (info != null)
1065             {
1066                 StatusMessage = info.Title;
1067                 return;
1068             }
1069                         if (String.IsNullOrWhiteSpace(notification.Message) && String.IsNullOrWhiteSpace(notification.Title))
1070                                 return;
1071
1072             if (notification.Level <= TraceLevel.Warning)
1073                             ShowBalloonFor(notification);
1074                 }
1075
1076             private void ShowBalloonFor(Notification notification)
1077             {
1078             Contract.Requires(notification!=null);
1079             
1080             if (!Settings.ShowDesktopNotifications) 
1081                 return;
1082             
1083             BalloonIcon icon;
1084                 switch (notification.Level)
1085                 {
1086                 case TraceLevel.Verbose:
1087                         return;
1088                     case TraceLevel.Info:                   
1089                         icon = BalloonIcon.Info;
1090                         break;
1091                 case TraceLevel.Error:
1092                     icon = BalloonIcon.Error;
1093                     break;
1094                 case TraceLevel.Warning:
1095                         icon = BalloonIcon.Warning;
1096                         break;
1097                     default:
1098                         return;
1099                 }
1100
1101                 var tv = (ShellView) GetView();
1102                 System.Action clickAction = null;
1103                 if (notification is ExpirationNotification)
1104                 {
1105                     clickAction = () => ShowPreferences("AccountTab");
1106                 }
1107                 var balloon = new PithosBalloon
1108                                   {
1109                                       Title = notification.Title,
1110                                       Message = notification.Message,
1111                                       Icon = icon,
1112                                       ClickAction = clickAction
1113                                   };
1114                 tv.TaskbarView.ShowCustomBalloon(balloon, PopupAnimation.Fade, 4000);
1115             }
1116
1117             #endregion
1118
1119                 public void Handle(ShowFilePropertiesEvent message)
1120                 {
1121                         if (message == null)
1122                                 throw new ArgumentNullException("message");
1123                         if (String.IsNullOrWhiteSpace(message.FileName) )
1124                                 throw new ArgumentException("message");
1125                         Contract.EndContractBlock();
1126
1127                         var fileName = message.FileName;
1128                         //TODO: Display file properties for non-container folders
1129                         if (File.Exists(fileName))
1130                                 //Retrieve the full name with exact casing. Pithos names are case sensitive                             
1131                                 ShowFileProperties(FileInfoExtensions.GetProperFilePathCapitalization(fileName));
1132                         else if (Directory.Exists(fileName))
1133                                 //Retrieve the full name with exact casing. Pithos names are case sensitive
1134                         {
1135                                 var path = FileInfoExtensions.GetProperDirectoryCapitalization(fileName);
1136                                 if (IsContainer(path))
1137                                         ShowContainerProperties(path);
1138                                 else
1139                                         ShowFileProperties(path);
1140                         }
1141                 }
1142
1143                 private bool IsContainer(string path)
1144                 {
1145                         var matchingFolders = from account in _accounts
1146                                                                   from rootFolder in Directory.GetDirectories(account.AccountPath)
1147                                                                   where rootFolder.Equals(path, StringComparison.InvariantCultureIgnoreCase)
1148                                                                   select rootFolder;
1149                         return matchingFolders.Any();
1150                 }
1151
1152                 public FileStatus GetFileStatus(string localFileName)
1153                 {
1154                         if (String.IsNullOrWhiteSpace(localFileName))
1155                                 throw new ArgumentNullException("localFileName");
1156                         Contract.EndContractBlock();
1157                         
1158                         var statusKeeper = IoC.Get<IStatusKeeper>();
1159                         var status=statusKeeper.GetFileStatus(localFileName);
1160                         return status;
1161                 }
1162
1163             public void RemoveAccountFromDatabase(AccountSettings account)
1164             {
1165             var statusKeeper = IoC.Get<IStatusKeeper>();
1166             statusKeeper.ClearFolderStatus(account.RootPath);           
1167             }
1168         }
1169 }