sunox 0.0.14

Generate AI music from your terminal via direct Suno web workflows
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

//! Suno web endpoint client, endpoint methods, auth retry, and response mapping.

use std::time::Duration;

pub mod billing;
pub mod challenge;
pub mod clip_info;
pub mod concat;
pub mod cover;
pub mod delete;
pub mod download;
pub mod edit;
pub mod extend;
pub mod feed;
pub mod generate;
pub mod inspiration;
pub mod lyrics;
pub mod metadata;
pub mod persona;
pub mod playlist;
pub mod prompts;
pub mod remaster;
pub mod speed;
pub mod stems;
pub mod types;
pub mod upload;

mod auth_retry;
mod client;
mod headers;
mod response;

#[cfg(test)]
mod endpoint_tests;

pub use client::SunoClient;

#[derive(Clone, Copy)]
pub struct PollingOptions {
    pub timeout: Duration,
    pub interval: Duration,
}

impl PollingOptions {
    pub(crate) fn validate(self) -> Result<(), crate::core::CliError> {
        crate::core::ensure_poll_interval(self.interval)?;
        crate::core::ensure_poll_timeout(self.timeout)
    }

    pub(crate) fn deadline(self) -> Result<tokio::time::Instant, crate::core::CliError> {
        self.validate()?;
        crate::core::deadline_after(self.timeout)
    }
}

#[cfg(test)]
mod polling_tests {
    use std::time::Duration;

    use super::PollingOptions;
    use crate::core::CliError;

    #[test]
    fn polling_options_reject_a_zero_interval() {
        let error = PollingOptions {
            timeout: Duration::from_secs(1),
            interval: Duration::ZERO,
        }
        .deadline()
        .expect_err("zero polling interval must be rejected");

        assert!(
            matches!(error, CliError::Config(message) if message.contains("poll interval") && message.contains("greater than 0"))
        );
    }
}