mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-27 21:46:04 +02:00
31 lines
1.4 KiB
Scala
31 lines
1.4 KiB
Scala
|
package com.twitter.home_mixer.functional_component.decorator.urt.builder
|
||
|
|
|
||
|
|
import com.twitter.product_mixer.component_library.premarshaller.urt.builder.AlwaysInclude
|
||
|
|
import com.twitter.product_mixer.component_library.premarshaller.urt.builder.IncludeInstruction
|
||
|
|
import com.twitter.product_mixer.component_library.premarshaller.urt.builder.UrtInstructionBuilder
|
||
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.AddEntriesTimelineInstruction
|
||
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.Cover
|
||
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.ShowAlert
|
||
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.TimelineEntry
|
||
|
|
import com.twitter.product_mixer.core.pipeline.PipelineQuery
|
||
|
|
|
||
|
|
case class AddEntriesWithReplaceAndShowAlertAndCoverInstructionBuilder[Query <: PipelineQuery](
|
||
|
|
override val includeInstruction: IncludeInstruction[Query] = AlwaysInclude)
|
||
|
|
extends UrtInstructionBuilder[Query, AddEntriesTimelineInstruction] {
|
||
|
|
|
||
|
|
override def build(
|
||
|
|
query: Query,
|
||
|
|
entries: Seq[TimelineEntry]
|
||
|
|
): Seq[AddEntriesTimelineInstruction] = {
|
||
|
|
if (includeInstruction(query, entries)) {
|
||
|
|
val entriesToAdd = entries
|
||
|
|
.filterNot(_.isInstanceOf[ShowAlert])
|
||
|
|
.filterNot(_.isInstanceOf[Cover])
|
||
|
|
.filter(_.entryIdToReplace.isEmpty)
|
||
|
|
if (entriesToAdd.nonEmpty) Seq(AddEntriesTimelineInstruction(entriesToAdd))
|
||
|
|
else Seq.empty
|
||
|
|
} else
|
||
|
|
Seq.empty
|
||
|
|
}
|
||
|
|
}
|