sunox 0.0.13

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

use std::future::Future;

use tokio::time::{Instant, timeout_at};

use super::CliError;

pub(crate) fn ensure_poll_interval(interval: std::time::Duration) -> Result<(), CliError> {
    if interval.is_zero() {
        return Err(CliError::Config(
            "poll interval must be greater than 0 seconds".into(),
        ));
    }
    Ok(())
}

pub(crate) fn ensure_poll_timeout(timeout: std::time::Duration) -> Result<(), CliError> {
    checked_deadline(timeout).map(|_| ())
}

pub(crate) fn deadline_after(timeout: std::time::Duration) -> Result<Instant, CliError> {
    checked_deadline(timeout)
}

fn checked_deadline(timeout: std::time::Duration) -> Result<Instant, CliError> {
    if timeout.is_zero() {
        return Err(CliError::Config(
            "poll timeout must be greater than 0 seconds".into(),
        ));
    }
    Instant::now()
        .checked_add(timeout)
        .ok_or_else(|| CliError::Config("poll timeout is too large".into()))
}

pub(crate) async fn run_before_deadline<T, F>(
    deadline: Instant,
    future: F,
    timeout_error: CliError,
) -> Result<T, CliError>
where
    F: Future<Output = Result<T, CliError>>,
{
    match timeout_at(deadline, future).await {
        Ok(result) => result,
        Err(_) => Err(timeout_error),
    }
}

pub(crate) async fn sleep_before_deadline(
    deadline: Instant,
    interval: std::time::Duration,
) -> bool {
    let remaining = deadline.saturating_duration_since(Instant::now());
    if remaining.is_zero() {
        return false;
    }
    tokio::time::sleep(interval.min(remaining)).await;
    Instant::now() < deadline
}