rust_anticaptcha/instruments/
image_instrument.rs

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
108
109
110
//! # ImageInstrument module
//!
//! Module contains struct and functions to read and
//! encode file from local system or web to base64 string
//!
//! ## Basic example with image file reading and encode in base64 string
//!
//! ```
//! use rust_anticaptcha::instruments::image_instrument::ImageInstrument;
//!
//! let image_instrument = ImageInstrument::new();
//! let base64_str = image_instrument.read_image_file("files/captcha-image.jpg".to_string());
//! ```

use std::fs;
use std::io::ErrorKind;

use base64::alphabet::STANDARD;
use base64::engine::general_purpose;
use base64::engine::Engine;
use base64::engine::GeneralPurpose;

use crate::core::request_interface::RequestInterface;

/// Struct help in encoding image from file/url to base64 string
///
/// # Examples
///
/// ```
/// use rust_anticaptcha::instruments::image_instrument::ImageInstrument;
///
/// let image_instrument = ImageInstrument::new();
/// ```
pub struct ImageInstrument {
    engine: GeneralPurpose,
}
impl ImageInstrument {
    /// Method init new ImageInstrument struct with base64 engine
    ///
    /// # Examples
    ///
    /// ```
    /// use rust_anticaptcha::instruments::image_instrument::ImageInstrument;
    ///
    /// let image_instrument = ImageInstrument::new();
    /// ```
    ///
    /// # Returns
    /// Method return `ImageInstrument` instance
    ///
    pub fn new() -> Self {
        ImageInstrument {
            engine: GeneralPurpose::new(&STANDARD, general_purpose::PAD),
        }
    }

    /// Method read image file and return base64 string
    ///
    /// # Arguments
    /// `file_path` - path to image on local system
    ///
    /// # Examples
    ///
    /// ```
    /// use rust_anticaptcha::instruments::image_instrument::ImageInstrument;
    ///
    /// let image_instrument = ImageInstrument::new();
    /// let base64_str = image_instrument.read_image_file("files/captcha-image.jpg".to_string());
    /// ```
    ///
    /// # Returns
    /// Method return image as base64 string
    ///
    pub fn read_image_file(&self, file_path: String) -> String {
        let contents = match fs::read(&file_path) {
            Ok(content) => content,
            Err(error) => match error.kind() {
                ErrorKind::NotFound => panic!("File `{file_path}` is not exists!"),
                _ => panic!("Can't read file - {error}"),
            },
        };
        self.engine.encode(contents)
    }

    /// Method read image file from link and return base64 string
    ///
    /// # Arguments
    /// `file_url` - image URL
    ///
    /// # Examples
    ///
    /// ```
    /// use rust_anticaptcha::instruments::image_instrument::ImageInstrument;
    ///
    /// let image_instrument = ImageInstrument::new();
    /// let base64_str = image_instrument.read_image_link("https://some-file-url.jpg".to_string());
    /// ```
    ///
    /// # Returns
    /// Method return image as base64 string
    ///
    pub async fn read_image_link(&self, file_url: String) -> String {
        let request_client = RequestInterface::new();
        let result = match request_client.send_get_request(&file_url).await {
            Ok(result) => result,
            Err(error) => panic!("{}", error),
        };
        self.engine.encode(result.bytes().await.unwrap())
    }
}