1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
//! media item stuff
use {
crate::serve::media::{metadata::PostMetadata, types::MediaType},
serde::{Deserialize, Serialize},
};
/// a media item
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct MediaItem {
/// file path relative to media root
path: String,
/// display name of the item
name: String,
/// type of media (image, video, etc)
#[serde(rename = "media_type")]
media_type: MediaType,
/// file size in bytes
size: u64,
/// optional post metadata
#[serde(skip_serializing_if = "Option::is_none")]
metadata: Option<PostMetadata>,
}
impl MediaItem {
/// make a new media item
pub fn new(path: String, name: String, media_type: MediaType, size: u64) -> Self {
Self {
path,
name,
media_type,
size,
metadata: None,
}
}
/// set the post metadata
pub fn with_metadata(mut self, metadata: PostMetadata) -> Self {
self.metadata = Some(metadata);
self
}
/// get the file path
pub fn path(&self) -> &str {
&self.path
}
/// get the display name
pub fn name(&self) -> &str {
&self.name
}
/// get the media type
pub fn media_type(&self) -> &MediaType {
&self.media_type
}
/// get the file size in bytes
pub fn size(&self) -> u64 {
self.size
}
/// get the post metadata if present
pub fn metadata(&self) -> Option<&PostMetadata> {
self.metadata.as_ref()
}
/// check if this item matches a search query
pub fn matches_search(&self, query: &str) -> bool {
let query_lower = query.to_lowercase();
if self.name.to_lowercase().contains(&query_lower) {
return true;
}
if let Some(meta) = &self.metadata {
if meta.id.to_string().contains(&query_lower) {
return true;
}
if meta
.artists
.iter()
.any(|a| a.to_lowercase().contains(&query_lower))
{
return true;
}
if meta
.tags
.iter()
.any(|t| t.to_lowercase().contains(&query_lower))
|| meta
.character_tags
.iter()
.any(|t| t.to_lowercase().contains(&query_lower))
|| meta
.species_tags
.iter()
.any(|t| t.to_lowercase().contains(&query_lower))
{
return true;
}
}
false
}
}