imgui_test_harness_service.cc

Go to the documentation of this file.

#include "app/service/imgui_test_harness_service.h"
#include "app/application.h"
#include "app/service/imgui_test_harness_internal.h"
 
#ifdef YAZE_WITH_GRPC
 
#include "app/platform/sdl_compat.h"
 
#include <algorithm>
#include <chrono>
#include <deque>
#include <fstream>
#include <iostream>
#include <limits>
#include <optional>
#include <thread>
 
#include "absl/base/thread_annotations.h"
#include "absl/container/flat_hash_map.h"
#include "absl/strings/ascii.h"
#include "absl/strings/numbers.h"
#include "absl/strings/str_cat.h"
#include "absl/strings/str_format.h"
#include "absl/strings/str_replace.h"
#include "absl/synchronization/mutex.h"
#include "absl/time/clock.h"
#include "absl/time/time.h"
#include "app/gui/automation/widget_id_registry.h"
#include "app/service/screenshot_utils.h"
#include "app/testing/test_manager.h"
#include "app/testing/test_script_parser.h"
#include "protos/imgui_test_harness.grpc.pb.h"
#include "protos/imgui_test_harness.pb.h"
#include "yaze.h"  // For YAZE_VERSION_STRING
 
// The Windows SDK defines Yield as a function-like macro, which collides with
// ImGuiTestContext::Yield when this translation unit is built with clang-cl.
#ifdef Yield
#undef Yield
#endif
 
#if defined(YAZE_ENABLE_IMGUI_TEST_ENGINE) && YAZE_ENABLE_IMGUI_TEST_ENGINE
#include "imgui_test_engine/imgui_te_context.h"
#include "imgui_test_engine/imgui_te_engine.h"
 
// Helper to register and run a test dynamically
namespace {
struct DynamicTestData {
  std::function<void(ImGuiTestContext*)> test_func;
};
 
absl::Mutex g_dynamic_tests_mutex;
std::deque<std::shared_ptr<DynamicTestData>> g_dynamic_tests
    ABSL_GUARDED_BY(g_dynamic_tests_mutex);
 
void KeepDynamicTestData(const std::shared_ptr<DynamicTestData>& data) {
  absl::MutexLock lock(&g_dynamic_tests_mutex);
  // ImGuiTest stores UserData as a raw pointer and registered tests remain
  // available for replay for the lifetime of the engine. Retain every backing
  // closure for that same lifetime; evicting an older entry can otherwise
  // leave a queued or re-runnable test with dangling UserData.
  g_dynamic_tests.push_back(data);
}
 
void RunDynamicTest(ImGuiTestContext* ctx) {
  auto* data = (DynamicTestData*)ctx->Test->UserData;
  if (data && data->test_func) {
    data->test_func(ctx);
  }
}
 
// Helper to check if a test has completed (not queued or running)
bool IsTestCompleted(ImGuiTest* test) {
  return test->Output.Status != ImGuiTestStatus_Queued &&
         test->Output.Status != ImGuiTestStatus_Running;
}
 
// Thread-safe state for RPC communication
template <typename T>
struct RPCState {
  std::atomic<bool> completed{false};
  std::mutex data_mutex;
  T result;
  std::string message;
 
  void SetResult(const T& res, const std::string& msg) {
    std::lock_guard<std::mutex> lock(data_mutex);
    result = res;
    message = msg;
    completed.store(true);
  }
 
  void GetResult(T& res, std::string& msg) {
    std::lock_guard<std::mutex> lock(data_mutex);
    res = result;
    msg = message;
  }
};
 
}  // namespace
#endif
 
namespace {
 
::yaze::test::GetTestStatusResponse_TestStatus ConvertHarnessStatus(
    ::yaze::test::HarnessTestStatus status) {
  switch (status) {
    case ::yaze::test::HarnessTestStatus::kQueued:
      return ::yaze::test::GetTestStatusResponse::TEST_STATUS_QUEUED;
    case ::yaze::test::HarnessTestStatus::kRunning:
      return ::yaze::test::GetTestStatusResponse::TEST_STATUS_RUNNING;
    case ::yaze::test::HarnessTestStatus::kPassed:
      return ::yaze::test::GetTestStatusResponse::TEST_STATUS_PASSED;
    case ::yaze::test::HarnessTestStatus::kFailed:
      return ::yaze::test::GetTestStatusResponse::TEST_STATUS_FAILED;
    case ::yaze::test::HarnessTestStatus::kTimeout:
      return ::yaze::test::GetTestStatusResponse::TEST_STATUS_TIMEOUT;
    case ::yaze::test::HarnessTestStatus::kUnspecified:
    default:
      return ::yaze::test::GetTestStatusResponse::TEST_STATUS_UNSPECIFIED;
  }
}
 
int64_t ToUnixMillisSafe(absl::Time timestamp) {
  if (timestamp == absl::InfinitePast()) {
    return 0;
  }
  return absl::ToUnixMillis(timestamp);
}
 
int32_t ClampDurationToInt32(absl::Duration duration) {
  int64_t millis = absl::ToInt64Milliseconds(duration);
  if (millis > std::numeric_limits<int32_t>::max()) {
    return std::numeric_limits<int32_t>::max();
  }
  if (millis < std::numeric_limits<int32_t>::min()) {
    return std::numeric_limits<int32_t>::min();
  }
  return static_cast<int32_t>(millis);
}
 
}  // namespace
 
#include <grpcpp/grpcpp.h>
#include <grpcpp/server_builder.h>
 
