// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

import 'package:amplify_authenticator/amplify_authenticator.dart';
import 'package:amplify_authenticator_test/amplify_authenticator_test.dart';
import 'package:amplify_authenticator_test/src/configs/email_config.dart';
import 'package:amplify_authenticator_test/src/configs/email_or_phone_config.dart';
import 'package:amplify_authenticator_test/src/configs/phone_config.dart';
import 'package:amplify_authenticator_test/src/configs/social_providers_config.dart';
import 'package:amplify_authenticator_test/src/configs/username_with_attributes.dart';
import 'package:flutter/material.dart';
import 'package:flutter_test/flutter_test.dart';
import 'package:golden_toolkit/golden_toolkit.dart';

import 'utils.dart';

enum TestConfig {
  email(emailConfig),
  emailOrPhone(emailOrPhoneConfig),
  socialProvider(socialProviderConfig),
  phoneNumber(phoneNumberConfig),
  usernameWithAttributes(usernameWithAttributesConfig);

  const TestConfig(this.config);

  final String config;
}

enum ScreenGeometry {
  // Geometry based off of an iPhone SE 2022
  mini(size: Size(750, 1334), pixelRatio: 2),

  // Geometry based off of an iPhone 14
  mobile(size: Size(1170, 2532), pixelRatio: 3),

  // Geometry based off of an iPad 10
  tablet(size: Size(1620, 2160), pixelRatio: 2),

  // Geometry based off of a Macbook Air/Pro 13" (2020)
  laptop(size: Size(2560, 1600), pixelRatio: 2),

  // Geometry based off of an ultra HD monitor
  desktop(size: Size(2560, 1440), pixelRatio: 1);

  const ScreenGeometry({required this.size, required this.pixelRatio});

  final Size size;

  final double pixelRatio;
}

// TODO(Jordan-Nelson): Update all themes to use material3
enum TestTheme {
  defaultMaterial(),
  material3(),
  highContrast(),
  customSwatch(),
  custom();

  ThemeData get lightTheme {
    switch (this) {
      case TestTheme.defaultMaterial:
        return ThemeData.light(useMaterial3: false);
      case TestTheme.material3:
        return ThemeData.light(useMaterial3: true);
      case TestTheme.highContrast:
        return ThemeData.from(
          useMaterial3: false,
          colorScheme: const ColorScheme.highContrastLight(),
        );
      case TestTheme.customSwatch:
        return ThemeData.from(
          useMaterial3: false,
          colorScheme: ColorScheme.fromSwatch(
            primarySwatch: Colors.red,
            backgroundColor: Colors.white,
          ),
        ).copyWith(
          indicatorColor: Colors.red,
        );
      case TestTheme.custom:
        return ThemeData.light(useMaterial3: false).copyWith(
          tabBarTheme: const TabBarTheme(
            labelColor: Colors.amber,
          ),
          indicatorColor: Colors.pink,
        );
    }
  }

  ThemeData get darkTheme {
    switch (this) {
      case TestTheme.defaultMaterial:
        return ThemeData.dark(useMaterial3: false);
      case TestTheme.material3:
        return ThemeData.dark(useMaterial3: true);
      case TestTheme.highContrast:
        return ThemeData.from(
          useMaterial3: false,
          colorScheme: const ColorScheme.highContrastDark(),
        );
      case TestTheme.customSwatch:
        return ThemeData.from(
          useMaterial3: false,
          colorScheme: ColorScheme.fromSwatch(
            primarySwatch: Colors.red,
            backgroundColor: Colors.black,
            brightness: Brightness.dark,
          ),
        );
      case TestTheme.custom:
        return ThemeData.dark(useMaterial3: false).copyWith(
          tabBarTheme: const TabBarTheme(
            labelColor: Colors.amber,
          ),
          indicatorColor: Colors.pink,
        );
    }
  }
}

void main() {
  final binding = TestWidgetsFlutterBinding.ensureInitialized();

  setUp(() async {
    // flutter's default test font does not accurately reflect font size,
    // which leads to pixel over flows.
    await loadAppFonts();
  });

  group('UI Tests', () {
    setUp(binding.platformDispatcher.clearPlatformBrightnessTestValue);

    // Tests the layout for each config/step/geometry.
    group('layout', () {
      testMatrix3(
        TestConfig.values,
        [
          AuthenticatorStep.signIn,
          AuthenticatorStep.signUp,
          AuthenticatorStep.resetPassword,
        ],
        ScreenGeometry.values,
        (
          TestConfig testConfig,
          AuthenticatorStep step,
          ScreenGeometry geometry,
        ) {
          final configName = '${testConfig.name}Config';
          final stepName = '${step.name}Step';
          final geometryName = '${geometry.name}Geometry';
          final testName = 'layout_${configName}_${stepName}_$geometryName';

          setUp(() {
            // TODO(Jordan-Nelson): Migrate to WidgetTester
            // ignore: deprecated_member_use
            binding.window.devicePixelRatioTestValue = geometry.pixelRatio;
            // ignore: deprecated_member_use
            binding.window.physicalSizeTestValue = geometry.size;
          });

          testWidgets('matches snapshot', (WidgetTester tester) async {
            await tester.pumpWidget(
              MockAuthenticatorApp(
                initialStep: step,
                config: testConfig.config,
              ),
            );
            await tester.pumpAndSettle();
            await expectGoldenMatches(authenticatorFinder, '$testName.png');
          });
        },
      );
    });

    // Tests the theme/brightness for a limited set of configs/geometries.
    group('theme', () {
      testMatrix5(
        [TestConfig.email, TestConfig.socialProvider, TestConfig.emailOrPhone],
        [AuthenticatorStep.signIn, AuthenticatorStep.signUp],
        TestTheme.values,
        Brightness.values,
        [ScreenGeometry.mobile, ScreenGeometry.laptop],
        (
          TestConfig testConfig,
          AuthenticatorStep step,
          TestTheme theme,
          Brightness brightness,
          ScreenGeometry geometry,
        ) {
          final configName = '${testConfig.name}Config';
          final themeName = '${theme.name}Theme';
          final brightnessName = '${brightness.name}Mode';
          final stepName = '${step.name}Step';
          final geometryName = '${geometry.name}Geometry';
          final testName =
              'theme_${configName}_${stepName}_${themeName}_${brightnessName}_$geometryName';

          setUp(() {
            binding.platformDispatcher.platformBrightnessTestValue = brightness;

            // TODO(Jordan-Nelson): Migrate to WidgetTester
            // ignore: deprecated_member_use
            binding.window.devicePixelRatioTestValue = geometry.pixelRatio;
            // ignore: deprecated_member_use
            binding.window.physicalSizeTestValue = geometry.size;
          });

          testWidgets('matches snapshot', (WidgetTester tester) async {
            await tester.pumpWidget(
              MockAuthenticatorApp(
                initialStep: step,
                lightTheme: theme.lightTheme,
                darkTheme: theme.darkTheme,
                config: testConfig.config,
              ),
            );
            await tester.pumpAndSettle();
            await expectGoldenMatches(authenticatorFinder, '$testName.png');
          });
        },
      );
    });
  });
}