lemmy/server/src/feeds.rs

extern crate rss;
extern crate htmlescape;

use super::*;
use crate::Settings;
use crate::db::{establish_connection, ListingType, SortType};
use crate::db::community_view::SiteView;
use crate::db::post_view::PostView;
use crate::db::user::User_;
use crate::db::community::Community;
use actix_web::{HttpResponse, web, Result};
use actix_web::body::Body;
use rss::{ChannelBuilder, Item, ItemBuilder};
use diesel::result::Error;
use std::str::FromStr;
use self::rss::Guid;
use serde::Deserialize;
use strum::ParseError;

#[derive(Deserialize)]
pub struct Params {
  sort: Option<String>,
}

enum RequestType {
  All,
  Community,
  User,
}

pub fn get_all_feed(info: web::Query<Params>) -> HttpResponse<Body> {
  let sort_type = get_sort_type(info);
  if sort_type.is_err() {
    return HttpResponse::BadRequest().finish();
  }

  let result = get_feed_internal(&sort_type.unwrap(), RequestType::All, None);
  return match result {
    Ok(rss) => HttpResponse::Ok()
    .content_type("application/rss+xml")
    .body(rss),
    Err(_) => HttpResponse::InternalServerError().finish(),
  };
}

pub fn get_feed(path: web::Path<(char, String)>, info: web::Query<Params>) -> HttpResponse<Body> {
  let sort_type = get_sort_type(info);
  if sort_type.is_err() {
    return HttpResponse::BadRequest().finish();
  }

  let request_type = match path.0 {
    'u' => RequestType::User,
    'c' =>  RequestType::Community,
    _ => return HttpResponse::NotFound().finish(),
  };

  let result = get_feed_internal(&sort_type.unwrap(), request_type, Some(path.1.clone()));
  if result.is_ok() {
    let rss = result.unwrap();
    return HttpResponse::Ok()
      .content_type("application/rss+xml")
      .body(rss);
  } else {
    let error = result.err().unwrap();
    return match error {
      Error::NotFound => HttpResponse::NotFound().finish(),
      _ => HttpResponse::InternalServerError().finish(),
    }
  }
}

fn get_sort_type(info: web::Query<Params>) -> Result<SortType, ParseError> {
  let sort_query = info.sort.clone().unwrap_or(SortType::Hot.to_string());
  return SortType::from_str(&sort_query);
}