namespace yaze {
namespace test {
 
namespace {
 
std::string ClickTypeToString(ClickRequest::ClickType type) {
  switch (type) {
    case ClickRequest::CLICK_TYPE_RIGHT:
      return "right";
    case ClickRequest::CLICK_TYPE_MIDDLE:
      return "middle";
    case ClickRequest::CLICK_TYPE_DOUBLE:
      return "double";
    case ClickRequest::CLICK_TYPE_LEFT:
    case ClickRequest::CLICK_TYPE_UNSPECIFIED:
    default:
      return "left";
  }
}
 
ClickRequest::ClickType ClickTypeFromString(absl::string_view type) {
  const std::string lower = absl::AsciiStrToLower(std::string(type));
  if (lower == "right") {
    return ClickRequest::CLICK_TYPE_RIGHT;
  }
  if (lower == "middle") {
    return ClickRequest::CLICK_TYPE_MIDDLE;
  }
  if (lower == "double" || lower == "double_click" || lower == "dbl") {
    return ClickRequest::CLICK_TYPE_DOUBLE;
  }
  return ClickRequest::CLICK_TYPE_LEFT;
}
 
HarnessTestStatus HarnessStatusFromString(absl::string_view status) {
  const std::string lower = absl::AsciiStrToLower(std::string(status));
  if (lower == "passed" || lower == "success") {
    return HarnessTestStatus::kPassed;
  }
  if (lower == "failed" || lower == "fail") {
    return HarnessTestStatus::kFailed;
  }
  if (lower == "timeout") {
    return HarnessTestStatus::kTimeout;
  }
  if (lower == "queued") {
    return HarnessTestStatus::kQueued;
  }
  if (lower == "running") {
    return HarnessTestStatus::kRunning;
  }
  return HarnessTestStatus::kUnspecified;
}
 
const char* HarnessStatusToString(HarnessTestStatus status) {
  switch (status) {
    case HarnessTestStatus::kPassed:
      return "passed";
    case HarnessTestStatus::kFailed:
      return "failed";
    case HarnessTestStatus::kTimeout:
      return "timeout";
    case HarnessTestStatus::kQueued:
      return "queued";
    case HarnessTestStatus::kRunning:
      return "running";
    case HarnessTestStatus::kUnspecified:
    default:
      return "unknown";
  }
}
 
std::string ApplyOverrides(
    const std::string& value,
    const absl::flat_hash_map<std::string, std::string>& overrides) {
  if (overrides.empty() || value.empty()) {
    return value;
  }
  std::string result = value;
  for (const auto& [key, replacement] : overrides) {
    const std::string placeholder = absl::StrCat("{{", key, "}}");
    result = absl::StrReplaceAll(result, {{placeholder, replacement}});
  }
  return result;
}
 
void MaybeRecordStep(TestRecorder* recorder, TestRecorder::RecordedStep step) {
  if (!recorder || !recorder->IsRecording()) {
    return;
  }
  if (step.captured_at == absl::InfinitePast()) {
    step.captured_at = absl::Now();
  }
  recorder->RecordStep(step);
}
 
absl::Status WaitForHarnessTestCompletion(TestManager* manager,
                                          const std::string& test_id,
                                          HarnessTestExecution* execution) {
  if (!manager) {
    return absl::FailedPreconditionError("TestManager unavailable");
  }
  if (test_id.empty()) {
    return absl::InvalidArgumentError("Missing harness test identifier");
  }
 
  const absl::Time deadline = absl::Now() + absl::Seconds(20);
  while (absl::Now() < deadline) {
    absl::StatusOr<HarnessTestExecution> current =
        manager->GetHarnessTestExecution(test_id);
    if (!current.ok()) {
      absl::SleepFor(absl::Milliseconds(75));
      continue;
    }
 
    if (execution) {
      *execution = std::move(current.value());
    }
 
    if (current->status == HarnessTestStatus::kQueued ||
        current->status == HarnessTestStatus::kRunning) {
      absl::SleepFor(absl::Milliseconds(75));
      continue;
    }
    return absl::OkStatus();
  }
 
  return absl::DeadlineExceededError(absl::StrFormat(
      "Harness test %s did not reach a terminal state", test_id));
}
 
struct ParsedCondition {
  std::string type;
  std::string target;
};
 
std::string ExtractLabelFromPath(absl::string_view path) {
  size_t colon = path.rfind(':');
  if (colon != absl::string_view::npos && colon + 1 < path.size()) {
    return std::string(path.substr(colon + 1));
  }
  size_t slash = path.rfind('/');
  if (slash != absl::string_view::npos && slash + 1 < path.size()) {
    return std::string(path.substr(slash + 1));
  }
  return std::string(path);
}
 
std::string ExtractTypeFromPath(absl::string_view path) {
  size_t slash = path.rfind('/');
  absl::string_view segment =
      slash == absl::string_view::npos ? path : path.substr(slash + 1);
  size_t colon = segment.find(':');
  if (colon == absl::string_view::npos || colon == 0) {
    return "widget";
  }
  return std::string(segment.substr(0, colon));
}
 
absl::StatusOr<internal::ParsedTarget> ParseTargetString(
    absl::string_view target) {
  if (target.empty()) {
    return absl::InvalidArgumentError(
        "Missing target. Provide 'type:label' or widget_key.");
  }
  size_t colon_pos = target.find(':');
  if (colon_pos == absl::string_view::npos || colon_pos == 0 ||
      colon_pos + 1 >= target.size()) {
    return absl::InvalidArgumentError(
        "Invalid target format. Use 'type:label' (e.g. 'button:Open ROM').");
  }
 
  internal::ParsedTarget parsed;
  parsed.type = std::string(target.substr(0, colon_pos));
  parsed.label = std::string(target.substr(colon_pos + 1));
  return parsed;
}
 
}  // namespace
 
namespace internal {
 
absl::StatusOr<ResolvedWidgetSelector> ResolveWidgetSelector(
    absl::string_view target, absl::string_view widget_key) {
  ResolvedWidgetSelector resolved;
  if (!widget_key.empty()) {
    const std::string key(widget_key);
    std::optional<gui::WidgetIdRegistry::WidgetInfo> info =
        gui::WidgetIdRegistry::Instance().GetWidgetInfo(key);
    if (!info) {
      return absl::NotFoundError(absl::StrFormat(
          "widget_key '%s' was not found in WidgetIdRegistry", key));
    }
 
    resolved.resolved_widget_key = key;
    resolved.resolved_path =
        info->full_path.empty() ? key : std::string(info->full_path);
    resolved.imgui_id = info->imgui_id;
    if (resolved.imgui_id == 0) {
      return absl::FailedPreconditionError(absl::StrFormat(
          "widget_key '%s' resolved without a valid ImGui ID", key));
    }
    resolved.target.type = info->type.empty()
                               ? ExtractTypeFromPath(resolved.resolved_path)
                               : std::string(info->type);
    resolved.target.label = info->label.empty()
                                ? ExtractLabelFromPath(resolved.resolved_path)
                                : std::string(info->label);
    if (resolved.target.label.empty()) {
      return absl::FailedPreconditionError(absl::StrFormat(
          "widget_key '%s' resolved without a usable label", key));
    }
    if (resolved.target.type.empty()) {
      resolved.target.type = "widget";
    }
    return resolved;
  }
 
  absl::StatusOr<ParsedTarget> parsed = ParseTargetString(target);
  if (!parsed.ok()) {
    return parsed.status();
  }
  resolved.target = *parsed;
  return resolved;
}
 
void ApplyTerminalHarnessExecution(const HarnessTestExecution& execution,
                                   bool* step_success,
                                   std::string* step_message) {
  if (step_success) {
    *step_success = execution.status == HarnessTestStatus::kPassed;
  }
  if (step_message && !execution.error_message.empty()) {
    *step_message = execution.error_message;
  }
}
 
}  // namespace internal
 
namespace {
 
absl::StatusOr<ParsedCondition> ParseConditionString(absl::string_view value) {
  if (value.empty()) {
    return absl::InvalidArgumentError(
        "Missing condition. Provide 'type:target' or widget_key.");
  }
 
  size_t colon_pos = value.find(':');
  if (colon_pos == absl::string_view::npos || colon_pos == 0) {
    return absl::InvalidArgumentError(
        "Invalid condition format. Use 'type:target'.");
  }
 
  ParsedCondition parsed;
  parsed.type = std::string(value.substr(0, colon_pos));
  parsed.target = std::string(value.substr(colon_pos + 1));
  return parsed;
}
 
absl::StatusOr<ParsedCondition> ResolveCondition(
    absl::string_view condition, absl::string_view widget_key,
    absl::string_view default_type_for_widget,
    internal::ResolvedWidgetSelector* resolved_selector) {
  std::optional<internal::ResolvedWidgetSelector> widget_selector;
  if (!widget_key.empty()) {
    absl::StatusOr<internal::ResolvedWidgetSelector> resolved =
        internal::ResolveWidgetSelector("", widget_key);
    if (!resolved.ok()) {
      return resolved.status();
    }
    widget_selector = *resolved;
    if (resolved_selector) {
      *resolved_selector = *resolved;
    }
  } else if (resolved_selector) {
    *resolved_selector = internal::ResolvedWidgetSelector{};
  }
 
  if (condition.empty()) {
    if (!widget_selector.has_value()) {
      return absl::InvalidArgumentError(
          "Missing condition. Provide 'type:target' or widget_key.");
    }
    ParsedCondition parsed;
    parsed.type = std::string(default_type_for_widget);
    parsed.target = widget_selector->target.label;
    return parsed;
  }
 
  absl::StatusOr<ParsedCondition> parsed = ParseConditionString(condition);
  if (!parsed.ok()) {
    return parsed.status();
  }
 
  if (widget_selector.has_value() && parsed->target.empty()) {
    parsed->target = widget_selector->target.label;
  }
 
  if (parsed->target.empty()) {
    return absl::InvalidArgumentError(
        "Condition target is empty. Provide 'type:target'.");
  }
 
  return parsed;
}
 
struct UiSyncSnapshot {
  int64_t frame_id = 0;
  int32_t pending_editor_actions = 0;
  int32_t pending_layout_actions = 0;
  bool layout_rebuild_pending = false;
  int32_t pending_harness_tests = 0;
  int32_t queued_harness_tests = 0;
  int32_t running_harness_tests = 0;
};
 
editor::EditorManager* GetEditorManagerForSync() {
  Controller* controller = Application::Instance().GetController();
  if (!controller) {
    return nullptr;
  }
  return controller->editor_manager();
}
 
UiSyncSnapshot CaptureUiSyncSnapshot(TestManager* test_manager) {
  UiSyncSnapshot snapshot;
  if (editor::EditorManager* editor_manager = GetEditorManagerForSync()) {
    const editor::EditorManager::UiSyncState state =
        editor_manager->GetUiSyncStateSnapshot();
    snapshot.frame_id = static_cast<int64_t>(state.frame_id);
    snapshot.pending_editor_actions = state.pending_editor_actions;
    snapshot.pending_layout_actions = state.pending_layout_actions;
    snapshot.layout_rebuild_pending = state.layout_rebuild_pending;
  }
 
  if (test_manager) {
    const std::vector<HarnessTestSummary> summaries =
        test_manager->ListHarnessTestSummaries();
    for (const HarnessTestSummary& summary : summaries) {
      switch (summary.latest_execution.status) {
        case HarnessTestStatus::kQueued:
          snapshot.queued_harness_tests++;
          break;
        case HarnessTestStatus::kRunning:
          snapshot.running_harness_tests++;
          break;
        default:
          break;
      }
    }
  }
 
  snapshot.pending_harness_tests =
      snapshot.queued_harness_tests + snapshot.running_harness_tests;
  return snapshot;
}
 
bool IsUiIdle(const UiSyncSnapshot& snapshot) {
  return snapshot.pending_editor_actions == 0 &&
         snapshot.pending_layout_actions == 0 &&
         !snapshot.layout_rebuild_pending &&
         snapshot.pending_harness_tests == 0;
}
 
std::string BuildIdleTimeoutReason(const UiSyncSnapshot& snapshot) {
  return absl::StrFormat(
      "pending_editor=%d pending_layout=%d layout_rebuild=%s queued_tests=%d "
      "running_tests=%d",
      snapshot.pending_editor_actions, snapshot.pending_layout_actions,
      snapshot.layout_rebuild_pending ? "true" : "false",
      snapshot.queued_harness_tests, snapshot.running_harness_tests);
}
 
int32_t NormalizeTimeoutMs(int32_t timeout_ms, int32_t fallback_ms) {
  return timeout_ms <= 0 ? fallback_ms : timeout_ms;
}
 
int32_t NormalizePollIntervalMs(int32_t poll_interval_ms) {
  if (poll_interval_ms <= 0) {
    return 25;
  }
  return std::clamp(poll_interval_ms, 5, 1000);
}
 
}  // namespace
 
// gRPC service wrapper that forwards to our implementation
class ImGuiTestHarnessServiceGrpc final : public ImGuiTestHarness::Service {
 public:
  explicit ImGuiTestHarnessServiceGrpc(ImGuiTestHarnessServiceImpl* impl)
      : impl_(impl) {}
 
  ::grpc::Status Ping(::grpc::ServerContext* context,
                      const PingRequest* request,
                      PingResponse* response) override {
    return ConvertStatus(impl_->Ping(request, response));
  }
 
  ::grpc::Status Click(::grpc::ServerContext* context,
                       const ClickRequest* request,
                       ClickResponse* response) override {
    return ConvertStatus(impl_->Click(request, response));
  }
 
  ::grpc::Status Type(::grpc::ServerContext* context,
                      const TypeRequest* request,
                      TypeResponse* response) override {
    return ConvertStatus(impl_->Type(request, response));
  }
 
  ::grpc::Status Wait(::grpc::ServerContext* context,
                      const WaitRequest* request,
                      WaitResponse* response) override {
    return ConvertStatus(impl_->Wait(request, response));
  }
 
  ::grpc::Status Assert(::grpc::ServerContext* context,
                        const AssertRequest* request,
                        AssertResponse* response) override {
    return ConvertStatus(impl_->Assert(request, response));
  }
 
  ::grpc::Status FlushUiActions(::grpc::ServerContext* context,
                                const FlushUiActionsRequest* request,
                                FlushUiActionsResponse* response) override {
    return ConvertStatus(impl_->FlushUiActions(request, response));
  }
 
  ::grpc::Status WaitForIdle(::grpc::ServerContext* context,
                             const WaitForIdleRequest* request,
                             WaitForIdleResponse* response) override {
    return ConvertStatus(impl_->WaitForIdle(request, response));
  }
 
  ::grpc::Status GetUiSyncState(::grpc::ServerContext* context,
                                const GetUiSyncStateRequest* request,
                                GetUiSyncStateResponse* response) override {
    return ConvertStatus(impl_->GetUiSyncState(request, response));
  }
 
  ::grpc::Status Screenshot(::grpc::ServerContext* context,
                            const ScreenshotRequest* request,
                            ScreenshotResponse* response) override {
    return ConvertStatus(impl_->Screenshot(request, response));
  }
 
  ::grpc::Status GetTestStatus(::grpc::ServerContext* context,
                               const GetTestStatusRequest* request,
                               GetTestStatusResponse* response) override {
    return ConvertStatus(impl_->GetTestStatus(request, response));
  }
 
  ::grpc::Status ListTests(::grpc::ServerContext* context,
                           const ListTestsRequest* request,
                           ListTestsResponse* response) override {
    return ConvertStatus(impl_->ListTests(request, response));
  }
 
  ::grpc::Status GetTestResults(::grpc::ServerContext* context,
                                const GetTestResultsRequest* request,
                                GetTestResultsResponse* response) override {
    return ConvertStatus(impl_->GetTestResults(request, response));
  }
 
  ::grpc::Status DiscoverWidgets(::grpc::ServerContext* context,
                                 const DiscoverWidgetsRequest* request,
                                 DiscoverWidgetsResponse* response) override {
    return ConvertStatus(impl_->DiscoverWidgets(request, response));
  }
 
  ::grpc::Status StartRecording(::grpc::ServerContext* context,
                                const StartRecordingRequest* request,
                                StartRecordingResponse* response) override {
    return ConvertStatus(impl_->StartRecording(request, response));
  }
 
  ::grpc::Status StopRecording(::grpc::ServerContext* context,
                               const StopRecordingRequest* request,
                               StopRecordingResponse* response) override {
    return ConvertStatus(impl_->StopRecording(request, response));
  }
 
  ::grpc::Status ReplayTest(::grpc::ServerContext* context,
                            const ReplayTestRequest* request,
                            ReplayTestResponse* response) override {
    return ConvertStatus(impl_->ReplayTest(request, response));
  }
 
