pastor 0.14.0

The pastebin that hopefully doesn't suck
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

use std::{
    path::{Path, PathBuf},
    time::Duration,
};

use tracing::{debug, error};

use crate::file::{Paste, PasteError};

pub async fn cleanup_routine(storage: PathBuf) {
    let mut ticker = tokio::time::interval(Duration::from_secs(60));
    loop {
        ticker.tick().await;
        if let Err(err) = cleanup(&storage).await {
            error!("failure during cleanup: {err}");
        }
    }
}

pub async fn cleanup(storage: &Path) -> Result<(), PasteError> {
    for entry in std::fs::read_dir(storage)? {
        let path = entry?.path();
        match Paste::load_from_path(&path, None).await {
            Ok((paste, _)) => {
                if !paste.expired()? {
                    return Ok(());
                }

                debug!("Deleting: {paste:?}. (Now: {})", chrono::Utc::now().timestamp());
            }
            Err(err) => {
                error!("got error {err} while cleanup pastes, deleting path {path:?} anyways")
            }
        };

        tokio::fs::remove_file(path).await?;
    }

    Ok(())
}

#[cfg(test)]
mod tests {
    use std::path::Path;

    use tempfile::tempdir;
    use tracing_test::traced_test;

    use crate::{file::Paste, id::PasteId};

    async fn write_paste(storage: &Path, expires: i64) -> Paste {
        let now = chrono::Utc::now().timestamp();
        let paste = Paste {
            id: PasteId::new(),
            created: now,
            expires: now + expires,
            token: "secrettoken".to_string(),
            mime: "text/none".to_string(),
        };

        paste.write(storage).await.unwrap();

        paste
    }

    #[tokio::test]
    #[traced_test]
    async fn cleanup() {
        let dir = tempdir().unwrap();
        let storage = dir.path();
        let paste1 = write_paste(storage, -10).await;
        let paste2 = write_paste(storage, 60).await;

        crate::cleanup::cleanup(storage).await.unwrap();
        assert!(!paste1.path(storage).exists());
        assert!(paste2.path(storage).exists());
    }

    #[tokio::test]
    #[traced_test]
    async fn cleanup_broken_file() {
        let dir = tempdir().unwrap();
        let storage = dir.path();

        tokio::fs::write(storage.join("testfile"), "").await.unwrap();
        crate::cleanup::cleanup(storage).await.unwrap();
        assert!(!storage.join("testfile").exists());
    }
}