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

22 lines
1.1 KiB
Scala

package com.twitter.product_mixer.core.functional_component.marshaller.response.urt.item.topic
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.topic.BasicTopicDisplayType
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.topic.NoIconTopicDisplayType
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.topic.PillTopicDisplayType
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.topic.PillWithoutActionIconDisplayType
import com.twitter.product_mixer.core.model.marshalling.response.urt.item.topic.TopicDisplayType
import com.twitter.timelines.render.{thriftscala => urt}
import javax.inject.Inject
import javax.inject.Singleton
@Singleton
class TopicDisplayTypeMarshaller @Inject() () {
def apply(topicDisplayType: TopicDisplayType): urt.TopicDisplayType = topicDisplayType match {
case BasicTopicDisplayType => urt.TopicDisplayType.Basic
case PillTopicDisplayType => urt.TopicDisplayType.Pill
case NoIconTopicDisplayType => urt.TopicDisplayType.NoIcon
case PillWithoutActionIconDisplayType => urt.TopicDisplayType.PillWithoutActionIcon
}
}