gem-rs 0.1.1

A Rust library that serves as a wrapper around the Gemini API, providing support for streaming
Documentation 
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

//! Utility functions for the Gem-rs library.
//!
//! This module contains various utility functions used internally by the Gem-rs library,
//! primarily for handling MIME types of different file formats.

use std::path::Path;

/// Determines the MIME type of a file based on its extension.
///
/// This function takes a file path and attempts to determine its MIME type
/// by examining the file extension. It supports various file formats including
/// documents, images, audio, and video files.
///
/// # Arguments
///
/// * `file_path` - A reference to a `Path` representing the file path.
///
/// # Returns
///
/// An `Option<String>` containing the MIME type if recognized, or `None` if the
/// file type is not supported or recognized.
///
/// # Examples
///
/// ```
/// use std::path::Path;
/// use gem_rs::utils::get_mime_type;
///
/// let path = Path::new("example.pdf");
/// assert_eq!(get_mime_type(path), Some("application/pdf".to_string()));
///
/// let path = Path::new("image.png");
/// assert_eq!(get_mime_type(path), Some("image/png".to_string()));
///
/// let path = Path::new("unknown.xyz");
/// assert_eq!(get_mime_type(path), None);
/// ```
///
/// # Notes
///
/// - Currently, all document types are treated as PDF. This may change in future updates.
/// - GIF files are currently not supported, as there's a TODO to implement GIF to video conversion.
/// - All text types (including code files) are currently treated as plain text.
pub(crate) fn get_mime_type(file_path: &Path) -> Option<String> {
    match file_path.extension().and_then(|ext| ext.to_str()) {
        //TODO: Convert all document types to PDF
        Some("pdf") => Some("application/pdf".to_string()),

        Some("png") => Some("image/png".to_string()),
        Some("jpg") | Some("jpeg") => Some("image/jpeg".to_string()),

        Some("mp3") => Some("audio/mpeg".to_string()),
        Some("wav") => Some("audio/wav".to_string()),

        Some("mp4") => Some("video/mp4".to_string()),
        Some("mov") => Some("video/quicktime".to_string()),
        Some("mpeg") => Some("video/mpeg".to_string()),
        Some("mpg") => Some("video/mpeg".to_string()),
        Some("avi") => Some("video/x-msvideo".to_string()),
        Some("wmv") => Some("video/x-ms-wmv".to_string()),
        Some("mpegps") => Some("video/mpeg".to_string()),
        Some("flv") => Some("video/x-flv".to_string()),
        // Some("gif") => Some("image/gif".to_string()), TODO: implement gif to video conversion

        //TODO: Convert all text types (including codes) to plain text
        Some("txt") => Some("text/plain".to_string()),
        _ => None,
    }
}