mirror of
https://github.com/storytold/artcraft.git
synced 2026-11-05 09:48:44 +00:00
something
This commit is contained in:
1 parent
c5c1187e24
commit
1cffbbcf35
6 files changed
+145
-26
No files matched your search
@@ -1,5 +1,6 @@
|
||||
use chrono::{DateTime, Utc};
|
||||
use sqlx::{MySql, MySqlPool, QueryBuilder};
|
||||
use log::warn;
|
||||
use sqlx::{Execute, MySql, MySqlPool, QueryBuilder};
|
||||
|
||||
use enums::by_table::media_files::media_file_origin_category::MediaFileOriginCategory;
|
||||
use enums::by_table::media_files::media_file_origin_model_type::MediaFileOriginModelType;
|
||||
@@ -8,6 +9,7 @@ use enums::by_table::media_files::media_file_type::MediaFileType;
|
||||
use enums::common::visibility::Visibility;
|
||||
use errors::AnyhowResult;
|
||||
use tokens::tokens::media_files::MediaFileToken;
|
||||
use tokens::tokens::users::UserToken;
|
||||
|
||||
pub struct MediaFileListPage {
|
||||
pub records: Vec<MediaFileListItem>,
|
||||
@@ -34,6 +36,10 @@ pub struct MediaFileListItem {
|
||||
pub maybe_public_bucket_prefix: Option<String>,
|
||||
pub maybe_public_bucket_extension: Option<String>,
|
||||
|
||||
pub creator_user_token: UserToken,
|
||||
pub creator_username: String,
|
||||
pub creator_display_name: String,
|
||||
pub creator_email_gravatar_hash: String,
|
||||
pub creator_set_visibility: Visibility,
|
||||
|
||||
pub created_at: DateTime<Utc>,
|
||||
@@ -42,13 +48,11 @@ pub struct MediaFileListItem {
|
||||
|
||||
#[derive(Clone, Copy)]
|
||||
pub enum ViewAs {
|
||||
Author,
|
||||
Moderator,
|
||||
AnotherUser,
|
||||
}
|
||||
|
||||
pub struct ListMediaFilesArgs<'a> {
|
||||
pub username: &'a str,
|
||||
pub limit: usize,
|
||||
pub maybe_filter_media_type: Option<MediaFileType>,
|
||||
pub maybe_offset: Option<usize>,
|
||||
@@ -61,7 +65,6 @@ pub async fn list_media_files(args: ListMediaFilesArgs<'_>) -> AnyhowResult<Medi
|
||||
|
||||
let mut query = query_builder(
|
||||
args.maybe_filter_media_type,
|
||||
args.username,
|
||||
args.limit,
|
||||
args.maybe_offset,
|
||||
args.cursor_is_reversed,
|
||||
@@ -71,7 +74,6 @@ pub async fn list_media_files(args: ListMediaFilesArgs<'_>) -> AnyhowResult<Medi
|
||||
let query = query.build_query_as::<MediaFileListItemInternal>();
|
||||
|
||||
let results = query.fetch_all(args.mysql_pool).await?;
|
||||
|
||||
let first_id = results.first()
|
||||
.map(|raw_result| raw_result.id);
|
||||
|
||||
@@ -81,7 +83,7 @@ pub async fn list_media_files(args: ListMediaFilesArgs<'_>) -> AnyhowResult<Medi
|
||||
let results = results.into_iter()
|
||||
.map(|record| {
|
||||
MediaFileListItem {
|
||||
token: record.token,
|
||||
token: MediaFileToken::new_from_str(&record.media_file_token),
|
||||
origin_category: record.origin_category,
|
||||
origin_product_category: record.origin_product_category,
|
||||
maybe_origin_model_type: record.maybe_origin_model_type,
|
||||
@@ -90,6 +92,10 @@ pub async fn list_media_files(args: ListMediaFilesArgs<'_>) -> AnyhowResult<Medi
|
||||
public_bucket_directory_hash: record.public_bucket_directory_hash,
|
||||
maybe_public_bucket_prefix: record.maybe_public_bucket_prefix,
|
||||
maybe_public_bucket_extension: record.maybe_public_bucket_extension,
|
||||
creator_user_token: UserToken::new_from_str(&record.creator_user_token),
|
||||
creator_username: record.creator_username,
|
||||
creator_display_name: record.creator_display_name,
|
||||
creator_email_gravatar_hash: record.creator_email_gravatar_hash,
|
||||
creator_set_visibility: record.creator_set_visibility,
|
||||
created_at: record.created_at,
|
||||
updated_at: record.updated_at,
|
||||
@@ -107,19 +113,18 @@ pub async fn list_media_files(args: ListMediaFilesArgs<'_>) -> AnyhowResult<Medi
|
||||
|
||||
fn query_builder<'a>(
|
||||
maybe_filter_media_type: Option<MediaFileType>,
|
||||
username: &'a str,
|
||||
limit: usize,
|
||||
maybe_offset: Option<usize>,
|
||||
cursor_is_reversed: bool,
|
||||
view_as: ViewAs,
|
||||
) -> QueryBuilder<'a, MySql> {
|
||||
|
||||
// NB: Query cannot be statically checked by sqlx
|
||||
let mut query_builder: QueryBuilder<MySql> = QueryBuilder::new(
|
||||
let mut query_string = String::from(
|
||||
r#"
|
||||
|
||||
SELECT
|
||||
m.id,
|
||||
m.token as `token: tokens::tokens::media_files::MediaFileToken`,
|
||||
m.token as media_file_token,
|
||||
|
||||
m.origin_category as `origin_category: enums::by_table::media_files::media_file_origin_category::MediaFileOriginCategory`,
|
||||
m.origin_product_category as `origin_product_category: enums::by_table::media_files::media_file_origin_product_category::MediaFileOriginProductCategory`,
|
||||
@@ -133,21 +138,25 @@ SELECT
|
||||
m.maybe_public_bucket_prefix,
|
||||
m.maybe_public_bucket_extension,
|
||||
|
||||
users.token as creator_user_token,
|
||||
users.username as creator_username,
|
||||
users.display_name as creator_display_name,
|
||||
users.email_gravatar_hash as creator_email_gravatar_hash,
|
||||
m.creator_set_visibility as `creator_set_visibility: enums::common::visibility::Visibility`,
|
||||
m.created_at,
|
||||
m.updated_at
|
||||
|
||||
FROM media_files AS m
|
||||
LEFT OUTER JOIN users AS u
|
||||
ON m.maybe_creator_user_token = u.token
|
||||
|
||||
WHERE u.username = ?
|
||||
LEFT OUTER JOIN users
|
||||
ON m.maybe_creator_user_token = users.token
|
||||
WHERE m.maybe_creator_user_token IS NOT NULL
|
||||
AND m.user_deleted_at IS NULL
|
||||
AND m.mod_deleted_at IS NULL
|
||||
"#
|
||||
);
|
||||
|
||||
query_builder.push_bind(username);
|
||||
// NB: Query cannot be statically checked by sqlx
|
||||
let mut query_builder: QueryBuilder<MySql> = QueryBuilder::new(query_string);
|
||||
|
||||
if let Some(media_type) = maybe_filter_media_type {
|
||||
query_builder.push(" AND m.media_file_type = ? ");
|
||||
@@ -155,7 +164,6 @@ AND m.mod_deleted_at IS NULL
|
||||
}
|
||||
|
||||
match view_as {
|
||||
ViewAs::Author => {}
|
||||
ViewAs::Moderator => {}
|
||||
ViewAs::AnotherUser => {
|
||||
query_builder.push(" AND m.creator_set_visibility = 'public' ");
|
||||
@@ -184,7 +192,7 @@ AND m.mod_deleted_at IS NULL
|
||||
#[derive(sqlx::FromRow)]
|
||||
struct MediaFileListItemInternal {
|
||||
id: i64,
|
||||
token: MediaFileToken,
|
||||
media_file_token: String,
|
||||
|
||||
origin_category: MediaFileOriginCategory,
|
||||
origin_product_category: MediaFileOriginProductCategory,
|
||||
@@ -196,6 +204,10 @@ struct MediaFileListItemInternal {
|
||||
maybe_public_bucket_prefix: Option<String>,
|
||||
maybe_public_bucket_extension: Option<String>,
|
||||
|
||||
creator_user_token: String,
|
||||
creator_username: String,
|
||||
creator_display_name: String,
|
||||
creator_email_gravatar_hash: String,
|
||||
creator_set_visibility: Visibility,
|
||||
|
||||
created_at: DateTime<Utc>,
|
||||
|
||||
@@ -167,3 +167,6 @@ diesel = { version = "2.1.3", features = ["mysql"] }
|
||||
|
||||
[dev-dependencies]
|
||||
speculoos = "0.9.0" # Fluent assertions
|
||||
|
||||
[profile.dev.package.sqlx-macros]
|
||||
opt-level = 3
|
||||
+96
@@ -0,0 +1,96 @@
|
||||
use tokens::tokens::media_files::MediaFileToken;
|
||||
|
||||
|
||||
/// Everything we need to refer to a media_file on the public web interface.
|
||||
#[derive(Clone, Serialize)]
|
||||
pub struct MediaFileSocialMetaLight {
|
||||
/// The token for the media_file
|
||||
pub media_file_token: MediaFileToken,
|
||||
|
||||
pub favorites_count: u64,
|
||||
pub comments_count: u64,
|
||||
pub signed_in_user_favorited: bool,
|
||||
}
|
||||
|
||||
impl MediaFileSocialMetaLight {
|
||||
|
||||
pub fn from_db_fields(
|
||||
media_file_token: &MediaFileToken,
|
||||
favorites_count: u64,
|
||||
comments_count: u64,
|
||||
signed_in_user_favorited: bool,
|
||||
) -> Self {
|
||||
Self {
|
||||
media_file_token: media_file_token.clone(),
|
||||
favorites_count,
|
||||
comments_count,
|
||||
signed_in_user_favorited,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn from_optional_db_fields(
|
||||
maybe_media_file_token: Option<&MediaFileToken>,
|
||||
favorites_count: u64,
|
||||
comments_count: u64,
|
||||
signed_in_user_favorited: bool,
|
||||
) -> Option<Self> {
|
||||
Self::from_optional_db_fields_owned(
|
||||
maybe_media_file_token.map(|u| u.clone()),
|
||||
favorites_count,
|
||||
comments_count,
|
||||
signed_in_user_favorited,
|
||||
)
|
||||
}
|
||||
|
||||
pub fn from_optional_db_fields_owned(
|
||||
maybe_media_file_token: Option<MediaFileToken>,
|
||||
favorites_count: u64,
|
||||
comments_count: u64,
|
||||
signed_in_user_favorited: bool,
|
||||
) -> Option<Self> {
|
||||
match (maybe_media_file_token) {
|
||||
(Some(media_file_token)) => {
|
||||
Some(Self {
|
||||
media_file_token,
|
||||
favorites_count,
|
||||
comments_count,
|
||||
signed_in_user_favorited,
|
||||
})
|
||||
}
|
||||
_ => {
|
||||
None
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Clone, Serialize)]
|
||||
pub struct DefaultAvatarInfo {
|
||||
pub image_index: u8,
|
||||
pub color_index: u8,
|
||||
}
|
||||
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use tokens::tokens::media_files::MediaFileToken;
|
||||
|
||||
use crate::http_server::common_responses::media_file_social_meta_lite::MediaFileSocialMetaLight;
|
||||
|
||||
#[test]
|
||||
fn test_from_optional_db_fields() {
|
||||
let media_file_token = MediaFileToken::new_from_str("token");
|
||||
let media_filename = "media_filename";
|
||||
let display_name = "display_name";
|
||||
let gravatar_hash= "adsf";
|
||||
|
||||
let media_file_details = MediaFileSocialMetaLight::from_optional_db_fields(
|
||||
Some(&media_file_token),
|
||||
0,0,false
|
||||
);
|
||||
|
||||
let media_file_details = media_file_details.expect("Should not be optional.");
|
||||
|
||||
assert_eq!(media_file_details.media_file_token, media_file_token);
|
||||
}
|
||||
}
|
||||
@@ -1,3 +1,4 @@
|
||||
pub mod pagination_cursors;
|
||||
pub mod pagination_page;
|
||||
pub mod user_details_lite;
|
||||
pub mod user_details_lite;
|
||||
pub mod media_file_social_meta_lite;
|
||||
+10
-8
@@ -12,7 +12,7 @@ use enums::by_table::media_files::media_file_origin_model_type::MediaFileOriginM
|
||||
use enums::by_table::media_files::media_file_origin_product_category::MediaFileOriginProductCategory;
|
||||
use enums::by_table::media_files::media_file_type::MediaFileType;
|
||||
use enums::common::visibility::Visibility;
|
||||
use mysql_queries::queries::media_files::list_media_files_for_user::{list_media_files_for_user, ListMediaFileForUserArgs, ViewAs};
|
||||
use mysql_queries::queries::media_files::list_media_files::{list_media_files, ListMediaFilesArgs, ViewAs};
|
||||
use tokens::tokens::media_files::MediaFileToken;
|
||||
|
||||
use crate::http_server::common_responses::pagination_cursors::PaginationCursors;
|
||||
@@ -50,7 +50,7 @@ pub struct MediaFileListItem {
|
||||
pub maybe_public_bucket_prefix: Option<String>,
|
||||
pub maybe_public_bucket_extension: Option<String>,
|
||||
|
||||
pub maybe_creator: Option<UserDetailsLight>,
|
||||
pub creator: UserDetailsLight,
|
||||
|
||||
pub creator_set_visibility: Visibility,
|
||||
|
||||
@@ -91,7 +91,6 @@ pub async fn list_media_files_handler(
|
||||
server_state: web::Data<Arc<ServerState>>
|
||||
) -> Result<HttpResponse, ErrorResponse>
|
||||
{
|
||||
info!("Fetching inference results for user: {}", &path.username);
|
||||
|
||||
let maybe_user_session = server_state
|
||||
.session_checker
|
||||
@@ -108,7 +107,6 @@ pub async fn list_media_files_handler(
|
||||
match maybe_user_session {
|
||||
None => {},
|
||||
Some(session) => {
|
||||
is_author = session.username == path.username;
|
||||
is_mod = session.can_ban_users;
|
||||
},
|
||||
};
|
||||
@@ -130,15 +128,13 @@ pub async fn list_media_files_handler(
|
||||
None
|
||||
};
|
||||
|
||||
let view_as = if is_author {
|
||||
ViewAs::Author
|
||||
} else if is_mod {
|
||||
let view_as = if is_mod {
|
||||
ViewAs::Moderator
|
||||
} else {
|
||||
ViewAs::AnotherUser
|
||||
};
|
||||
|
||||
let query_results = list_media_files_for_user(ListMediaFileForUserArgs {
|
||||
let query_results = list_media_files(ListMediaFilesArgs {
|
||||
limit,
|
||||
maybe_filter_media_type: query.filter_media_type,
|
||||
maybe_offset: cursor,
|
||||
@@ -189,6 +185,12 @@ pub async fn list_media_files_handler(
|
||||
public_bucket_directory_hash: record.public_bucket_directory_hash,
|
||||
maybe_public_bucket_prefix: record.maybe_public_bucket_prefix,
|
||||
maybe_public_bucket_extension: record.maybe_public_bucket_extension,
|
||||
creator: UserDetailsLight::from_db_fields(
|
||||
&record.creator_user_token,
|
||||
&record.creator_username,
|
||||
&record.creator_display_name,
|
||||
&record.creator_email_gravatar_hash,
|
||||
),
|
||||
creator_set_visibility: record.creator_set_visibility,
|
||||
created_at: record.created_at,
|
||||
updated_at: record.updated_at,
|
||||
|
||||
@@ -44,6 +44,7 @@ use crate::http_server::endpoints::investor_demo::enable_demo_mode_handler::enab
|
||||
use crate::http_server::endpoints::leaderboard::get_leaderboard::leaderboard_handler;
|
||||
use crate::http_server::endpoints::media_files::delete_media_file::delete_media_file_handler;
|
||||
use crate::http_server::endpoints::media_files::get_media_file::get_media_file_handler;
|
||||
use crate::http_server::endpoints::media_files::list_media_files::list_media_files_handler;
|
||||
use crate::http_server::endpoints::media_files::list_media_files_for_user::list_media_files_for_user_handler;
|
||||
use crate::http_server::endpoints::media_uploads::list_user_media_uploads_of_type::list_user_media_uploads_of_type_handler;
|
||||
use crate::http_server::endpoints::media_uploads::upload_audio::upload_audio_handler;
|
||||
@@ -1007,6 +1008,10 @@ fn add_media_file_routes<T, B> (app: App<T>) -> App<T>
|
||||
.route(web::delete().to(delete_media_file_handler))
|
||||
.route(web::head().to(|| HttpResponse::Ok()))
|
||||
)
|
||||
.service(web::resource("/feed")
|
||||
.route(web::get().to(list_media_files_handler))
|
||||
.route(web::head().to(|| HttpResponse::Ok()))
|
||||
)
|
||||
.service(web::resource("/user/{username}")
|
||||
.route(web::get().to(list_media_files_for_user_handler))
|
||||
.route(web::head().to(|| HttpResponse::Ok()))
|
||||
|
||||
Reference in new issue
Block a user