the-algorithm/product-mixer/core/src/main/scala/com/twitter/product_mixer/core/functional_component/marshaller/response/urt/media/MediaMarshaller.scala
twitter-team ef4c5eb65e Twitter Recommendation Algorithm
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
2023-03-31 17:36:31 -05:00

24 lines
867 B
Scala

package com.twitter.product_mixer.core.functional_component.marshaller.response.urt.media
import com.twitter.product_mixer.core.model.marshalling.response.urt.media.Media
import com.twitter.timelines.render.{thriftscala => urt}
import javax.inject.Inject
import javax.inject.Singleton
@Singleton
class MediaMarshaller @Inject() (
mediaEntityMarshaller: MediaEntityMarshaller,
mediaKeyMarshaller: MediaKeyMarshaller,
rectMarshaller: RectMarshaller,
aspectRatioMarshaller: AspectRatioMarshaller) {
def apply(media: Media): urt.Media = urt.Media(
mediaEntity = media.mediaEntity.map(mediaEntityMarshaller(_)),
mediaKey = media.mediaKey.map(mediaKeyMarshaller(_)),
imagePossibleCropping = media.imagePossibleCropping.map { rects =>
rects.map(rectMarshaller(_))
},
aspectRatio = media.aspectRatio.map(aspectRatioMarshaller(_))
)
}