lib/src/handler/infer_handler.dart

import 'dart:convert';

import '../common/capabilities.dart';
import '../common/exception.dart';
import '../common/request.dart';
import '../common/session.dart';
import '../common/spec.dart';
import '../common/webdriver_handler.dart';
import 'json_wire/session.dart';
import 'w3c/session.dart';

/// A [WebDriverHandler] that is only used when creating new session /
/// getting existing session info without given the spec.
class InferWebDriverHandler extends WebDriverHandler {
  @override
  final SessionHandler session = InferSessionHandler();

  @override
  CoreHandler get core =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  ElementHandler get element =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  ElementFinder get elementFinder =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  KeyboardHandler get keyboard =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  MouseHandler get mouse =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  AlertHandler get alert =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  NavigationHandler get navigation =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  WindowHandler get window =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  FrameHandler get frame =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  CookiesHandler get cookies =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  TimeoutsHandler get timeouts =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  LogsHandler get logs =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  WebDriverRequest buildGeneralRequest(HttpMethod method, String uri,
          [params]) =>
      throw UnsupportedError('Unsupported for InferHandler');

  @override
  dynamic parseGeneralResponse(
          WebDriverResponse response, dynamic Function(String) createElement) =>
      throw UnsupportedError('Unsupported for InferHandler');
}

class InferSessionHandler extends SessionHandler {
  @override
  WebDriverRequest buildCreateRequest({Map<String, dynamic>? desired}) =>
      WebDriverRequest.postRequest('session', {
        'desiredCapabilities': desired,
        'capabilities': {'alwaysMatch': desired}
      });

  @override
  SessionInfo parseCreateResponse(WebDriverResponse response) {
    Map<String, dynamic> responseBody;
    try {
      responseBody = json.decode(response.body!) as Map<String, dynamic>;
    } catch (e) {
      final rawBody = response.body == null || response.body!.isEmpty
          ? '<empty response>'
          : response.body;
      throw WebDriverException(
          response.statusCode, 'Error parsing response body: $rawBody');
    }

    // JSON responses have multiple keys.
    if (responseBody.keys.length > 1) {
      return JsonWireSessionHandler().parseCreateResponse(response);
      // W3C responses have only one key, value.
    } else if (responseBody.keys.length == 1) {
      return W3cSessionHandler().parseCreateResponse(response);
    }

    throw WebDriverException(
      response.statusCode,
      'Unexpected response structure: ${response.body}',
    );
  }

  @override
  WebDriverRequest buildInfoRequest(String id) =>
      WebDriverRequest.getRequest('session/$id');

  @override
  SessionInfo parseInfoResponse(WebDriverResponse response,
      [String? sessionId]) {
    if (response.statusCode == 404) {
      // May be W3C, as it will throw an unknown command exception.
      Map<String, dynamic>? body;
      try {
        body = json.decode(response.body!)['value'] as Map<String, dynamic>?;
      } catch (e) {
        final rawBody = response.body?.isEmpty != false
            ? '<empty response>'
            : response.body;
        throw WebDriverException(
          response.statusCode,
          'Error parsing response body: $rawBody',
        );
      }

      if (body == null ||
          body['error'] != 'unknown command' ||
          body['message'] is! String) {
        throw WebDriverException(
          response.statusCode,
          'Unexpected response body (expecting `unexpected command error` '
          'produced by W3C WebDriver): ${response.body}',
        );
      }

      return SessionInfo(sessionId!, WebDriverSpec.W3c, Capabilities.empty);
    }

    return JsonWireSessionHandler().parseInfoResponse(response);
  }
}