Skip to content

Commit

Permalink
Merge pull request #530 from serde-rs/derive
Browse files Browse the repository at this point in the history
Macros 1.1
  • Loading branch information
dtolnay authored Sep 2, 2016
2 parents 4bb9279 + ac1128a commit ffa2f80
Show file tree
Hide file tree
Showing 4 changed files with 103 additions and 35 deletions.
86 changes: 51 additions & 35 deletions serde_codegen/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -36,51 +36,67 @@ include!(concat!(env!("OUT_DIR"), "/lib.rs"));
include!("lib.rs.in");

#[cfg(feature = "with-syntex")]
pub fn expand<S, D>(src: S, dst: D) -> Result<(), syntex::Error>
where S: AsRef<Path>,
D: AsRef<Path>,
{
let src = src.as_ref().to_owned();
let dst = dst.as_ref().to_owned();

let expand_thread = move || {
use syntax::{ast, fold};

/// Strip the serde attributes from the crate.
#[cfg(feature = "with-syntex")]
fn strip_attributes(krate: ast::Crate) -> ast::Crate {
/// Helper folder that strips the serde attributes after the extensions have been expanded.
struct StripAttributeFolder;

impl fold::Folder for StripAttributeFolder {
fn fold_attribute(&mut self, attr: ast::Attribute) -> Option<ast::Attribute> {
match attr.node.value.node {
ast::MetaItemKind::List(ref n, _) if n == &"serde" => { return None; }
_ => {}
}

Some(attr)
fn syntex_registry() -> syntex::Registry {
use syntax::{ast, fold};

/// Strip the serde attributes from the crate.
#[cfg(feature = "with-syntex")]
fn strip_attributes(krate: ast::Crate) -> ast::Crate {
/// Helper folder that strips the serde attributes after the extensions have been expanded.
struct StripAttributeFolder;

impl fold::Folder for StripAttributeFolder {
fn fold_attribute(&mut self, attr: ast::Attribute) -> Option<ast::Attribute> {
match attr.node.value.node {
ast::MetaItemKind::List(ref n, _) if n == &"serde" => { return None; }
_ => {}
}

fn fold_mac(&mut self, mac: ast::Mac) -> ast::Mac {
fold::noop_fold_mac(mac, self)
}
Some(attr)
}

fold::Folder::fold_crate(&mut StripAttributeFolder, krate)
fn fold_mac(&mut self, mac: ast::Mac) -> ast::Mac {
fold::noop_fold_mac(mac, self)
}
}

let mut reg = syntex::Registry::new();
fold::Folder::fold_crate(&mut StripAttributeFolder, krate)
}

let mut reg = syntex::Registry::new();

reg.add_attr("feature(custom_derive)");
reg.add_attr("feature(custom_attribute)");

reg.add_decorator("derive_Serialize", ser::expand_derive_serialize);
reg.add_decorator("derive_Deserialize", de::expand_derive_deserialize);

reg.add_post_expansion_pass(strip_attributes);

reg
}

reg.add_attr("feature(custom_derive)");
reg.add_attr("feature(custom_attribute)");
#[cfg(feature = "with-syntex")]
pub fn expand_str(src: &str) -> Result<String, syntex::Error> {
let src = src.to_owned();

reg.add_decorator("derive_Serialize", ser::expand_derive_serialize);
reg.add_decorator("derive_Deserialize", de::expand_derive_deserialize);
let expand_thread = move || {
syntex_registry().expand_str("", "", &src)
};

reg.add_post_expansion_pass(strip_attributes);
syntex::with_extra_stack(expand_thread)
}

reg.expand("", src, dst)
#[cfg(feature = "with-syntex")]
pub fn expand<S, D>(src: S, dst: D) -> Result<(), syntex::Error>
where S: AsRef<Path>,
D: AsRef<Path>,
{
let src = src.as_ref().to_owned();
let dst = dst.as_ref().to_owned();

let expand_thread = move || {
syntex_registry().expand("", src, dst)
};

syntex::with_extra_stack(expand_thread)
Expand Down
23 changes: 23 additions & 0 deletions serde_derive/Cargo.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
[package]
name = "serde_derive"
version = "0.8.5"
authors = ["Erick Tryzelaar <erick.tryzelaar@gmail.com>"]
license = "MIT/Apache-2.0"
description = "Macros 1.1 implementation of #[derive(Serialize, Deserialize)]"
homepage = "https://serde.rs"
repository = "https://github.com/serde-rs/serde"
documentation = "https://serde.rs/codegen.html"
keywords = ["serde", "serialization"]
include = ["Cargo.toml", "src/**/*.rs"]

[lib]
name = "serde_derive"
rustc-macro = true

[dependencies]
serde_codegen = { version = "=0.8.5", path = "../serde_codegen" }

[dev-dependencies]
fnv = "1.0"
serde = { version = "0.8.5", path = "../serde" }
serde_test = { version = "0.8.5", path = "../serde_test" }
21 changes: 21 additions & 0 deletions serde_derive/src/lib.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
#![feature(rustc_macro, rustc_macro_lib)]
#![cfg(not(test))]

extern crate rustc_macro;
extern crate serde_codegen;

use rustc_macro::TokenStream;

#[rustc_macro_derive(Serialize)]
pub fn derive_serialize(input: TokenStream) -> TokenStream {
let item = format!("#[derive(Serialize)]\n{}", input);
let expanded = serde_codegen::expand_str(&item).unwrap();
expanded.parse().unwrap()
}

#[rustc_macro_derive(Deserialize)]
pub fn derive_deserialize(input: TokenStream) -> TokenStream {
let item = format!("#[derive(Deserialize)]\n{}", input);
let expanded = serde_codegen::expand_str(&item).unwrap();
expanded.parse().unwrap()
}
8 changes: 8 additions & 0 deletions serde_derive/tests/test.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
#![feature(test, rustc_macro, rustc_attrs)]

#[macro_use]
extern crate serde_derive;

extern crate test;

include!("../../testing/tests/test.rs.in");

0 comments on commit ffa2f80

Please sign in to comment.