match parent.kind() {
STRUCT_DEF => HighlightTag::Struct.into(),
ENUM_DEF => HighlightTag::Enum.into(),
- UNION_KW => HighlightTag::Union.into(),
+ UNION_DEF => HighlightTag::Union.into(),
TRAIT_DEF => HighlightTag::Trait.into(),
TYPE_ALIAS_DEF => HighlightTag::TypeAlias.into(),
TYPE_PARAM => HighlightTag::TypeParam.into(),
let def: hir::Struct = sema.to_def(&it)?;
Some(from_module_def(def.into()))
},
+ ast::UnionDef(it) => {
+ let def: hir::Union = sema.to_def(&it)?;
+ Some(from_module_def(def.into()))
+ },
ast::EnumDef(it) => {
let def: hir::Enum = sema.to_def(&it)?;
Some(from_module_def(def.into()))