 private:
  static ::grpc::Status ConvertStatus(const absl::Status& status) {
    if (status.ok()) {
      return ::grpc::Status::OK;
    }
 
    ::grpc::StatusCode code = ::grpc::StatusCode::UNKNOWN;
    switch (status.code()) {
      case absl::StatusCode::kCancelled:
        code = ::grpc::StatusCode::CANCELLED;
        break;
      case absl::StatusCode::kUnknown:
        code = ::grpc::StatusCode::UNKNOWN;
        break;
      case absl::StatusCode::kInvalidArgument:
        code = ::grpc::StatusCode::INVALID_ARGUMENT;
        break;
      case absl::StatusCode::kDeadlineExceeded:
        code = ::grpc::StatusCode::DEADLINE_EXCEEDED;
        break;
      case absl::StatusCode::kNotFound:
        code = ::grpc::StatusCode::NOT_FOUND;
        break;
      case absl::StatusCode::kAlreadyExists:
        code = ::grpc::StatusCode::ALREADY_EXISTS;
        break;
      case absl::StatusCode::kPermissionDenied:
        code = ::grpc::StatusCode::PERMISSION_DENIED;
        break;
      case absl::StatusCode::kResourceExhausted:
        code = ::grpc::StatusCode::RESOURCE_EXHAUSTED;
        break;
      case absl::StatusCode::kFailedPrecondition:
        code = ::grpc::StatusCode::FAILED_PRECONDITION;
        break;
      case absl::StatusCode::kAborted:
        code = ::grpc::StatusCode::ABORTED;
        break;
      case absl::StatusCode::kOutOfRange:
        code = ::grpc::StatusCode::OUT_OF_RANGE;
        break;
      case absl::StatusCode::kUnimplemented:
        code = ::grpc::StatusCode::UNIMPLEMENTED;
        break;
      case absl::StatusCode::kInternal:
        code = ::grpc::StatusCode::INTERNAL;
        break;
      case absl::StatusCode::kUnavailable:
        code = ::grpc::StatusCode::UNAVAILABLE;
        break;
      case absl::StatusCode::kDataLoss:
        code = ::grpc::StatusCode::DATA_LOSS;
        break;
      case absl::StatusCode::kUnauthenticated:
        code = ::grpc::StatusCode::UNAUTHENTICATED;
        break;
      default:
        code = ::grpc::StatusCode::UNKNOWN;
        break;
    }
 
    return ::grpc::Status(
        code, std::string(status.message().data(), status.message().size()));
  }
 
  ImGuiTestHarnessServiceImpl* impl_;
};
 
std::unique_ptr<::grpc::Service> CreateImGuiTestHarnessServiceGrpc(
    ImGuiTestHarnessServiceImpl* impl) {
  return std::make_unique<ImGuiTestHarnessServiceGrpc>(impl);
}
 
// ============================================================================
// ImGuiTestHarnessServiceImpl - RPC Handlers
// ============================================================================
 
absl::Status ImGuiTestHarnessServiceImpl::Ping(const PingRequest* request,
                                               PingResponse* response) {
  // Echo back the message with "Pong: " prefix
  response->set_message(absl::StrFormat("Pong: %s", request->message()));
 
  // Add current timestamp
  auto now = std::chrono::system_clock::now();
  auto ms = std::chrono::duration_cast<std::chrono::milliseconds>(
      now.time_since_epoch());
  response->set_timestamp_ms(ms.count());
 
  // Add YAZE version
  response->set_yaze_version(YAZE_VERSION_STRING);
 
  return absl::OkStatus();
}
 
absl::Status ImGuiTestHarnessServiceImpl::Click(const ClickRequest* request,
                                                ClickResponse* response) {
  auto start = std::chrono::steady_clock::now();
 
  TestRecorder::RecordedStep recorded_step;
  recorded_step.type = TestRecorder::ActionType::kClick;
  if (request) {
    recorded_step.widget_key = request->widget_key();
    if (!request->widget_key().empty()) {
      recorded_step.target = absl::StrCat("widget_key:", request->widget_key());
    } else {
      recorded_step.target = request->target();
    }
    recorded_step.click_type = ClickTypeToString(request->type());
  }
 
  auto finalize = [&](const absl::Status& status) {
    recorded_step.success = response->success();
    recorded_step.message = response->message();
    recorded_step.execution_time_ms = response->execution_time_ms();
    recorded_step.test_id = response->test_id();
    MaybeRecordStep(&test_recorder_, recorded_step);
    return status;
  };
 
  if (!test_manager_) {
    response->set_success(false);
    response->set_message("TestManager not available");
    response->set_execution_time_ms(0);
    return finalize(absl::FailedPreconditionError("TestManager not available"));
  }
 
  const std::string requested_target = request ? request->target() : "";
  const std::string requested_widget_key = request ? request->widget_key() : "";
  const std::string request_summary =
      !requested_widget_key.empty()
          ? absl::StrFormat("widget_key:%s", requested_widget_key)
          : requested_target;
 
  const std::string test_id = test_manager_->RegisterHarnessTest(
      absl::StrFormat("Click: %s", request_summary), "grpc");
  response->set_test_id(test_id);
  recorded_step.test_id = test_id;
  test_manager_->AppendHarnessTestLog(
      test_id, absl::StrCat("Queued click request: ", request_summary));
 
  absl::StatusOr<internal::ResolvedWidgetSelector> resolved_selector =
      internal::ResolveWidgetSelector(requested_target, requested_widget_key);
  if (!resolved_selector.ok()) {
    auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
        std::chrono::steady_clock::now() - start);
    std::string message = std::string(resolved_selector.status().message());
    response->set_success(false);
    response->set_message(message);
    response->set_execution_time_ms(elapsed.count());
    test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                            message);
    test_manager_->AppendHarnessTestLog(test_id, message);
    return finalize(absl::OkStatus());
  }
 
  if (!resolved_selector->resolved_widget_key.empty()) {
    response->set_resolved_widget_key(resolved_selector->resolved_widget_key);
  }
  if (!resolved_selector->resolved_path.empty()) {
    response->set_resolved_path(resolved_selector->resolved_path);
  }
 
  const std::string widget_type = resolved_selector->target.type;
  const std::string widget_label = resolved_selector->target.label;
  const ImGuiID widget_id = resolved_selector->imgui_id;
 
#if defined(YAZE_ENABLE_IMGUI_TEST_ENGINE) && YAZE_ENABLE_IMGUI_TEST_ENGINE
  ImGuiTestEngine* engine = test_manager_->GetUITestEngine();
  if (!engine) {
    auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
        std::chrono::steady_clock::now() - start);
    std::string message = "ImGuiTestEngine not initialized";
    response->set_success(false);
    response->set_message(message);
    response->set_execution_time_ms(elapsed.count());
    test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                            message);
    return finalize(absl::OkStatus());
  }
 
  const ClickRequest::ClickType click_type =
      request ? request->type() : ClickRequest::CLICK_TYPE_LEFT;
 
  ImGuiMouseButton mouse_button = ImGuiMouseButton_Left;
  switch (click_type) {
    case ClickRequest::CLICK_TYPE_UNSPECIFIED:
    case ClickRequest::CLICK_TYPE_LEFT:
      mouse_button = ImGuiMouseButton_Left;
      break;
    case ClickRequest::CLICK_TYPE_RIGHT:
      mouse_button = ImGuiMouseButton_Right;
      break;
    case ClickRequest::CLICK_TYPE_MIDDLE:
      mouse_button = ImGuiMouseButton_Middle;
      break;
    case ClickRequest::CLICK_TYPE_DOUBLE:
      // handled below
      break;
  }
 
  auto test_data = std::make_shared<DynamicTestData>();
  TestManager* manager = test_manager_;
  test_data->test_func = [manager, captured_id = test_id, widget_type,
                          widget_label, widget_id, click_type,
                          mouse_button](ImGuiTestContext* ctx) {
    manager->MarkHarnessTestRunning(captured_id);
    try {
      if (widget_type == "shortcut") {
        ImGuiKeyChord chord = 0;
        if (widget_label == "Save ROM") {
          chord = ImGuiMod_Shortcut | ImGuiKey_S;
        } else if (widget_label == "Quit") {
          // A successful quit destroys the harness before a simulated key
          // press can finish and report its result. Invoke the same guarded
          // application action directly, fail if unsaved work blocks it, and
          // let the caller prove process and listener teardown.
          auto* controller = Application::Instance().GetController();
          auto* editor_manager =
              controller != nullptr ? controller->editor_manager() : nullptr;
          if (editor_manager == nullptr) {
            const std::string error_message =
                "Application quit is unavailable before editor startup";
            manager->AppendHarnessTestLog(captured_id, error_message);
            manager->MarkHarnessTestCompleted(
                captured_id, HarnessTestStatus::kFailed, error_message);
            return;
          }
 
          editor_manager->Quit();
          if (!editor_manager->quit()) {
            const std::string error_message = absl::StrCat(
                "Application quit was blocked: ",
                editor_manager->GetPendingUnsavedSessionActionPrompt());
            manager->AppendHarnessTestLog(captured_id, error_message);
            manager->MarkHarnessTestCompleted(
                captured_id, HarnessTestStatus::kFailed, error_message);
            return;
          }
 
          const std::string success_message = "Requested application quit";
          manager->AppendHarnessTestLog(captured_id, success_message);
          manager->MarkHarnessTestCompleted(
              captured_id, HarnessTestStatus::kPassed, success_message);
          return;
        } else {
          const std::string error_message = absl::StrFormat(
              "Unsupported harness shortcut '%s'", widget_label);
          manager->AppendHarnessTestLog(captured_id, error_message);
          manager->MarkHarnessTestCompleted(
              captured_id, HarnessTestStatus::kFailed, error_message);
          return;
        }
 
        ctx->KeyPress(chord);
        ctx->Yield(2);
        if (ctx->IsError()) {
          const std::string error_message =
              absl::StrFormat("Shortcut failed for '%s'", widget_label);
          manager->AppendHarnessTestLog(captured_id, error_message);
          manager->MarkHarnessTestCompleted(
              captured_id, HarnessTestStatus::kFailed, error_message);
          return;
        }
        const std::string success_message =
            absl::StrFormat("Pressed shortcut '%s'", widget_label);
        manager->AppendHarnessTestLog(captured_id, success_message);
        manager->MarkHarnessTestCompleted(
            captured_id, HarnessTestStatus::kPassed, success_message);
        return;
      }
 
      const ImGuiTestRef widget_ref = widget_id != 0
                                          ? ImGuiTestRef(widget_id)
                                          : ImGuiTestRef(widget_label.c_str());
      const ImGuiTestItemInfo item =
          ctx->ItemInfo(widget_ref, ImGuiTestOpFlags_NoError);
      if (item.ID == 0) {
        const std::string error_message =
            absl::StrFormat("Click target '%s' not found", widget_label);
        manager->AppendHarnessTestLog(captured_id, error_message);
        manager->MarkHarnessTestCompleted(
            captured_id, HarnessTestStatus::kFailed, error_message);
        return;
      }
 
      if (click_type == ClickRequest::CLICK_TYPE_DOUBLE) {
        ctx->ItemDoubleClick(ImGuiTestRef(item.ID));
      } else {
        ctx->ItemClick(ImGuiTestRef(item.ID), mouse_button);
      }
      ctx->Yield();
      if (ctx->IsError()) {
        const std::string error_message =
            absl::StrFormat("Click failed for '%s'", widget_label);
        manager->AppendHarnessTestLog(captured_id, error_message);
        manager->MarkHarnessTestCompleted(
            captured_id, HarnessTestStatus::kFailed, error_message);
        return;
      }
      const std::string success_message =
          absl::StrFormat("Clicked %s '%s'", widget_type, widget_label);
      manager->AppendHarnessTestLog(captured_id, success_message);
      manager->MarkHarnessTestCompleted(captured_id, HarnessTestStatus::kPassed,
                                        success_message);
    } catch (const std::exception& e) {
      const std::string error_message =
          absl::StrFormat("Click failed: %s", e.what());
      manager->AppendHarnessTestLog(captured_id, error_message);
      manager->MarkHarnessTestCompleted(captured_id, HarnessTestStatus::kFailed,
                                        error_message);
    }
  };
 
  std::string test_name = absl::StrFormat(
      "grpc_click_%lld",
      static_cast<long long>(
          std::chrono::system_clock::now().time_since_epoch().count()));
 
  ImGuiTest* test = IM_REGISTER_TEST(engine, "grpc", test_name.c_str());
  test->TestFunc = RunDynamicTest;
  test->UserData = test_data.get();
 
  ImGuiTestEngine_QueueTest(engine, test, ImGuiTestRunFlags_RunFromGui);
  KeepDynamicTestData(test_data);
 
  auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
      std::chrono::steady_clock::now() - start);
  std::string message =
      absl::StrFormat("Queued click on %s '%s'", widget_type, widget_label);
  response->set_success(true);
  response->set_message(message);
  response->set_execution_time_ms(elapsed.count());
  test_manager_->AppendHarnessTestLog(test_id, message);
 
