mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-06-01 08:48:46 +02:00
ef4c5eb65e
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.
33 lines
1.2 KiB
Scala
33 lines
1.2 KiB
Scala
package com.twitter.product_mixer.component_library.premarshaller.urt.builder
|
|
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.MarkEntriesUnreadInstruction
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.TimelineEntry
|
|
import com.twitter.product_mixer.core.pipeline.PipelineQuery
|
|
import com.twitter.product_mixer.core.model.marshalling.response.urt.metadata.MarkUnreadableEntry
|
|
|
|
/**
|
|
* Build a MarkUnreadEntries instruction
|
|
*
|
|
* Note that this implementation currently supports top-level entries, but not module item entries.
|
|
*/
|
|
case class MarkUnreadInstructionBuilder[Query <: PipelineQuery](
|
|
override val includeInstruction: IncludeInstruction[Query] = AlwaysInclude)
|
|
extends UrtInstructionBuilder[Query, MarkEntriesUnreadInstruction] {
|
|
|
|
override def build(
|
|
query: Query,
|
|
entries: Seq[TimelineEntry]
|
|
): Seq[MarkEntriesUnreadInstruction] = {
|
|
if (includeInstruction(query, entries)) {
|
|
val filteredEntries = entries.collect {
|
|
case entry: MarkUnreadableEntry if entry.isMarkUnread.contains(true) =>
|
|
entry.entryIdentifier
|
|
}
|
|
if (filteredEntries.nonEmpty) Seq(MarkEntriesUnreadInstruction(filteredEntries))
|
|
else Seq.empty
|
|
} else {
|
|
Seq.empty
|
|
}
|
|
}
|
|
}
|