esetres 0.1.1

A self hosted file storage server.
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
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
114
115
116
117
118
119
120
121
122
123
124
125

use std::fmt::Display;

use super::connect;
use bcrypt::hash;
use clap::ValueEnum;
use rusqlite::types::{FromSql, FromSqlError};
use tokio::task;

#[derive(Debug, Clone)]
pub struct Token {
    pub id: i32,
    pub name: String,
    pub bucket_scope: String,
    pub access: Access,
    pub token: String,
}

#[derive(Clone, Debug, PartialEq, ValueEnum)]
pub enum Access {
    READ,
    WRITE,
    FULL,
}

impl FromSql for Access {
    fn column_result(value: rusqlite::types::ValueRef<'_>) -> rusqlite::types::FromSqlResult<Self> {
        let str = value.as_str()?;

        match str {
            "full" => Ok(Access::FULL),
            "write" => Ok(Access::WRITE),
            "read" => Ok(Access::READ),
            &_ => Err(FromSqlError::InvalidType),
        }
    }
}

impl Access {
    fn to_string(&self) -> String {
        match self {
            Access::READ => "read".to_string(),
            Access::WRITE => "write".to_string(),
            Access::FULL => "full".to_string(),
        }
    }
}

impl Display for Access {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Access::READ => write!(f, "read"),
            Access::WRITE => write!(f, "write"),
            Access::FULL => write!(f, "full"),
        }
    }
}

impl Token {
    pub async fn get_all() -> Result<Vec<Token>, Box<dyn std::error::Error>> {
        let tokens = task::spawn_blocking(move || -> rusqlite::Result<Vec<Token>> {
            let conn = connect()?;

            let mut stmt = conn.prepare(
                "SELECT id, name, bucket_scope, access, token FROM tokens ORDER BY Id DESC",
            )?;
            let tokens_iter = stmt.query_map([], |row| {
                Ok(Token {
                    id: row.get(0)?,
                    name: row.get(1)?,
                    bucket_scope: row.get(2)?,
                    access: row.get(3)?,
                    token: row.get(4)?,
                })
            })?;

            let mut tokens: Vec<Token> = vec![];

            for token in tokens_iter {
                if let Ok(t) = token {
                    tokens.push(t);
                }
            }

            Ok(tokens)
        })
        .await??;

        Ok(tokens)
    }

    pub async fn create(self) -> Result<(), Box<dyn std::error::Error>> {
        let token_hash = hash(&self.token, 6)?;
        task::spawn_blocking(move || -> rusqlite::Result<()> {
            let conn = connect()?;

            conn.execute(
                "INSERT INTO tokens (name, token, access, bucket_scope) VALUES (?1, ?2, ?3, ?4)",
                [
                    &self.name,
                    &token_hash,
                    &self.access.to_string(),
                    &self.bucket_scope,
                ],
            )?;

            Ok(())
        })
        .await??;

        Ok(())
    }

    pub async fn delete(name: String) -> Result<(), Box<dyn std::error::Error>> {
        task::spawn_blocking(move || -> rusqlite::Result<()> {
            let conn = connect()?;

            conn.execute("DELETE FROM tokens WHERE name = ?1", [name])?;

            Ok(())
        })
        .await??;

        Ok(())
    }
}