vetis-compio 0.1.1

Very Tiny Server with compio runtime
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
111
112
113

//! Path module for handling different types of paths in the server
use std::sync::Arc;
use vetis::{
    errors::{HandlerError, HostError, VetisError},
    host::path::Path,
    HandlerFn, Request, Response, VetisFutureResult,
};

/// Builder for handler path
pub struct HandlerPathBuilder {
    uri: Arc<String>,
    handler: Option<HandlerFn>,
}

impl HandlerPathBuilder {
    /// Allow set handler uri path
    ///
    /// # Arguments
    ///
    /// * `uri` - The uri of the handler path
    ///
    /// # Returns
    ///
    /// * `Self` - The builder
    pub fn uri(mut self, uri: &str) -> Self {
        self.uri = Arc::from(uri.to_string());
        self
    }

    /// Allow set handler function
    ///
    /// # Arguments
    ///
    /// * `handler` - The handler function
    ///
    /// # Returns
    ///
    /// * `Self` - The builder
    pub fn handler(mut self, handler: HandlerFn) -> Self {
        self.handler = Some(handler);
        self
    }

    /// Build the handler path
    ///
    /// # Returns
    ///
    /// * `Result<HostPath, VetisError>` - The handler path or error
    pub fn build(self) -> Result<HandlerPath, VetisError> {
        if self.uri.is_empty() {
            return Err(VetisError::Host(HostError::Handler(HandlerError::Uri(
                "URI cannot be empty".to_string(),
            ))));
        }

        let handler = match self.handler {
            Some(handler) => handler,
            None => {
                return Err(VetisError::Host(HostError::Handler(HandlerError::Handler(
                    "Handler must be set".to_string(),
                ))))
            }
        };

        Ok(HandlerPath { uri: self.uri, handler })
    }
}

/// Handler path
pub struct HandlerPath {
    uri: Arc<String>,
    handler: HandlerFn,
}

impl HandlerPath {
    /// Allow create a new handler path builder
    ///
    /// # Returns
    ///
    /// * `HandlerPathBuilder` - The builder
    pub fn builder() -> HandlerPathBuilder {
        HandlerPathBuilder { uri: Arc::from("/".to_string()), handler: None }
    }
}

impl Path for HandlerPath {
    /// Allow get handler uri path
    ///
    /// # Returns
    ///
    /// * `&str` - The uri of the handler path
    fn uri(&self) -> &str {
        self.uri.as_ref()
    }

    /// Handles the request for the path
    ///
    /// # Arguments
    ///
    /// * `request` - The request to handle
    /// * `uri` - The URI of the path
    ///
    /// # Returns
    ///
    /// * `VetisFutureResult<'a, Response>` - The future that will handle the request
    fn handle<'a>(
        &'a self,
        request: Request,
        _uri: Arc<String>,
    ) -> VetisFutureResult<'a, Response> {
        (self.handler)(request)
    }
}