crates/websocket/src/lib.rs

   1 #[macro_use]
   2 extern crate strum_macros;
   3
   4 use crate::chat_server::ChatServer;
   5 use actix::Addr;
   6 use background_jobs::QueueHandle;
   7 use lemmy_db_schema::{source::secret::Secret, DbPool};
   8 use lemmy_utils::{settings::structs::Settings, LemmyError};
   9 use reqwest_middleware::ClientWithMiddleware;
  10 use serde::Serialize;
  11
  12 pub mod chat_server;
  13 pub mod handlers;
  14 pub mod messages;
  15 pub mod routes;
  16 pub mod send;
  17
  18 pub struct LemmyContext {
  19   pool: DbPool,
  20   chat_server: Addr<ChatServer>,
  21   client: ClientWithMiddleware,
  22   activity_queue: QueueHandle,
  23   settings: Settings,
  24   secret: Secret,
  25 }
  26
  27 impl LemmyContext {
  28   pub fn create(
  29     pool: DbPool,
  30     chat_server: Addr<ChatServer>,
  31     client: ClientWithMiddleware,
  32     activity_queue: QueueHandle,
  33     settings: Settings,
  34     secret: Secret,
  35   ) -> LemmyContext {
  36     LemmyContext {
  37       pool,
  38       chat_server,
  39       client,
  40       activity_queue,
  41       settings,
  42       secret,
  43     }
  44   }
  45   pub fn pool(&self) -> &DbPool {
  46     &self.pool
  47   }
  48   pub fn chat_server(&self) -> &Addr<ChatServer> {
  49     &self.chat_server
  50   }
  51   pub fn client(&self) -> &ClientWithMiddleware {
  52     &self.client
  53   }
  54   pub fn activity_queue(&self) -> &QueueHandle {
  55     &self.activity_queue
  56   }
  57   pub fn settings(&self) -> Settings {
  58     // TODO hacky solution to be able to hotload the settings.
  59     Settings::get()
  60   }
  61   pub fn secret(&self) -> &Secret {
  62     &self.secret
  63   }
  64 }
  65
  66 impl Clone for LemmyContext {
  67   fn clone(&self) -> Self {
  68     LemmyContext {
  69       pool: self.pool.clone(),
  70       chat_server: self.chat_server.clone(),
  71       client: self.client.clone(),
  72       activity_queue: self.activity_queue.clone(),
  73       settings: self.settings.clone(),
  74       secret: self.secret.clone(),
  75     }
  76   }
  77 }
  78
  79 #[derive(Serialize)]
  80 struct WebsocketResponse<T> {
  81   op: String,
  82   data: T,
  83 }
  84
  85 pub fn serialize_websocket_message<OP, Response>(
  86   op: &OP,
  87   data: &Response,
  88 ) -> Result<String, LemmyError>
  89 where
  90   Response: Serialize,
  91   OP: ToString,
  92 {
  93   let response = WebsocketResponse {
  94     op: op.to_string(),
  95     data,
  96   };
  97   Ok(serde_json::to_string(&response)?)
  98 }
  99
 100 #[derive(EnumString, Display, Debug, Clone)]
 101 pub enum UserOperation {
 102   Login,
 103   GetCaptcha,
 104   MarkCommentAsRead,
 105   SaveComment,
 106   CreateCommentLike,
 107   CreateCommentReport,
 108   ResolveCommentReport,
 109   ListCommentReports,
 110   CreatePostLike,
 111   LockPost,
 112   StickyPost,
 113   MarkPostAsRead,
 114   SavePost,
 115   CreatePostReport,
 116   ResolvePostReport,
 117   ListPostReports,
 118   GetReportCount,
 119   GetUnreadCount,
 120   FollowCommunity,
 121   GetReplies,
 122   GetPersonMentions,
 123   MarkPersonMentionAsRead,
 124   GetModlog,
 125   BanFromCommunity,
 126   AddModToCommunity,
 127   AddAdmin,
 128   BanPerson,
 129   Search,
 130   ResolveObject,
 131   MarkAllAsRead,
 132   SaveUserSettings,
 133   TransferCommunity,
 134   TransferSite,
 135   PasswordReset,
 136   PasswordChange,
 137   MarkPrivateMessageAsRead,
 138   UserJoin,
 139   GetSiteConfig,
 140   SaveSiteConfig,
 141   PostJoin,
 142   CommunityJoin,
 143   ModJoin,
 144   ChangePassword,
 145   GetSiteMetadata,
 146   BlockCommunity,
 147   BlockPerson,
 148 }
 149
 150 #[derive(EnumString, Display, Debug, Clone)]
 151 pub enum UserOperationCrud {
 152   // Site
 153   CreateSite,
 154   GetSite,
 155   EditSite,
 156   // Community
 157   CreateCommunity,
 158   ListCommunities,
 159   GetCommunity,
 160   EditCommunity,
 161   DeleteCommunity,
 162   RemoveCommunity,
 163   // Post
 164   CreatePost,
 165   GetPost,
 166   GetPosts,
 167   EditPost,
 168   DeletePost,
 169   RemovePost,
 170   // Comment
 171   CreateComment,
 172   GetComment,
 173   GetComments,
 174   EditComment,
 175   DeleteComment,
 176   RemoveComment,
 177   // User
 178   Register,
 179   GetPersonDetails,
 180   DeleteAccount,
 181   // Private Message
 182   CreatePrivateMessage,
 183   GetPrivateMessages,
 184   EditPrivateMessage,
 185   DeletePrivateMessage,
 186 }
 187
 188 pub trait OperationType {}
 189
 190 impl OperationType for UserOperationCrud {}
 191
 192 impl OperationType for UserOperation {}