79 lines
3.2 KiB
Scala
79 lines
3.2 KiB
Scala
package com.twitter.home_mixer.functional_component.decorator.urt.builder
|
|
|
|
import com.twitter.home_mixer.model.HomeFeatures.FocalTweetAuthorIdFeature
|
|
import com.twitter.home_mixer.model.HomeFeatures.FocalTweetInNetworkFeature
|
|
import com.twitter.home_mixer.model.HomeFeatures.FocalTweetRealNamesFeature
|
|
import com.twitter.home_mixer.model.HomeFeatures.InNetworkFeature
|
|
import com.twitter.home_mixer.product.following.model.HomeMixerExternalStrings
|
|
import com.twitter.product_mixer.component_library.model.candidate.TweetCandidate
|
|
import com.twitter.product_mixer.core.feature.featuremap.FeatureMap
|
|
import com.twitter.product_mixer.core.functional_component.decorator.urt.builder.social_context.BaseSocialContextBuilder
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.metadata.SocialContext
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.metadata._
|
|
import com.twitter.product_mixer.core.pipeline.PipelineQuery
|
|
import com.twitter.product_mixer.core.product.guice.scope.ProductScoped
|
|
import com.twitter.stringcenter.client.StringCenter
|
|
import javax.inject.Inject
|
|
import javax.inject.Provider
|
|
import javax.inject.Singleton
|
|
|
|
/**
|
|
* Use '@A replied' when the root tweet is out-of-network and the reply is in network.
|
|
*
|
|
* This function should only be called for the root Tweet of convo modules. This is enforced by
|
|
* [[HomeTweetSocialContextBuilder]].
|
|
*/
|
|
@Singleton
|
|
case class ExtendedReplySocialContextBuilder @Inject() (
|
|
externalStrings: HomeMixerExternalStrings,
|
|
@ProductScoped stringCenterProvider: Provider[StringCenter])
|
|
extends BaseSocialContextBuilder[PipelineQuery, TweetCandidate] {
|
|
|
|
private val stringCenter = stringCenterProvider.get()
|
|
private val extendedReplyString = externalStrings.socialContextExtendedReply
|
|
|
|
def apply(
|
|
query: PipelineQuery,
|
|
candidate: TweetCandidate,
|
|
candidateFeatures: FeatureMap
|
|
): Option[SocialContext] = {
|
|
|
|
// If these values are missing default to not showing an extended reply banner
|
|
val inNetworkRoot = candidateFeatures.getOrElse(InNetworkFeature, true)
|
|
|
|
val inNetworkFocalTweet =
|
|
candidateFeatures.getOrElse(FocalTweetInNetworkFeature, None).getOrElse(false)
|
|
|
|
if (!inNetworkRoot && inNetworkFocalTweet) {
|
|
|
|
val focalTweetAuthorIdOpt = candidateFeatures.getOrElse(FocalTweetAuthorIdFeature, None)
|
|
val focalTweetRealNames =
|
|
candidateFeatures
|
|
.getOrElse(FocalTweetRealNamesFeature, None).getOrElse(Map.empty[Long, String])
|
|
val focalTweetAuthorNameOpt = focalTweetAuthorIdOpt.flatMap(focalTweetRealNames.get)
|
|
|
|
(focalTweetAuthorIdOpt, focalTweetAuthorNameOpt) match {
|
|
case (Some(focalTweetAuthorId), Some(focalTweetAuthorName)) =>
|
|
Some(
|
|
GeneralContext(
|
|
contextType = ConversationGeneralContextType,
|
|
text = stringCenter
|
|
.prepare(extendedReplyString, placeholders = Map("user1" -> focalTweetAuthorName)),
|
|
url = None,
|
|
contextImageUrls = None,
|
|
landingUrl = Some(
|
|
Url(
|
|
urlType = DeepLink,
|
|
url = "",
|
|
urtEndpointOptions = None
|
|
))
|
|
))
|
|
case _ =>
|
|
None
|
|
}
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
}
|