Search in sources :

Example 1 with LegacyObjectMapType

use of com.google.template.soy.types.LegacyObjectMapType 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);
    }
}
Also used : UnionType(com.google.template.soy.types.UnionType) EnumSet(java.util.EnumSet) LinkedHashSet(java.util.LinkedHashSet) ImmutableSortedSet(com.google.common.collect.ImmutableSortedSet) ImmutableSet(com.google.common.collect.ImmutableSet) Set(java.util.Set) LegacyObjectMapType(com.google.template.soy.types.LegacyObjectMapType) MapType(com.google.template.soy.types.MapType) RecordType(com.google.template.soy.types.RecordType) CodeChunk(com.google.template.soy.jssrc.dsl.CodeChunk) ListType(com.google.template.soy.types.ListType) Kind(com.google.template.soy.types.SoyType.Kind) SoyBackendKind(com.google.template.soy.base.SoyBackendKind) SanitizedContentKind(com.google.template.soy.base.internal.SanitizedContentKind) Optional(com.google.common.base.Optional) LegacyObjectMapType(com.google.template.soy.types.LegacyObjectMapType) SoyProtoType(com.google.template.soy.types.SoyProtoType) SoyType(com.google.template.soy.types.SoyType) SoyProtoEnumType(com.google.template.soy.types.SoyProtoEnumType) LinkedHashMap(java.util.LinkedHashMap) Map(java.util.Map) ImmutableMap(com.google.common.collect.ImmutableMap) EnumMap(java.util.EnumMap) Generator(com.google.template.soy.jssrc.dsl.CodeChunk.Generator)

Example 2 with LegacyObjectMapType

use of com.google.template.soy.types.LegacyObjectMapType in project closure-templates by google.

the class GenerateParseInfoVisitor method findProtoTypesRecurse.

/**
 * Recursively search for protocol buffer types within the given type.
 *
 * @param type The type to search.
 * @param protoTypes Output set.
 */
private static void findProtoTypesRecurse(SoyType type, SortedSet<String> protoTypes) {
    if (type.getKind() == SoyType.Kind.PROTO) {
        protoTypes.add(((SoyProtoType) type).getDescriptorExpression());
    } else if (type.getKind() == SoyType.Kind.PROTO_ENUM) {
        protoTypes.add(((SoyProtoEnumType) type).getDescriptorExpression());
    } else {
        switch(type.getKind()) {
            case UNION:
                for (SoyType member : ((UnionType) type).getMembers()) {
                    findProtoTypesRecurse(member, protoTypes);
                }
                break;
            case LIST:
                {
                    ListType listType = (ListType) type;
                    findProtoTypesRecurse(listType.getElementType(), protoTypes);
                    break;
                }
            case LEGACY_OBJECT_MAP:
                {
                    LegacyObjectMapType mapType = (LegacyObjectMapType) type;
                    findProtoTypesRecurse(mapType.getKeyType(), protoTypes);
                    findProtoTypesRecurse(mapType.getValueType(), protoTypes);
                    break;
                }
            case RECORD:
                {
                    RecordType recordType = (RecordType) type;
                    for (SoyType fieldType : recordType.getMembers().values()) {
                        findProtoTypesRecurse(fieldType, protoTypes);
                    }
                    break;
                }
            default:
                break;
        }
    }
}
Also used : RecordType(com.google.template.soy.types.RecordType) SoyType(com.google.template.soy.types.SoyType) ListType(com.google.template.soy.types.ListType) SoyProtoEnumType(com.google.template.soy.types.SoyProtoEnumType) LegacyObjectMapType(com.google.template.soy.types.LegacyObjectMapType)

Example 3 with LegacyObjectMapType

use of com.google.template.soy.types.LegacyObjectMapType in project closure-templates by google.

the class TofuTypeChecksTest method testMapTypeIsInstance.

@Test
public void testMapTypeIsInstance() {
    LegacyObjectMapType mapOfStringToAny = LegacyObjectMapType.of(StringType.getInstance(), AnyType.getInstance());
    assertIsInstance(mapOfStringToAny, MAP_DATA, LIST_DATA);
    assertIsNotInstance(mapOfStringToAny, NULL_DATA, BOOLEAN_DATA, STRING_DATA, INTEGER_DATA, FLOAT_DATA, HTML_DATA, ATTRIBUTES_DATA, CSS_DATA, URI_DATA, TRUSTED_RESOURCE_URI_DATA, JS_DATA);
}
Also used : LegacyObjectMapType(com.google.template.soy.types.LegacyObjectMapType) Test(org.junit.Test)

