use con_::Con;
use java::{BlockComment, Modifier};
use {Cons, Element, IntoTokens, Java, Tokens};
#[derive(Debug, Clone)]
pub struct Field<'el> {
pub annotations: Tokens<'el, Java<'el>>,
pub modifiers: Vec<Modifier>,
pub comments: Vec<Cons<'el>>,
ty: Java<'el>,
name: Cons<'el>,
initializer: Option<Tokens<'el, Java<'el>>>,
}
impl<'el> Field<'el> {
pub fn new<T, N>(ty: T, name: N) -> Field<'el>
where
T: Into<Java<'el>>,
N: Into<Cons<'el>>,
{
use self::Modifier::*;
Field {
annotations: Tokens::new(),
modifiers: vec![Private, Final],
comments: vec![],
ty: ty.into(),
name: name.into(),
initializer: None,
}
}
pub fn annotation<A>(&mut self, annotation: A)
where
A: IntoTokens<'el, Java<'el>>,
{
self.annotations.push(annotation.into_tokens());
}
pub fn initializer<I>(&mut self, initializer: I)
where
I: IntoTokens<'el, Java<'el>>,
{
self.initializer = Some(initializer.into_tokens());
}
pub fn var(&self) -> Cons<'el> {
self.name.clone()
}
pub fn ty(&self) -> Java<'el> {
self.ty.clone()
}
}
into_tokens_impl_from!(Field<'el>, Java<'el>);
impl<'el> IntoTokens<'el, Java<'el>> for Field<'el> {
fn into_tokens(self) -> Tokens<'el, Java<'el>> {
let mut tokens = Tokens::new();
tokens.push_unless_empty(BlockComment(self.comments));
if !self.annotations.is_empty() {
tokens.push(self.annotations);
tokens.append(Element::PushSpacing);
}
tokens.append({
let mut sig = Tokens::new();
sig.extend(self.modifiers.into_tokens());
sig.append(self.ty);
sig.append(self.name);
if let Some(initializer) = self.initializer {
sig.append("=");
sig.append(initializer);
}
sig.join_spacing()
});
tokens
}
}
impl<'el> From<Field<'el>> for Element<'el, Java<'el>> {
fn from(f: Field<'el>) -> Self {
Element::Append(Con::Owned(f.into_tokens()))
}
}
#[cfg(test)]
mod tests {
use java::{Field, INTEGER};
use tokens::Tokens;
fn field() -> Field<'static> {
Field::new(INTEGER, "foo")
}
#[test]
fn test_with_comments() {
let mut c = field();
c.comments.push("Hello World".into());
let t: Tokens<_> = c.into();
assert_eq!(
Ok(String::from(
"/**\n * Hello World\n */\nprivate final int foo",
)),
t.to_string()
);
}
#[test]
fn test_no_comments() {
let t = Tokens::from(field());
assert_eq!(Ok(String::from("private final int foo")), t.to_string());
}
}