mirror of
https://github.com/fanchenio/DawnLauncher.git
synced 2025-07-13 21:02:10 +08:00
优化右键菜单'以管理员身份运行'功能,改为右键时动态判断文件是否有此权限。
This commit is contained in:
parent
f365005bd8
commit
2ba01403a5
@ -195,8 +195,10 @@ export default function () {
|
|||||||
// "打开"菜单
|
// "打开"菜单
|
||||||
let openMenu = false;
|
let openMenu = false;
|
||||||
if (
|
if (
|
||||||
(item.type === 0 && ext && (ext === "exe" || ext === "bat")) ||
|
(item.type === 0 &&
|
||||||
item.type === 4
|
global.addon.hasRunas(parsePath(item.data.target))) ||
|
||||||
|
item.type === 4 ||
|
||||||
|
(item.type === 3 && item.data.target === "cmd.exe")
|
||||||
) {
|
) {
|
||||||
menuList.push(
|
menuList.push(
|
||||||
new MenuItem({
|
new MenuItem({
|
||||||
|
@ -201,3 +201,12 @@ fn shell_execute(operation: String, file: String, params: String, start_location
|
|||||||
fn system_item_execute(target: String, params: Option<String>) {
|
fn system_item_execute(target: String, params: Option<String>) {
|
||||||
windows::system_item_execute(&target, params.as_deref())
|
windows::system_item_execute(&target, params.as_deref())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 判断文件是否有以管理员身份运行权限
|
||||||
|
*/
|
||||||
|
#[allow(dead_code)]
|
||||||
|
#[napi]
|
||||||
|
fn has_runas(path: String) -> bool {
|
||||||
|
windows::has_runas(&path)
|
||||||
|
}
|
||||||
|
@ -15,7 +15,7 @@ use std::{
|
|||||||
sync::atomic::{AtomicBool, Ordering},
|
sync::atomic::{AtomicBool, Ordering},
|
||||||
};
|
};
|
||||||
use windows::{
|
use windows::{
|
||||||
core::{ComInterface, HSTRING, PCSTR, PCWSTR},
|
core::{ComInterface, HSTRING, PCSTR, PCWSTR, PSTR},
|
||||||
w,
|
w,
|
||||||
Win32::{
|
Win32::{
|
||||||
Foundation::{HWND, LPARAM, LRESULT, MAX_PATH, POINT, RECT, SIZE, WPARAM},
|
Foundation::{HWND, LPARAM, LRESULT, MAX_PATH, POINT, RECT, SIZE, WPARAM},
|
||||||
@ -42,14 +42,15 @@ use windows::{
|
|||||||
Shell::{
|
Shell::{
|
||||||
BHID_SFUIObject, IContextMenu, IShellItem, IShellItemImageFactory, IShellLinkW,
|
BHID_SFUIObject, IContextMenu, IShellItem, IShellItemImageFactory, IShellLinkW,
|
||||||
SHCreateItemFromParsingName, SHEmptyRecycleBinW, SHQueryUserNotificationState,
|
SHCreateItemFromParsingName, SHEmptyRecycleBinW, SHQueryUserNotificationState,
|
||||||
ShellLink, CMF_NORMAL, CMINVOKECOMMANDINFO, QUNS_ACCEPTS_NOTIFICATIONS, QUNS_APP,
|
ShellLink, CMF_NORMAL, CMINVOKECOMMANDINFO, GCS_VERBA, QUNS_ACCEPTS_NOTIFICATIONS,
|
||||||
QUNS_BUSY, QUNS_NOT_PRESENT, QUNS_PRESENTATION_MODE, QUNS_QUIET_TIME,
|
QUNS_APP, QUNS_BUSY, QUNS_NOT_PRESENT, QUNS_PRESENTATION_MODE, QUNS_QUIET_TIME,
|
||||||
QUNS_RUNNING_D3D_FULL_SCREEN, SHERB_NOSOUND, SIIGBF_ICONONLY, SLGP_UNCPRIORITY,
|
QUNS_RUNNING_D3D_FULL_SCREEN, SHERB_NOSOUND, SIIGBF_ICONONLY, SLGP_UNCPRIORITY,
|
||||||
},
|
},
|
||||||
WindowsAndMessaging::{
|
WindowsAndMessaging::{
|
||||||
CallNextHookEx, CreatePopupMenu, DestroyMenu, FindWindowW, GetClassNameW,
|
CallNextHookEx, CreatePopupMenu, DestroyMenu, FindWindowW, GetClassNameW,
|
||||||
GetCursorPos, GetForegroundWindow, GetSystemMetrics, GetWindowRect, SendMessageW,
|
GetCursorPos, GetForegroundWindow, GetMenuItemCount, GetMenuItemInfoA,
|
||||||
SetForegroundWindow, SetWindowsHookExW, TrackPopupMenu, WindowFromPoint, HHOOK,
|
GetSystemMetrics, GetWindowRect, SendMessageW, SetForegroundWindow,
|
||||||
|
SetWindowsHookExW, TrackPopupMenu, WindowFromPoint, HHOOK, MENUITEMINFOA, MIIM_ID,
|
||||||
MSLLHOOKSTRUCT, SC_MONITORPOWER, SM_CXSCREEN, SM_CYSCREEN, SW_NORMAL, TPM_NONOTIFY,
|
MSLLHOOKSTRUCT, SC_MONITORPOWER, SM_CXSCREEN, SM_CYSCREEN, SW_NORMAL, TPM_NONOTIFY,
|
||||||
TPM_RETURNCMD, WH_MOUSE_LL, WM_LBUTTONDOWN, WM_LBUTTONUP, WM_MBUTTONDOWN,
|
TPM_RETURNCMD, WH_MOUSE_LL, WM_LBUTTONDOWN, WM_LBUTTONUP, WM_MBUTTONDOWN,
|
||||||
WM_MBUTTONUP, WM_MOUSEHWHEEL, WM_MOUSEMOVE, WM_MOUSEWHEEL, WM_RBUTTONDOWN,
|
WM_MBUTTONUP, WM_MOUSEHWHEEL, WM_MOUSEMOVE, WM_MOUSEWHEEL, WM_RBUTTONDOWN,
|
||||||
@ -788,3 +789,68 @@ pub fn system_item_execute(target: &str, params: Option<&str>) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 判断文件是否有以管理员身份运行权限
|
||||||
|
*/
|
||||||
|
pub fn has_runas(path: &str) -> bool {
|
||||||
|
// IShellItem
|
||||||
|
let path = HSTRING::from(path);
|
||||||
|
if let Ok(shell_item) =
|
||||||
|
unsafe { SHCreateItemFromParsingName::<_, _, IShellItem>(PCWSTR(path.as_ptr()), None) }
|
||||||
|
{
|
||||||
|
// IContextMenu
|
||||||
|
if let Ok(context_menu) =
|
||||||
|
unsafe { shell_item.BindToHandler::<_, IContextMenu>(None, &BHID_SFUIObject) }
|
||||||
|
{
|
||||||
|
// Menu
|
||||||
|
if let Ok(menu) = unsafe { CreatePopupMenu() } {
|
||||||
|
// 写入菜单
|
||||||
|
if let Ok(()) =
|
||||||
|
unsafe { context_menu.QueryContextMenu(menu, 0, 1, 0x7FFF, CMF_NORMAL) }
|
||||||
|
{
|
||||||
|
// 获取菜单总数
|
||||||
|
let count = unsafe { GetMenuItemCount(menu) };
|
||||||
|
// 循环每一项菜单
|
||||||
|
for i in 0..count {
|
||||||
|
// 获取菜单信息
|
||||||
|
let mut menu_info = MENUITEMINFOA::default();
|
||||||
|
menu_info.cbSize = std::mem::size_of::<MENUITEMINFOA>() as u32;
|
||||||
|
menu_info.fMask = MIIM_ID;
|
||||||
|
if unsafe { GetMenuItemInfoA(menu, i as u32, true, &mut menu_info) }
|
||||||
|
.as_bool()
|
||||||
|
{
|
||||||
|
// idcmd
|
||||||
|
let idcmd = menu_info.wID as usize - 1;
|
||||||
|
// 菜单项比如在指定范围内
|
||||||
|
if idcmd >= 1 && idcmd <= 0x7FFF {
|
||||||
|
// 获取菜单命令信息
|
||||||
|
let mut verb = [0u8; 256];
|
||||||
|
let pstr = PSTR(verb.as_mut_ptr());
|
||||||
|
if let Ok(()) = unsafe {
|
||||||
|
context_menu.GetCommandString(
|
||||||
|
menu_info.wID as usize - 1,
|
||||||
|
GCS_VERBA,
|
||||||
|
None,
|
||||||
|
pstr,
|
||||||
|
verb.len() as u32,
|
||||||
|
)
|
||||||
|
} {
|
||||||
|
if let Ok(command) = unsafe { pstr.to_string() } {
|
||||||
|
if command.to_lowercase() == "runas" {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
unsafe {
|
||||||
|
DestroyMenu(menu);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
false
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user