pub use self::err::TimelineErr; pub use self::inner::{Content, Reach, Scope, Stream, UserData}; use super::query::Query; use lru::LruCache; use warp::reject::Rejection; mod err; mod inner; type Result = std::result::Result; #[derive(Clone, Debug, Copy, Eq, Hash, PartialEq)] pub struct Timeline(pub Stream, pub Reach, pub Content); impl Timeline { pub fn empty() -> Self { Self(Stream::Unset, Reach::Local, Content::Notification) } pub fn to_redis_raw_timeline(&self, hashtag: Option<&String>) -> Result { use {Content::*, Reach::*, Stream::*}; Ok(match self { Timeline(Public, Federated, All) => "timeline:public".into(), Timeline(Public, Local, All) => "timeline:public:local".into(), Timeline(Public, Federated, Media) => "timeline:public:media".into(), Timeline(Public, Local, Media) => "timeline:public:local:media".into(), // TODO -- would `.push_str` be faster here? Timeline(Hashtag(_id), Federated, All) => format!( "timeline:hashtag:{}", hashtag.ok_or(TimelineErr::MissingHashtag)? ), Timeline(Hashtag(_id), Local, All) => format!( "timeline:hashtag:{}:local", hashtag.ok_or(TimelineErr::MissingHashtag)? ), Timeline(User(id), Federated, All) => format!("timeline:{}", id), Timeline(User(id), Federated, Notification) => format!("timeline:{}:notification", id), Timeline(List(id), Federated, All) => format!("timeline:list:{}", id), Timeline(Direct(id), Federated, All) => format!("timeline:direct:{}", id), Timeline(_one, _two, _three) => Err(TimelineErr::InvalidInput)?, }) } pub fn from_redis_text(timeline: &str, cache: &mut LruCache) -> Result { use {Content::*, Reach::*, Stream::*, TimelineErr::*}; let mut tag_id = |t: &str| cache.get(&t.to_string()).map_or(Err(BadTag), |id| Ok(*id)); Ok(match &timeline.split(':').collect::>()[..] { ["public"] => Timeline(Public, Federated, All), ["public", "local"] => Timeline(Public, Local, All), ["public", "media"] => Timeline(Public, Federated, Media), ["public", "local", "media"] => Timeline(Public, Local, Media), ["hashtag", tag] => Timeline(Hashtag(tag_id(tag)?), Federated, All), ["hashtag", tag, "local"] => Timeline(Hashtag(tag_id(tag)?), Local, All), [id] => Timeline(User(id.parse()?), Federated, All), [id, "notification"] => Timeline(User(id.parse()?), Federated, Notification), ["list", id] => Timeline(List(id.parse()?), Federated, All), ["direct", id] => Timeline(Direct(id.parse()?), Federated, All), // Other endpoints don't exist: [..] => Err(InvalidInput)?, }) } pub fn from_query_and_user(q: &Query, user: &UserData) -> std::result::Result { use {warp::reject::custom, Content::*, Reach::*, Scope::*, Stream::*}; Ok(match q.stream.as_ref() { "public" => match q.media { true => Timeline(Public, Federated, Media), false => Timeline(Public, Federated, All), }, "public:local" => match q.media { true => Timeline(Public, Local, Media), false => Timeline(Public, Local, All), }, "public:media" => Timeline(Public, Federated, Media), "public:local:media" => Timeline(Public, Local, Media), "hashtag" => Timeline(Hashtag(0), Federated, All), "hashtag:local" => Timeline(Hashtag(0), Local, All), "user" => match user.scopes.contains(&Statuses) { true => Timeline(User(user.id), Federated, All), false => Err(custom("Error: Missing access token"))?, }, "user:notification" => match user.scopes.contains(&Statuses) { true => Timeline(User(user.id), Federated, Notification), false => Err(custom("Error: Missing access token"))?, }, "list" => match user.scopes.contains(&Lists) { true => Timeline(List(q.list), Federated, All), false => Err(warp::reject::custom("Error: Missing access token"))?, }, "direct" => match user.scopes.contains(&Statuses) { true => Timeline(Direct(*user.id), Federated, All), false => Err(custom("Error: Missing access token"))?, }, other => { log::warn!("Request for nonexistent endpoint: `{}`", other); Err(custom("Error: Nonexistent endpoint"))? } }) } }