#else
  const std::string message = "ImGuiTestEngine is not available in this build";
  test_manager_->MarkHarnessTestRunning(test_id);
  test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                          message);
  test_manager_->AppendHarnessTestLog(test_id, message);
 
  auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
      std::chrono::steady_clock::now() - start);
  response->set_success(false);
  response->set_message(message);
  response->set_execution_time_ms(elapsed.count());
#endif
 
  return finalize(absl::OkStatus());
}
 
absl::Status ImGuiTestHarnessServiceImpl::Type(const TypeRequest* request,
                                               TypeResponse* response) {
  auto start = std::chrono::steady_clock::now();
 
  TestRecorder::RecordedStep recorded_step;
  recorded_step.type = TestRecorder::ActionType::kType;
  if (request) {
    recorded_step.widget_key = request->widget_key();
    if (!request->widget_key().empty()) {
      recorded_step.target = absl::StrCat("widget_key:", request->widget_key());
    } else {
      recorded_step.target = request->target();
    }
    recorded_step.text = request->text();
    recorded_step.clear_first = request->clear_first();
  }
 
  auto finalize = [&](const absl::Status& status) {
    recorded_step.success = response->success();
    recorded_step.message = response->message();
    recorded_step.execution_time_ms = response->execution_time_ms();
    recorded_step.test_id = response->test_id();
    MaybeRecordStep(&test_recorder_, recorded_step);
    return status;
  };
 
  if (!test_manager_) {
    response->set_success(false);
    response->set_message("TestManager not available");
    response->set_execution_time_ms(0);
    return finalize(absl::FailedPreconditionError("TestManager not available"));
  }
 
  const std::string requested_target = request ? request->target() : "";
  const std::string requested_widget_key = request ? request->widget_key() : "";
  const std::string request_summary =
      !requested_widget_key.empty()
          ? absl::StrFormat("widget_key:%s", requested_widget_key)
          : requested_target;
 
  const std::string test_id = test_manager_->RegisterHarnessTest(
      absl::StrFormat("Type: %s", request_summary), "grpc");
  response->set_test_id(test_id);
  recorded_step.test_id = test_id;
  test_manager_->AppendHarnessTestLog(
      test_id, absl::StrFormat("Queued type request: %s", request_summary));
 
  absl::StatusOr<internal::ResolvedWidgetSelector> resolved_selector =
      internal::ResolveWidgetSelector(requested_target, requested_widget_key);
  if (!resolved_selector.ok()) {
    auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
        std::chrono::steady_clock::now() - start);
    std::string message = std::string(resolved_selector.status().message());
    response->set_success(false);
    response->set_message(message);
    response->set_execution_time_ms(elapsed.count());
    test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                            message);
    test_manager_->AppendHarnessTestLog(test_id, message);
    return finalize(absl::OkStatus());
  }
 
  if (!resolved_selector->resolved_widget_key.empty()) {
    response->set_resolved_widget_key(resolved_selector->resolved_widget_key);
  }
  if (!resolved_selector->resolved_path.empty()) {
    response->set_resolved_path(resolved_selector->resolved_path);
  }
 
  const std::string widget_type = resolved_selector->target.type;
  const std::string widget_label = resolved_selector->target.label;
  const ImGuiID widget_id = resolved_selector->imgui_id;
 
#if defined(YAZE_ENABLE_IMGUI_TEST_ENGINE) && YAZE_ENABLE_IMGUI_TEST_ENGINE
  ImGuiTestEngine* engine = test_manager_->GetUITestEngine();
  if (!engine) {
    auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
        std::chrono::steady_clock::now() - start);
    std::string message = "ImGuiTestEngine not initialized";
    response->set_success(false);
    response->set_message(message);
    response->set_execution_time_ms(elapsed.count());
    test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                            message);
    return finalize(absl::OkStatus());
  }
  std::string text = request->text();
  bool clear_first = request->clear_first();
 
  auto rpc_state = std::make_shared<RPCState<bool>>();
  auto test_data = std::make_shared<DynamicTestData>();
  TestManager* manager = test_manager_;
  test_data->test_func = [manager, captured_id = test_id, widget_type,
                          widget_label, widget_id, clear_first, text,
                          rpc_state](ImGuiTestContext* ctx) {
    manager->MarkHarnessTestRunning(captured_id);
    try {
      const ImGuiTestRef widget_ref = widget_id != 0
                                          ? ImGuiTestRef(widget_id)
                                          : ImGuiTestRef(widget_label.c_str());
      ImGuiTestItemInfo item =
          ctx->ItemInfo(widget_ref, ImGuiTestOpFlags_NoError);
      if (item.ID == 0) {
        std::string error_message =
            absl::StrFormat("Input field '%s' not found", widget_label);
        manager->AppendHarnessTestLog(captured_id, error_message);
        manager->MarkHarnessTestCompleted(
            captured_id, HarnessTestStatus::kFailed, error_message);
        rpc_state->SetResult(false, error_message);
        return;
      }
 
      ctx->ItemClick(ImGuiTestRef(item.ID));
      if (clear_first) {
        ctx->KeyPress(ImGuiMod_Shortcut | ImGuiKey_A);
        ctx->KeyPress(ImGuiKey_Delete);
      }
 
      ctx->ItemInputValue(ImGuiTestRef(item.ID), text.c_str());
      ctx->Yield();
      if (ctx->IsError()) {
        const std::string error_message =
            absl::StrFormat("Type failed for '%s'", widget_label);
        manager->AppendHarnessTestLog(captured_id, error_message);
        manager->MarkHarnessTestCompleted(
            captured_id, HarnessTestStatus::kFailed, error_message);
        rpc_state->SetResult(false, error_message);
        return;
      }
 
      std::string success_message =
          absl::StrFormat("Typed '%s' into %s '%s'%s", text, widget_type,
                          widget_label, clear_first ? " (cleared first)" : "");
      manager->AppendHarnessTestLog(captured_id, success_message);
      manager->MarkHarnessTestCompleted(captured_id, HarnessTestStatus::kPassed,
                                        success_message);
      rpc_state->SetResult(true, success_message);
    } catch (const std::exception& e) {
      std::string error_message = absl::StrFormat("Type failed: %s", e.what());
      manager->AppendHarnessTestLog(captured_id, error_message);
      manager->MarkHarnessTestCompleted(captured_id, HarnessTestStatus::kFailed,
                                        error_message);
      rpc_state->SetResult(false, error_message);
    }
  };
 
  std::string test_name = absl::StrFormat(
      "grpc_type_%lld",
      static_cast<long long>(
          std::chrono::system_clock::now().time_since_epoch().count()));
 
  ImGuiTest* test = IM_REGISTER_TEST(engine, "grpc", test_name.c_str());
  test->TestFunc = RunDynamicTest;
  test->UserData = test_data.get();
 
  ImGuiTestEngine_QueueTest(engine, test, ImGuiTestRunFlags_RunFromGui);
  KeepDynamicTestData(test_data);
 
  auto timeout = std::chrono::seconds(5);
  auto wait_start = std::chrono::steady_clock::now();
  while (!rpc_state->completed.load()) {
    if (std::chrono::steady_clock::now() - wait_start > timeout) {
      std::string error_message =
          "Test timeout - input field not found or unresponsive";
      manager->AppendHarnessTestLog(test_id, error_message);
      manager->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kTimeout,
                                        error_message);
      rpc_state->SetResult(false, error_message);
      break;
    }
    std::this_thread::sleep_for(std::chrono::milliseconds(100));
  }
 
  bool success = false;
  std::string message;
  rpc_state->GetResult(success, message);
  auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
      std::chrono::steady_clock::now() - start);
 
  response->set_success(success);
  response->set_message(message);
  response->set_execution_time_ms(elapsed.count());
  if (!message.empty()) {
    test_manager_->AppendHarnessTestLog(test_id, message);
  }
 
#else
  test_manager_->MarkHarnessTestRunning(test_id);
  const std::string message = "ImGuiTestEngine is not available in this build";
  test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                          message);
  test_manager_->AppendHarnessTestLog(test_id, message);
 
  auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
      std::chrono::steady_clock::now() - start);
  response->set_success(false);
  response->set_message(message);
  response->set_execution_time_ms(elapsed.count());
#endif
 
  return finalize(absl::OkStatus());
}
 
absl::Status ImGuiTestHarnessServiceImpl::Wait(const WaitRequest* request,
                                               WaitResponse* response) {
  auto start = std::chrono::steady_clock::now();
 
  TestRecorder::RecordedStep recorded_step;
  recorded_step.type = TestRecorder::ActionType::kWait;
  if (request) {
    recorded_step.widget_key = request->widget_key();
    if (!request->widget_key().empty() && request->condition().empty()) {
      recorded_step.condition =
          absl::StrCat("widget_key:", request->widget_key());
    } else {
      recorded_step.condition = request->condition();
    }
    recorded_step.timeout_ms = request->timeout_ms();
  }
 
  auto finalize = [&](const absl::Status& status) {
    recorded_step.success = response->success();
    recorded_step.message = response->message();
    recorded_step.execution_time_ms = response->elapsed_ms();
    recorded_step.test_id = response->test_id();
    MaybeRecordStep(&test_recorder_, recorded_step);
    return status;
  };
 
  if (!test_manager_) {
    response->set_success(false);
    response->set_message("TestManager not available");
    response->set_elapsed_ms(0);
    return finalize(absl::FailedPreconditionError("TestManager not available"));
  }
 
  const std::string requested_condition = request ? request->condition() : "";
  const std::string requested_widget_key = request ? request->widget_key() : "";
  const std::string request_summary =
      !requested_widget_key.empty()
          ? absl::StrFormat("%s [widget_key:%s]",
                            requested_condition.empty()
                                ? "element_visible:<auto>"
                                : requested_condition,
                            requested_widget_key)
          : requested_condition;
 
  const std::string test_id = test_manager_->RegisterHarnessTest(
      absl::StrFormat("Wait: %s", request_summary), "grpc");
  response->set_test_id(test_id);
  recorded_step.test_id = test_id;
  test_manager_->AppendHarnessTestLog(
      test_id, absl::StrFormat("Queued wait condition: %s", request_summary));
 
  internal::ResolvedWidgetSelector resolved_selector;
  absl::StatusOr<ParsedCondition> parsed_condition = ResolveCondition(
      requested_condition, requested_widget_key,
      /*default_type_for_widget=*/"element_visible", &resolved_selector);
  if (!parsed_condition.ok()) {
    auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
        std::chrono::steady_clock::now() - start);
    std::string message = std::string(parsed_condition.status().message());
    response->set_success(false);
    response->set_message(message);
    response->set_elapsed_ms(elapsed.count());
    test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                            message);
    test_manager_->AppendHarnessTestLog(test_id, message);
    return finalize(absl::OkStatus());
  }
 
  if (!resolved_selector.resolved_widget_key.empty()) {
    response->set_resolved_widget_key(resolved_selector.resolved_widget_key);
  }
  if (!resolved_selector.resolved_path.empty()) {
    response->set_resolved_path(resolved_selector.resolved_path);
  }
 
  const std::string condition_type = parsed_condition->type;
  const std::string condition_target = parsed_condition->target;
  const ImGuiID widget_id = resolved_selector.imgui_id;
  const std::string resolved_condition =
      absl::StrFormat("%s:%s", condition_type, condition_target);
 
