impl gen hash for enums

This commit is contained in:
Yoshua Wuyts 2021-08-10 11:59:41 +02:00
parent e652545472
commit ec2535e9ce
3 changed files with 85 additions and 0 deletions

View File

@ -366,6 +366,33 @@ impl Default for Foo {
Self { }
}
}
"#,
)
}
#[test]
fn add_custom_impl_hash_enum() {
check_assist(
replace_derive_with_manual_impl,
r#"
//- minicore: hash
#[derive(Has$0h)]
enum Foo {
Bar,
Baz,
}
"#,
r#"
enum Foo {
Bar,
Baz,
}
impl core::hash::Hash for Foo {
$0fn hash<H: core::hash::Hasher>(&self, state: &mut H) {
core::mem::discriminant(self).hash(state);
}
}
"#,
)
}

View File

@ -18,6 +18,7 @@ pub(crate) fn gen_trait_fn_body(
match trait_path.segment()?.name_ref()?.text().as_str() {
"Debug" => gen_debug_impl(adt, func),
"Default" => gen_default_impl(adt, func),
"Hash" => gen_hash_impl(adt, func),
_ => None,
}
}
@ -151,3 +152,49 @@ fn gen_default_call() -> ast::Expr {
}
}
}
/// Generate a `Hash` impl based on the fields and members of the target type.
fn gen_hash_impl(adt: &ast::Adt, func: &ast::Fn) -> Option<()> {
let body = match adt {
// `Hash` cannot be derived for unions, so no default impl can be provided.
ast::Adt::Union(_) => return None,
// => std::mem::discriminant(self).hash(state);
ast::Adt::Enum(_) => {
let root = make::ext::ident_path("core");
let submodule = make::ext::ident_path("mem");
let fn_name = make::ext::ident_path("discriminant");
let fn_name = make::path_concat(submodule, fn_name);
let fn_name = make::expr_path(make::path_concat(root, fn_name));
let arg = make::expr_path(make::ext::ident_path("self"));
let fn_call = make::expr_call(fn_name, make::arg_list(Some(arg)));
let method = make::name_ref("hash");
let arg = make::expr_path(make::ext::ident_path("state"));
let expr = make::expr_method_call(fn_call, method, make::arg_list(Some(arg)));
let stmt = make::expr_stmt(expr);
make::block_expr(Some(stmt.into()), None).indent(ast::edit::IndentLevel(1))
}
ast::Adt::Struct(strukt) => match strukt.field_list() {
// => self.<field>.hash(state);*
Some(ast::FieldList::RecordFieldList(field_list)) => {
// let mut stmts = vec![];
for field in field_list.fields() {}
todo!();
}
// => self.<field_index>.hash(state);*
Some(ast::FieldList::TupleFieldList(field_list)) => {
todo!();
}
// No fields in the body means there's nothing to hash.
None => make::ext::empty_block_expr(),
},
};
ted::replace(func.body()?.syntax(), body.clone_for_update().syntax());
Some(())
}

View File

@ -25,6 +25,7 @@
//! iterator: option
//! iterators: iterator, fn
//! default: sized
//! hash:
//! clone: sized
//! copy: clone
//! from: sized
@ -87,6 +88,16 @@ pub trait Default: Sized {
}
// endregion:default
// region:hash
pub mod hash {
pub trait Hasher {}
pub trait Hash {
fn hash<H: Hasher>(&self, state: &mut H);
}
}
// endregion:hash
// region:clone
pub mod clone {
#[lang = "clone"]