From 05253e8d03613bd86825d8d169edea9059ae86d2 Mon Sep 17 00:00:00 2001
From: Zichuan Li <34680029+river-li@users.noreply.github.com>
Date: Thu, 18 Jul 2024 14:33:58 -0400
Subject: [PATCH] feat: register action for users to define GUID types

registered a new action in plugin context menu, users could query the
GUID database and define EFI_GUID with names now
---
 __init__.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 4284b83..e6b3d83 100644
--- a/__init__.py
+++ b/__init__.py
@@ -10,7 +10,8 @@
     define_mm_locate_protocol_types,
     define_smm_locate_protocol_types,
     define_mm_handle_protocol_types,
-    define_smm_handle_protocol_types
+    define_smm_handle_protocol_types,
+    lookup_and_define_guid
 )
 
 from .system_table import propagate_function_param_types, set_windows_bootloader_type
@@ -110,4 +111,5 @@ def run(self):
     Task(bv).start()
 
 
-PluginCommand.register("Resolve EFI Protocols", "Automatically resolve usage of EFI protocols", resolve_efi)
+PluginCommand.register("EFI Resolver\Resolve EFI Protocols", "Automatically resolve usage of EFI protocols", resolve_efi)
+PluginCommand.register_for_address("EFI Resolver\Define GUID", "Lookup the database and define GUID at the given address", lookup_and_define_guid)