use of com.google.template.soy.types.UnionType in project closure-templates by google.
the class JsType method forSoyType.
/**
* Returns a {@link JsType} corresponding to the given {@link SoyType}
*
* <p>TODO(lukes): consider adding a cache for all the computed types. The same type is probably
* accessed many many times.
*
* @param soyType the soy type
* @param isIncrementalDom whether or not this is for incremental dom.
*/
static JsType forSoyType(SoyType soyType, boolean isIncrementalDom) {
switch(soyType.getKind()) {
case NULL:
return NULL_OR_UNDEFINED_TYPE;
case ANY:
return ANY_TYPE;
case UNKNOWN:
return UNKNOWN_TYPE;
case BOOL:
return BOOLEAN_TYPE;
case PROTO_ENUM:
SoyProtoEnumType enumType = (SoyProtoEnumType) soyType;
String enumTypeName = enumType.getNameForBackend(SoyBackendKind.JS_SRC);
return builder().addType("number").addType(enumTypeName).addRequire(GoogRequire.create(enumTypeName)).setPredicate(GOOG_IS_NUMBER).build();
case FLOAT:
case INT:
return NUMBER_TYPE;
case STRING:
return STRING_OR_UNSANITIZED_TEXT;
case ATTRIBUTES:
if (isIncrementalDom) {
// idom has a different strategy for handling these
return IDOM_ATTRIBUTES;
}
// fall through
case HTML:
if (isIncrementalDom) {
// idom has a different strategy for handling these
return IDOM_HTML;
}
// fall-through
case CSS:
case JS:
case URI:
case TRUSTED_RESOURCE_URI:
return STRICT_TYPES.get(((SanitizedType) soyType).getContentKind());
case LIST:
ListType listType = (ListType) soyType;
if (listType.getElementType().getKind() == SoyType.Kind.ANY) {
return RAW_ARRAY_TYPE;
}
JsType element = forSoyType(listType.getElementType(), isIncrementalDom);
return builder().addType("!Array<" + element.typeExpr() + ">").addRequires(element.getGoogRequires()).setPredicate(GOOG_IS_ARRAY).build();
case LEGACY_OBJECT_MAP:
{
LegacyObjectMapType mapType = (LegacyObjectMapType) soyType;
if (mapType.getKeyType().getKind() == SoyType.Kind.ANY && mapType.getValueType().getKind() == SoyType.Kind.ANY) {
return RAW_OBJECT_TYPE;
}
JsType keyTypeName = forSoyType(mapType.getKeyType(), isIncrementalDom);
JsType valueTypeName = forSoyType(mapType.getValueType(), isIncrementalDom);
return builder().addType(String.format("!Object<%s,%s>", keyTypeName.typeExpr(), valueTypeName.typeExpr())).addRequires(keyTypeName.getGoogRequires()).addRequires(valueTypeName.getGoogRequires()).setPredicate(GOOG_IS_OBJECT).build();
}
case MAP:
{
MapType mapType = (MapType) soyType;
SoyType keyType = mapType.getKeyType();
SoyType.Kind keyKind = keyType.getKind();
Preconditions.checkState(MapType.isAllowedKeyType(keyType));
// Soy key type of string should translate to a JS key type of string.
// forSoyType(StringType.getInstance()) normally translates to
// string|!goog.soy.data.UnsanitizedText, but ES6 Maps always use instance equality for
// lookups. Using UnsanitizedText instances as keys in Soy maps would cause unexpected
// behavior (usually a failed map lookup), so don't generate signatures that allow it.
JsType keyTypeName = keyKind == SoyType.Kind.STRING ? STRING_TYPE : forSoyType(keyType, isIncrementalDom);
JsType valueTypeName = forSoyType(mapType.getValueType(), isIncrementalDom);
return builder().addType(String.format("!soy.map.Map<%s,%s>", keyTypeName.typeExpr(), valueTypeName.typeExpr())).addRequires(keyTypeName.getGoogRequires()).addRequires(valueTypeName.getGoogRequires()).addRequire(GoogRequire.create("soy.map")).setPredicate(TypePredicate.NO_OP).build();
}
case PROTO:
final SoyProtoType protoType = (SoyProtoType) soyType;
final String protoTypeName = protoType.getNameForBackend(SoyBackendKind.JS_SRC);
// isn't clear that this is very useful for users.
return builder().addType(protoTypeName).addRequire(GoogRequire.create(protoTypeName)).addCoercionStrategy(ValueCoercionStrategy.PROTO).setPredicate(new TypePredicate() {
@Override
public Optional<CodeChunk.WithValue> maybeCheck(CodeChunk.WithValue value, Generator codeGenerator) {
return Optional.of(value.instanceof_(JsRuntime.protoConstructor(protoType)));
}
}).build();
case RECORD:
{
RecordType recordType = (RecordType) soyType;
if (recordType.getMembers().isEmpty()) {
return RAW_OBJECT_TYPE;
}
Builder builder = builder();
Map<String, String> members = new LinkedHashMap<>();
for (Map.Entry<String, SoyType> member : recordType.getMembers().entrySet()) {
JsType forSoyType = forSoyType(member.getValue(), isIncrementalDom);
builder.addRequires(forSoyType.getGoogRequires());
members.put(member.getKey(), forSoyType.typeExprForRecordMember(/* isOptional= */
false));
}
return builder.addType("{" + Joiner.on(", ").withKeyValueSeparator(": ").join(members) + ",}").setPredicate(GOOG_IS_OBJECT).build();
}
case UNION:
{
UnionType unionType = (UnionType) soyType;
Builder builder = builder();
final Set<JsType> types = new LinkedHashSet<>();
final boolean isNullable = unionType.isNullable();
// handle null first so that if other type tests dereference the param they won't fail
if (isNullable) {
builder.addTypes(NULL_OR_UNDEFINED_TYPE.typeExpressions);
builder.addCoercionStrategy(ValueCoercionStrategy.NULL);
types.add(NULL_OR_UNDEFINED_TYPE);
}
for (SoyType member : unionType.getMembers()) {
if (member.getKind() == Kind.NULL) {
// handled above
continue;
}
JsType memberType = forSoyType(member, isIncrementalDom);
builder.addRequires(memberType.extraRequires);
builder.addTypes(memberType.typeExpressions);
builder.addCoercionStrategies(memberType.coercionStrategies);
types.add(memberType);
}
return builder.setPredicate(new TypePredicate() {
@Override
public Optional<CodeChunk.WithValue> maybeCheck(CodeChunk.WithValue value, Generator codeGenerator) {
CodeChunk.WithValue result = null;
// this automatically.
for (JsType memberType : types) {
Optional<CodeChunk.WithValue> typeAssertion = memberType.getTypeAssertion(value, codeGenerator);
if (!typeAssertion.isPresent()) {
return Optional.absent();
}
if (result == null) {
result = typeAssertion.get();
} else {
result = result.or(typeAssertion.get(), codeGenerator);
}
}
return Optional.of(result);
}
}).build();
}
default:
throw new AssertionError("unhandled soytype: " + soyType);
}
}
use of com.google.template.soy.types.UnionType in project closure-templates by google.
the class TranslateExprNodeVisitor method genCodeForFieldAccess.
/**
* Generates the code for a field access, e.g. {@code .foo} or {@code .getFoo()}.
*
* @param baseType The type of the object that contains the field.
* @param fieldAccessNode The field access node.
* @param fieldName The field name.
*/
private FieldAccess genCodeForFieldAccess(SoyType baseType, FieldAccessNode fieldAccessNode, String fieldName) {
Preconditions.checkNotNull(baseType);
// type, and then see if they all agree.
if (baseType.getKind() == SoyType.Kind.UNION) {
// TODO(msamuel): We will need to generate fallback code for each variant.
UnionType unionType = (UnionType) baseType;
FieldAccess fieldAccess = null;
for (SoyType memberType : unionType.getMembers()) {
if (memberType.getKind() != SoyType.Kind.NULL) {
FieldAccess fieldAccessForType = genCodeForFieldAccess(memberType, fieldAccessNode, fieldName);
if (fieldAccess == null) {
fieldAccess = fieldAccessForType;
} else if (!fieldAccess.equals(fieldAccessForType)) {
errorReporter.report(fieldAccessNode.getSourceLocation(), UNION_ACCESSOR_MISMATCH, fieldName, baseType);
}
}
}
return fieldAccess;
}
if (baseType.getKind() == SoyType.Kind.PROTO) {
SoyProtoType protoType = (SoyProtoType) baseType;
FieldDescriptor desc = protoType.getFieldDescriptor(fieldName);
Preconditions.checkNotNull(desc, "Error in proto %s, field not found: %s", protoType.getDescriptor().getFullName(), fieldName);
return FieldAccess.protoCall(fieldName, desc);
}
return FieldAccess.id(fieldName);
}
use of com.google.template.soy.types.UnionType in project closure-templates by google.
the class TranslateExprNodeVisitor method genMapKeyCode.
/**
* Soy strings can be represented by SanitizedContent objects at runtime, so care needs to be
* taken when indexing into a map with a Soy "string". For pre-ES6 object maps, this isn't a
* problem, since bracket access implicitly calls toString() on the key, and SanitizedContent
* overrides toString appropriately. But ES6 Maps and jspb.Maps don't do this automatically, so we
* need to set it up.
*/
private CodeChunk.WithValue genMapKeyCode(ExprNode keyNode) {
CodeChunk.WithValue key = visit(keyNode);
// We need to coerce if the value could possibly a sanitizedcontent object
boolean needsRuntimeCoercionLogic = false;
SoyType type = keyNode.getType();
for (SoyType member : (type instanceof UnionType ? ((UnionType) type).getMembers() : ImmutableList.of(type))) {
Kind kind = member.getKind();
needsRuntimeCoercionLogic |= kind.isKnownStringOrSanitizedContent() || kind == Kind.UNKNOWN || kind == Kind.UNION || kind == Kind.ANY;
}
return needsRuntimeCoercionLogic ? SOY_MAP_MAYBE_COERCE_KEY_TO_STRING.call(key) : key;
}
Aggregations