Example 4 with LegacyObjectMapType

use of com.google.template.soy.types.LegacyObjectMapType in project closure-templates by google.

the class TofuTypeChecksTest method testRecordTypeIsInstance.

@Test
public void testRecordTypeIsInstance() {
    LegacyObjectMapType mapOfStringToAny = LegacyObjectMapType.of(StringType.getInstance(), AnyType.getInstance());
    assertIsInstance(mapOfStringToAny, MAP_DATA, DICT_DATA);
    assertIsNotInstance(mapOfStringToAny, NULL_DATA, BOOLEAN_DATA, STRING_DATA, INTEGER_DATA, FLOAT_DATA, HTML_DATA, ATTRIBUTES_DATA, CSS_DATA, URI_DATA, JS_DATA);
}
Also used : LegacyObjectMapType(com.google.template.soy.types.LegacyObjectMapType) Test(org.junit.Test)

Example 5 with LegacyObjectMapType

use of com.google.template.soy.types.LegacyObjectMapType in project closure-templates by google.

the class MapToLegacyObjectMapFunction method computeForJbcSrc.

@Override
public SoyExpression computeForJbcSrc(JbcSrcPluginContext context, List<SoyExpression> args) {
    SoyExpression soyExpression = Iterables.getOnlyElement(args);
    SoyType originalType = soyExpression.soyRuntimeType().soyType();
    LegacyObjectMapType newType;
    if (originalType instanceof MapType) {
        newType = LegacyObjectMapType.of(((MapType) originalType).getKeyType(), ((MapType) originalType).getValueType());
    } else if (originalType instanceof UnknownType) {
        newType = LegacyObjectMapType.of(UnknownType.getInstance(), UnknownType.getInstance());
    } else {
        throw new IllegalArgumentException("mapToLegacyObjectMap() expects input to be MAP, get " + originalType.getKind());
    }
    return SoyExpression.forLegacyObjectMap(newType, JbcSrcMethods.MAP_TO_LEGACY_OBJECT_MAP.invoke(soyExpression.box().checkedCast(SoyMap.class)));
}
Also used : UnknownType(com.google.template.soy.types.UnknownType) SoyExpression(com.google.template.soy.jbcsrc.restricted.SoyExpression) SoyType(com.google.template.soy.types.SoyType) LegacyObjectMapType(com.google.template.soy.types.LegacyObjectMapType) LegacyObjectMapType(com.google.template.soy.types.LegacyObjectMapType) MapType(com.google.template.soy.types.MapType)

Aggregations

LegacyObjectMapType (com.google.template.soy.types.LegacyObjectMapType)7 SoyType (com.google.template.soy.types.SoyType)4 SoyExpression (com.google.template.soy.jbcsrc.restricted.SoyExpression)3 MapType (com.google.template.soy.types.MapType)3 ListType (com.google.template.soy.types.ListType)2 RecordType (com.google.template.soy.types.RecordType)2 SoyProtoEnumType (com.google.template.soy.types.SoyProtoEnumType)2 UnknownType (com.google.template.soy.types.UnknownType)2 Test (org.junit.Test)2 Optional (com.google.common.base.Optional)1 ImmutableMap (com.google.common.collect.ImmutableMap)1 ImmutableSet (com.google.common.collect.ImmutableSet)1 ImmutableSortedSet (com.google.common.collect.ImmutableSortedSet)1 SoyBackendKind (com.google.template.soy.base.SoyBackendKind)1 SanitizedContentKind (com.google.template.soy.base.internal.SanitizedContentKind)1 Expression (com.google.template.soy.jbcsrc.restricted.Expression)1 CodeChunk (com.google.template.soy.jssrc.dsl.CodeChunk)1 Generator (com.google.template.soy.jssrc.dsl.CodeChunk.Generator)1 SoyProtoType (com.google.template.soy.types.SoyProtoType)1 Kind (com.google.template.soy.types.SoyType.Kind)1