Context menu work

Author: Decimation

SmartImage.Lib 3/Resources.Designer.cs

@@ -193,6 +193,12 @@
     <value>/usr/share/applications</value>
   </data>
   <data name="Linux_Desktop_File" xml:space="preserve">
-    <value>SmartImage.desktop</value>
+    <value>smartimage.desktop</value>
+  </data>
+  <data name="Linux_Launch_Args" xml:space="preserve">
+    <value>%u</value>
+  </data>
+  <data name="Reg_Launch_Args" xml:space="preserve">
+    <value>"%1"</value>
   </data>
 </root>

SmartImage.Lib 3/Resources.resx

@@ -2,7 +2,7 @@
 // Date: 2024/05/22 @ 15:05:58
 
 #nullable disable
-global using USI=JetBrains.Annotations.UsedImplicitlyAttribute;
+global using USI = JetBrains.Annotations.UsedImplicitlyAttribute;
 using System.ComponentModel.DataAnnotations;
 using System.Diagnostics;
 using System.Reflection;
@@ -34,10 +34,6 @@ public static class AppUtil
 	[SupportedOSPlatformGuard(OS_WIN)]
 	internal static readonly bool IsWindows = OperatingSystem.IsWindows();
 
-	public static readonly Assembly Assembly = Assembly.GetExecutingAssembly();
-
-	public static readonly Version Version = Assembly.GetName().Version;
-
 	[CBN]
 	internal static string GetOSName()
 	{
@@ -122,21 +118,28 @@ public static bool IsContextMenuAdded
 	}
 
 	/// <returns><c>true</c> if operation succeeded; <c>false</c> otherwise</returns>
-	public static bool HandleContextMenu(bool option)
+	public static bool? HandleContextMenu(bool option, [CanBeNull] string args = null)
 	{
 		if (IsWindows) {
-			HandleContextMenuWindows(option);
+			return HandleContextMenuWindows(option, args);
 		}
 		else if (IsLinux) {
-			HandleContextMenuLinux(option);
+			return HandleContextMenuLinux(option, args);
 		}
 
-		throw new InvalidOperationException();
+		return null;
 	}
 
 	[SupportedOSPlatform(OS_LINUX)]
-	public static bool HandleContextMenuLinux(bool option)
+	internal static bool HandleContextMenuLinux(bool option, [CanBeNull] string args = null)
 	{
+		if (!FileSystem.IsRoot) {
+			throw new SmartImageException("Root permissions required");
+
+		}
+
+		args ??= R1.Linux_Launch_Args;
+
 		if (option) {
 			string dsk = $"""
 			              [Desktop Entry]
@@ -145,14 +148,15 @@ [Desktop Entry]
 			              Version=1.0
 			              Name=SmartImage
 			              Terminal=true
-			              Exec={ExeLocation} %u
+			              Exec={ExeLocation} {args}
 			              """;
 			File.WriteAllText(LinuxDesktopFile, dsk);
 
 		}
 		else {
-			
-			File.Delete(LinuxDesktopFile);
+			if (File.Exists(LinuxDesktopFile)) {
+				File.Delete(LinuxDesktopFile);
+			}
 		}
 
 		// Console.WriteLine(Path.GetFullPath(s));
@@ -164,13 +168,16 @@ [Desktop Entry]
 	}
 
 	[SupportedOSPlatform(OS_WIN)]
-	public static bool HandleContextMenuWindows(bool option)
+	internal static bool HandleContextMenuWindows(bool option,[CanBeNull] string args = null)
 	{
 		/*
 		 * New context menu
 		 */
+		bool ok = false;
 		switch (option) {
 			case true:
+				
+				args ??= R1.Reg_Launch_Args;
 
 				RegistryKey regMenu = null;
 				RegistryKey regCmd  = null;
@@ -184,12 +191,15 @@ public static bool HandleContextMenuWindows(bool option)
 
 					regCmd = Registry.CurrentUser.CreateSubKey(R1.Reg_Shell_Cmd);
 
-					regCmd?.SetValue(String.Empty,
-					                 $"\"{fullPath}\" -i \"%1\" -auto -s");
+					regCmd?.SetValue(String.Empty, $"\"{fullPath}\" {args}");
+					// regCmd?.SetValue(String.Empty, $"\"{fullPath}\" \"%1\"");
+					// regCmd?.SetValue(String.Empty, $"\"{fullPath}\" -i \"%1\" -auto -s");
+					ok = true;
 				}
 				catch (Exception ex) {
 					Trace.WriteLine($"{ex.Message}");
-					return false;
+					// return false;
+					ok = false;
 				}
 				finally {
 					regMenu?.Close();
@@ -214,18 +224,23 @@ public static bool HandleContextMenuWindows(bool option)
 						reg.Close();
 						Registry.CurrentUser.DeleteSubKey(R1.Reg_Shell);
 					}
+
+					// return true;
+					ok = true;
+					break;
 				}
 				catch (Exception ex) {
 					Trace.WriteLine($"{ex.Message}");
-
-					return false;
+					ok = false;
+					// return false;
+					break;
 				}
 
 				break;
 
 		}
 
