// Copyright (c) 2014, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'package:analyzer/dart/element/element.dart';
import 'package:analyzer/dart/element/nullability_suffix.dart';
import 'package:analyzer/src/dart/element/type_provider.dart';
import 'package:analyzer/src/generated/engine.dart' show AnalysisContext;
import 'package:analyzer/src/generated/source.dart' show Source;
import 'package:analyzer/src/generated/source.dart';
import 'package:analyzer/src/test_utilities/mock_sdk_elements.dart';

/**
 * A type provider that can be used by tests without creating the element model
 * for the core library.
 */
class TestTypeProvider extends TypeProviderImpl {
  factory TestTypeProvider([
    AnalysisContext context,
    Object analysisDriver,
    NullabilitySuffix nullabilitySuffix = NullabilitySuffix.star,
  ]) {
    context ??= _MockAnalysisContext();
    var sdkElements = MockSdkElements(context, nullabilitySuffix);
    return TestTypeProvider._(
      nullabilitySuffix,
      sdkElements.coreLibrary,
      sdkElements.asyncLibrary,
    );
  }

  TestTypeProvider._(
    NullabilitySuffix nullabilitySuffix,
    LibraryElement coreLibrary,
    LibraryElement asyncLibrary,
  ) : super(coreLibrary, asyncLibrary, nullabilitySuffix: nullabilitySuffix);
}

class _MockAnalysisContext implements AnalysisContext {
  @override
  final SourceFactory sourceFactory = _MockSourceFactory();

  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
}

class _MockSource implements Source {
  @override
  final Uri uri;

  _MockSource(this.uri);

  @override
  String get encoding => '$uri';

  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
}

class _MockSourceFactory implements SourceFactory {
  @override
  Source forUri(String uriStr) {
    var uri = Uri.parse(uriStr);
    return _MockSource(uri);
  }

  noSuchMethod(Invocation invocation) => super.noSuchMethod(invocation);
}