Add hooks sub system

Author: Your Name

README.md

@@ -21,6 +21,7 @@ LLMs are a part of our lives from here on out so join us in learning about and c
 * [TUI Configuration](https://github.com/dwash96/cecli/blob/main/cecli/website/docs/config/tui.md)
 * [Skills](https://github.com/dwash96/cecli/blob/main/cecli/website/docs/config/skills.md)
 * [Session Management](https://github.com/dwash96/cecli/blob/main/cecli/website/docs/sessions.md)
+* [Hooks](https://github.com/dwash96/cecli/blob/main/cecli/website/docs/config/hooks.md)
 * [Custom Commands](https://github.com/dwash96/cecli/blob/main/cecli/website/docs/config/custom-commands.md)
 * [Custom System Prompts](https://github.com/dwash96/cecli/blob/main/cecli/website/docs/config/custom-system-prompts.md)
 * [Custom Tools](https://github.com/dwash96/cecli/blob/main/cecli/website/docs/config/agent-mode.md#creating-custom-tools)

cecli/args.py

@@ -310,6 +310,12 @@ def get_parser(default_config_files, git_root):
         help="Specify Agent Mode configuration as a JSON string",
         default=None,
     )
+    group.add_argument(
+        "--hooks",
+        metavar="HOOKS_CONFIG_JSON",
+        help="Specify hooks configuration as a JSON string",
+        default=None,
+    )
     group.add_argument(
         "--agent-model",
         metavar="AGENT_MODEL",

cecli/coders/agent_coder.py

@@ -26,6 +26,7 @@
     normalize_vector,
 )
 from cecli.helpers.skills import SkillsManager
+from cecli.hooks import HookIntegration
 from cecli.llm import litellm
 from cecli.mcp import LocalServer, McpServerManager
 from cecli.tools.utils.registry import ToolRegistry
@@ -241,6 +242,17 @@ async def _execute_local_tool_calls(self, tool_calls_list):
             try:
                 args_string = tool_call.function.arguments.strip()
                 parsed_args_list = []
+
+                if not await HookIntegration.call_pre_tool_hooks(self, tool_name, args_string):
+                    tool_responses.append(
+                        {
+                            "role": "tool",
+                            "tool_call_id": tool_call.id,
+                            "content": "Tool Request Aborted.",
+                        }
+                    )
+                    continue
+
                 if args_string:
                     json_chunks = utils.split_concatenated_json(args_string)
                     for chunk in json_chunks:
@@ -293,6 +305,19 @@ async def _execute_local_tool_calls(self, tool_calls_list):
                 if tasks:
                     task_results = await asyncio.gather(*tasks)
                     all_results_content.extend(str(res) for res in task_results)
+
+                if not await HookIntegration.call_post_tool_hooks(
+                    self, tool_name, args_string, "\n\n".join(all_results_content)
+                ):
+                    tool_responses.append(
+                        {
+                            "role": "tool",
+                            "tool_call_id": tool_call.id,
+                            "content": "Tool Response Redacted.",
+                        }
+                    )
+                    continue
+
                 result_message = "\n\n".join(all_results_content)
             except Exception as e:
                 result_message = f"Error executing {tool_name}: {e}"

cecli/coders/base_coder.py

@@ -48,6 +48,7 @@
 )
 from cecli.helpers.profiler import TokenProfiler
 from cecli.history import ChatSummary
+from cecli.hooks import HookIntegration
 from cecli.io import ConfirmGroup, InputOutput
 from cecli.linter import Linter
 from cecli.llm import litellm
@@ -1590,6 +1591,10 @@ async def preproc_user_input(self, inp):
     async def run_one(self, user_message, preproc):
         self.init_before_message()
 
+        if not await HookIntegration.call_start_hooks(self):
+            self.io.tool_warning("Execution stopped by start hook")
+            return
+
         if preproc:
             message = await self.preproc_user_input(user_message)
         else:
@@ -1637,6 +1642,10 @@ async def run_one(self, user_message, preproc):
 
             await self.auto_save_session(force=True)
 
+        if not await HookIntegration.call_end_hooks(self):
+            self.io.tool_warning("Execution stopped by end hook")
+            return
+
     def _is_url_allowed(self, url):
         allowed_domains = self.security_config.get("allowed-domains")
         if not allowed_domains:
@@ -2274,7 +2283,7 @@ async def send_message(self, inp):
 
         self.io.tool_output()
         self.show_usage_report()
-        self.add_assistant_reply_to_cur_messages()
+        await self.add_assistant_reply_to_cur_messages()
 
         if exhausted:
             cur_messages = ConversationManager.get_messages_dict(MessageTag.CUR)
@@ -2596,6 +2605,13 @@ async def _exec_server_tools(server, tool_calls_list):
                             new_tool_call = copy_tool_call(tool_call)
                             new_tool_call.function.arguments = json.dumps(args)
 
+                            if not await HookIntegration.call_pre_tool_hooks(
+                                self, new_tool_call.function.name, args
+                            ):
+                                self.io.tool_warning("Tool call skipped by pre-tool call hook")
+                                all_results_content.append("Tool Request Aborted.")
+                                continue
+
                             call_result = await experimental_mcp_client.call_openai_tool(
                                 session=session,
                                 openai_tool=new_tool_call,
@@ -2628,6 +2644,16 @@ async def _exec_server_tools(server, tool_calls_list):
                                         content_parts.append(item.text)
 
                             result_text = "".join(content_parts)
+
+                            if not await HookIntegration.call_post_tool_hooks(
+                                self, new_tool_call.function.name, args, result_text
+                            ):
+                                self.io.tool_warning(
+                                    "Tool call output skipped by post-tool call hook"
+                                )
+                                all_results_content.append("Tool Response Redacted.")
+                                continue
+
                             all_results_content.append(result_text)
 
                         tool_responses.append(
@@ -2803,7 +2829,7 @@ def __del__(self):
         """Cleanup when the Coder object is destroyed."""
         self.ok_to_warm_cache = False
 
-    def add_assistant_reply_to_cur_messages(self):
+    async def add_assistant_reply_to_cur_messages(self):
         """
         Add the assistant's reply to `cur_messages`.
         Handles model-specific quirks, like Deepseek which requires `content`
@@ -2845,6 +2871,10 @@ def add_assistant_reply_to_cur_messages(self):
             or msg.get("tool_calls", None)
             or msg.get("function_call", None)
         ):
+            if not await HookIntegration.call_end_message_hooks(self, str(msg)):
+                self.io.tool_warning("Execution stopped by end message hook")
+                return
+
             ConversationManager.add_message(
                 message_dict=msg,
                 tag=MessageTag.CUR,
@@ -2972,7 +3002,7 @@ async def send(self, messages, model=None, functions=None, tools=None):
                 async for chunk in self.show_send_output_stream(completion):
                     yield chunk
             else:
-                self.show_send_output(completion)
+                await self.show_send_output(completion)
 
             response, func_err, content_err = self.consolidate_chunks()
 
@@ -3002,7 +3032,7 @@ async def send(self, messages, model=None, functions=None, tools=None):
                 if args:
                     self.io.ai_output(json.dumps(args, indent=4))
 
-    def show_send_output(self, completion):
+    async def show_send_output(self, completion):
         if self.verbose:
             print(completion)
 
@@ -3018,6 +3048,10 @@ def show_send_output(self, completion):
 
         response, func_err, content_err = self.consolidate_chunks()
 
+        if not await HookIntegration.call_on_message_hooks(self, self.partial_response_content):
+            self.io.tool_warning("Execution stopped by on message hook")
+            return
+
         resp_hash = dict(
             function_call=str(self.partial_response_function_call),
             content=self.partial_response_content,
@@ -3183,6 +3217,10 @@ async def show_send_output_stream(self, completion):
         # The Part Doing the Heavy Lifting Now
         self.consolidate_chunks()
 
+        if not await HookIntegration.call_on_message_hooks(self, self.partial_response_content):
+            self.io.tool_warning("Execution stopped by on message hook")
+            return
+
         if not received_content and len(self.partial_response_tool_calls) == 0:
             self.io.tool_warning("Empty response received from LLM. Check your provider account?")
 

cecli/coders/copypaste_coder.py

@@ -146,7 +146,7 @@ async def send(self, messages, model=None, functions=None, tools=None):
         try:
             hash_object, completion = self.copy_paste_completion(messages, model)
             self.chat_completion_call_hashes.append(hash_object.hexdigest())
-            self.show_send_output(completion)
+            await self.show_send_output(completion)
             self.calculate_and_show_tokens_and_cost(messages, completion)
         finally:
             self.preprocess_response()

cecli/coders/single_wholefile_func_coder.py

@@ -38,7 +38,7 @@ class SingleWholeFileFunctionCoder(Coder):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
-    def add_assistant_reply_to_cur_messages(self, edited):
+    async def add_assistant_reply_to_cur_messages(self, edited):
         if edited:
             # Always add to conversation manager
             ConversationManager.add_message(

cecli/coders/wholefile_func_coder.py

@@ -49,7 +49,7 @@ def __init__(self, *args, **kwargs):
 
         super().__init__(*args, **kwargs)
 
-    def add_assistant_reply_to_cur_messages(self, edited):
+    async def add_assistant_reply_to_cur_messages(self, edited):
         if edited:
             # Always add to conversation manager
             ConversationManager.add_message(

cecli/commands/__init__.py

@@ -30,9 +30,11 @@
 from .hashline import HashlineCommand
 from .help import HelpCommand
 from .history_search import HistorySearchCommand
+from .hooks import HooksCommand
 from .lint import LintCommand
 from .list_sessions import ListSessionsCommand
 from .load import LoadCommand
+from .load_hook import LoadHookCommand
 from .load_mcp import LoadMcpCommand
 from .load_session import LoadSessionCommand
 from .load_skill import LoadSkillCommand
@@ -47,6 +49,7 @@
 from .read_only import ReadOnlyCommand
 from .read_only_stub import ReadOnlyStubCommand
 from .reasoning_effort import ReasoningEffortCommand
+from .remove_hook import RemoveHookCommand
 from .remove_mcp import RemoveMcpCommand
 from .remove_skill import RemoveSkillCommand
 from .report import ReportCommand
@@ -102,9 +105,11 @@
 CommandRegistry.register(HashlineCommand)
 CommandRegistry.register(HelpCommand)
 CommandRegistry.register(HistorySearchCommand)
+CommandRegistry.register(HooksCommand)
 CommandRegistry.register(LintCommand)
 CommandRegistry.register(ListSessionsCommand)
 CommandRegistry.register(LoadCommand)
+CommandRegistry.register(LoadHookCommand)
 CommandRegistry.register(LoadMcpCommand)
 CommandRegistry.register(LoadSessionCommand)
 CommandRegistry.register(LoadSkillCommand)
@@ -119,6 +124,7 @@
 CommandRegistry.register(ReadOnlyCommand)
 CommandRegistry.register(ReadOnlyStubCommand)
 CommandRegistry.register(ReasoningEffortCommand)
+CommandRegistry.register(RemoveHookCommand)
 CommandRegistry.register(RemoveMcpCommand)
 CommandRegistry.register(RemoveSkillCommand)
 CommandRegistry.register(ReportCommand)
@@ -171,9 +177,11 @@
     "HashlineCommand",
     "HelpCommand",
     "HistorySearchCommand",
+    "HookCommand",
     "LintCommand",
     "ListSessionsCommand",
     "LoadCommand",
+    "LoadHookCommand",
     "LoadMcpCommand",
     "LoadSessionCommand",
     "LoadSkillCommand",
@@ -190,6 +198,7 @@
     "ReadOnlyCommand",
     "ReadOnlyStubCommand",
     "ReasoningEffortCommand",
+    "RemoveHookCommand",
     "RemoveMcpCommand",
     "RemoveSkillCommand",
     "ReportCommand",