Java Code Examples for com.google.javascript.rhino.jstype.FunctionType#getInstanceType()

/**
 * Returns the type for a type assertion, or null if the function asserts
 * that the node must not be null or undefined.
 */
@Override
public JSType getAssertedType(Node call, JSTypeRegistry registry) {
  if (call.getChildCount() > 2) {
    Node constructor = call.getFirstChild().getNext().getNext();
    if (constructor != null) {
      JSType ownerType = constructor.getJSType();
      if (ownerType != null
          && ownerType.isFunctionType()
          && ownerType.isConstructor()) {
        FunctionType functionType = ((FunctionType) ownerType);
        return functionType.getInstanceType();
      }
    }
  }
  return super.getAssertedType(call, registry);
}
 

public void testAbstractMethod() {
  testSame(
      "/** @type {!Function} */ var abstractMethod;" +
      "/** @constructor */ function Foo() {}" +
      "/** @param {number} x */ Foo.prototype.bar = abstractMethod;");
  assertEquals(
      "Function", findNameType("abstractMethod", globalScope).toString());

  FunctionType ctor = (FunctionType) findNameType("Foo", globalScope);
  ObjectType instance = ctor.getInstanceType();
  assertEquals("Foo", instance.toString());

  ObjectType proto = instance.getImplicitPrototype();
  assertEquals("Foo.prototype", proto.toString());

  assertEquals(
      "function (this:Foo, number): ?",
      proto.getPropertyType("bar").toString());
}
 

public void testAbstractMethod4() {
  testSame(
      "/** @type {!Function} */ var abstractMethod;" +
      "/** @constructor */ function Foo() {}" +
      "Foo.prototype = {/** @param {number} x */ bar: abstractMethod};");
  assertEquals(
      "Function", findNameType("abstractMethod", globalScope).toString());

  FunctionType ctor = (FunctionType) findNameType("Foo", globalScope);
  ObjectType instance = ctor.getInstanceType();
  assertEquals("Foo", instance.toString());

  ObjectType proto = instance.getImplicitPrototype();
  assertEquals("Foo.prototype", proto.toString());

  assertEquals(
      // should be: "function (this:Foo, number): ?"
      "function (this:Foo, number): ?",
      proto.getPropertyType("bar").toString());
}
 

/**
 * Expect that all properties on interfaces that this type implements are
 * implemented and correctly typed.
 */
void expectAllInterfaceProperties(NodeTraversal t, Node n,
    FunctionType type) {
  ObjectType instance = type.getInstanceType();
  for (ObjectType implemented : type.getAllImplementedInterfaces()) {
    if (implemented.getImplicitPrototype() != null) {
      for (String prop :
           implemented.getImplicitPrototype().getOwnPropertyNames()) {
        expectInterfaceProperty(t, n, instance, implemented, prop);
      }
    }
  }
}
 

/**
 * Expect that all properties on interfaces that this type implements are
 * implemented and correctly typed.
 */
void expectAllInterfaceProperties(NodeTraversal t, Node n,
    FunctionType type) {
  ObjectType instance = type.getInstanceType();
  for (ObjectType implemented : type.getAllImplementedInterfaces()) {
    if (implemented.getImplicitPrototype() != null) {
      for (String prop :
           implemented.getImplicitPrototype().getOwnPropertyNames()) {
        expectInterfaceProperty(t, n, instance, implemented, prop);
      }
    }
  }
}
 

/**
 * Expect that all properties on interfaces that this type implements are
 * implemented and correctly typed.
 */
void expectAllInterfaceProperties(NodeTraversal t, Node n,
    FunctionType type) {
  ObjectType instance = type.getInstanceType();
  for (ObjectType implemented : type.getAllImplementedInterfaces()) {
    if (implemented.getImplicitPrototype() != null) {
      for (String prop :
           implemented.getImplicitPrototype().getOwnPropertyNames()) {
        expectInterfaceProperty(t, n, instance, implemented, prop);
      }
    }
  }
}
 

/**
 * Expect that all properties on interfaces that this type implements are
 * implemented and correctly typed.
 */
void expectAllInterfaceProperties(NodeTraversal t, Node n,
    FunctionType type) {
  ObjectType instance = type.getInstanceType();
  for (ObjectType implemented : type.getAllImplementedInterfaces()) {
    if (implemented.getImplicitPrototype() != null) {
      for (String prop :
           implemented.getImplicitPrototype().getOwnPropertyNames()) {
        expectInterfaceProperty(t, n, instance, implemented, prop);
      }
    }
  }
}
 

