accel-derive 0.3.0

Procedual macro for writing GPGPU kernel
Documentation 
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

use failure::*;
use maplit::hashmap;
use quote::ToTokens;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;

#[derive(Debug, Serialize)]
pub struct MetaData {
    package: HashMap<&'static str, String>,
    lib: HashMap<&'static str, Vec<&'static str>>,
    dependencies: HashMap<String, Depenency>,
}

impl MetaData {
    fn new(name: &str) -> Self {
        MetaData {
            package: hashmap! { "version" => "0.0.0".into(), "name" => name.into(), "edition" => "2018".into() },
            lib: hashmap! { "crate-type" => vec![ "cdylib" ] },
            dependencies: HashMap::new(),
        }
    }

    pub fn name(&self) -> &str {
        &self.package["name"]
    }

    pub fn from_token(func: &syn::ItemFn) -> Fallible<Self> {
        let attrs = &func.attrs;
        let mut kernel_attrs = MetaData::new(&func.sig.ident.to_string());
        for attr in attrs {
            let path = attr.path.to_token_stream().to_string();
            match path.as_ref() {
                "dependencies" => {
                    let dep = parse_dependency(
                        attr.tokens
                            .to_string()
                            .trim_start_matches('(')
                            .trim_end_matches(')'),
                    )?;
                    for (key, val) in dep {
                        kernel_attrs.dependencies.insert(key, val);
                    }
                }
                "name" => {
                    let token = attr.tokens.to_string();
                    let name = token.trim_start_matches('(').trim_end_matches(')').trim();
                    kernel_attrs.package.insert("name", name.into());
                }
                _ => {
                    bail!("Unsupported attribute: {}", path);
                }
            }
        }
        kernel_attrs
            .dependencies
            .entry("accel-core".into())
            .or_insert_with(|| Depenency::Version("0.3.0-alpha.4".into()));
        Ok(kernel_attrs)
    }
}

// Should I use `cargo::core::dependency::Depenency`?
// https://docs.rs/cargo/0.41.0/cargo/core/dependency/struct.Dependency.html
#[derive(Debug, PartialEq, Serialize, Deserialize)]
#[serde(untagged, deny_unknown_fields)]
enum Depenency {
    Version(String),
    VersionTable {
        version: String,
        #[serde(default)]
        features: Vec<String>,
    },
    Git {
        git: String,
        branch: Option<String>,
        tag: Option<String>,
        hash: Option<String>,
        #[serde(default)]
        features: Vec<String>,
    },
    Path {
        path: String,
        #[serde(default)]
        features: Vec<String>,
    },
}

fn parse_dependency(dep: &str) -> Fallible<HashMap<String, Depenency>> {
    Ok(toml::from_str(&dep.replace("\n", ""))?)
}

#[cfg(test)]
mod tests {
    #[test]
    fn parse_dependency() {
        let map = super::parse_dependency(r#"accel-core = "0.1.1""#).unwrap();
        dbg!(map);
        let map = super::parse_dependency(r#"accel-core = { version = "0.1.1" }"#).unwrap();
        dbg!(map);

        let map = super::parse_dependency(
            r#"accel-core = { git = "https://github.com/rust-accel/accel" }"#,
        )
        .unwrap();
        dbg!(map);

        let map = super::parse_dependency(
            r#"accel-core = { git = "https://github.com/rust-accel/accel", branch = "master" }"#,
        )
        .unwrap();
        dbg!(map);

        // `git` is lacked
        assert!(super::parse_dependency(r#"accel-core = { branch = "master" }"#,).is_err());

        // Unsupported tag
        assert!(super::parse_dependency(
            r#"accel-core = { git = "https://github.com/rust-accel/accel", homhom = "master" }"#,
        )
        .is_err());
    }
}