#if defined(YAZE_ENABLE_IMGUI_TEST_ENGINE) && YAZE_ENABLE_IMGUI_TEST_ENGINE
  ImGuiTestEngine* engine = test_manager_->GetUITestEngine();
  if (!engine) {
    auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
        std::chrono::steady_clock::now() - start);
    std::string message = "ImGuiTestEngine not initialized";
    response->set_success(false);
    response->set_message(message);
    response->set_elapsed_ms(elapsed.count());
    test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                            message);
    return finalize(absl::OkStatus());
  }
  int timeout_ms = request->timeout_ms() > 0 ? request->timeout_ms() : 5000;
  int poll_interval_ms =
      request->poll_interval_ms() > 0 ? request->poll_interval_ms() : 100;
 
  auto test_data = std::make_shared<DynamicTestData>();
  TestManager* manager = test_manager_;
  test_data->test_func = [manager, captured_id = test_id, condition_type,
                          condition_target, widget_id, timeout_ms,
                          poll_interval_ms](ImGuiTestContext* ctx) {
    manager->MarkHarnessTestRunning(captured_id);
    auto poll_start = std::chrono::steady_clock::now();
    auto timeout = std::chrono::milliseconds(timeout_ms);
 
    for (int i = 0; i < 10; ++i) {
      ctx->Yield();
    }
 
    try {
      while (std::chrono::steady_clock::now() - poll_start < timeout) {
        bool current_state = false;
        const ImGuiTestRef widget_ref =
            widget_id != 0 ? ImGuiTestRef(widget_id)
                           : ImGuiTestRef(condition_target.c_str());
 
        if (condition_type == "window_visible") {
          ImGuiTestItemInfo window_info =
              ctx->WindowInfo(widget_ref, ImGuiTestOpFlags_NoError);
          current_state = (window_info.ID != 0);
        } else if (condition_type == "element_visible") {
          ImGuiTestItemInfo item =
              ctx->ItemInfo(widget_ref, ImGuiTestOpFlags_NoError);
          current_state = (item.ID != 0 && item.RectClipped.GetWidth() > 0 &&
                           item.RectClipped.GetHeight() > 0);
        } else if (condition_type == "element_enabled") {
          ImGuiTestItemInfo item =
              ctx->ItemInfo(widget_ref, ImGuiTestOpFlags_NoError);
          current_state =
              (item.ID != 0 && !(item.ItemFlags & ImGuiItemFlags_Disabled));
        } else {
          std::string error_message =
              absl::StrFormat("Unknown condition type: %s", condition_type);
          manager->AppendHarnessTestLog(captured_id, error_message);
          manager->MarkHarnessTestCompleted(
              captured_id, HarnessTestStatus::kFailed, error_message);
          return;
        }
 
        if (current_state) {
          auto elapsed_ms =
              std::chrono::duration_cast<std::chrono::milliseconds>(
                  std::chrono::steady_clock::now() - poll_start);
          std::string success_message = absl::StrFormat(
              "Condition '%s:%s' met after %lld ms", condition_type,
              condition_target, static_cast<long long>(elapsed_ms.count()));
          manager->AppendHarnessTestLog(captured_id, success_message);
          manager->MarkHarnessTestCompleted(
              captured_id, HarnessTestStatus::kPassed, success_message);
          return;
        }
 
        std::this_thread::sleep_for(
            std::chrono::milliseconds(poll_interval_ms));
        ctx->Yield();
      }
 
      std::string timeout_message =
          absl::StrFormat("Condition '%s:%s' not met after %d ms timeout",
                          condition_type, condition_target, timeout_ms);
      manager->AppendHarnessTestLog(captured_id, timeout_message);
      manager->MarkHarnessTestCompleted(
          captured_id, HarnessTestStatus::kTimeout, timeout_message);
    } catch (const std::exception& e) {
      std::string error_message = absl::StrFormat("Wait failed: %s", e.what());
      manager->AppendHarnessTestLog(captured_id, error_message);
      manager->MarkHarnessTestCompleted(captured_id, HarnessTestStatus::kFailed,
                                        error_message);
    }
  };
 
  std::string test_name = absl::StrFormat(
      "grpc_wait_%lld",
      static_cast<long long>(
          std::chrono::system_clock::now().time_since_epoch().count()));
 
  ImGuiTest* test = IM_REGISTER_TEST(engine, "grpc", test_name.c_str());
  test->TestFunc = RunDynamicTest;
  test->UserData = test_data.get();
 
  ImGuiTestEngine_QueueTest(engine, test, ImGuiTestRunFlags_RunFromGui);
  KeepDynamicTestData(test_data);
 
  auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
      std::chrono::steady_clock::now() - start);
  std::string message =
      absl::StrFormat("Queued wait for '%s'", resolved_condition);
  response->set_success(true);
  response->set_message(message);
  response->set_elapsed_ms(elapsed.count());
  test_manager_->AppendHarnessTestLog(test_id, message);
 
#else
  test_manager_->MarkHarnessTestRunning(test_id);
  const std::string message = "ImGuiTestEngine is not available in this build";
  test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                          message);
  test_manager_->AppendHarnessTestLog(test_id, message);
 
  auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
      std::chrono::steady_clock::now() - start);
  response->set_success(false);
  response->set_message(message);
  response->set_elapsed_ms(elapsed.count());
#endif
 
  return finalize(absl::OkStatus());
}
 
absl::Status ImGuiTestHarnessServiceImpl::Assert(const AssertRequest* request,
                                                 AssertResponse* response) {
  TestRecorder::RecordedStep recorded_step;
  recorded_step.type = TestRecorder::ActionType::kAssert;
  if (request) {
    recorded_step.widget_key = request->widget_key();
    if (!request->widget_key().empty() && request->condition().empty()) {
      recorded_step.condition =
          absl::StrCat("widget_key:", request->widget_key());
    } else {
      recorded_step.condition = request->condition();
    }
  }
 
  auto finalize = [&](const absl::Status& status) {
    recorded_step.success = response->success();
    recorded_step.message = response->message();
    recorded_step.expected_value = response->expected_value();
    recorded_step.actual_value = response->actual_value();
    recorded_step.test_id = response->test_id();
    MaybeRecordStep(&test_recorder_, recorded_step);
    return status;
  };
 
  if (!test_manager_) {
    response->set_success(false);
    response->set_message("TestManager not available");
    response->set_actual_value("N/A");
    response->set_expected_value("N/A");
    return finalize(absl::FailedPreconditionError("TestManager not available"));
  }
 
  const std::string requested_condition = request ? request->condition() : "";
  const std::string requested_widget_key = request ? request->widget_key() : "";
  const std::string request_summary =
      !requested_widget_key.empty()
          ? absl::StrFormat("%s [widget_key:%s]",
                            requested_condition.empty() ? "exists:<auto>"
                                                        : requested_condition,
                            requested_widget_key)
          : requested_condition;
 
  const std::string test_id = test_manager_->RegisterHarnessTest(
      absl::StrFormat("Assert: %s", request_summary), "grpc");
  response->set_test_id(test_id);
  recorded_step.test_id = test_id;
  test_manager_->AppendHarnessTestLog(
      test_id, absl::StrFormat("Queued assertion: %s", request_summary));
 
  internal::ResolvedWidgetSelector resolved_selector;
  absl::StatusOr<ParsedCondition> parsed_condition = ResolveCondition(
      requested_condition, requested_widget_key,
      /*default_type_for_widget=*/"exists", &resolved_selector);
  if (!parsed_condition.ok()) {
    std::string message = std::string(parsed_condition.status().message());
    response->set_success(false);
    response->set_message(message);
    response->set_actual_value("N/A");
    response->set_expected_value("N/A");
    test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                            message);
    test_manager_->AppendHarnessTestLog(test_id, message);
    return finalize(absl::OkStatus());
  }
 
  if (!resolved_selector.resolved_widget_key.empty()) {
    response->set_resolved_widget_key(resolved_selector.resolved_widget_key);
  }
  if (!resolved_selector.resolved_path.empty()) {
    response->set_resolved_path(resolved_selector.resolved_path);
  }
 
  const std::string assertion_type = parsed_condition->type;
  const std::string assertion_target = parsed_condition->target;
  const ImGuiID widget_id = resolved_selector.imgui_id;
  const std::string widget_label = resolved_selector.target.label;
 
  if (assertion_type == "value_equals" && widget_id == 0) {
    const std::string message =
        "value_equals requires widget_key so the input is unambiguous";
    response->set_success(false);
    response->set_message(message);
    response->set_actual_value("N/A");
    response->set_expected_value(assertion_target);
    test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                            message);
    test_manager_->AppendHarnessTestLog(test_id, message);
    return finalize(absl::OkStatus());
  }
 
