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

use reqwest::{
    header::{HeaderMap, HeaderValue},
    Client,
};
use url::Url;

pub mod build;
pub mod config;
pub mod model;

use build::builder::Builder;
use config::SupabaseConfig;

/// A struct representing a Storage with an associated client and headers.
pub struct Storage {
    url: url::Url,
    headers: HeaderMap,
    client: Client,
}

impl Storage {
    /// Creates a new `Storage` instance with the provided URL.
    ///
    /// # Arguments
    ///
    /// * `url` - The base URL for the storage.
    ///
    /// # Example
    ///
    /// ```
    /// use supabase_storage::Storage;
    ///
    /// let _ = Storage::new("https://your_project_path/storage/v1");
    /// ```
    pub fn new<T>(url: T) -> Self
    where
        T: Into<String>,
    {
        Self {
            url: Url::parse(&url.into()).unwrap(),
            headers: HeaderMap::new(),
            client: Client::new(),
        }
    }

    /// Creates a new `Storage` instance with provided configuration.
    ///
    /// # Arguments
    ///
    /// * `config` - The `SupabaseConfig` containing the necessary configuration for Supabase.
    ///
    /// # Example
    /// ```
    /// use supabase_storage::{Storage, config::SupabaseConfig};
    /// use dotenv::dotenv;
    ///
    /// dotenv().ok();
    /// let config = SupabaseConfig::default();  // load values from .env file using the 'envy' crate
    /// let storage = Storage::new_with_config(config);
    /// ```
    pub fn new_with_config(config: SupabaseConfig) -> Self {
        let mut headers = HeaderMap::new();
        if let Some(api_key) = config.supabase_api_key {
            headers.insert(
                "Authorization",
                HeaderValue::from_str(&format!("Bearer {}", api_key))
                    .expect("header value is invalid"),
            );
            headers.insert(
                "apiKey",
                HeaderValue::from_str(&format!("{}", api_key)).expect("header value is invalid"),
            );
        }

        Self {
            url: Url::parse(&config.supabase_url_storage).unwrap(),
            headers,
            client: Client::new(),
        }
    }

    /// Creates a new `Builder` instance to build an action
    ///
    /// # Example
    /// ```
    /// use supabase_storage::Storage;
    ///
    /// let storage = Storage::new("https://your_project_path/storage/v1");
    /// let builder = storage.from();
    /// ```
    pub fn from(&self) -> Builder {
        Builder::new(self.url.clone(), self.headers.clone(), self.client.clone())
    }
}