/**
 * Normalize the type of a constructor, its instance, and its prototype
 * all down to the same type (the instance type).
 */
private JSType normalizeClassType(JSType type) {
  if (type == null || type.isUnknownType()) {
    return type;
  } else if (type.isConstructor()) {
    return ((FunctionType) type).getInstanceType();
  } else if (type.isFunctionPrototypeType()) {
    FunctionType owner = ((FunctionPrototypeType) type).getOwnerFunction();
    if (owner.isConstructor()) {
      return owner.getInstanceType();
    }
  }
  return type;
}
 

/**
 * Normalize the type of a constructor, its instance, and its prototype
 * all down to the same type (the instance type).
 */
private JSType normalizeClassType(JSType type) {
  if (type == null || type.isUnknownType()) {
    return type;
  } else if (type.isConstructor()) {
    return ((FunctionType) type).getInstanceType();
  } else if (type.isFunctionPrototypeType()) {
    FunctionType owner = ((FunctionPrototypeType) type).getOwnerFunction();
    if (owner.isConstructor()) {
      return owner.getInstanceType();
    }
  }
  return type;
}
 

/**
 * Expect that all properties on interfaces that this type implements are
 * implemented and correctly typed.
 */
void expectAllInterfaceProperties(NodeTraversal t, Node n,
    FunctionType type) {
  ObjectType instance = type.getInstanceType();
  for (ObjectType implemented : type.getAllImplementedInterfaces()) {
    if (implemented.getImplicitPrototype() != null) {
      for (String prop :
           implemented.getImplicitPrototype().getOwnPropertyNames()) {
        expectInterfaceProperty(t, n, instance, implemented, prop);
      }
    }
  }
}
 

private JSType maybeGetInstanceTypeFromPrototypeRef(Node src) {
  JSType ownerType = getJSType(src.getFirstChild());
  if (ownerType.isFunctionType() && ownerType.isConstructor()) {
    FunctionType functionType = ((FunctionType) ownerType);
    return functionType.getInstanceType();
  }
  return null;
}
 

/**
 * Normalize the type of a constructor, its instance, and its prototype
 * all down to the same type (the instance type).
 */
private JSType normalizeClassType(JSType type) {
  if (type == null || type.isUnknownType()) {
    return type;
  } else if (type.isNominalConstructor()) {
    return (type.toMaybeFunctionType()).getInstanceType();
  } else if (type.isFunctionPrototypeType()) {
    FunctionType owner = ((ObjectType) type).getOwnerFunction();
    if (owner.isConstructor()) {
      return owner.getInstanceType();
    }
  }
  return type;
}
 

@Override
public JSType caseUnknownType() {
  FunctionType funcTarget = JSType.toMaybeFunctionType(target);
  if (funcTarget != null && funcTarget.hasInstanceType()) {
    return funcTarget.getInstanceType();
  }
  return getNativeType(UNKNOWN_TYPE);
}
 

/**
 * Adds the instance of the given constructor, its implicit prototype and all
 * its related types to the given bit set.
 */
private void addRelatedInstance(
    FunctionType constructor, JSTypeBitSet related) {
  // TODO(user): A constructor which doesn't have an instance type
  // (e.g. it's missing the @constructor annotation) should be an invalidating
  // type which doesn't reach this code path.
  if (constructor.hasInstanceType()) {
    ObjectType instanceType = constructor.getInstanceType();
    related.set(getIntForType(instanceType.getImplicitPrototype()));
    computeRelatedTypes(instanceType);
    related.or(relatedBitsets.get(instanceType));
  }
}
 

private JSType getInstanceType(final JSTypeRegistry jsRegistry, FunctionType ctor) {
  ObjectType instance = ctor.getInstanceType();
  if (ctor.getJSDocInfo() != null && !ctor.getJSDocInfo().getTemplateTypeNames().isEmpty()) {
    ImmutableList<JSType> templateTypes =
        ctor.getJSDocInfo()
            .getTemplateTypeNames()
            .stream()
            .map(jsRegistry::createTemplateType)
            .collect(toImmutableList());
    instance = jsRegistry.createTemplatizedType(instance, templateTypes);
  }
  return instance;
}
 

/**
 * Gets the symbol for the given constructor or interface.
 */
public Symbol getSymbolDeclaredBy(FunctionType fn) {
  Preconditions.checkState(fn.isConstructor() || fn.isInterface());
  ObjectType instanceType = fn.getInstanceType();
  return getSymbolForName(fn.getSource(), instanceType.getReferenceName());
}