anstyle-progress 0.1.0

/// Determines whether the terminal supports ANSI OSC 9;4.
pub fn supports_term_progress(is_terminal: bool) -> bool {
    if !is_terminal {
        return false;
    }

    // Terminal feature detection, includes
    // - iTerm support in v3.6.6
    let term_features = std::env::var("TERM_FEATURES")
        .ok()
        .map(|v| term_features_has_progress(&v))
        .unwrap_or(false);
    if term_features {
        return true;
    }

    // https://github.com/wezterm/wezterm/issues/6581
    // https://ghostty.org/docs/install/release-notes/1-2-0#graphical-progress-bars
    let term_program = std::env::var("TERM_PROGRAM").ok();
    if matches!(term_program.as_deref(), Some("WezTerm") | Some("ghostty")) {
        return true;
    }

    // https://github.com/microsoft/terminal/pull/8055
    if std::env::var("WT_SESSION").is_ok() {
        return true;
    }

    // https://conemu.github.io/en/AnsiEscapeCodes.html#ConEmu_specific_OSC
    if std::env::var("ConEmuANSI").ok() == Some("ON".into()) {
        return true;
    }

    // Ptyxis added OSC 9;4 support in 48.0.
    // See https://gitlab.gnome.org/chergert/ptyxis/-/issues/305
    let ptyxis = std::env::var("PTYXIS_VERSION")
        .ok()
        .and_then(|version| version.split(".").next()?.parse::<i32>().ok())
        .map(|major_version| major_version >= 48)
        .unwrap_or(false);
    if ptyxis {
        return true;
    }

    false
}

// For iTerm, the TERM_FEATURES value "P" indicates OSC 9;4 support.
//
// Context: https://iterm2.com/feature-reporting/
fn term_features_has_progress(value: &str) -> bool {
    let mut current = String::new();

    for ch in value.chars() {
        if !ch.is_ascii_alphanumeric() {
            break;
        }
        if ch.is_ascii_uppercase() {
            if current == "P" {
                return true;
            }
            current.clear();
            current.push(ch);
        } else {
            current.push(ch);
        }
    }
    current == "P"
}

#[cfg(test)]
mod tests {
    use super::term_features_has_progress;

    #[test]
    fn term_features_progress_detection() {
        // With PROGRESS feature ("P")
        assert!(term_features_has_progress("MBT2ScP"));

        // Without PROGRESS feature
        assert!(!term_features_has_progress("MBT2Sc"));
    }
}