mirror of
https://github.com/QL-Win/QuickLook.git
synced 2025-09-11 17:59:17 +00:00
Code Cleanup
This commit is contained in:
ema
@@ -15,138 +15,136 @@
|
||||
// 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 QuickLook.Common.Helpers;
|
||||
using QuickLook.Common.Plugin;
|
||||
using QuickLook.Plugin.HtmlViewer;
|
||||
using System;
|
||||
using System.Diagnostics;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using System.Net;
|
||||
using System.Text;
|
||||
using System.Windows;
|
||||
using System.Windows.Threading;
|
||||
using QuickLook.Common.Helpers;
|
||||
using QuickLook.Common.Plugin;
|
||||
using QuickLook.Plugin.HtmlViewer;
|
||||
using UtfUnknown;
|
||||
|
||||
namespace QuickLook.Plugin.MarkdownViewer
|
||||
namespace QuickLook.Plugin.MarkdownViewer;
|
||||
|
||||
public class Plugin : IViewer
|
||||
{
|
||||
public class Plugin : IViewer
|
||||
private WebpagePanel? _panel;
|
||||
private string? _currentHtmlPath;
|
||||
|
||||
private static readonly string _resourcePath = Path.Combine(SettingHelper.LocalDataPath, "QuickLook.Plugin.MarkdownViewer");
|
||||
private static readonly string _resourcePrefix = "QuickLook.Plugin.MarkdownViewer.Resources.";
|
||||
private static readonly ResourceManager _resourceManager = new(_resourcePath, _resourcePrefix);
|
||||
|
||||
public int Priority => 0;
|
||||
|
||||
public void Init()
|
||||
{
|
||||
private WebpagePanel? _panel;
|
||||
private string? _currentHtmlPath;
|
||||
// Initialize resources and handle versioning
|
||||
_resourceManager.InitializeResources();
|
||||
|
||||
private static readonly string _resourcePath = Path.Combine(SettingHelper.LocalDataPath, "QuickLook.Plugin.MarkdownViewer");
|
||||
private static readonly string _resourcePrefix = "QuickLook.Plugin.MarkdownViewer.Resources.";
|
||||
private static readonly ResourceManager _resourceManager = new ResourceManager(_resourcePath, _resourcePrefix);
|
||||
// Clean up any temporary HTML files if QuickLook was forcibly terminated
|
||||
CleanupTempFiles();
|
||||
}
|
||||
|
||||
public int Priority => 0;
|
||||
public bool CanHandle(string path)
|
||||
{
|
||||
return !Directory.Exists(path) && new[] { ".md", ".mdown", ".rmd", ".markdown" }.Any(path.ToLower().EndsWith);
|
||||
}
|
||||
|
||||
public void Init()
|
||||
public void Prepare(string path, ContextObject context)
|
||||
{
|
||||
context.PreferredSize = new Size(1000, 600);
|
||||
}
|
||||
|
||||
public void View(string path, ContextObject context)
|
||||
{
|
||||
_panel = new WebpagePanel();
|
||||
context.ViewerContent = _panel;
|
||||
context.Title = Path.GetFileName(path);
|
||||
|
||||
var htmlPath = GenerateMarkdownHtml(path);
|
||||
_panel.NavigateToFile(htmlPath, Path.GetDirectoryName(path));
|
||||
_panel.Dispatcher.Invoke(() => { context.IsBusy = false; }, DispatcherPriority.Loaded);
|
||||
}
|
||||
|
||||
private string GenerateMarkdownHtml(string path)
|
||||
{
|
||||
var templatePath = Path.Combine(_resourcePath, "md2html.html");
|
||||
|
||||
if (!File.Exists(templatePath))
|
||||
throw new FileNotFoundException($"Required template file md2html.html not found in extracted resources at {templatePath}");
|
||||
|
||||
var bytes = File.ReadAllBytes(path);
|
||||
var encoding = CharsetDetector.DetectFromBytes(bytes).Detected?.Encoding ?? Encoding.Default;
|
||||
var content = encoding.GetString(bytes);
|
||||
|
||||
var template = File.ReadAllText(templatePath);
|
||||
var html = template.Replace("{{content}}", content);
|
||||
|
||||
// Generate unique filename and ensure it doesn't exist
|
||||
string outputPath;
|
||||
do
|
||||
{
|
||||
// Initialize resources and handle versioning
|
||||
_resourceManager.InitializeResources();
|
||||
var uniqueId = Guid.NewGuid().ToString("N").Substring(0, 8);
|
||||
var outputFileName = $"temp_{uniqueId}.html";
|
||||
outputPath = Path.Combine(_resourcePath, outputFileName);
|
||||
} while (File.Exists(outputPath));
|
||||
|
||||
// Clean up any temporary HTML files if QuickLook was forcibly terminated
|
||||
CleanupTempFiles();
|
||||
}
|
||||
// Clean up previous file if it exists
|
||||
CleanupTempHtmlFile();
|
||||
|
||||
public bool CanHandle(string path)
|
||||
File.WriteAllText(outputPath, html);
|
||||
_currentHtmlPath = outputPath;
|
||||
|
||||
return outputPath;
|
||||
}
|
||||
|
||||
#region Cleanup
|
||||
|
||||
private void CleanupTempHtmlFile()
|
||||
{
|
||||
if (!string.IsNullOrEmpty(_currentHtmlPath) && File.Exists(_currentHtmlPath))
|
||||
{
|
||||
return !Directory.Exists(path) && new[] { ".md", ".mdown", ".rmd", ".markdown" }.Any(path.ToLower().EndsWith);
|
||||
}
|
||||
|
||||
public void Prepare(string path, ContextObject context)
|
||||
{
|
||||
context.PreferredSize = new Size(1000, 600);
|
||||
}
|
||||
|
||||
public void View(string path, ContextObject context)
|
||||
{
|
||||
_panel = new WebpagePanel();
|
||||
context.ViewerContent = _panel;
|
||||
context.Title = Path.GetFileName(path);
|
||||
|
||||
var htmlPath = GenerateMarkdownHtml(path);
|
||||
_panel.NavigateToFile(htmlPath, Path.GetDirectoryName(path));
|
||||
_panel.Dispatcher.Invoke(() => { context.IsBusy = false; }, DispatcherPriority.Loaded);
|
||||
}
|
||||
|
||||
private string GenerateMarkdownHtml(string path)
|
||||
{
|
||||
var templatePath = Path.Combine(_resourcePath, "md2html.html");
|
||||
|
||||
if (!File.Exists(templatePath))
|
||||
throw new FileNotFoundException($"Required template file md2html.html not found in extracted resources at {templatePath}");
|
||||
|
||||
var bytes = File.ReadAllBytes(path);
|
||||
var encoding = CharsetDetector.DetectFromBytes(bytes).Detected?.Encoding ?? Encoding.Default;
|
||||
var content = encoding.GetString(bytes);
|
||||
|
||||
var template = File.ReadAllText(templatePath);
|
||||
var html = template.Replace("{{content}}", content);
|
||||
|
||||
// Generate unique filename and ensure it doesn't exist
|
||||
string outputPath;
|
||||
do
|
||||
try
|
||||
{
|
||||
var uniqueId = Guid.NewGuid().ToString("N").Substring(0, 8);
|
||||
var outputFileName = $"temp_{uniqueId}.html";
|
||||
outputPath = Path.Combine(_resourcePath, outputFileName);
|
||||
} while (File.Exists(outputPath));
|
||||
|
||||
// Clean up previous file if it exists
|
||||
CleanupTempHtmlFile();
|
||||
|
||||
File.WriteAllText(outputPath, html);
|
||||
_currentHtmlPath = outputPath;
|
||||
|
||||
return outputPath;
|
||||
File.Delete(_currentHtmlPath);
|
||||
}
|
||||
catch (IOException) { } // Ignore deletion errors
|
||||
}
|
||||
}
|
||||
|
||||
#region Cleanup
|
||||
|
||||
private void CleanupTempHtmlFile()
|
||||
private void CleanupTempFiles()
|
||||
{
|
||||
try
|
||||
{
|
||||
if (!string.IsNullOrEmpty(_currentHtmlPath) && File.Exists(_currentHtmlPath))
|
||||
var tempFiles = Directory.GetFiles(_resourcePath, "temp_*.html");
|
||||
foreach (var file in tempFiles)
|
||||
{
|
||||
try
|
||||
{
|
||||
File.Delete(_currentHtmlPath);
|
||||
File.Delete(file);
|
||||
}
|
||||
catch (IOException) { } // Ignore deletion errors
|
||||
}
|
||||
}
|
||||
|
||||
private void CleanupTempFiles()
|
||||
catch (Exception ex)
|
||||
{
|
||||
try
|
||||
{
|
||||
var tempFiles = Directory.GetFiles(_resourcePath, "temp_*.html");
|
||||
foreach (var file in tempFiles)
|
||||
{
|
||||
try
|
||||
{
|
||||
File.Delete(file);
|
||||
}
|
||||
catch (IOException) { } // Ignore deletion errors
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Debug.WriteLine($"Failed to clean up temporary HTML files: {ex.Message}");
|
||||
}
|
||||
Debug.WriteLine($"Failed to clean up temporary HTML files: {ex.Message}");
|
||||
}
|
||||
|
||||
public void Cleanup()
|
||||
{
|
||||
GC.SuppressFinalize(this);
|
||||
|
||||
CleanupTempHtmlFile();
|
||||
|
||||
_panel?.Dispose();
|
||||
_panel = null;
|
||||
}
|
||||
|
||||
#endregion Cleanup
|
||||
}
|
||||
|
||||
public void Cleanup()
|
||||
{
|
||||
GC.SuppressFinalize(this);
|
||||
|
||||
CleanupTempHtmlFile();
|
||||
|
||||
_panel?.Dispose();
|
||||
_panel = null;
|
||||
}
|
||||
|
||||
#endregion Cleanup
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user