the-algorithm/product-mixer/core/src/main/scala/com/twitter/product_mixer/core/functional_component/marshaller/response/urt/item/tweet/TweetHighlightsMarshaller.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

23 lines
1011 B
Scala

package com.twitter.product_mixer.core.functional_component.marshaller.response.urt.item.tweet
import com.twitter.product_mixer.core.functional_component.marshaller.response.urt.item.highlight.HighlightedSectionMarshaller
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.tweet.TweetHighlights
import com.twitter.timelines.render.{thriftscala => urt}
import javax.inject.Inject
import javax.inject.Singleton
@Singleton
class TweetHighlightsMarshaller @Inject() (
highlightedSectionMarshaller: HighlightedSectionMarshaller) {
def apply(tweetHighlights: TweetHighlights): urt.TweetHighlights =
urt.TweetHighlights(
textHighlights = tweetHighlights.textHighlights
.map(_.map(highlightedSectionMarshaller(_))),
cardTitleHighlights = tweetHighlights.cardTitleHighlights
.map(_.map(highlightedSectionMarshaller(_))),
cardDescriptionHighlights = tweetHighlights.cardDescriptionHighlights
.map(_.map(highlightedSectionMarshaller(_)))
)
}