HCK-CI · kostyanf14 · Apr 3, 2026 · Apr 3, 2026 · Apr 3, 2026 · gemini-code-assist
diff --git a/lib/all.rb b/lib/all.rb
@@ -36,6 +36,7 @@ module AutoHCK
   autoload_relative :Downloader, 'auxiliary/downloader'
   autoload_relative :Engine, 'engines/engine'
   autoload_relative :EngineError, 'engines/exceptions'
+  autoload_relative :ExternalScript, 'notificationmanagers/external_script/external_script'
   autoload_relative :ExtraSoftwareBrokenConfig, 'auxiliary/extra_software/exceptions'
   autoload_relative :ExtraSoftwareManager, 'auxiliary/extra_software/manager'
   autoload_relative :ExtraSoftwareMissingConfig, 'auxiliary/extra_software/exceptions'
@@ -56,6 +57,8 @@ module AutoHCK
   autoload_relative :MachineRunError, 'setupmanagers/exceptions'
   autoload_relative :Models, 'models'
   autoload_relative :MultiLogger, 'auxiliary/multi_logger'
+  autoload_relative :NotificationManager, 'notificationmanagers/notification_manager'
+  autoload_relative :NotificationProviderInterface, 'notificationmanagers/notification_provider_interface'
   autoload_relative :NotImplementedError, 'exceptions'
   autoload_relative :OpenJsonError, 'exceptions'
   autoload_relative :PackageManager, 'auxiliary/package_manager'

diff --git a/lib/engines/config_manager/config_manager.rb b/lib/engines/config_manager/config_manager.rb
@@ -12,6 +12,7 @@ def initialize(project)
       @project = project
       @logger = project.logger
       @project.append_multilog("#{self.class.tag}.log")
+      @project.notification_manager.post_engine_init(self)
     end
 
     def self.tag(*)
@@ -23,11 +24,14 @@ def test_steps
     end
 
     def run
+      @project.notification_manager.pre_engine_run(self)
       ResourceScope.open do |scope|
         @project.logger.info('Stating result uploader token initialization')
         @result_uploader = ResultUploader.new(scope, @project)
         @result_uploader.ask_token
       end
+    ensure
+      @project.notification_manager.post_engine_run(self)
     end
 
     def drivers
@@ -41,5 +45,9 @@ def self.platform(*)
     def result_uploader_needed?
       false
     end
+
+    def target
+      nil
+    end
   end
 end
diff --git a/lib/engines/hckinstall/hckinstall.rb b/lib/engines/hckinstall/hckinstall.rb
@@ -28,6 +28,7 @@ def initialize(project)
       validate_paths
       prepare_extra_sw
       @logger.debug('HCKInstall: initialized')
+      @project.notification_manager.post_engine_init(self)
     end
 
     def test_steps
@@ -438,6 +439,7 @@ def plan_client
     end
 
     def run
+      @project.notification_manager.pre_engine_run(self)
       @logger.debug('HCKInstall: run')
 
       prepare_setup_scripts_config
@@ -449,6 +451,8 @@ def run
         run_first(studio:, client:)
         run_second(client:)
       end
+    ensure
+      @project.notification_manager.post_engine_run(self)
     end
   end
 end
diff --git a/lib/engines/hcktest/hcktest.rb b/lib/engines/hcktest/hcktest.rb
@@ -28,6 +28,7 @@ def initialize(project)
       @extensions = find_extensions
       prepare_extra_sw
       validate_paths unless @driver_path.nil?
+      @project.notification_manager.post_engine_init(self)
     end
 
     def test_steps
@@ -345,6 +346,7 @@ def group_tests_by_config
     end
 
     def run
+      @project.notification_manager.pre_engine_run(self)
       upload_driver_package unless @driver_path.nil?
 
       if @project.options.test.dump
@@ -358,6 +360,8 @@ def run
       else
         auto_run
       end
+    ensure
+      @project.notification_manager.post_engine_run(self)
     end
 
     def result_uploader_needed?

diff --git a/lib/engines/hcktest/tests.rb b/lib/engines/hcktest/tests.rb
@@ -308,6 +308,7 @@ def perform_queue_test_commands(test)
 
     sig { params(test: Models::HLK::Test, wait: T::Boolean).void }
     def queue_test(test, wait: false)