#if defined(YAZE_ENABLE_IMGUI_TEST_ENGINE) && YAZE_ENABLE_IMGUI_TEST_ENGINE
  ImGuiTestEngine* engine = test_manager_->GetUITestEngine();
  if (!engine) {
    std::string message = "ImGuiTestEngine not initialized";
    response->set_success(false);
    response->set_message(message);
    response->set_actual_value("N/A");
    response->set_expected_value("N/A");
    test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                            message);
    return finalize(absl::OkStatus());
  }
 
  auto test_data = std::make_shared<DynamicTestData>();
  TestManager* manager = test_manager_;
  test_data->test_func = [manager, captured_id = test_id, assertion_type,
                          assertion_target, widget_id,
                          widget_label](ImGuiTestContext* ctx) {
    manager->MarkHarnessTestRunning(captured_id);
 
    auto complete_with = [manager, captured_id](bool passed,
                                                const std::string& message,
                                                const std::string& actual,
                                                const std::string& expected,
                                                HarnessTestStatus status) {
      manager->AppendHarnessTestLog(captured_id, message);
      if (!actual.empty() || !expected.empty()) {
        manager->AppendHarnessTestLog(
            captured_id,
            absl::StrFormat("Actual: %s | Expected: %s", actual, expected));
      }
      manager->MarkHarnessTestCompleted(captured_id, status,
                                        passed ? "" : message);
    };
 
    try {
      bool passed = false;
      std::string actual_value;
      std::string expected_value;
      std::string message;
      const ImGuiTestRef widget_ref =
          widget_id != 0 ? ImGuiTestRef(widget_id)
                         : ImGuiTestRef(assertion_target.c_str());
 
      if (assertion_type == "visible") {
        const ImGuiTestItemInfo item =
            widget_id != 0
                ? ctx->ItemInfo(widget_ref, ImGuiTestOpFlags_NoError)
                : ctx->WindowInfo(widget_ref, ImGuiTestOpFlags_NoError);
        const bool is_visible =
            item.ID != 0 &&
            (widget_id == 0 || (item.RectClipped.GetWidth() > 0 &&
                                item.RectClipped.GetHeight() > 0));
        passed = is_visible;
        actual_value = is_visible ? "visible" : "hidden";
        expected_value = "visible";
        const std::string target_description =
            widget_id != 0 ? widget_label : assertion_target;
        message =
            passed ? absl::StrFormat("'%s' is visible", target_description)
                   : absl::StrFormat("'%s' is not visible", target_description);
      } else if (assertion_type == "enabled") {
        ImGuiTestItemInfo item =
            ctx->ItemInfo(widget_ref, ImGuiTestOpFlags_NoError);
        bool is_enabled =
            (item.ID != 0 && !(item.ItemFlags & ImGuiItemFlags_Disabled));
        passed = is_enabled;
        actual_value = is_enabled ? "enabled" : "disabled";
        expected_value = "enabled";
        message =
            passed ? absl::StrFormat("'%s' is enabled", assertion_target)
                   : absl::StrFormat("'%s' is not enabled", assertion_target);
      } else if (assertion_type == "exists") {
        ImGuiTestItemInfo item =
            ctx->ItemInfo(widget_ref, ImGuiTestOpFlags_NoError);
        bool exists = (item.ID != 0);
        passed = exists;
        actual_value = exists ? "exists" : "not found";
        expected_value = "exists";
        message = passed ? absl::StrFormat("'%s' exists", assertion_target)
                         : absl::StrFormat("'%s' not found", assertion_target);
      } else if (assertion_type == "value_equals") {
        const ImGuiTestItemInfo item =
            ctx->ItemInfo(widget_ref, ImGuiTestOpFlags_NoError);
        if (item.ID == 0) {
          passed = false;
          actual_value = "not found";
          expected_value = assertion_target;
          message = absl::StrFormat("Input '%s' not found", widget_label);
        } else {
          const char* value = ctx->ItemReadAsString(ImGuiTestRef(item.ID));
          actual_value = value ? value : "";
          expected_value = assertion_target;
          passed = !ctx->IsError() && actual_value == expected_value;
          message = passed ? absl::StrFormat("'%s' equals '%s'", widget_label,
                                             expected_value)
                           : absl::StrFormat(
                                 "'%s' does not equal '%s' (actual: '%s')",
                                 widget_label, expected_value, actual_value);
        }
      } else if (assertion_type == "text_contains") {
        size_t second_colon = assertion_target.find(':');
        if (widget_id == 0 && second_colon == std::string::npos) {
          std::string error_message =
              "text_contains requires format "
              "'text_contains:target:expected_text'";
          complete_with(false, error_message, "N/A", "N/A",
                        HarnessTestStatus::kFailed);
          return;
        }
 
        const std::string input_target =
            widget_id != 0 ? widget_label
                           : assertion_target.substr(0, second_colon);
        const std::string expected_text =
            widget_id != 0 ? assertion_target
                           : assertion_target.substr(second_colon + 1);
        const ImGuiTestRef input_ref = widget_id != 0
                                           ? ImGuiTestRef(widget_id)
                                           : ImGuiTestRef(input_target.c_str());
 
        ImGuiTestItemInfo item =
            ctx->ItemInfo(input_ref, ImGuiTestOpFlags_NoError);
        if (item.ID != 0) {
          const char* value = ctx->ItemReadAsString(ImGuiTestRef(item.ID));
          const std::string actual_text = value ? value : "";
          passed = actual_text.find(expected_text) != std::string::npos;
          actual_value = actual_text;
          expected_value = absl::StrFormat("contains '%s'", expected_text);
          message = passed ? absl::StrFormat("'%s' contains '%s'", input_target,
                                             expected_text)
                           : absl::StrFormat(
                                 "'%s' does not contain '%s' (actual: '%s')",
                                 input_target, expected_text, actual_text);
        } else {
          passed = false;
          actual_value = "not found";
          expected_value = expected_text;
          message = absl::StrFormat("Input '%s' not found", input_target);
        }
      } else {
        std::string error_message =
            absl::StrFormat("Unknown assertion type: %s", assertion_type);
        complete_with(false, error_message, "N/A", "N/A",
                      HarnessTestStatus::kFailed);
        return;
      }
 
      complete_with(
          passed, message, actual_value, expected_value,
          passed ? HarnessTestStatus::kPassed : HarnessTestStatus::kFailed);
    } catch (const std::exception& e) {
      std::string error_message =
          absl::StrFormat("Assertion failed: %s", e.what());
      manager->AppendHarnessTestLog(captured_id, error_message);
      manager->MarkHarnessTestCompleted(captured_id, HarnessTestStatus::kFailed,
                                        error_message);
    }
  };
 
  std::string test_name = absl::StrFormat(
      "grpc_assert_%lld",
      static_cast<long long>(
          std::chrono::system_clock::now().time_since_epoch().count()));
 
  ImGuiTest* test = IM_REGISTER_TEST(engine, "grpc", test_name.c_str());
  test->TestFunc = RunDynamicTest;
  test->UserData = test_data.get();
 
  ImGuiTestEngine_QueueTest(engine, test, ImGuiTestRunFlags_RunFromGui);
  KeepDynamicTestData(test_data);
 
  response->set_success(true);
  std::string message = absl::StrFormat("Queued assertion for '%s:%s'",
                                        assertion_type, assertion_target);
  response->set_message(message);
  response->set_actual_value("(async)");
  response->set_expected_value("(async)");
  test_manager_->AppendHarnessTestLog(test_id, message);
 
#else
  test_manager_->MarkHarnessTestRunning(test_id);
  const std::string message = "ImGuiTestEngine is not available in this build";
  test_manager_->MarkHarnessTestCompleted(test_id, HarnessTestStatus::kFailed,
                                          message);
  test_manager_->AppendHarnessTestLog(test_id, message);
 
  response->set_success(false);
  response->set_message(message);
  response->set_actual_value("(stub)");
  response->set_expected_value("(stub)");
#endif
 
  return finalize(absl::OkStatus());
}
 
absl::Status ImGuiTestHarnessServiceImpl::FlushUiActions(
    const FlushUiActionsRequest* request, FlushUiActionsResponse* response) {
  if (!response) {
    return absl::InvalidArgumentError("response cannot be null");
  }
 
  const int32_t timeout_ms =
      NormalizeTimeoutMs(request ? request->timeout_ms() : 0, 2000);
  const absl::Duration timeout = absl::Milliseconds(timeout_ms);
  const absl::Time deadline = absl::Now() + timeout;
  const absl::Time started_at = absl::Now();
 
  UiSyncSnapshot initial = CaptureUiSyncSnapshot(test_manager_);
  UiSyncSnapshot snapshot = initial;
 
  if (initial.pending_editor_actions == 0 &&
      initial.pending_layout_actions == 0 && !initial.layout_rebuild_pending) {
    response->set_success(true);
    response->set_message("UI queues already idle");
    response->set_elapsed_ms(0);
    response->set_frame_id(initial.frame_id);
    response->set_pending_editor_actions(initial.pending_editor_actions);
    response->set_pending_layout_actions(initial.pending_layout_actions);
    response->set_layout_rebuild_pending(initial.layout_rebuild_pending);
    response->set_pending_harness_tests(initial.pending_harness_tests);
    response->set_queued_harness_tests(initial.queued_harness_tests);
    response->set_running_harness_tests(initial.running_harness_tests);
    return absl::OkStatus();
  }
 
  while (absl::Now() <= deadline) {
    snapshot = CaptureUiSyncSnapshot(test_manager_);
    const bool frame_advanced = snapshot.frame_id > initial.frame_id;
    const bool queues_flushed = snapshot.pending_editor_actions == 0 &&
                                snapshot.pending_layout_actions == 0 &&
                                !snapshot.layout_rebuild_pending;
    if (frame_advanced && queues_flushed) {
      break;
    }
    absl::SleepFor(absl::Milliseconds(10));
  }
 
  const int32_t elapsed_ms = ClampDurationToInt32(absl::Now() - started_at);
  const bool success = snapshot.pending_editor_actions == 0 &&
                       snapshot.pending_layout_actions == 0 &&
                       !snapshot.layout_rebuild_pending;
  response->set_success(success);
  response->set_message(success ? "Deferred UI actions flushed"
                                : "Timed out waiting for UI action flush");
  response->set_elapsed_ms(elapsed_ms);
  response->set_frame_id(snapshot.frame_id);
  response->set_pending_editor_actions(snapshot.pending_editor_actions);
  response->set_pending_layout_actions(snapshot.pending_layout_actions);
  response->set_layout_rebuild_pending(snapshot.layout_rebuild_pending);
  response->set_pending_harness_tests(snapshot.pending_harness_tests);
  response->set_queued_harness_tests(snapshot.queued_harness_tests);
  response->set_running_harness_tests(snapshot.running_harness_tests);
  return absl::OkStatus();
}
 
absl::Status ImGuiTestHarnessServiceImpl::WaitForIdle(
    const WaitForIdleRequest* request, WaitForIdleResponse* response) {
  if (!response) {
    return absl::InvalidArgumentError("response cannot be null");
  }
 
  const int32_t timeout_ms =
      NormalizeTimeoutMs(request ? request->timeout_ms() : 0, 5000);
  const int32_t required_stable_frames =
      std::max(1, request ? request->stable_frames() : 2);
  const int32_t poll_interval_ms =
      NormalizePollIntervalMs(request ? request->poll_interval_ms() : 25);
  const bool flush_first = request ? request->flush_first() : false;
 
  const absl::Time started_at = absl::Now();
  const absl::Time deadline = started_at + absl::Milliseconds(timeout_ms);
 
  if (flush_first) {
    FlushUiActionsRequest flush_request;
    flush_request.set_timeout_ms(std::min(timeout_ms, 2000));
    FlushUiActionsResponse flush_response;
    absl::Status flush_status = FlushUiActions(&flush_request, &flush_response);
    if (!flush_status.ok()) {
      return flush_status;
    }
  }
 
  UiSyncSnapshot snapshot = CaptureUiSyncSnapshot(test_manager_);
  int32_t stable_frames_observed = 0;
  int64_t last_frame_id = -1;
 
  while (absl::Now() <= deadline) {
    snapshot = CaptureUiSyncSnapshot(test_manager_);
 
    const bool is_idle_now = IsUiIdle(snapshot);
    if (snapshot.frame_id != last_frame_id) {
      last_frame_id = snapshot.frame_id;
      stable_frames_observed = is_idle_now ? (stable_frames_observed + 1) : 0;
    } else if (!is_idle_now) {
      stable_frames_observed = 0;
    }
 
    if (stable_frames_observed >= required_stable_frames) {
      response->set_success(true);
      response->set_message(absl::StrFormat(
          "UI idle for %d consecutive frame(s)", stable_frames_observed));
      response->set_elapsed_ms(ClampDurationToInt32(absl::Now() - started_at));
      response->set_last_frame_id(snapshot.frame_id);
      response->set_stable_frames_observed(stable_frames_observed);
      response->set_pending_editor_actions(snapshot.pending_editor_actions);
      response->set_pending_layout_actions(snapshot.pending_layout_actions);
      response->set_layout_rebuild_pending(snapshot.layout_rebuild_pending);
      response->set_pending_harness_tests(snapshot.pending_harness_tests);
      response->set_queued_harness_tests(snapshot.queued_harness_tests);
      response->set_running_harness_tests(snapshot.running_harness_tests);
      response->set_timeout_reason("");
      return absl::OkStatus();
    }
 
    absl::SleepFor(absl::Milliseconds(poll_interval_ms));
  }
 
  const std::string timeout_reason = BuildIdleTimeoutReason(snapshot);
  response->set_success(false);
  response->set_message("Timed out waiting for deterministic UI idle");
  response->set_elapsed_ms(ClampDurationToInt32(absl::Now() - started_at));
  response->set_last_frame_id(snapshot.frame_id);
  response->set_stable_frames_observed(stable_frames_observed);
  response->set_pending_editor_actions(snapshot.pending_editor_actions);
  response->set_pending_layout_actions(snapshot.pending_layout_actions);
  response->set_layout_rebuild_pending(snapshot.layout_rebuild_pending);
  response->set_pending_harness_tests(snapshot.pending_harness_tests);
  response->set_queued_harness_tests(snapshot.queued_harness_tests);
  response->set_running_harness_tests(snapshot.running_harness_tests);
  response->set_timeout_reason(timeout_reason);
  return absl::OkStatus();
}
 
