80aef61aed
More fixes - fixed docker builds - fixed mentions regex test - fixed DATABASE_URL stuff - change schema path in diesel.toml Address review comments - add jsonb column back into activity table - remove authors field from cargo.toml - adjust LEMMY_DATABASE_URL env var usage - rename all occurences of LEMMY_DATABASE_URL to DATABASE_URL Decouple utils and db Split code into cargo workspaces Co-authored-by: Felix Ableitner <me@nutomic.com> Reviewed-on: https://yerbamate.dev/LemmyNet/lemmy/pulls/67
73 lines
2 KiB
Rust
73 lines
2 KiB
Rust
use diesel::{result::Error, PgConnection};
|
|
use jsonwebtoken::{decode, encode, DecodingKey, EncodingKey, Header, TokenData, Validation};
|
|
use lemmy_db::{user::User_, Crud};
|
|
use lemmy_utils::{is_email_regex, settings::Settings};
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
type Jwt = String;
|
|
|
|
#[derive(Debug, Serialize, Deserialize)]
|
|
pub struct Claims {
|
|
pub id: i32,
|
|
pub username: String,
|
|
pub iss: String,
|
|
pub show_nsfw: bool,
|
|
pub theme: String,
|
|
pub default_sort_type: i16,
|
|
pub default_listing_type: i16,
|
|
pub lang: String,
|
|
pub avatar: Option<String>,
|
|
pub show_avatars: bool,
|
|
}
|
|
|
|
impl Claims {
|
|
pub fn decode(jwt: &str) -> Result<TokenData<Claims>, jsonwebtoken::errors::Error> {
|
|
let v = Validation {
|
|
validate_exp: false,
|
|
..Validation::default()
|
|
};
|
|
decode::<Claims>(
|
|
&jwt,
|
|
&DecodingKey::from_secret(Settings::get().jwt_secret.as_ref()),
|
|
&v,
|
|
)
|
|
}
|
|
|
|
pub fn jwt(user: User_, hostname: String) -> Jwt {
|
|
let my_claims = Claims {
|
|
id: user.id,
|
|
username: user.name.to_owned(),
|
|
iss: hostname,
|
|
show_nsfw: user.show_nsfw,
|
|
theme: user.theme.to_owned(),
|
|
default_sort_type: user.default_sort_type,
|
|
default_listing_type: user.default_listing_type,
|
|
lang: user.lang.to_owned(),
|
|
avatar: user.avatar.to_owned(),
|
|
show_avatars: user.show_avatars.to_owned(),
|
|
};
|
|
encode(
|
|
&Header::default(),
|
|
&my_claims,
|
|
&EncodingKey::from_secret(Settings::get().jwt_secret.as_ref()),
|
|
)
|
|
.unwrap()
|
|
}
|
|
|
|
// TODO: move these into user?
|
|
pub fn find_by_email_or_username(
|
|
conn: &PgConnection,
|
|
username_or_email: &str,
|
|
) -> Result<User_, Error> {
|
|
if is_email_regex(username_or_email) {
|
|
User_::find_by_email(conn, username_or_email)
|
|
} else {
|
|
User_::find_by_username(conn, username_or_email)
|
|
}
|
|
}
|
|
|
|
pub fn find_by_jwt(conn: &PgConnection, jwt: &str) -> Result<User_, Error> {
|
|
let claims: Claims = Claims::decode(&jwt).expect("Invalid token").claims;
|
|
User_::read(&conn, claims.id)
|
|
}
|
|
}
|