// Copyright © 2017 Paddy Xu

//

// This file is part of QuickLook program.

//

// This program is free software: you can redistribute it and/or modify

// it under the terms of the GNU General Public License as published by

// the Free Software Foundation, either version 3 of the License, or

// (at your option) any later version.

//

// This program is distributed in the hope that it will be useful,

// but WITHOUT ANY WARRANTY; without even the implied warranty of

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

// GNU General Public License for more details.

//

// You should have received a copy of the GNU General Public License

// along with this program. If not, see <http://www.gnu.org/licenses/>.

using System;

using System.Diagnostics;

using System.IO;

using System.Runtime.ExceptionServices;

using System.Windows;

using QuickLook.Common.Helpers;

using QuickLook.Common.Plugin;

namespace QuickLook

{

internal class ViewWindowManager : IDisposable

{

private static ViewWindowManager _instance;

private string _invokedPath = string.Empty;

private ViewerWindow _viewerWindow;

internal ViewWindowManager()

{

InitNewViewerWindow();

}

public void Dispose()

{

StopFocusMonitor();

}

public void RunAndClosePreview()

{

if (_viewerWindow.Visibility != Visibility.Visible)

return;

// if the current focus is in Desktop or explorer windows, just close the preview window and leave the task to System.

var focus = NativeMethods.QuickLook.GetFocusedWindowType();

if (focus != NativeMethods.QuickLook.FocusedWindowType.Invalid)

{

StopFocusMonitor();

_viewerWindow.Close();

return;

}

// if the focus is in the preview window, run it

if (!WindowHelper.IsForegroundWindowBelongToSelf())

return;

StopFocusMonitor();

_viewerWindow.RunAndClose();

}

public void ClosePreview()

{

if (_viewerWindow.Visibility != Visibility.Visible)

return;

StopFocusMonitor();

_viewerWindow.Close();

}

public void TogglePreview(string path = null)

{

if (string.IsNullOrEmpty(path))

path = NativeMethods.QuickLook.GetCurrentSelection();

if (_viewerWindow.Visibility == Visibility.Visible && (string.IsNullOrEmpty(path) || path == _invokedPath))

ClosePreview();

else

InvokePreview(path);

}

private void RunFocusMonitor()

{

FocusMonitor.GetInstance().Start();

}

private void StopFocusMonitor()

{

FocusMonitor.GetInstance().Stop();

}

internal void ForgetCurrentWindow()

{

StopFocusMonitor();

_viewerWindow.Pinned = true;

InitNewViewerWindow();

}

public void SwitchPreview(string path = null)

{

if (_viewerWindow.Visibility != Visibility.Visible)

return;

if (string.IsNullOrEmpty(path))

path = NativeMethods.QuickLook.GetCurrentSelection();

if (string.IsNullOrEmpty(path))

return;

InvokePreview(path);

}

public void InvokePreview(string path = null)

{

if (string.IsNullOrEmpty(path))

path = NativeMethods.QuickLook.GetCurrentSelection();

if (string.IsNullOrEmpty(path))

return;

if (_viewerWindow.Visibility == Visibility.Visible && path == _invokedPath)

return;

if (!Directory.Exists(path) && !File.Exists(path))

return;

_invokedPath = path;

RunFocusMonitor();

var matchedPlugin = PluginManager.GetInstance().FindMatch(path);

BeginShowNewWindow(path, matchedPlugin);

}

private void BeginShowNewWindow(string path, IViewer matchedPlugin)

{

_viewerWindow.UnloadPlugin();

_viewerWindow.BeginShow(matchedPlugin, path, CurrentPluginFailed);

}

private void CurrentPluginFailed(string path, ExceptionDispatchInfo e)

{

var plugin = _viewerWindow.Plugin?.GetType();

_viewerWindow.Close();

TrayIconManager.ShowNotification($"Failed to preview {Path.GetFileName(path)}",

"Consider reporting this incident to QuickLook’s author.", true);

Debug.WriteLine(e.SourceException.ToString());

ProcessHelper.WriteLog(e.SourceException.ToString());

if (plugin != PluginManager.GetInstance().DefaultPlugin.GetType())

BeginShowNewWindow(path, PluginManager.GetInstance().DefaultPlugin);

else

e.Throw();

}

private void InitNewViewerWindow()

{

_viewerWindow = new ViewerWindow();

_viewerWindow.Closed += (sender, e) =>

{

if (ProcessHelper.IsShuttingDown())

return;

if (!(sender is ViewerWindow w) || w.Pinned)

return; // Pinned window has already been forgotten

StopFocusMonitor();

InitNewViewerWindow();

};

}

internal static ViewWindowManager GetInstance()

{

return _instance ?? (_instance = new ViewWindowManager());

}

}

}