absl::Status ImGuiTestHarnessServiceImpl::GetUiSyncState(
    const GetUiSyncStateRequest* /*request*/,
    GetUiSyncStateResponse* response) {
  if (!response) {
    return absl::InvalidArgumentError("response cannot be null");
  }
 
  const UiSyncSnapshot snapshot = CaptureUiSyncSnapshot(test_manager_);
  response->set_frame_id(snapshot.frame_id);
  response->set_pending_editor_actions(snapshot.pending_editor_actions);
  response->set_pending_layout_actions(snapshot.pending_layout_actions);
  response->set_layout_rebuild_pending(snapshot.layout_rebuild_pending);
  response->set_pending_harness_tests(snapshot.pending_harness_tests);
  response->set_queued_harness_tests(snapshot.queued_harness_tests);
  response->set_running_harness_tests(snapshot.running_harness_tests);
  response->set_sampled_at_ms(absl::ToUnixMillis(absl::Now()));
  response->set_idle(IsUiIdle(snapshot));
  return absl::OkStatus();
}
 
absl::Status ImGuiTestHarnessServiceImpl::Screenshot(
    const ScreenshotRequest* request, ScreenshotResponse* response) {
  if (!response) {
    return absl::InvalidArgumentError("response cannot be null");
  }
 
  const std::string requested_path =
      request ? request->output_path() : std::string();
 
  // We must execute capture on the main thread to avoid Metal/OpenGL context errors.
  // Use Controller's request queue.
  struct State {
    std::atomic<bool> done{false};
    absl::StatusOr<ScreenshotArtifact> result =
        absl::UnknownError("Not captured");
  };
  auto state = std::make_shared<State>();
 
  Application::Instance().GetController()->RequestScreenshot(
      {.preferred_path = requested_path,
       .reveal_to_user = false,
       .callback = [state](absl::StatusOr<ScreenshotArtifact> result) {
         state->result = std::move(result);
         state->done.store(true);
       }});
 
  // Wait for main thread to process (timeout after 5s)
  auto start = std::chrono::steady_clock::now();
  while (!state->done.load()) {
    if (std::chrono::steady_clock::now() - start > std::chrono::seconds(5)) {
      return absl::DeadlineExceededError(
          "Timed out waiting for screenshot capture on main thread");
    }
    std::this_thread::sleep_for(std::chrono::milliseconds(10));
  }
 
  if (!state->result.ok()) {
    response->set_success(false);
    response->set_message(std::string(state->result.status().message()));
    return state->result.status();
  }
 
  const ScreenshotArtifact& artifact = *state->result;
  response->set_success(true);
  response->set_message(absl::StrFormat("Screenshot saved to %s (%dx%d)",
                                        artifact.file_path, artifact.width,
                                        artifact.height));
  response->set_file_path(artifact.file_path);
  response->set_file_size_bytes(artifact.file_size_bytes);
 
  return absl::OkStatus();
}
 
absl::Status ImGuiTestHarnessServiceImpl::GetTestStatus(
    const GetTestStatusRequest* request, GetTestStatusResponse* response) {
  if (!test_manager_) {
    return absl::FailedPreconditionError("TestManager not available");
  }
 
  if (request->test_id().empty()) {
    return absl::InvalidArgumentError("test_id must be provided");
  }
 
  auto execution_or =
      test_manager_->GetHarnessTestExecution(request->test_id());
  if (!execution_or.ok()) {
    response->set_status(GetTestStatusResponse::TEST_STATUS_UNSPECIFIED);
    response->set_error_message(std::string(execution_or.status().message()));
    return absl::OkStatus();
  }
 
  const auto& execution = execution_or.value();
  response->set_status(ConvertHarnessStatus(execution.status));
  response->set_queued_at_ms(ToUnixMillisSafe(execution.queued_at));
  response->set_started_at_ms(ToUnixMillisSafe(execution.started_at));
  response->set_completed_at_ms(ToUnixMillisSafe(execution.completed_at));
  response->set_execution_time_ms(ClampDurationToInt32(execution.duration));
  if (!execution.error_message.empty()) {
    response->set_error_message(execution.error_message);
  } else {
    response->clear_error_message();
  }
 
  response->clear_assertion_failures();
  for (const auto& failure : execution.assertion_failures) {
    response->add_assertion_failures(failure);
  }
 
  return absl::OkStatus();
}
 
absl::Status ImGuiTestHarnessServiceImpl::ListTests(
    const ListTestsRequest* request, ListTestsResponse* response) {
  if (!test_manager_) {
    return absl::FailedPreconditionError("TestManager not available");
  }
 
  if (request->page_size() < 0) {
    return absl::InvalidArgumentError("page_size cannot be negative");
  }
 
  int page_size = request->page_size() > 0 ? request->page_size() : 100;
  constexpr int kMaxPageSize = 500;
  if (page_size > kMaxPageSize) {
    page_size = kMaxPageSize;
  }
 
  size_t start_index = 0;
  if (!request->page_token().empty()) {
    int64_t token_value = 0;
    if (!absl::SimpleAtoi(request->page_token(), &token_value) ||
        token_value < 0) {
      return absl::InvalidArgumentError("Invalid page_token");
    }
    start_index = static_cast<size_t>(token_value);
  }
 
  auto summaries =
      test_manager_->ListHarnessTestSummaries(request->category_filter());
 
  response->set_total_count(static_cast<int32_t>(summaries.size()));
 
  if (start_index >= summaries.size()) {
    response->clear_tests();
    response->clear_next_page_token();
    return absl::OkStatus();
  }
 
  size_t end_index =
      std::min(start_index + static_cast<size_t>(page_size), summaries.size());
 
  for (size_t i = start_index; i < end_index; ++i) {
    const auto& summary = summaries[i];
    auto* test_info = response->add_tests();
    const auto& exec = summary.latest_execution;
 
    test_info->set_test_id(exec.test_id);
    test_info->set_name(exec.name);
    test_info->set_category(exec.category);
 
    int64_t last_run_ms = ToUnixMillisSafe(exec.completed_at);
    if (last_run_ms == 0) {
      last_run_ms = ToUnixMillisSafe(exec.started_at);
    }
    if (last_run_ms == 0) {
      last_run_ms = ToUnixMillisSafe(exec.queued_at);
    }
    test_info->set_last_run_timestamp_ms(last_run_ms);
 
    test_info->set_total_runs(summary.total_runs);
    test_info->set_pass_count(summary.pass_count);
    test_info->set_fail_count(summary.fail_count);
 
    int32_t average_duration_ms = 0;
    if (summary.total_runs > 0) {
      absl::Duration average_duration =
          summary.total_duration / summary.total_runs;
      average_duration_ms = ClampDurationToInt32(average_duration);
    }
    test_info->set_average_duration_ms(average_duration_ms);
  }
 
  if (end_index < summaries.size()) {
    response->set_next_page_token(absl::StrCat(end_index));
  } else {
    response->clear_next_page_token();
  }
 
  return absl::OkStatus();
}
 
absl::Status ImGuiTestHarnessServiceImpl::GetTestResults(
    const GetTestResultsRequest* request, GetTestResultsResponse* response) {
  if (!test_manager_) {
    return absl::FailedPreconditionError("TestManager not available");
  }
 
  if (request->test_id().empty()) {
    return absl::InvalidArgumentError("test_id must be provided");
  }
 
  auto execution_or =
      test_manager_->GetHarnessTestExecution(request->test_id());
  if (!execution_or.ok()) {
    return execution_or.status();
  }
 
  const auto& execution = execution_or.value();
  response->set_success(execution.status == HarnessTestStatus::kPassed);
  response->set_test_name(execution.name);
  response->set_category(execution.category);
 
  int64_t executed_at_ms = ToUnixMillisSafe(execution.completed_at);
  if (executed_at_ms == 0) {
    executed_at_ms = ToUnixMillisSafe(execution.started_at);
  }
  if (executed_at_ms == 0) {
    executed_at_ms = ToUnixMillisSafe(execution.queued_at);
  }
  response->set_executed_at_ms(executed_at_ms);
  response->set_duration_ms(ClampDurationToInt32(execution.duration));
 
  response->clear_assertions();
  if (!execution.assertion_failures.empty()) {
    for (const auto& failure : execution.assertion_failures) {
      auto* assertion = response->add_assertions();
      assertion->set_description(failure);
      assertion->set_passed(false);
      assertion->set_error_message(failure);
    }
  } else if (!execution.error_message.empty()) {
    auto* assertion = response->add_assertions();
    assertion->set_description("Execution error");
    assertion->set_passed(false);
    assertion->set_error_message(execution.error_message);
  }
 
  if (request->include_logs()) {
    for (const auto& log_entry : execution.logs) {
      response->add_logs(log_entry);
    }
  }
 
  auto* metrics_map = response->mutable_metrics();
  for (const auto& [key, value] : execution.metrics) {
    (*metrics_map)[key] = value;
  }
 
  // IT-08b: Include failure diagnostics if available
  if (!execution.screenshot_path.empty()) {
    response->set_screenshot_path(execution.screenshot_path);
    response->set_screenshot_size_bytes(execution.screenshot_size_bytes);
  }
  if (!execution.failure_context.empty()) {
    response->set_failure_context(execution.failure_context);
  }
  if (!execution.widget_state.empty()) {
    response->set_widget_state(execution.widget_state);
  }
 
  return absl::OkStatus();
}
 
absl::Status ImGuiTestHarnessServiceImpl::DiscoverWidgets(
    const DiscoverWidgetsRequest* request, DiscoverWidgetsResponse* response) {
  if (!request) {
    return absl::InvalidArgumentError("request cannot be null");
  }
  if (!response) {
    return absl::InvalidArgumentError("response cannot be null");
  }
 
  if (!test_manager_) {
    return absl::FailedPreconditionError("TestManager not available");
  }
 
  widget_discovery_service_.CollectWidgets(/*ctx=*/nullptr, *request, response);
  return absl::OkStatus();
}
 
absl::Status ImGuiTestHarnessServiceImpl::StartRecording(
    const StartRecordingRequest* request, StartRecordingResponse* response) {
  if (!request) {
    return absl::InvalidArgumentError("request cannot be null");
  }
  if (!response) {
    return absl::InvalidArgumentError("response cannot be null");
  }
 
  TestRecorder::RecordingOptions options;
  options.output_path = request->output_path();
  options.session_name = request->session_name();
  options.description = request->description();
 
  if (options.output_path.empty()) {
    response->set_success(false);
    response->set_message("output_path is required to start recording");
    return absl::InvalidArgumentError("output_path cannot be empty");
  }
 
  absl::StatusOr<std::string> recording_id = test_recorder_.Start(options);
  if (!recording_id.ok()) {
    response->set_success(false);
    response->set_message(std::string(recording_id.status().message()));
    return recording_id.status();
  }
 
  response->set_success(true);
  response->set_message("Recording started");
  response->set_recording_id(*recording_id);
  response->set_started_at_ms(absl::ToUnixMillis(absl::Now()));
  return absl::OkStatus();
}
 
absl::Status ImGuiTestHarnessServiceImpl::StopRecording(
    const StopRecordingRequest* request, StopRecordingResponse* response) {
  if (!request) {
    return absl::InvalidArgumentError("request cannot be null");
  }
  if (!response) {
    return absl::InvalidArgumentError("response cannot be null");
  }
 
  absl::StatusOr<TestRecorder::StopRecordingSummary> summary =
      test_recorder_.Stop(request->recording_id(), request->discard());
  if (!summary.ok()) {
    response->set_success(false);
    response->set_message(std::string(summary.status().message()));
    return summary.status();
  }
 
  response->set_success(true);
  if (summary->saved) {
    response->set_message("Recording saved");
  } else {
    response->set_message("Recording discarded");
  }
  response->set_output_path(summary->output_path);
  response->set_step_count(summary->step_count);
  response->set_duration_ms(absl::ToInt64Milliseconds(summary->duration));
  return absl::OkStatus();
}
 
