src/api_routes.rs

   1 use actix_web::{error::ErrorBadRequest, *};
   2 use lemmy_api::Perform;
   3 use lemmy_api_common::{comment::*, community::*, person::*, post::*, site::*, websocket::*};
   4 use lemmy_api_crud::PerformCrud;
   5 use lemmy_utils::rate_limit::RateLimit;
   6 use lemmy_websocket::LemmyContext;
   7 use serde::Deserialize;
   8
   9 pub fn config(cfg: &mut web::ServiceConfig, rate_limit: &RateLimit) {
  10   cfg.service(
  11     web::scope("/api/v2")
  12       // Site
  13       .service(
  14         web::scope("/site")
  15           .wrap(rate_limit.message())
  16           .route("", web::get().to(route_get_crud::<GetSite>))
  17           // Admin Actions
  18           .route("", web::post().to(route_post_crud::<CreateSite>))
  19           .route("", web::put().to(route_post_crud::<EditSite>))
  20           .route("/transfer", web::post().to(route_post::<TransferSite>))
  21           .route("/config", web::get().to(route_get::<GetSiteConfig>))
  22           .route("/config", web::put().to(route_post::<SaveSiteConfig>)),
  23       )
  24       .service(
  25         web::resource("/modlog")
  26           .wrap(rate_limit.message())
  27           .route(web::get().to(route_get::<GetModlog>)),
  28       )
  29       .service(
  30         web::resource("/search")
  31           .wrap(rate_limit.message())
  32           .route(web::get().to(route_get::<Search>)),
  33       )
  34       // Community
  35       .service(
  36         web::resource("/community")
  37           .guard(guard::Post())
  38           .wrap(rate_limit.register())
  39           .route(web::post().to(route_post_crud::<CreateCommunity>)),
  40       )
  41       .service(
  42         web::scope("/community")
  43           .wrap(rate_limit.message())
  44           .route("", web::get().to(route_get_crud::<GetCommunity>))
  45           .route("", web::put().to(route_post_crud::<EditCommunity>))
  46           .route("/list", web::get().to(route_get_crud::<ListCommunities>))
  47           .route("/follow", web::post().to(route_post::<FollowCommunity>))
  48           .route(
  49             "/delete",
  50             web::post().to(route_post_crud::<DeleteCommunity>),
  51           )
  52           // Mod Actions
  53           .route(
  54             "/remove",
  55             web::post().to(route_post_crud::<RemoveCommunity>),
  56           )
  57           .route("/transfer", web::post().to(route_post::<TransferCommunity>))
  58           .route("/ban_user", web::post().to(route_post::<BanFromCommunity>))
  59           .route("/mod", web::post().to(route_post::<AddModToCommunity>))
  60           .route("/join", web::post().to(route_post::<CommunityJoin>))
  61           .route("/mod/join", web::post().to(route_post::<ModJoin>)),
  62       )
  63       // Post
  64       .service(
  65         // Handle POST to /post separately to add the post() rate limitter
  66         web::resource("/post")
  67           .guard(guard::Post())
  68           .wrap(rate_limit.post())
  69           .route(web::post().to(route_post_crud::<CreatePost>)),
  70       )
  71       .service(
  72         web::scope("/post")
  73           .wrap(rate_limit.message())
  74           .route("", web::get().to(route_get_crud::<GetPost>))
  75           .route("", web::put().to(route_post_crud::<EditPost>))
  76           .route("/delete", web::post().to(route_post_crud::<DeletePost>))
  77           .route("/remove", web::post().to(route_post_crud::<RemovePost>))
  78           .route("/lock", web::post().to(route_post::<LockPost>))
  79           .route("/sticky", web::post().to(route_post::<StickyPost>))
  80           .route("/list", web::get().to(route_get_crud::<GetPosts>))
  81           .route("/like", web::post().to(route_post::<CreatePostLike>))
  82           .route("/save", web::put().to(route_post::<SavePost>))
  83           .route("/join", web::post().to(route_post::<PostJoin>))
  84           .route("/report", web::post().to(route_post::<CreatePostReport>))
  85           .route(
  86             "/report/resolve",
  87             web::put().to(route_post::<ResolvePostReport>),
  88           )
  89           .route("/report/list", web::get().to(route_get::<ListPostReports>)),
  90       )
  91       // Comment
  92       .service(
  93         web::scope("/comment")
  94           .wrap(rate_limit.message())
  95           .route("", web::post().to(route_post_crud::<CreateComment>))
  96           .route("", web::put().to(route_post_crud::<EditComment>))
  97           .route("/delete", web::post().to(route_post_crud::<DeleteComment>))
  98           .route("/remove", web::post().to(route_post_crud::<RemoveComment>))
  99           .route(
 100             "/mark_as_read",
 101             web::post().to(route_post::<MarkCommentAsRead>),
 102           )
 103           .route("/like", web::post().to(route_post::<CreateCommentLike>))
 104           .route("/save", web::put().to(route_post::<SaveComment>))
 105           .route("/list", web::get().to(route_get_crud::<GetComments>))
 106           .route("/report", web::post().to(route_post::<CreateCommentReport>))
 107           .route(
 108             "/report/resolve",
 109             web::put().to(route_post::<ResolveCommentReport>),
 110           )
 111           .route(
 112             "/report/list",
 113             web::get().to(route_get::<ListCommentReports>),
 114           ),
 115       )
 116       // Private Message
 117       .service(
 118         web::scope("/private_message")
 119           .wrap(rate_limit.message())
 120           .route("/list", web::get().to(route_get_crud::<GetPrivateMessages>))
 121           .route("", web::post().to(route_post_crud::<CreatePrivateMessage>))
 122           .route("", web::put().to(route_post_crud::<EditPrivateMessage>))
 123           .route(
 124             "/delete",
 125             web::post().to(route_post_crud::<DeletePrivateMessage>),
 126           )
 127           .route(
 128             "/mark_as_read",
 129             web::post().to(route_post::<MarkPrivateMessageAsRead>),
 130           ),
 131       )
 132       // User
 133       .service(
 134         // Account action, I don't like that it's in /user maybe /accounts
 135         // Handle /user/register separately to add the register() rate limitter
 136         web::resource("/user/register")
 137           .guard(guard::Post())
 138           .wrap(rate_limit.register())
 139           .route(web::post().to(route_post_crud::<Register>)),
 140       )
 141       // User actions
 142       .service(
 143         web::scope("/user")
 144           .wrap(rate_limit.message())
 145           .route("", web::get().to(route_get_crud::<GetPersonDetails>))
 146           .route("/mention", web::get().to(route_get::<GetPersonMentions>))
 147           .route(
 148             "/mention/mark_as_read",
 149             web::post().to(route_post::<MarkPersonMentionAsRead>),
 150           )
 151           .route("/replies", web::get().to(route_get::<GetReplies>))
 152           .route(
 153             "/followed_communities",
 154             web::get().to(route_get::<GetFollowedCommunities>),
 155           )
 156           .route("/join", web::post().to(route_post::<UserJoin>))
 157           // Admin action. I don't like that it's in /user
 158           .route("/ban", web::post().to(route_post::<BanPerson>))
 159           // Account actions. I don't like that they're in /user maybe /accounts
 160           .route("/login", web::post().to(route_post::<Login>))
 161           .route("/get_captcha", web::get().to(route_get::<GetCaptcha>))
 162           .route(
 163             "/delete_account",
 164             web::post().to(route_post_crud::<DeleteAccount>),
 165           )
 166           .route(
 167             "/password_reset",
 168             web::post().to(route_post::<PasswordReset>),
 169           )
 170           .route(
 171             "/password_change",
 172             web::post().to(route_post::<PasswordChange>),
 173           )
 174           // mark_all_as_read feels off being in this section as well
 175           .route(
 176             "/mark_all_as_read",
 177             web::post().to(route_post::<MarkAllAsRead>),
 178           )
 179           .route(
 180             "/save_user_settings",
 181             web::put().to(route_post::<SaveUserSettings>),
 182           )
 183           .route("/report_count", web::get().to(route_get::<GetReportCount>)),
 184       )
 185       // Admin Actions
 186       .service(
 187         web::resource("/admin/add")
 188           .wrap(rate_limit.message())
 189           .route(web::post().to(route_post::<AddAdmin>)),
 190       ),
 191   );
 192 }
 193
 194 async fn perform<Request>(
 195   data: Request,
 196   context: web::Data<LemmyContext>,
 197 ) -> Result<HttpResponse, Error>
 198 where
 199   Request: Perform,
 200   Request: Send + 'static,
 201 {
 202   let res = data
 203     .perform(&context, None)
 204     .await
 205     .map(|json| HttpResponse::Ok().json(json))
 206     .map_err(ErrorBadRequest)?;
 207   Ok(res)
 208 }
 209
 210 async fn route_get<'a, Data>(
 211   data: web::Query<Data>,
 212   context: web::Data<LemmyContext>,
 213 ) -> Result<HttpResponse, Error>
 214 where
 215   Data: Deserialize<'a> + Send + 'static + Perform,
 216 {
 217   perform::<Data>(data.0, context).await
 218 }
 219
 220 async fn route_post<'a, Data>(
 221   data: web::Json<Data>,
 222   context: web::Data<LemmyContext>,
 223 ) -> Result<HttpResponse, Error>
 224 where
 225   Data: Deserialize<'a> + Send + 'static + Perform,
 226 {
 227   perform::<Data>(data.0, context).await
 228 }
 229
 230 async fn perform_crud<Request>(
 231   data: Request,
 232   context: web::Data<LemmyContext>,
 233 ) -> Result<HttpResponse, Error>
 234 where
 235   Request: PerformCrud,
 236   Request: Send + 'static,
 237 {
 238   let res = data
 239     .perform(&context, None)
 240     .await
 241     .map(|json| HttpResponse::Ok().json(json))
 242     .map_err(ErrorBadRequest)?;
 243   Ok(res)
 244 }
 245
 246 async fn route_get_crud<'a, Data>(
 247   data: web::Query<Data>,
 248   context: web::Data<LemmyContext>,
 249 ) -> Result<HttpResponse, Error>
 250 where
 251   Data: Deserialize<'a> + Send + 'static + PerformCrud,
 252 {
 253   perform_crud::<Data>(data.0, context).await
 254 }
 255
 256 async fn route_post_crud<'a, Data>(
 257   data: web::Json<Data>,
 258   context: web::Data<LemmyContext>,
 259 ) -> Result<HttpResponse, Error>
 260 where
 261   Data: Deserialize<'a> + Send + 'static + PerformCrud,
 262 {
 263   perform_crud::<Data>(data.0, context).await
 264 }