From 1fbe130c9a6e16f5c3d2f716e94f18f03f90a454 Mon Sep 17 00:00:00 2001
From: jurialmunkey <jurialmunkey@kodi.tv>
Date: Mon, 20 Nov 2023 00:58:29 +1100
Subject: [PATCH] :sparkles: Add widget adder

---
 resources/lib/method.py | 34 +++++++++++++++++++---------------
 1 file changed, 19 insertions(+), 15 deletions(-)

diff --git a/resources/lib/method.py b/resources/lib/method.py
index 3865ff1..8823a5d 100644
--- a/resources/lib/method.py
+++ b/resources/lib/method.py
@@ -53,22 +53,26 @@ def get_target():
     if x == -1:
         return
     menu = menunodes[x]
-    item = None
-    mode = None
-    guid = None
-
     from resources.lib.shortcuts.node import ListGetShortcutsNode
-    lgsn = ListGetShortcutsNode(-1, '')
-    lgsn.get_directory(menu=menu, skin=skin, item=item, mode=mode, guid=guid, func='node')
-    x = xbmcgui.Dialog().select('Choose menu', [parse_localize(i.get('label') or '') for i in lgsn.menunode] + ['Add here...'], useDetails=True)
-    if x == -1:
-        return
-    if x == len(lgsn.menunode):
-        lgsn.menunode.append(new_item)
-        lgsn.write_meta_to_file()
-        lgsn.do_refresh()
-        return
-    return
+
+    available_modes = ['submenu', 'widgets']
+
+    def get_menu_node(mode, guid):
+        lgsn = ListGetShortcutsNode(-1, '')
+        lgsn.get_directory(menu=menu, skin=skin, item=None, mode=mode, guid=guid, func='node')
+        x = xbmcgui.Dialog().select('Choose menu', [parse_localize(i.get('label') or '') for i in lgsn.menunode] + ['Add here...'])
+        if x == -1:
+            return
+        if x == len(lgsn.menunode):
+            lgsn.menunode.append(new_item)
+            lgsn.write_meta_to_file()
+            lgsn.do_refresh()
+            return
+        y = xbmcgui.Dialog().select('Choose menu', available_modes)
+        if y == -1:
+            return get_menu_node(mode, guid)
+        return get_menu_node(available_modes[y], lgsn.menunode[x].get('guid'))
+    return get_menu_node('submenu', None)
 
 
 def set_shortcut(set_shortcut):