_extension_web.dart 1.71 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
// Copyright 2014 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

import 'dart:async';
import 'dart:convert';
import 'dart:html' as html;
import 'dart:js';
import 'dart:js_util' as js_util;

11
/// The web implementation of [registerWebServiceExtension].
12
///
13 14
/// Adds a hidden `window.$flutterDriver` JavaScript function that's called by
/// `FlutterWebDriver` to fulfill FlutterDriver commands.
15 16 17 18
///
/// See also:
///
///  * `_extension_io.dart`, which has the dart:io implementation
19
void registerWebServiceExtension(Future<Map<String, dynamic>> Function(Map<String, String>) callback) {
20 21 22 23 24
  // Define the result variable because packages/flutter_driver/lib/src/driver/web_driver.dart
  // checks for this value to become non-null when waiting for the result. If this value is
  // undefined at the time of the check, WebDriver throws an exception.
  context[r'$flutterDriverResult'] = null;

25
  js_util.setProperty(html.window, r'$flutterDriver', allowInterop((dynamic message) async {
26 27 28 29 30 31 32
    try {
      final Map<String, dynamic> messageJson = jsonDecode(message as String) as Map<String, dynamic>;
      final Map<String, String> params = messageJson.cast<String, String>();
      final Map<String, dynamic> result = await callback(params);
      context[r'$flutterDriverResult'] = json.encode(result);
    } catch (error, stackTrace) {
      // Encode the error in the same format the FlutterDriver extension uses.
33
      // See //packages/flutter_driver/lib/src/extension/extension.dart
34 35 36 37 38
      context[r'$flutterDriverResult'] = json.encode(<String, dynamic>{
        'isError': true,
        'response': '$error\n$stackTrace',
      });
    }
39 40
  }));
}