From f9dd1c3389238c2a7fc5a620f7f59915cf62ac8c Mon Sep 17 00:00:00 2001
From: Alexander Yakushev <yakushev.alex@gmail.com>
Date: Thu, 1 Mar 2012 19:33:22 +0200
Subject: [PATCH] awful.prompt: Introduce changed_callback and
 keypressed_callback to prompt.run

changed_callback allows to execute arbitrary code any time the
command string changes.
keypressed_callback allows to intercept keypresses before
awful.prompt.run code handles them and run arbitrary code depending on
the key pressed and modificators.

Signed-off-by: Uli Schlachter <psychon@znc.in>
---
 lib/awful/prompt.lua.in | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

diff --git a/lib/awful/prompt.lua.in b/lib/awful/prompt.lua.in
index 5d294ae5..31865e89 100644
--- a/lib/awful/prompt.lua.in
+++ b/lib/awful/prompt.lua.in
@@ -163,7 +163,9 @@ end
 -- @param history_path Optional parameter: file path where the history should be saved, set nil to disable history
 -- @param history_max Optional parameter: set the maximum entries in history file, 50 by default
 -- @param done_callback Optional parameter: the callback function to always call without arguments, regardless of whether the prompt was cancelled.
-function run(args, textbox, exe_callback, completion_callback, history_path, history_max, done_callback)
+-- @param changed_callback Optional parameter: the callback function to call with command as argument when a command was changed.
+-- @param keypressed_callback Optional parameter: the callback function to call with mod table, key and command as arguments when a key was pressed.
+function run(args, textbox, exe_callback, completion_callback, history_path, history_max, done_callback, changed_callback, keypressed_callback)
     local theme = beautiful.get()
     if not args then args = {} end
     local command = args.text or ""
@@ -217,6 +219,31 @@ function run(args, textbox, exe_callback, completion_callback, history_path, his
         -- Convert index array to hash table
         local mod = {}
         for k, v in ipairs(modifiers) do mod[v] = true end
+
+        -- Call the user specified callback. If it returns true as
+        -- the first result then return from the function. Treat the
+        -- second and third results as a new command and new prompt
+        -- to be set (if provided)
+        if keypressed_callback then
+            local user_catched, new_command, new_prompt =
+                keypressed_callback(mod, key, command)
+            if new_command or new_prompt then
+                if new_command then
+                    command = new_command
+                end
+                if new_prompt then
+                    prettyprompt = new_prompt
+                end
+                update()
+            end
+            if user_catched then
+                if changed_callback then
+                    changed_callback(command)
+                end
+                return
+            end
+        end
+
         -- Get out cases
         if (mod.Control and (key == "c" or key == "g"))
             or (not mod.Control and key == "Escape") then
@@ -430,6 +457,10 @@ function run(args, textbox, exe_callback, completion_callback, history_path, his
             cur_pos = cur_pos - 1
             success = pcall(update)
         end
+
+        if changed_callback then
+            changed_callback(command)
+        end
     end)
 end