rtic/macros/src/check.rs

use std::collections::HashMap;

use quote::Tokens;
use rtfm_syntax::{Idents, Idle, Init, Statics};
use syn::Ident;

use error::*;

pub struct App {
    pub device: Tokens,
    pub idle: Idle,
    pub init: Init,
    pub resources: Statics,
    pub tasks: Tasks,
}

pub type Tasks = HashMap<Ident, Task>;

pub struct Task {
    pub enabled: Option<bool>,
    pub priority: u8,
    pub resources: Idents,
}

pub fn app(app: ::rtfm_syntax::App) -> Result<App> {
    let mut tasks = HashMap::new();

    for (k, v) in app.tasks {
        let name = k.clone();
        tasks.insert(
            k,
            ::check::task(v)
                .chain_err(|| format!("checking task `{}`", name))?,
        );
    }

    let app = App {
        device: app.device,
        idle: app.idle,
        init: app.init,
        resources: app.resources,
        tasks,
    };

    ::check::resources(&app)?;

    Ok(app)
}

fn resources(app: &App) -> Result<()> {
    for resource in app.resources.keys() {
        if app.idle.resources.contains(resource) {
            continue;
        }

        if app.tasks
            .values()
            .any(|task| task.resources.contains(resource))
        {
            continue;
        }

        bail!("resource `{}` is unused", resource);
    }

    Ok(())
}

fn task(task: ::rtfm_syntax::Task) -> Result<Task> {
    if let Some(priority) = task.priority {
        Ok(Task {
            enabled: task.enabled,
            priority,
            resources: task.resources,
        })
    } else {
        bail!("should contain a `priority` field")
    }
}