absl::Status ImGuiTestHarnessServiceImpl::ReplayTest(
    const ReplayTestRequest* request, ReplayTestResponse* response) {
  if (!request) {
    return absl::InvalidArgumentError("request cannot be null");
  }
  if (!response) {
    return absl::InvalidArgumentError("response cannot be null");
  }
 
  response->clear_logs();
  response->clear_assertions();
 
  if (request->script_path().empty()) {
    response->set_success(false);
    response->set_message("script_path is required");
    return absl::InvalidArgumentError("script_path cannot be empty");
  }
 
  absl::StatusOr<TestScript> script_or =
      TestScriptParser::ParseFromFile(request->script_path());
  if (!script_or.ok()) {
    response->set_success(false);
    response->set_message(std::string(script_or.status().message()));
    return script_or.status();
  }
  TestScript script = std::move(*script_or);
 
  absl::flat_hash_map<std::string, std::string> overrides;
  for (const auto& entry : request->parameter_overrides()) {
    overrides[entry.first] = entry.second;
  }
 
  response->set_replay_session_id(absl::StrFormat(
      "replay_%s",
      absl::FormatTime("%Y%m%dT%H%M%S", absl::Now(), absl::UTCTimeZone())));
 
  auto suspension = test_recorder_.Suspend();
 
  std::vector<std::string> logs;
  logs.reserve(script.steps.size() * 2 + 4);
 
  bool overall_success = true;
  std::string overall_message = "Replay completed successfully";
  int steps_executed = 0;
  absl::Status overall_rpc_status = absl::OkStatus();
 
  for (const auto& step : script.steps) {
    ++steps_executed;
    std::string action_label =
        absl::StrFormat("Step %d: %s", steps_executed, step.action);
    logs.push_back(action_label);
 
    absl::Status status = absl::OkStatus();
    bool step_success = false;
    std::string step_message;
    HarnessTestExecution execution;
    bool have_execution = false;
 
    if (step.action == "click") {
      ClickRequest sub_request;
      sub_request.set_target(ApplyOverrides(step.target, overrides));
      sub_request.set_widget_key(ApplyOverrides(step.widget_key, overrides));
      sub_request.set_type(ClickTypeFromString(step.click_type));
      ClickResponse sub_response;
      status = Click(&sub_request, &sub_response);
      step_success = sub_response.success();
      step_message = sub_response.message();
      if (status.ok() && !sub_response.test_id().empty()) {
        absl::Status wait_status = WaitForHarnessTestCompletion(
            test_manager_, sub_response.test_id(), &execution);
        if (wait_status.ok()) {
          have_execution = true;
          internal::ApplyTerminalHarnessExecution(execution, &step_success,
                                                  &step_message);
        } else {
          status = wait_status;
          step_success = false;
          step_message = std::string(wait_status.message());
        }
      }
    } else if (step.action == "type") {
      TypeRequest sub_request;
      sub_request.set_target(ApplyOverrides(step.target, overrides));
      sub_request.set_widget_key(ApplyOverrides(step.widget_key, overrides));
      sub_request.set_text(ApplyOverrides(step.text, overrides));
      sub_request.set_clear_first(step.clear_first);
      TypeResponse sub_response;
      status = Type(&sub_request, &sub_response);
      step_success = sub_response.success();
      step_message = sub_response.message();
      if (status.ok() && !sub_response.test_id().empty()) {
        absl::Status wait_status = WaitForHarnessTestCompletion(
            test_manager_, sub_response.test_id(), &execution);
        if (wait_status.ok()) {
          have_execution = true;
          internal::ApplyTerminalHarnessExecution(execution, &step_success,
                                                  &step_message);
        } else {
          status = wait_status;
          step_success = false;
          step_message = std::string(wait_status.message());
        }
      }
    } else if (step.action == "wait") {
      WaitRequest sub_request;
      sub_request.set_condition(ApplyOverrides(step.condition, overrides));
      sub_request.set_widget_key(ApplyOverrides(step.widget_key, overrides));
      if (step.timeout_ms > 0) {
        sub_request.set_timeout_ms(step.timeout_ms);
      }
      WaitResponse sub_response;
      status = Wait(&sub_request, &sub_response);
      step_success = sub_response.success();
      step_message = sub_response.message();
      if (status.ok() && !sub_response.test_id().empty()) {
        absl::Status wait_status = WaitForHarnessTestCompletion(
            test_manager_, sub_response.test_id(), &execution);
        if (wait_status.ok()) {
          have_execution = true;
          internal::ApplyTerminalHarnessExecution(execution, &step_success,
                                                  &step_message);
        } else {
          status = wait_status;
          step_success = false;
          step_message = std::string(wait_status.message());
        }
      }
    } else if (step.action == "assert") {
      AssertRequest sub_request;
      sub_request.set_condition(ApplyOverrides(step.condition, overrides));
      sub_request.set_widget_key(ApplyOverrides(step.widget_key, overrides));
      AssertResponse sub_response;
      status = Assert(&sub_request, &sub_response);
      step_success = sub_response.success();
      step_message = sub_response.message();
      if (status.ok() && !sub_response.test_id().empty()) {
        absl::Status wait_status = WaitForHarnessTestCompletion(
            test_manager_, sub_response.test_id(), &execution);
        if (wait_status.ok()) {
          have_execution = true;
          internal::ApplyTerminalHarnessExecution(execution, &step_success,
                                                  &step_message);
        } else {
          status = wait_status;
          step_success = false;
          step_message = std::string(wait_status.message());
        }
      }
    } else if (step.action == "screenshot") {
      ScreenshotRequest sub_request;
      sub_request.set_window_title(ApplyOverrides(step.target, overrides));
      if (!step.region.empty()) {
        sub_request.set_output_path(ApplyOverrides(step.region, overrides));
      }
      ScreenshotResponse sub_response;
      status = Screenshot(&sub_request, &sub_response);
      step_success = sub_response.success();
      step_message = sub_response.message();
    } else {
      status = absl::InvalidArgumentError(
          absl::StrFormat("Unsupported action '%s'", step.action));
      step_message =
          std::string(status.message().data(), status.message().size());
    }
 
    auto* assertion = response->add_assertions();
    assertion->set_description(
        absl::StrFormat("Step %d (%s)", steps_executed, step.action));
 
    if (!status.ok()) {
      assertion->set_passed(false);
      assertion->set_error_message(
          std::string(status.message().data(), status.message().size()));
      overall_success = false;
      overall_message = step_message;
      logs.push_back(absl::StrFormat("  Error: %s", status.message()));
      overall_rpc_status = status;
      break;
    }
 
    bool expectations_met = (step_success == step.expect_success);
    std::string expectation_error;
 
    if (!expectations_met) {
      expectation_error =
          absl::StrFormat("Expected success=%s but got %s",
                          step.expect_success ? "true" : "false",
                          step_success ? "true" : "false");
    }
 
    if (!step.expect_status.empty()) {
      HarnessTestStatus expected_status =
          ::yaze::test::HarnessStatusFromString(step.expect_status);
      if (!have_execution) {
        expectations_met = false;
        if (!expectation_error.empty()) {
          expectation_error.append("; ");
        }
        expectation_error.append("No execution details available");
      } else if (expected_status != HarnessTestStatus::kUnspecified &&
                 execution.status != expected_status) {
        expectations_met = false;
        if (!expectation_error.empty()) {
          expectation_error.append("; ");
        }
        expectation_error.append(absl::StrFormat(
            "Expected status %s but observed %s", step.expect_status,
            ::yaze::test::HarnessStatusToString(execution.status)));
      }
      if (have_execution) {
        assertion->set_actual_value(
            ::yaze::test::HarnessStatusToString(execution.status));
        assertion->set_expected_value(step.expect_status);
      }
    }
 
    if (!step.expect_message.empty()) {
      std::string actual_message = step_message;
      if (have_execution && !execution.error_message.empty()) {
        actual_message = execution.error_message;
      }
      if (actual_message.find(step.expect_message) == std::string::npos) {
        expectations_met = false;
        if (!expectation_error.empty()) {
          expectation_error.append("; ");
        }
        expectation_error.append(
            absl::StrFormat("Expected message containing '%s' but got '%s'",
                            step.expect_message, actual_message));
      }
    }
 
    if (!expectations_met) {
      assertion->set_passed(false);
      assertion->set_error_message(expectation_error);
      overall_success = false;
      overall_message = expectation_error;
      logs.push_back(
          absl::StrFormat("  Failed expectations: %s", expectation_error));
      if (request->ci_mode()) {
        break;
      }
    } else {
      assertion->set_passed(true);
      logs.push_back(absl::StrFormat("  Result: %s", step_message));
    }
 
    if (have_execution && !execution.assertion_failures.empty()) {
      for (const auto& failure : execution.assertion_failures) {
        logs.push_back(absl::StrFormat("  Assertion failure: %s", failure));
      }
    }
 
    if (!overall_success && request->ci_mode()) {
      break;
    }
  }
 
  response->set_steps_executed(steps_executed);
  response->set_success(overall_success);
  response->set_message(overall_message);
  for (const auto& log_entry : logs) {
    response->add_logs(log_entry);
  }
 
  return overall_rpc_status;
}
 
// ============================================================================
// ImGuiTestHarnessServer - Server Lifecycle
// ============================================================================
 
ImGuiTestHarnessServer& ImGuiTestHarnessServer::Instance() {
  static ImGuiTestHarnessServer* instance = new ImGuiTestHarnessServer();
  return *instance;
}
 
ImGuiTestHarnessServer::~ImGuiTestHarnessServer() {
  Shutdown();
}
 
absl::Status ImGuiTestHarnessServer::Start(int port,
                                           TestManager* test_manager) {
  if (server_) {
    return absl::FailedPreconditionError("Server already running");
  }
 
  if (!test_manager) {
    return absl::InvalidArgumentError("TestManager cannot be null");
  }
 
  // Create the service implementation with TestManager reference
  service_ = std::make_unique<ImGuiTestHarnessServiceImpl>(test_manager);
 
  // Create the gRPC service wrapper (store as member to prevent it from going
  // out of scope)
  grpc_service_ = std::make_unique<ImGuiTestHarnessServiceGrpc>(service_.get());
 
  std::string server_address = absl::StrFormat("0.0.0.0:%d", port);
 
  grpc::ServerBuilder builder;
 
  // Listen on all interfaces (use 0.0.0.0 to avoid IPv6/IPv4 binding conflicts)
  builder.AddListeningPort(server_address, grpc::InsecureServerCredentials());
 
  // Register service
  builder.RegisterService(grpc_service_.get());
 
  // Build and start
  server_ = builder.BuildAndStart();
 
  if (!server_) {
    return absl::InternalError(
        absl::StrFormat("Failed to start gRPC server on %s", server_address));
  }
 
  port_ = port;
 
  std::cout << "✓ ImGuiTestHarness gRPC server listening on " << server_address
            << " (with TestManager integration)\n";
  std::cout << "  Use 'grpcurl -plaintext -d '{\"message\":\"test\"}' "
            << server_address << " yaze.test.ImGuiTestHarness/Ping' to test\n";
 
  return absl::OkStatus();
}
 
void ImGuiTestHarnessServer::Shutdown() {
  if (server_) {
    std::cout << "⏹ Shutting down ImGuiTestHarness gRPC server...\n";
    server_->Shutdown();
    server_.reset();
    service_.reset();
    port_ = 0;
    std::cout << "✓ ImGuiTestHarness gRPC server stopped\n";
  }
}
 
}  // namespace test
}  // namespace yaze
 
#endif  // YAZE_WITH_GRPC