Load env in all cargo tasks

[rbdr/lyricli] / src / sources / mod.rs

use std::io::{Error, ErrorKind::Other, Result};

#[cfg(target_os = "macos")]
mod apple_music;
#[cfg(target_os = "macos")]
mod spotify;

#[cfg(target_os = "linux")]
mod dbus;

#[cfg(target_os = "macos")]
use apple_music::AppleMusic;
#[cfg(target_os = "macos")]
use spotify::Spotify;

#[cfg(target_os = "linux")]
use dbus::Dbus;

use crate::Track;

pub trait LyricsSource {
    fn name(&self) -> String;

    fn current_track(&self) -> Option<Track>;

    fn disable(&self) -> Result<()>;
    fn enable(&self) -> Result<()>;
    fn reset(&self) -> Result<()>;
}

pub fn list() -> Vec<String> {
    available_sources()
        .into_iter()
        .map(|source| source.name())
        .collect()
}

pub fn enable(source_name: &String) -> Result<()> {
    let sources = available_sources();
    for source in sources {
        if &source.name() == source_name {
            return source.enable();
        }
    }
    Err(Error::new(Other, "No such source was available."))
}

pub fn disable(source_name: &String) -> Result<()> {
    let sources = available_sources();
    for source in sources {
        if &source.name() == source_name {
            return source.disable();
        }
    }
    Err(Error::new(Other, "No such source was available."))
}

pub fn reset(source_name: &String) -> Result<()> {
    let sources = available_sources();
    for source in sources {
        if &source.name() == source_name {
            return source.reset();
        }
    }
    Err(Error::new(Other, "No such source was available."))
}

pub fn get_track() -> Option<Track> {
    let sources = available_sources();
    for source in sources {
        if let Some(track) = source.current_track() {
            return Some(track);
        }
    }
    return None;
}

pub fn available_sources() -> Vec<Box<dyn LyricsSource>> {
    let mut sources: Vec<Box<dyn LyricsSource>> = Vec::new();
    #[cfg(target_os = "macos")]
    {
        sources.push(Box::new(AppleMusic::new()));
        sources.push(Box::new(Spotify::new()));
    }

    #[cfg(not(target_os = "macos"))]
    {
        sources.push(Box::new(Dbus::new()));
    }

    sources
}