commit 01c3cbe1e1a1e83a9311ed1f5b7763f9dc5fa8dc
author Konstantin Scheglov <scheglov@google.com> Wed Aug 05 08:27:42 2020 -0700
committer GitHub <noreply@github.com> Wed Aug 05 08:27:42 2020 -0700
tree 53d816513f7a09cbc0f1e7e75136cc46c0244db9
parent 2f1f19d42e583295db6cfc6ee0c2f7fe1fcbd2a4

Access to type using specific types and getters. (#2283)

lib/src/model/model_element.dart

diff --git a/lib/src/model/model_element.dart b/lib/src/model/model_element.dart
index c19737b..bdb52f9 100644
--- a/lib/src/model/model_element.dart
+++ b/lib/src/model/model_element.dart
@@ -947,6 +947,7 @@
       .renderLinkedParams(parameters, showMetadata: false, showNames: false);
 
   ElementType get modelType {
+    var element = this.element;
     if (_modelType == null) {
       // TODO(jcollins-g): Need an interface for a "member with a type" (or changed object model).
       if (_originalMember != null &&
@@ -962,13 +963,16 @@
           _modelType = ElementType.from(
               (_originalMember as ParameterMember).type, library, packageGraph);
         }
-      } else if (element is ExecutableElement ||
-          element is FunctionTypedElement ||
-          element is ParameterElement ||
-          element is TypeDefiningElement ||
-          element is PropertyInducingElement) {
-        _modelType =
-            ElementType.from((element as dynamic).type, library, packageGraph);
+      } else if (element is ClassElement) {
+        _modelType = ElementType.from(element.thisType, library, packageGraph);
+      } else if (element is FunctionTypedElement) {
+        _modelType = ElementType.from(element.type, library, packageGraph);
+      } else if (element is ParameterElement) {
+        _modelType = ElementType.from(element.type, library, packageGraph);
+      } else if (element is PropertyInducingElement) {
+        _modelType = ElementType.from(element.type, library, packageGraph);
+      } else {
+        throw UnimplementedError('(${element.runtimeType}) $element');
       }
     }
     return _modelType;