feat: Basic dialog autosize

in03 · in03 · commit 2c81b7e200e1 · 2022-12-15T10:38:31.000+10:00
diff --git a/src/patchwork/widgets/dialog_box.py b/src/patchwork/widgets/dialog_box.py
@@ -5,39 +5,47 @@
 
 def dialog_selection_callback(sender, unused, user_data) -> bool:
     if user_data[1]:
-        logger.info("User chose OK")
+        logger.debug("[magenta]User chose OK")
         dpg.delete_item(user_data[0])
         return True
     else:
-        logger.info("User chose Cancel")
+        logger.debug("[magenta]User chose Cancel")
         dpg.delete_item(user_data[0])
         return False
                    
-def prompt(message:str, title:str="Dialog", show_ok:bool=True, show_cancel:bool=False, wrap:int=400, **kwargs) -> bool|None:
+def prompt(message:str, title:str="Dialog", show_ok:bool=True, show_cancel:bool=False, width:int|None=None, height:int|None=None, wrap:int=400, **kwargs) -> bool|None:
     if dpg.does_item_exist("dialog"):
         dpg.delete_item("dialog")
+
+    viewport_width = dpg.get_viewport_client_width()
+    viewport_height = dpg.get_viewport_client_height()
     
-    with dpg.mutex():
     
-        viewport_width = dpg.get_viewport_client_width()
-        viewport_height = dpg.get_viewport_client_height()
+    # Calculate height
+    if not height:
+        height = 0
+        char_height_factor = 15
+        extra_for_buttons = 100
         
-        with dpg.window(label=title, modal=True, no_title_bar=True, tag="dialog", **kwargs):
-            dpg.add_text(message, wrap=wrap)
-            dpg.add_separator()
-            with dpg.group(horizontal=True, tag="dialog_buttons"):
-                
-                if show_ok:      
-                    dpg.add_button(label="Ok", user_data=("dialog", True), callback = dialog_selection_callback)
-                if show_cancel:
-                    dpg.add_button(label="Cancel", user_data=("dialog", False), callback = dialog_selection_callback)
-                
-    dpg.split_frame()
+        explicit_lines = message.split("\n")
+        height += (len(explicit_lines) * char_height_factor)
+        for line in explicit_lines:
+            if len(line) > wrap:
+                height += char_height_factor
+            
+        height+=extra_for_buttons
+        
+    if not width:
+        width = wrap
     
-    # CENTER DIALOG
-    width = dpg.get_item_width("dialog")
-    height = dpg.get_item_height("dialog")
-    if width and height:
-        dpg.set_item_pos("dialog", [viewport_width // 2 - width // 2, viewport_height // 2 - height // 2])
-        dpg.configure_item("dialog", show=True)
+    with dpg.window(label=title, modal=True, no_title_bar=True, tag="dialog", width=width, height=height, pos=[viewport_width // 2 - wrap // 2, viewport_height // 2 - height // 2], **kwargs):
+        dpg.add_text(message, wrap=wrap)
+        dpg.add_separator()
+        with dpg.group(horizontal=True, tag="dialog_buttons"):
+            
+            if show_ok:      
+                dpg.add_button(label="Ok", user_data=("dialog", True), callback = dialog_selection_callback)
+            if show_cancel:
+                dpg.add_button(label="Cancel", user_data=("dialog", False), callback = dialog_selection_callback)
+                
         