+      @project.notification_manager.pre_tests_queue_test(test, wait)
       retries = 0
       begin
         perform_queue_test_commands(test)
@@ -367,6 +368,7 @@ def done_test_results
 
     sig { params(test: Models::HLK::Test).void }
     def on_test_start(test)
+      @project.notification_manager.pre_tests_on_test_start(test)
       @logger.info(">>> Currently running: #{test.name} [#{test.estimatedruntime}]")
 
       test.started_at = DateTime.now
@@ -571,6 +573,7 @@ def collect_memory_dumps(test)
 
     sig { params(test: Models::HLK::Test, test_result: T::Hash[String, T.untyped]).void }
     def handle_finished_test_result(test, test_result)
+      @project.notification_manager.pre_tests_handle_finished_test_result(test, test_result)
       collect_memory_dumps(test)
 
       print_test_results(test, test_result)
@@ -590,6 +593,7 @@ def test_for_result(result)
     end
 
     def handle_finished_test_results(results)
+      @project.notification_manager.pre_tests_handle_finished_test_results(results, tests_stats)
       @project.update_test_stats(tests_stats)
 
       results.each do |result|
@@ -787,6 +791,7 @@ def retry_tests(current_run_tests)
 
     sig { params(tests: T::Array[Models::HLK::Test]).void }
     def run(tests)
+      @project.notification_manager.pre_tests_run(tests)
       load_clients_system_info
       update_summary_results_log
       tests.each do |test|
@@ -804,6 +809,8 @@ def run(tests)
       return if @project.run_terminated
 
       retry_tests(tests) unless @project.options.test.auto_retry_failed_tests.zero?
+    ensure
+      @project.notification_manager.post_tests_run(tests)
     end
   end
 end
diff --git a/lib/notificationmanagers/external_script/example_hook.sh b/lib/notificationmanagers/external_script/example_hook.sh
@@ -0,0 +1,17 @@
+#!/usr/bin/env bash
+# Example AutoHCK external notification hook.
+# Environment: AUTOHCK_NOTIFICATION_HOOK and one variable per Ruby argument,
+# e.g. AUTOHCK_NOTIFICATION_PROJECT holds JSON for the Project snapshot.
+
+set -euo pipefail
+
+log_default="${TMPDIR:-/tmp}/autohck_external_notification.log"
+log_path="${AUTOHCK_NOTIFICATION_LOG_PATH:-$log_default}"
+
+{
+  printf '%s\n' "=== $(date -Iseconds 2>/dev/null || date) ==="
+  env | LC_ALL=C sort | grep '^AUTOHCK_NOTIFICATION_' || true
+  printf '\n'
+} >>"$log_path"
+
+exit 0
diff --git a/lib/notificationmanagers/external_script/external_script.json b/lib/notificationmanagers/external_script/external_script.json
@@ -0,0 +1,27 @@
+{
+  "_readme": "Optional 'hooks' object maps Ruby notification method names to executable scripts. Paths are relative to this file's directory unless absolute. Omit hooks you do not need.",
+  "hooks": {
+    "post_project_init": "example_hook.sh",
+    "pre_project_run": "example_hook.sh",
+    "post_project_run": "example_hook.sh",
+    "post_engine_init": "example_hook.sh",
+    "pre_engine_run": "example_hook.sh",
+    "post_engine_run": "example_hook.sh",
+    "post_setup_manager_init": "example_hook.sh",
+    "pre_setup_manager_run_hck_studio": "example_hook.sh",
+    "post_setup_manager_run_hck_studio": "example_hook.sh",
+    "pre_setup_manager_run_studio": "example_hook.sh",
+    "post_setup_manager_run_studio": "example_hook.sh",
+    "pre_setup_manager_run_hck_client": "example_hook.sh",
+    "post_setup_manager_run_hck_client": "example_hook.sh",
+    "pre_setup_manager_run_client": "example_hook.sh",
+    "post_setup_manager_run_client": "example_hook.sh",
+    "pre_tests_run": "example_hook.sh",
+    "post_tests_run": "example_hook.sh",
+    "pre_tests_queue_test": "example_hook.sh",
+    "pre_tests_on_test_start": "example_hook.sh",
+    "pre_tests_handle_finished_test_results": "example_hook.sh",
+    "pre_tests_handle_finished_test_result": "example_hook.sh",
+    "close": "example_hook.sh"
+  }
+}