/*
 * Copyright 2011-2023 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.springframework.data.mongodb.core.convert;

import java.util.Set;

import org.bson.Document;
import org.bson.conversions.Bson;
import org.springframework.data.convert.TypeMapper;

/**
 * Mongo-specific {@link TypeMapper} exposing that {@link Document}s might contain a type key.
 *
 * @author Oliver Gierke
 * @author Christoph Strobl
 * @since 1.0
 */
public interface MongoTypeMapper extends TypeMapper<Bson> {

	/**
	 * Returns whether the given key is the type key.
	 *
	 * @return
	 */
	boolean isTypeKey(String key);

	/**
	 * Writes type restrictions to the given {@link Document}. This usually results in an {@code $in}-clause to be
	 * generated that restricts the type-key (e.g. {@code _class}) to be in the set of type aliases for the given
	 * {@code restrictedTypes}.
	 *
	 * @param result must not be {@literal null}
	 * @param restrictedTypes must not be {@literal null}
	 */
	void writeTypeRestrictions(Document result, Set<Class<?>> restrictedTypes);

	/**
	 * Compute the target type for a given source considering {@link org.springframework.data.convert.CustomConversions}.
	 *
	 * @param source the source type.
	 * @return never {@literal null}.
	 * @since 2.2
	 */
	default Class<?> getWriteTargetTypeFor(Class<?> source) {
		return source;
	}
}