-		return false;
+		return ok;
 
 	}
 
@@ -274,6 +289,7 @@ public static async Task<GHRelease> GetLatestReleaseAsync()
 
 		return r.OrderByDescending(x => x.published_at).First();
 	}
+
 }
 
 [USI(ImplicitUseTargetFlags.WithMembers)]

SmartImage.Lib 3/Utilities/AppUtil.cs

@@ -0,0 +1,31 @@
+// Author: Deci | Project: SmartImage.Rdx | Name: IntegrationCommand.cs
+// Date: 2024/05/22 @ 16:05:51
+
+using SmartImage.Lib.Utilities;
+using Spectre.Console.Cli;
+
+namespace SmartImage.Rdx;
+
+public class IntegrationCommand : Command<IntegrationCommandSettings>
+{
+
+	public override int Execute(CommandContext context, IntegrationCommandSettings settings)
+	{
+		try {
+			// AConsole.WriteLine($"{AppUtil.IsContextMenuAdded}");
+
+			if (settings.ContextMenu.HasValue) {
+				var rv = AppUtil.HandleContextMenu(settings.ContextMenu.Value);
+				AConsole.WriteLine($"Context menu change: {rv}");
+			}
+
+			AConsole.WriteLine($"Context menu enabled: {AppUtil.IsContextMenuAdded}");
+		}
+		catch (Exception e) {
+			AConsole.WriteException(e);
+		}
+
+		return SearchCommand.EC_OK;
+	}
+
+}

SmartImage.Rdx/IntegrationCommand.cs

@@ -0,0 +1,14 @@
+// Author: Deci | Project: SmartImage.Rdx | Name: IntegrationCommandSettings.cs
+// Date: 2024/05/22 @ 16:05:47
+
+using Spectre.Console.Cli;
+
+namespace SmartImage.Rdx;
+
+public class IntegrationCommandSettings : CommandSettings
+{
+
+	[CommandOption("--ctx-menu")]
+	public bool? ContextMenu { get; internal set; }
+
+}

SmartImage.Rdx/IntegrationCommandSettings.cs

@@ -9,7 +9,6 @@
 using SixLabors.ImageSharp;
 using SixLabors.ImageSharp.Formats;
 using SmartImage.Lib;
-using SmartImage.Lib.Utilities;
 using Spectre.Console;
 using Spectre.Console.Cli;
 using SmartImage.Rdx.Shell;
@@ -84,7 +83,8 @@ public static async Task<int> Main(string[] args)
 			c.PropagateExceptions();
 			var helpProvider = new CustomHelpProvider(c.Settings);
 			c.SetHelpProvider(helpProvider);
-			c.AddCommand<IntegrationCommand>("integrate");
+			c.AddCommand<IntegrationCommand>("integrate")
+				.WithDescription("Configure system integration such as context menu");
 		});
 
 		try {
@@ -102,38 +102,4 @@ public static async Task<int> Main(string[] args)
 		}
 	}
 
-}
-
-public class IntegrationCommandSettings : CommandSettings
-{
-
-	[CommandOption("--ctx-menu")]
-	public bool? ContextMenu { get; internal set; }
-
-}
-
-public class IntegrationCommand : Command<IntegrationCommandSettings>
-{
-
-	public void HandleContextMenu(bool option)
-	{
-
-	}
-	public override int Execute(CommandContext context, IntegrationCommandSettings settings)
-	{
-		try {
-			AConsole.WriteLine($"{AppUtil.IsContextMenuAdded}");
-
-			if (settings.ContextMenu.HasValue) {
-				AppUtil.HandleContextMenu(settings.ContextMenu.Value);
-
-			}
-		}
-		catch (Exception e) {
-			AConsole.WriteException(e);
-		}
-
-		return SearchCommand.EC_OK;
-	}
-
 }