mirror of
https://github.com/Hofknecht/SystemTrayMenu.git
synced 2024-10-01 01:27:33 +13:00
490 lines
16 KiB
C#
490 lines
16 KiB
C#
using IWshRuntimeLibrary;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.ComponentModel;
|
|
using System.Diagnostics;
|
|
using System.Drawing;
|
|
using System.Drawing.Imaging;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Runtime.InteropServices;
|
|
using System.Text;
|
|
using System.Windows.Forms;
|
|
using SystemTrayMenu.Handler;
|
|
using SystemTrayMenu.Helper;
|
|
using TAFactory.IconPack;
|
|
|
|
namespace SystemTrayMenu.Controls
|
|
{
|
|
public class RowData : IDisposable
|
|
{
|
|
public FileInfo FileInfo;
|
|
public Icon Icon;
|
|
public bool ContainsMenu;
|
|
public bool IsContextMenuOpen;
|
|
public bool ResolvedFileNotFound;
|
|
public string WorkingDirectory;
|
|
public string Arguments;
|
|
|
|
public bool IsResolvedLnk;
|
|
public string TargetFilePath;
|
|
public string Text;
|
|
|
|
|
|
public Menu SubMenu;
|
|
public int RowIndex;
|
|
|
|
public bool IsSelected;
|
|
public bool IsSelectedByKeyboard;
|
|
|
|
/// <summary>
|
|
/// Loads the icon
|
|
/// </summary>
|
|
/// <param name="isDirectory">True = directory, false = file</param>
|
|
/// <param name="resolvedLnkPath">Discovered path when functions returns true</param>
|
|
/// <returns>True when linking to a different directory, otherwise false</returns>
|
|
public bool ReadIcon(bool isDirectory, ref string resolvedLnkPath)
|
|
{
|
|
bool isLnkDirectory = false;
|
|
|
|
if (string.IsNullOrEmpty(TargetFilePath))
|
|
{
|
|
Log.Info($"TargetFilePath from {resolvedLnkPath} empty");
|
|
}
|
|
else if (isDirectory)
|
|
{
|
|
Icon = IconReader.GetFolderIcon(TargetFilePath,
|
|
IconReader.FolderType.Closed, false);
|
|
}
|
|
else
|
|
{
|
|
bool handled = false;
|
|
string fileExtension = Path.GetExtension(TargetFilePath);
|
|
|
|
if (fileExtension == ".lnk")
|
|
{
|
|
handled = SetLnk(ref isLnkDirectory,
|
|
ref resolvedLnkPath);
|
|
}
|
|
else if (fileExtension == ".url")
|
|
{
|
|
handled = SetUrl();
|
|
}
|
|
else if (fileExtension == ".sln")
|
|
{
|
|
handled = SetSln();
|
|
}
|
|
|
|
if (!handled)
|
|
{
|
|
try
|
|
{
|
|
Icon = IconReader.GetFileIconWithCache(TargetFilePath, false);
|
|
|
|
// other project -> fails sometimes
|
|
//icon = IconHelper.ExtractIcon(TargetFilePath, 0);
|
|
|
|
// standard way -> fails sometimes
|
|
//icon = Icon.ExtractAssociatedIcon(filePath);
|
|
|
|
// API Code Pack -> fails sometimes
|
|
//ShellFile shellFile = ShellFile.FromFilePath(filePath);
|
|
//Bitmap shellThumb = shellFile.Thumbnail.ExtraLargeBitmap;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Log.Error($"TargetFilePath:'{TargetFilePath}'", ex);
|
|
}
|
|
}
|
|
}
|
|
|
|
return isLnkDirectory;
|
|
}
|
|
|
|
private bool SetLnk(ref bool isLnkDirectory,
|
|
ref string resolvedLnkPath)
|
|
{
|
|
bool handled = false;
|
|
resolvedLnkPath = LnkHelper.ResolveShortcut(TargetFilePath);
|
|
if (LnkHelper.IsDirectory(resolvedLnkPath))
|
|
{
|
|
Icon = IconReader.GetFolderIcon(TargetFilePath,
|
|
IconReader.FolderType.Open, true);
|
|
handled = true;
|
|
isLnkDirectory = true;
|
|
}
|
|
else if (string.IsNullOrEmpty(resolvedLnkPath))
|
|
{
|
|
ResolvedFileNotFound = true;
|
|
Log.Info($"Resolve *.LNK '{TargetFilePath}' has no icon");
|
|
#warning [Feature] Resolve network root #48, start here
|
|
}
|
|
else
|
|
{
|
|
IWshShell shell = new WshShell();
|
|
var lnk = shell.CreateShortcut(TargetFilePath)
|
|
as IWshShortcut;
|
|
Arguments = lnk.Arguments;
|
|
WorkingDirectory = lnk.WorkingDirectory;
|
|
string iconLocation = lnk.IconLocation;
|
|
if (iconLocation.Length > 2)
|
|
{
|
|
iconLocation = iconLocation.Substring(0,
|
|
iconLocation.Length - 2);
|
|
if (System.IO.File.Exists(iconLocation))
|
|
{
|
|
try
|
|
{
|
|
Icon = Icon.ExtractAssociatedIcon(iconLocation);
|
|
handled = true;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Log.Error($"iconLocation:'{iconLocation}'", ex);
|
|
}
|
|
}
|
|
}
|
|
|
|
TargetFilePath = resolvedLnkPath;
|
|
}
|
|
|
|
SetText(Path.GetFileNameWithoutExtension(TargetFilePathOrig));
|
|
|
|
return handled;
|
|
}
|
|
|
|
private bool SetUrl()
|
|
{
|
|
bool handled = false;
|
|
string iconFile = string.Empty;
|
|
try
|
|
{
|
|
FileIni file = new FileIni(TargetFilePath);
|
|
iconFile = file.Value("IconFile", string.Empty);
|
|
if (string.IsNullOrEmpty(iconFile))
|
|
{
|
|
string browserPath = FileUrl.GetDefaultBrowserPath();
|
|
if (string.IsNullOrEmpty(browserPath))
|
|
{
|
|
Log.Info($"No default browser found!");
|
|
}
|
|
else
|
|
{
|
|
Icon = IconReader.GetFileIconWithCache(browserPath, false);
|
|
handled = true;
|
|
}
|
|
}
|
|
else if (System.IO.File.Exists(iconFile))
|
|
{
|
|
Icon = Icon.ExtractAssociatedIcon(iconFile);
|
|
handled = true;
|
|
}
|
|
else
|
|
{
|
|
Log.Info($"Resolve *.URL '{TargetFilePath}' has no icon");
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Log.Error($"TargetFilePath:'{TargetFilePath}', " +
|
|
$"iconFile:'{iconFile}'", ex);
|
|
}
|
|
|
|
SetText($"{FileInfo.Name.Substring(0, FileInfo.Name.Length - 4)}");
|
|
|
|
return handled;
|
|
}
|
|
|
|
[DllImport("shell32.dll")]
|
|
static extern int FindExecutable(string lpFile, string lpDirectory, [Out] StringBuilder lpResult);
|
|
private bool SetSln()
|
|
{
|
|
bool handled = false;
|
|
var executable = new StringBuilder(1024);
|
|
try
|
|
{
|
|
FindExecutable(TargetFilePath, string.Empty, executable);
|
|
// icon = IconReader.GetFileIcon(executable, false);
|
|
// e.g. VS 2019 icon, need another icom in imagelist
|
|
List<Icon> extractedIcons = IconHelper.ExtractAllIcons(
|
|
executable.ToString());
|
|
Icon = extractedIcons.Last();
|
|
handled = true;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Log.Error($"TargetFilePath:'{TargetFilePath}', " +
|
|
$"executable:'{executable.ToString()}'",ex);
|
|
}
|
|
|
|
return handled;
|
|
}
|
|
|
|
public void SetText(string text)
|
|
{
|
|
//text = $" {text}";
|
|
if (text.Length > MenuDefines.LengthMax)
|
|
{
|
|
text = $"{text.Substring(0, MenuDefines.LengthMax)}...";
|
|
}
|
|
Text = text;
|
|
}
|
|
|
|
|
|
#warning CodeBuity&Refactor #49 - sort this class and check for duplicated
|
|
|
|
public event Action<object, EventArgs> OpenMenu;
|
|
|
|
public bool IsLoading = false;
|
|
public bool RestartLoading = false;
|
|
public BackgroundWorker Reading = new BackgroundWorker();
|
|
|
|
Icon icon = null;
|
|
//FontFamily fontFamily = new FontFamily("Segoe UI");
|
|
//Font font = new Font(new FontFamily("Segoe UI"), 12F,
|
|
// FontStyle.Regular, GraphicsUnit.Pixel);
|
|
|
|
WaitMenuOpen waitMenuOpen = new WaitMenuOpen();
|
|
|
|
bool disposed = false;
|
|
internal string TargetFilePathOrig;
|
|
internal bool HiddenEntry;
|
|
|
|
public RowData()
|
|
{
|
|
Reading.WorkerSupportsCancellation = true;
|
|
|
|
Initialize();
|
|
void Initialize()
|
|
{
|
|
//Margin = new Padding(0, 0, 0, 0);
|
|
//FlatAppearance.BorderSize = 0;
|
|
//UseVisualStyleBackColor = true;
|
|
//FlatStyle = FlatStyle.Flat;
|
|
//BackColor = MenuDefines.File;
|
|
//FlatAppearance.BorderColor = MenuDefines.File;
|
|
//Anchor = (AnchorStyles.Left | AnchorStyles.Right);
|
|
//AutoSize = true;
|
|
//AutoSizeMode = AutoSizeMode.GrowAndShrink;
|
|
//Font = new Font(fontFamily, 7, FontStyle.Regular, GraphicsUnit.Pixel);
|
|
//ForeColor = Color.Black;
|
|
waitMenuOpen.DoOpen += WaitMenuOpen_DoOpen;
|
|
|
|
//MouseLeave += MenuButton_MouseLeave;
|
|
//void MenuButton_MouseLeave(object sender, EventArgs e)
|
|
//{
|
|
// if (Tag == null &&
|
|
// !isContextMenuOpen &&
|
|
// !ContainsMenu)
|
|
// {
|
|
// BackColor = MenuDefines.File;
|
|
// }
|
|
//}
|
|
|
|
//MouseEnter += Menubutton_MouseEnter;
|
|
//void Menubutton_MouseEnter(object sender, EventArgs e)
|
|
//{
|
|
// if (Tag == null &&
|
|
// !ContainsMenu)
|
|
// {
|
|
// BackColor = MenuDefines.FileHover;
|
|
// }
|
|
//}
|
|
|
|
//BackColorChanged += MenuButton_BackColorChanged;
|
|
//void MenuButton_BackColorChanged(object sender, EventArgs e)
|
|
//{
|
|
// FlatAppearance.BorderColor = BackColor;
|
|
//}
|
|
|
|
|
|
}
|
|
}
|
|
|
|
//DoubleClick += MenuButton_DoubleClick;
|
|
public void DoubleClick()
|
|
{
|
|
if (ContainsMenu)
|
|
{
|
|
try
|
|
{
|
|
Process.Start("explorer.exe", TargetFilePath);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Log.Error($"TargetFilePath:'{TargetFilePath}', " +
|
|
$"=>DirectoryNotFound?", ex);
|
|
ex = new DirectoryNotFoundException();
|
|
MessageBox.Show(ex.Message);
|
|
}
|
|
}
|
|
}
|
|
|
|
//MouseDown += MenuButton_MouseDown;
|
|
public void MouseDown(DataGridView dgv, MouseEventArgs e)
|
|
{
|
|
if (ContainsMenu)
|
|
{
|
|
TriggerMenuOpener(); // Touchscreen
|
|
}
|
|
|
|
if (IsLoading)
|
|
{
|
|
waitMenuOpen.Click();
|
|
}
|
|
|
|
if (e == null ||
|
|
e.Button == MouseButtons.Left &&
|
|
!ContainsMenu)
|
|
{
|
|
try
|
|
{
|
|
//https://stackoverflow.com/questions/31627801/
|
|
Process p = new Process();
|
|
p.StartInfo = new ProcessStartInfo(TargetFilePath);
|
|
p.StartInfo.Arguments = Arguments;
|
|
p.StartInfo.WorkingDirectory = WorkingDirectory;
|
|
p.StartInfo.CreateNoWindow = true;
|
|
p.Start();
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Log.Error($"TargetFilePath:'{ TargetFilePath}', " +
|
|
$"=>FileNotFound?", ex);
|
|
MessageBox.Show(ex.Message);
|
|
}
|
|
}
|
|
|
|
if (e != null &&
|
|
e.Button == MouseButtons.Right &&
|
|
FileInfo != null &&
|
|
dgv.Rows.Count > RowIndex)
|
|
{
|
|
IsContextMenuOpen = true;
|
|
|
|
#warning CodeBuity&Refactor #49 is there any other possiblity to raise selection changed event? dataGridView.ClearSelection(); seems to overwrite selected
|
|
IsSelected = true;
|
|
dgv.Rows[RowIndex].Selected = true;
|
|
|
|
try
|
|
{
|
|
ShellContextMenu ctxMnu = new ShellContextMenu();
|
|
Point location = dgv.FindForm().Location;
|
|
Point point = new Point(
|
|
e.X + location.X + dgv.Location.X,
|
|
e.Y + location.Y + dgv.Location.Y);
|
|
if (ContainsMenu)
|
|
{
|
|
DirectoryInfo[] dir = new DirectoryInfo[1];
|
|
dir[0] = new DirectoryInfo(TargetFilePathOrig);
|
|
ctxMnu.ShowContextMenu(dir, point);
|
|
}
|
|
else
|
|
{
|
|
FileInfo[] arrFI = new FileInfo[1];
|
|
arrFI[0] = new FileInfo(TargetFilePathOrig);
|
|
ctxMnu.ShowContextMenu(arrFI, point);
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
MessageBox.Show(ex.ToString() + TargetFilePath);
|
|
}
|
|
|
|
if (!dgv.IsDisposed)
|
|
{
|
|
IsSelected = false;
|
|
dgv.Rows[RowIndex].Selected = false;
|
|
}
|
|
|
|
IsContextMenuOpen = false;
|
|
}
|
|
}
|
|
|
|
public void MenuLoaded()
|
|
{
|
|
waitMenuOpen.MenuLoaded();
|
|
}
|
|
|
|
public void StartMenuOpener()
|
|
{
|
|
if (ContainsMenu)
|
|
{
|
|
IsLoading = true;
|
|
waitMenuOpen.Start();
|
|
}
|
|
}
|
|
|
|
private void TriggerMenuOpener()
|
|
{
|
|
if (ContainsMenu && IsLoading)
|
|
{
|
|
waitMenuOpen.Start();
|
|
}
|
|
}
|
|
|
|
public void StopLoadMenuAndStartWaitToOpenIt()
|
|
{
|
|
if (ContainsMenu)
|
|
{
|
|
waitMenuOpen.Stop();
|
|
//IsLoading = false;
|
|
}
|
|
}
|
|
|
|
private void WaitMenuOpen_DoOpen()
|
|
{
|
|
IsLoading = false;
|
|
OpenMenu?.Invoke(this, null);
|
|
}
|
|
|
|
public void SetData(RowData data, DataGridView dgv)
|
|
{
|
|
data.RowIndex = dgv.Rows.Add();
|
|
DataGridViewRow row = dgv.Rows[data.RowIndex];
|
|
|
|
if (Icon == null)
|
|
{
|
|
Icon = Properties.Resources.SystemTrayMenu;
|
|
}
|
|
DataGridViewImageCell cellIcon =
|
|
(DataGridViewImageCell)row.Cells[0];
|
|
|
|
if (HiddenEntry)
|
|
{
|
|
cellIcon.Value = AddIconOverlay(data.Icon, Properties.Resources.WhiteTransparency);
|
|
}
|
|
else
|
|
{
|
|
cellIcon.Value = data.Icon;
|
|
}
|
|
|
|
DataGridViewTextBoxCell cellName =
|
|
(DataGridViewTextBoxCell)row.Cells[1];
|
|
cellName.Value = data.Text;
|
|
|
|
row.Tag = data;
|
|
}
|
|
|
|
#warning CodeBuity&Refactor #49 either not public and as inline method or we want probably to move that code somewhere else
|
|
public Icon AddIconOverlay(Icon originalIcon, Icon overlay)
|
|
{
|
|
var target = new Bitmap(originalIcon.Width, originalIcon.Height, PixelFormat.Format32bppArgb);
|
|
var graphics = Graphics.FromImage(target);
|
|
graphics.DrawIcon(originalIcon, 0, 0);
|
|
graphics.DrawIcon(overlay, 0, 0);
|
|
target.MakeTransparent(target.GetPixel(1, 1));
|
|
return Icon.FromHandle(target.GetHicon());
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
if (!disposed)
|
|
{
|
|
icon?.Dispose();
|
|
waitMenuOpen.Dispose();
|
|
}
|
|
disposed = true;
|
|
}
|
|
}
|
|
}
|