summaryrefslogtreecommitdiff
path: root/ofborg/tickborg/src/commitstatus.rs
blob: 6747f3b048a6cc9a42ca7fe5dd65aac1de620d0e (plain)
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
use futures_util::future::TryFutureExt;
use tracing::warn;

pub struct CommitStatus {
    api: hubcaps::statuses::Statuses,
    sha: String,
    context: String,
    description: String,
    url: String,
}

impl CommitStatus {
    pub fn new(
        api: hubcaps::statuses::Statuses,
        sha: String,
        context: String,
        description: String,
        url: Option<String>,
    ) -> CommitStatus {
        let mut stat = CommitStatus {
            api,
            sha,
            context,
            description,
            url: "".to_owned(),
        };

        stat.set_url(url);

        stat
    }

    pub fn set_url(&mut self, url: Option<String>) {
        self.url = url.unwrap_or_else(|| String::from(""))
    }

    pub async fn set_with_description(
        &mut self,
        description: &str,
        state: hubcaps::statuses::State,
    ) -> Result<(), CommitStatusError> {
        self.set_description(description.to_owned());
        self.set(state).await
    }

    pub fn set_description(&mut self, description: String) {
        self.description = description;
    }

    pub async fn set(&self, state: hubcaps::statuses::State) -> Result<(), CommitStatusError> {
        let desc = if self.description.len() >= 140 {
            warn!(
                "description is over 140 char; truncating: {:?}",
                &self.description
            );
            self.description.chars().take(140).collect()
        } else {
            self.description.clone()
        };
        self.api
            .create(
                self.sha.as_ref(),
                &hubcaps::statuses::StatusOptions::builder(state)
                    .context(self.context.clone())
                    .description(desc)
                    .target_url(self.url.clone())
                    .build(),
            )
            .map_ok(|_| ())
            .map_err(|e| CommitStatusError::from(e))
            .await?;
        Ok(())
    }
}

#[derive(Debug)]
pub enum CommitStatusError {
    ExpiredCreds(hubcaps::Error),
    MissingSha(hubcaps::Error),
    Error(hubcaps::Error),
    InternalError(String),
}

impl From<hubcaps::Error> for CommitStatusError {
    fn from(e: hubcaps::Error) -> CommitStatusError {
        use http::status::StatusCode;
        use hubcaps::Error;
        match &e {
            Error::Fault { code, error }
                if code == &StatusCode::UNAUTHORIZED && error.message == "Bad credentials" =>
            {
                CommitStatusError::ExpiredCreds(e)
            }
            Error::Fault { code, error }
                if code == &StatusCode::UNPROCESSABLE_ENTITY
                    && error.message.starts_with("No commit found for SHA:") =>
            {
                CommitStatusError::MissingSha(e)
            }
            _otherwise => CommitStatusError::Error(e),
        }
    }
}