mirror of
https://github.com/QL-Win/QuickLook.git
synced 2026-02-27 01:00:11 +08: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
|
||||
}
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
using System;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Diagnostics;
|
||||
using System.IO;
|
||||
@@ -7,241 +7,239 @@ using System.Reflection;
|
||||
using System.Security.Cryptography;
|
||||
using System.Text;
|
||||
using System.Windows;
|
||||
using QuickLook.Common.Helpers;
|
||||
|
||||
namespace QuickLook.Plugin.MarkdownViewer
|
||||
namespace QuickLook.Plugin.MarkdownViewer;
|
||||
|
||||
internal class ResourceManager
|
||||
{
|
||||
internal class ResourceManager
|
||||
private readonly string _resourcePath;
|
||||
private readonly string _resourcePrefix;
|
||||
private readonly string _versionFilePath;
|
||||
private readonly string _noUpdateFilePath;
|
||||
private readonly string _embeddedHash;
|
||||
|
||||
public ResourceManager(string resourcePath, string resourcePrefix)
|
||||
{
|
||||
private readonly string _resourcePath;
|
||||
private readonly string _resourcePrefix;
|
||||
private readonly string _versionFilePath;
|
||||
private readonly string _noUpdateFilePath;
|
||||
private readonly string _embeddedHash;
|
||||
_resourcePath = resourcePath;
|
||||
_resourcePrefix = resourcePrefix;
|
||||
_versionFilePath = Path.Combine(_resourcePath, ".version");
|
||||
_noUpdateFilePath = Path.Combine(_resourcePath, ".noupdate");
|
||||
_embeddedHash = GetEmbeddedResourcesHash();
|
||||
}
|
||||
|
||||
public ResourceManager(string resourcePath, string resourcePrefix)
|
||||
public void InitializeResources()
|
||||
{
|
||||
// Extract resources for the first time
|
||||
if (!Directory.Exists(_resourcePath))
|
||||
{
|
||||
_resourcePath = resourcePath;
|
||||
_resourcePrefix = resourcePrefix;
|
||||
_versionFilePath = Path.Combine(_resourcePath, ".version");
|
||||
_noUpdateFilePath = Path.Combine(_resourcePath, ".noupdate");
|
||||
_embeddedHash = GetEmbeddedResourcesHash();
|
||||
ExtractResources();
|
||||
return;
|
||||
}
|
||||
|
||||
public void InitializeResources()
|
||||
// Check if updates are disabled
|
||||
if (File.Exists(_noUpdateFilePath))
|
||||
return;
|
||||
|
||||
// Check if resources need updating by comparing hashes
|
||||
var versionInfo = ReadVersionFile();
|
||||
|
||||
if (versionInfo == null)
|
||||
{
|
||||
// Extract resources for the first time
|
||||
if (!Directory.Exists(_resourcePath))
|
||||
{
|
||||
ExtractResources();
|
||||
return;
|
||||
}
|
||||
|
||||
// Check if updates are disabled
|
||||
if (File.Exists(_noUpdateFilePath))
|
||||
return;
|
||||
|
||||
// Check if resources need updating by comparing hashes
|
||||
var versionInfo = ReadVersionFile();
|
||||
|
||||
if (versionInfo == null)
|
||||
{
|
||||
// No version file exists, create it and extract resources
|
||||
ExtractResources();
|
||||
return;
|
||||
}
|
||||
|
||||
// If embedded hash matches stored hash, no update needed
|
||||
if (_embeddedHash == versionInfo.EmbeddedHash) return;
|
||||
|
||||
// Calculate current directory hash
|
||||
var currentDirectoryHash = CalculateDirectoryHash(_resourcePath);
|
||||
|
||||
// If current directory matches the stored extracted hash, user hasn't modified files
|
||||
if (currentDirectoryHash == versionInfo.ExtractedHash)
|
||||
{
|
||||
// Safe to update
|
||||
ExtractResources();
|
||||
return;
|
||||
}
|
||||
|
||||
// User has modified files, ask for permission to update
|
||||
var result = MessageBox.Show(
|
||||
"The MarkdownViewer resources have been updated. Would you like to update to the newest version?\n\n" +
|
||||
"Note: Your current resources appear to have been modified. Updating will overwrite your modifications.",
|
||||
"MarkdownViewer Update Available",
|
||||
MessageBoxButton.YesNo,
|
||||
MessageBoxImage.Question);
|
||||
|
||||
if (result == MessageBoxResult.Yes)
|
||||
{
|
||||
ExtractResources();
|
||||
}
|
||||
else
|
||||
{
|
||||
// Update the embedded hash in the version file to prevent any more prompts for this version
|
||||
UpdateVersionFileEmbeddedHash();
|
||||
}
|
||||
// No version file exists, create it and extract resources
|
||||
ExtractResources();
|
||||
return;
|
||||
}
|
||||
|
||||
private void ExtractResources()
|
||||
// If embedded hash matches stored hash, no update needed
|
||||
if (_embeddedHash == versionInfo.EmbeddedHash) return;
|
||||
|
||||
// Calculate current directory hash
|
||||
var currentDirectoryHash = CalculateDirectoryHash(_resourcePath);
|
||||
|
||||
// If current directory matches the stored extracted hash, user hasn't modified files
|
||||
if (currentDirectoryHash == versionInfo.ExtractedHash)
|
||||
{
|
||||
// Delete and recreate directory to ensure clean state
|
||||
if (Directory.Exists(_resourcePath))
|
||||
{
|
||||
try
|
||||
{
|
||||
Directory.Delete(_resourcePath, true);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Debug.WriteLine($"Failed to delete directory {_resourcePath}: {ex.Message}");
|
||||
// If we can't delete the directory, we'll try to continue with existing one
|
||||
}
|
||||
}
|
||||
Directory.CreateDirectory(_resourcePath);
|
||||
|
||||
var assembly = Assembly.GetExecutingAssembly();
|
||||
var resourceNames = assembly.GetManifestResourceNames();
|
||||
|
||||
foreach (var resourceName in resourceNames)
|
||||
{
|
||||
if (!resourceName.StartsWith(_resourcePrefix)) continue;
|
||||
|
||||
var relativePath = resourceName.Substring(_resourcePrefix.Length);
|
||||
if (relativePath.Equals("resources", StringComparison.OrdinalIgnoreCase)) continue; // Skip 'resources' binary file
|
||||
|
||||
var targetPath = Path.Combine(_resourcePath, relativePath.Replace('/', Path.DirectorySeparatorChar));
|
||||
|
||||
// Create directory if it doesn't exist
|
||||
var directory = Path.GetDirectoryName(targetPath);
|
||||
if (directory != null)
|
||||
Directory.CreateDirectory(directory);
|
||||
|
||||
// Extract the resource
|
||||
using (var stream = assembly.GetManifestResourceStream(resourceName))
|
||||
using (var fileStream = File.Create(targetPath))
|
||||
{
|
||||
if (stream == null) continue;
|
||||
stream.CopyTo(fileStream);
|
||||
}
|
||||
}
|
||||
|
||||
// Generate version file after extracting all resources
|
||||
GenerateVersionFile();
|
||||
|
||||
// Verify that md2html.html was extracted
|
||||
var htmlPath = Path.Combine(_resourcePath, "md2html.html");
|
||||
if (!File.Exists(htmlPath))
|
||||
{
|
||||
throw new FileNotFoundException($"Required template file md2html.html not found in resources. Available resources: {string.Join(", ", resourceNames)}");
|
||||
}
|
||||
// Safe to update
|
||||
ExtractResources();
|
||||
return;
|
||||
}
|
||||
|
||||
private class VersionInfo
|
||||
// User has modified files, ask for permission to update
|
||||
var result = MessageBox.Show(
|
||||
"The MarkdownViewer resources have been updated. Would you like to update to the newest version?\n\n" +
|
||||
"Note: Your current resources appear to have been modified. Updating will overwrite your modifications.",
|
||||
"MarkdownViewer Update Available",
|
||||
MessageBoxButton.YesNo,
|
||||
MessageBoxImage.Question);
|
||||
|
||||
if (result == MessageBoxResult.Yes)
|
||||
{
|
||||
public string EmbeddedHash { get; set; }
|
||||
public string ExtractedHash { get; set; }
|
||||
|
||||
public VersionInfo(string embeddedHash, string extractedHash)
|
||||
{
|
||||
EmbeddedHash = embeddedHash;
|
||||
ExtractedHash = extractedHash;
|
||||
}
|
||||
ExtractResources();
|
||||
}
|
||||
|
||||
private static string CalculateDirectoryHash(string directory)
|
||||
else
|
||||
{
|
||||
var files = Directory.GetFiles(directory, "*.*", SearchOption.AllDirectories)
|
||||
.Where(f => !f.EndsWith(".version"))
|
||||
.OrderBy(f => f)
|
||||
.ToList();
|
||||
|
||||
using (var sha256 = SHA256.Create())
|
||||
{
|
||||
var combinedBytes = new List<byte>();
|
||||
foreach (var file in files)
|
||||
{
|
||||
var relativePath = file.Substring(directory.Length + 1);
|
||||
var pathBytes = Encoding.UTF8.GetBytes(relativePath.ToLowerInvariant());
|
||||
combinedBytes.AddRange(pathBytes);
|
||||
|
||||
var contentBytes = File.ReadAllBytes(file);
|
||||
combinedBytes.AddRange(contentBytes);
|
||||
}
|
||||
|
||||
var hash = sha256.ComputeHash(combinedBytes.ToArray());
|
||||
return BitConverter.ToString(hash).Replace("-", "").ToLowerInvariant();
|
||||
}
|
||||
// Update the embedded hash in the version file to prevent any more prompts for this version
|
||||
UpdateVersionFileEmbeddedHash();
|
||||
}
|
||||
}
|
||||
|
||||
private string GetEmbeddedResourcesHash()
|
||||
private void ExtractResources()
|
||||
{
|
||||
// Delete and recreate directory to ensure clean state
|
||||
if (Directory.Exists(_resourcePath))
|
||||
{
|
||||
try
|
||||
{
|
||||
return EmbeddedResourcesHash.Hash;
|
||||
Directory.Delete(_resourcePath, true);
|
||||
}
|
||||
catch (Exception)
|
||||
catch (Exception ex)
|
||||
{
|
||||
Debug.WriteLine("QuickLook.Plugin.MarkdownViewer: Embedded resources hash file not found.");
|
||||
return CalculateEmbeddedResourcesHash();
|
||||
Debug.WriteLine($"Failed to delete directory {_resourcePath}: {ex.Message}");
|
||||
// If we can't delete the directory, we'll try to continue with existing one
|
||||
}
|
||||
}
|
||||
Directory.CreateDirectory(_resourcePath);
|
||||
|
||||
private string CalculateEmbeddedResourcesHash()
|
||||
var assembly = Assembly.GetExecutingAssembly();
|
||||
var resourceNames = assembly.GetManifestResourceNames();
|
||||
|
||||
foreach (var resourceName in resourceNames)
|
||||
{
|
||||
var assembly = Assembly.GetExecutingAssembly();
|
||||
using (var sha256 = SHA256.Create())
|
||||
if (!resourceName.StartsWith(_resourcePrefix)) continue;
|
||||
|
||||
var relativePath = resourceName.Substring(_resourcePrefix.Length);
|
||||
if (relativePath.Equals("resources", StringComparison.OrdinalIgnoreCase)) continue; // Skip 'resources' binary file
|
||||
|
||||
var targetPath = Path.Combine(_resourcePath, relativePath.Replace('/', Path.DirectorySeparatorChar));
|
||||
|
||||
// Create directory if it doesn't exist
|
||||
var directory = Path.GetDirectoryName(targetPath);
|
||||
if (directory != null)
|
||||
Directory.CreateDirectory(directory);
|
||||
|
||||
// Extract the resource
|
||||
using (var stream = assembly.GetManifestResourceStream(resourceName))
|
||||
using (var fileStream = File.Create(targetPath))
|
||||
{
|
||||
var combinedBytes = new List<byte>();
|
||||
var resourceNames = assembly.GetManifestResourceNames()
|
||||
.Where(name => name.StartsWith(_resourcePrefix) &&
|
||||
!name.EndsWith(".version"))
|
||||
.OrderBy(name => name)
|
||||
.ToList();
|
||||
|
||||
foreach (var resourceName in resourceNames)
|
||||
{
|
||||
var nameBytes = Encoding.UTF8.GetBytes(resourceName.ToLowerInvariant());
|
||||
combinedBytes.AddRange(nameBytes);
|
||||
|
||||
using (var stream = assembly.GetManifestResourceStream(resourceName))
|
||||
{
|
||||
if (stream == null) continue;
|
||||
var buffer = new byte[stream.Length];
|
||||
stream.Read(buffer, 0, buffer.Length);
|
||||
combinedBytes.AddRange(buffer);
|
||||
}
|
||||
}
|
||||
|
||||
var hash = sha256.ComputeHash(combinedBytes.ToArray());
|
||||
return BitConverter.ToString(hash).Replace("-", "").ToLowerInvariant();
|
||||
if (stream == null) continue;
|
||||
stream.CopyTo(fileStream);
|
||||
}
|
||||
}
|
||||
|
||||
private void GenerateVersionFile()
|
||||
// Generate version file after extracting all resources
|
||||
GenerateVersionFile();
|
||||
|
||||
// Verify that md2html.html was extracted
|
||||
var htmlPath = Path.Combine(_resourcePath, "md2html.html");
|
||||
if (!File.Exists(htmlPath))
|
||||
{
|
||||
var extractedHash = CalculateDirectoryHash(_resourcePath);
|
||||
var newVersionContent = $"{_embeddedHash}{Environment.NewLine}{extractedHash}";
|
||||
File.WriteAllText(_versionFilePath, newVersionContent);
|
||||
}
|
||||
|
||||
private void UpdateVersionFileEmbeddedHash()
|
||||
{
|
||||
var versionInfo = ReadVersionFile() ?? throw new InvalidOperationException("Cannot update version file: no existing version file found");
|
||||
|
||||
var newVersionContent = $"{_embeddedHash}{Environment.NewLine}{versionInfo.ExtractedHash}";
|
||||
File.WriteAllText(_versionFilePath, newVersionContent);
|
||||
}
|
||||
|
||||
private VersionInfo? ReadVersionFile()
|
||||
{
|
||||
if (!File.Exists(_versionFilePath)) return null;
|
||||
|
||||
var lines = File.ReadAllLines(_versionFilePath);
|
||||
if (lines.Length < 2) return null;
|
||||
|
||||
return new VersionInfo(lines[0], lines[1]);
|
||||
throw new FileNotFoundException($"Required template file md2html.html not found in resources. Available resources: {string.Join(", ", resourceNames)}");
|
||||
}
|
||||
}
|
||||
|
||||
private class VersionInfo
|
||||
{
|
||||
public string EmbeddedHash { get; set; }
|
||||
public string ExtractedHash { get; set; }
|
||||
|
||||
public VersionInfo(string embeddedHash, string extractedHash)
|
||||
{
|
||||
EmbeddedHash = embeddedHash;
|
||||
ExtractedHash = extractedHash;
|
||||
}
|
||||
}
|
||||
|
||||
private static string CalculateDirectoryHash(string directory)
|
||||
{
|
||||
var files = Directory.GetFiles(directory, "*.*", SearchOption.AllDirectories)
|
||||
.Where(f => !f.EndsWith(".version"))
|
||||
.OrderBy(f => f)
|
||||
.ToList();
|
||||
|
||||
using (var sha256 = SHA256.Create())
|
||||
{
|
||||
var combinedBytes = new List<byte>();
|
||||
foreach (var file in files)
|
||||
{
|
||||
var relativePath = file.Substring(directory.Length + 1);
|
||||
var pathBytes = Encoding.UTF8.GetBytes(relativePath.ToLowerInvariant());
|
||||
combinedBytes.AddRange(pathBytes);
|
||||
|
||||
var contentBytes = File.ReadAllBytes(file);
|
||||
combinedBytes.AddRange(contentBytes);
|
||||
}
|
||||
|
||||
var hash = sha256.ComputeHash(combinedBytes.ToArray());
|
||||
return BitConverter.ToString(hash).Replace("-", "").ToLowerInvariant();
|
||||
}
|
||||
}
|
||||
|
||||
private string GetEmbeddedResourcesHash()
|
||||
{
|
||||
try
|
||||
{
|
||||
return EmbeddedResourcesHash.Hash;
|
||||
}
|
||||
catch (Exception)
|
||||
{
|
||||
Debug.WriteLine("QuickLook.Plugin.MarkdownViewer: Embedded resources hash file not found.");
|
||||
return CalculateEmbeddedResourcesHash();
|
||||
}
|
||||
}
|
||||
|
||||
private string CalculateEmbeddedResourcesHash()
|
||||
{
|
||||
var assembly = Assembly.GetExecutingAssembly();
|
||||
using (var sha256 = SHA256.Create())
|
||||
{
|
||||
var combinedBytes = new List<byte>();
|
||||
var resourceNames = assembly.GetManifestResourceNames()
|
||||
.Where(name => name.StartsWith(_resourcePrefix) &&
|
||||
!name.EndsWith(".version"))
|
||||
.OrderBy(name => name)
|
||||
.ToList();
|
||||
|
||||
foreach (var resourceName in resourceNames)
|
||||
{
|
||||
var nameBytes = Encoding.UTF8.GetBytes(resourceName.ToLowerInvariant());
|
||||
combinedBytes.AddRange(nameBytes);
|
||||
|
||||
using (var stream = assembly.GetManifestResourceStream(resourceName))
|
||||
{
|
||||
if (stream == null) continue;
|
||||
var buffer = new byte[stream.Length];
|
||||
stream.Read(buffer, 0, buffer.Length);
|
||||
combinedBytes.AddRange(buffer);
|
||||
}
|
||||
}
|
||||
|
||||
var hash = sha256.ComputeHash(combinedBytes.ToArray());
|
||||
return BitConverter.ToString(hash).Replace("-", "").ToLowerInvariant();
|
||||
}
|
||||
}
|
||||
|
||||
private void GenerateVersionFile()
|
||||
{
|
||||
var extractedHash = CalculateDirectoryHash(_resourcePath);
|
||||
var newVersionContent = $"{_embeddedHash}{Environment.NewLine}{extractedHash}";
|
||||
File.WriteAllText(_versionFilePath, newVersionContent);
|
||||
}
|
||||
|
||||
private void UpdateVersionFileEmbeddedHash()
|
||||
{
|
||||
var versionInfo = ReadVersionFile() ?? throw new InvalidOperationException("Cannot update version file: no existing version file found");
|
||||
|
||||
var newVersionContent = $"{_embeddedHash}{Environment.NewLine}{versionInfo.ExtractedHash}";
|
||||
File.WriteAllText(_versionFilePath, newVersionContent);
|
||||
}
|
||||
|
||||
private VersionInfo? ReadVersionFile()
|
||||
{
|
||||
if (!File.Exists(_versionFilePath)) return null;
|
||||
|
||||
var lines = File.ReadAllLines(_versionFilePath);
|
||||
if (lines.Length < 2) return null;
|
||||
|
||||
return new VersionInfo(lines[0], lines[1]);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user