fn get_feed_internal(sort_type: &SortType, request_type: RequestType, name: Option<String>)
    -> Result<String, Error> {
  let conn = establish_connection();

  let mut community_id: Option<i32> = None;
  let mut creator_id: Option<i32> = None;
  match request_type {
    RequestType::All =>(),
    RequestType::Community =>  community_id = Some(Community::read_from_name(&conn,name.unwrap())?.id),
    RequestType::User => creator_id = Some(User_::find_by_email_or_username(&conn,&name.unwrap())?.id),
  }

  let post = PostView::list(&conn,
    ListingType::All,
    sort_type,
    community_id,
    creator_id,
    None,
    None,
    None,
    true,
    false,
    false,
    None,
    None,)?;

  let mut items: Vec<Item> = Vec::new();
  for p in post {
    let dt = DateTime::<Utc>::from_utc(p.published, Utc);
    let mut i = ItemBuilder::default();
    i.title(htmlescape::encode_minimal(&p.name));
    i.pub_date(htmlescape::encode_minimal(&dt.to_rfc2822()));

    let post_url = format!("https://{}/post/{}", Settings::get().hostname, p.id);
    let mut guid = Guid::default();
    guid.set_permalink(true);
    guid.set_value(&post_url);
    i.guid(guid);
    i.comments(post_url);

    if p.url.is_some() {
      i.link(p.url.unwrap());
    }
    if p.body.is_some() {
      i.content(p.body.unwrap());
    }
    items.push(i.build().unwrap());
  }

  let site_view = SiteView::read(&conn)?;
  let mut channel_builder = ChannelBuilder::default();
  channel_builder.title(htmlescape::encode_minimal(&site_view.name))
    .link(format!("https://{}", Settings::get().hostname))
    .items(items);
  if site_view.description.is_some() {
    channel_builder.description(htmlescape::encode_minimal(&site_view.description.unwrap()));
  }
  let channel = channel_builder.build().unwrap();
  channel.write_to(::std::io::sink()).unwrap();

  return Ok(channel.to_string());
}
Implement RSS feeds (fixes #118) 5 years ago			`extern crate rss;`
			`extern crate htmlescape;`

basic, working rss feeds 5 years ago			`use super::*;`
Implement RSS feeds (fixes #118) 5 years ago			`use crate::Settings;`
other sort options, sort by hot by default, adjust rss fields 5 years ago			`use crate::db::{establish_connection, ListingType, SortType};`
Implement RSS feeds (fixes #118) 5 years ago			`use crate::db::community_view::SiteView;`
			`use crate::db::post_view::PostView;`
basic, working rss feeds 5 years ago			`use crate::db::user::User_;`
			`use crate::db::community::Community;`
fix sort parameter handling, handle errors 5 years ago			`use actix_web::{HttpResponse, web, Result};`
basic, working rss feeds 5 years ago			`use actix_web::body::Body;`
			`use rss::{ChannelBuilder, Item, ItemBuilder};`
			`use diesel::result::Error;`
other sort options, sort by hot by default, adjust rss fields 5 years ago			`use std::str::FromStr;`
			`use self::rss::Guid;`
fix sort parameter handling, handle errors 5 years ago			`use serde::Deserialize;`
Add feed for all 5 years ago			`use strum::ParseError;`
basic, working rss feeds 5 years ago
fix sort parameter handling, handle errors 5 years ago			`#[derive(Deserialize)]`
			`pub struct Params {`
			`sort: Option<String>,`
			`}`

Add feed for all 5 years ago			`enum RequestType {`
			`All,`
			`Community,`
			`User,`
			`}`

			`pub fn get_all_feed(info: web::Query<Params>) -> HttpResponse<Body> {`
			`let sort_type = get_sort_type(info);`
			`if sort_type.is_err() {`
			`return HttpResponse::BadRequest().finish();`
			`}`

			`let result = get_feed_internal(&sort_type.unwrap(), RequestType::All, None);`
			`return match result {`
			`Ok(rss) => HttpResponse::Ok()`
			`.content_type("application/rss+xml")`
			`.body(rss),`
			`Err(_) => HttpResponse::InternalServerError().finish(),`
			`};`
			`}`

fix sort parameter handling, handle errors 5 years ago			`pub fn get_feed(path: web::Path<(char, String)>, info: web::Query<Params>) -> HttpResponse<Body> {`
Add feed for all 5 years ago			`let sort_type = get_sort_type(info);`
			`if sort_type.is_err() {`
			`return HttpResponse::BadRequest().finish();`
			`}`

			`let request_type = match path.0 {`
			`'u' => RequestType::User,`
			`'c' => RequestType::Community,`
			`_ => return HttpResponse::NotFound().finish(),`
other sort options, sort by hot by default, adjust rss fields 5 years ago			`};`

Add feed for all 5 years ago			`let result = get_feed_internal(&sort_type.unwrap(), request_type, Some(path.1.clone()));`
fix sort parameter handling, handle errors 5 years ago			`if result.is_ok() {`
			`let rss = result.unwrap();`
			`return HttpResponse::Ok()`
basic, working rss feeds 5 years ago			`.content_type("application/rss+xml")`
fix sort parameter handling, handle errors 5 years ago			`.body(rss);`
			`} else {`
			`let error = result.err().unwrap();`
			`return match error {`
			`Error::NotFound => HttpResponse::NotFound().finish(),`
			`_ => HttpResponse::InternalServerError().finish(),`
			`}`
basic, working rss feeds 5 years ago			`}`
			`}`
Implement RSS feeds (fixes #118) 5 years ago
Add feed for all 5 years ago			`fn get_sort_type(info: web::Query<Params>) -> Result<SortType, ParseError> {`
			`let sort_query = info.sort.clone().unwrap_or(SortType::Hot.to_string());`
			`return SortType::from_str(&sort_query);`
			`}`

			`fn get_feed_internal(sort_type: &SortType, request_type: RequestType, name: Option<String>)`
			`-> Result<String, Error> {`
Implement RSS feeds (fixes #118) 5 years ago			`let conn = establish_connection();`

basic, working rss feeds 5 years ago			`let mut community_id: Option<i32> = None;`
			`let mut creator_id: Option<i32> = None;`
Add feed for all 5 years ago			`match request_type {`
			`RequestType::All =>(),`
			`RequestType::Community => community_id = Some(Community::read_from_name(&conn,name.unwrap())?.id),`
			`RequestType::User => creator_id = Some(User_::find_by_email_or_username(&conn,&name.unwrap())?.id),`
basic, working rss feeds 5 years ago			`}`

			`let post = PostView::list(&conn,`
Implement RSS feeds (fixes #118) 5 years ago			`ListingType::All,`
other sort options, sort by hot by default, adjust rss fields 5 years ago			`sort_type,`
basic, working rss feeds 5 years ago			`community_id,`
			`creator_id,`
Implement RSS feeds (fixes #118) 5 years ago			`None,`
			`None,`
			`None,`
			`true,`
			`false,`
			`false,`
			`None,`
basic, working rss feeds 5 years ago			`None,)?;`
Implement RSS feeds (fixes #118) 5 years ago
			`let mut items: Vec<Item> = Vec::new();`
			`for p in post {`
basic, working rss feeds 5 years ago			`let dt = DateTime::<Utc>::from_utc(p.published, Utc);`
			`let mut i = ItemBuilder::default();`
			`i.title(htmlescape::encode_minimal(&p.name));`
			`i.pub_date(htmlescape::encode_minimal(&dt.to_rfc2822()));`
other sort options, sort by hot by default, adjust rss fields 5 years ago
			`let post_url = format!("https://{}/post/{}", Settings::get().hostname, p.id);`
			`let mut guid = Guid::default();`
			`guid.set_permalink(true);`
			`guid.set_value(&post_url);`
			`i.guid(guid);`
			`i.comments(post_url);`

basic, working rss feeds 5 years ago			`if p.url.is_some() {`
			`i.link(p.url.unwrap());`
			`}`
			`if p.body.is_some() {`
			`i.content(p.body.unwrap());`
			`}`
			`items.push(i.build().unwrap());`
Implement RSS feeds (fixes #118) 5 years ago			`}`

basic, working rss feeds 5 years ago			`let site_view = SiteView::read(&conn)?;`
			`let mut channel_builder = ChannelBuilder::default();`
			`channel_builder.title(htmlescape::encode_minimal(&site_view.name))`
Implement RSS feeds (fixes #118) 5 years ago			`.link(format!("https://{}", Settings::get().hostname))`
basic, working rss feeds 5 years ago			`.items(items);`
			`if site_view.description.is_some() {`
			`channel_builder.description(htmlescape::encode_minimal(&site_view.description.unwrap()));`
			`}`
			`let channel = channel_builder.build().unwrap();`
Implement RSS feeds (fixes #118) 5 years ago			`channel.write_to(::std::io::sink()).unwrap();`

basic, working rss feeds 5 years ago			`return Ok(channel.to_string());`
Implement RSS feeds (fixes #118) 5 years ago			`}`