flutter/dev/integration_tests/flutter_driver_screenshot_test/lib/main.dart

// 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 'package:flutter/material.dart';
import 'package:flutter_driver/driver_extension.dart';
import 'package:device_info/device_info.dart';
import './image_page.dart';
import './page.dart';

final List<PageWidget> _allPages = <PageWidget>[
  const ImagePage(),
];

void main() {
  enableFlutterDriverExtension(handler: driverDataHandler.handleMessage);
  runApp(_MyApp());
}

class _MyApp extends StatelessWidget {
  @override
  Widget build(BuildContext context) {
    return MaterialApp(
      title: 'Screenshot Tests',
      theme: ThemeData(
        primarySwatch: Colors.blue,
      ),
      home: const _MyHomePage(title: 'Screenshot Tests Home Page'),
    );
  }
}

class _MyHomePage extends StatefulWidget {
  const _MyHomePage({Key key, this.title}) : super(key: key);
  final String title;

  @override
  _MyHomePageState createState() => _MyHomePageState();
}

class _MyHomePageState extends State<_MyHomePage> {
  @override
  void initState() {
    driverDataHandler.handlerCompleter.complete(_handleDriverMessage);
    super.initState();
  }

  @override
  Widget build(BuildContext context) {
    return Scaffold(
      appBar: AppBar(
        title: Text(widget.title),
      ),
      body: ListView.builder(
        itemCount: _allPages.length,
        itemBuilder: (_, int index) => ListTile(
          title: Text(_allPages[index].title, key: _allPages[index].key),
          onTap: () => _pushPage(context, _allPages[index]),
        ),
      ),
    );
  }

  void _pushPage(BuildContext context, PageWidget page) {
    Navigator.of(context).push(MaterialPageRoute<void>(
      builder: (_) => page,
    ));
  }

  Future<String> _handleDriverMessage(String message) async {
    switch (message) {
      case 'device_model':
        final DeviceInfoPlugin deviceInfo = DeviceInfoPlugin();
        switch (Theme.of(context).platform) {
          case TargetPlatform.iOS:
            final IosDeviceInfo iosDeviceInfo = await deviceInfo.iosInfo;
            if (iosDeviceInfo.isPhysicalDevice) {
              return iosDeviceInfo.utsname.machine;
            }
            return 'sim_' + iosDeviceInfo.name;
          case TargetPlatform.android:
            return (await deviceInfo.androidInfo).model;
          case TargetPlatform.fuchsia:
            return 'fuchsia';
          case TargetPlatform.macOS:
          case TargetPlatform.linux:
          case TargetPlatform.windows:
            return 'unsupported';
            break;
        }
        assert(false, 'Unhandled Theme target platform ${Theme.of(context).platform}.');
        return 'unsupported';
    }
    return 'unknown message: "$message"';
  }
}

class _FutureDataHandler {
  final Completer<DataHandler> handlerCompleter = Completer<DataHandler>();

  Future<String> handleMessage(String message) async {
    final DataHandler handler = await handlerCompleter.future;
    return handler(message);
  }
}

_FutureDataHandler driverDataHandler = _FutureDataHandler();