Refactor to move some code to `state.rs`

1 files changed, 4 insertions, 3 deletions

diff --git a/src/construct/attention.rs b/src/construct/attention.rs
index d61813d..7e873ca 100644
--- a/src/construct/attention.rs
+++ b/src/construct/attention.rs
@@ -51,8 +51,9 @@
 //! [html-em]: https://html.spec.whatwg.org/multipage/text-level-semantics.html#the-em-element
 //! [html-strong]: https://html.spec.whatwg.org/multipage/text-level-semantics.html#the-strong-element
 
+use crate::state::{Name, State};
 use crate::token::Token;
-use crate::tokenizer::{Event, EventType, Point, State, StateName, Tokenizer};
+use crate::tokenizer::{Event, EventType, Point, Tokenizer};
 use crate::unicode::PUNCTUATION;
 use crate::util::slice::Slice;
 
@@ -120,7 +121,7 @@ pub fn start(tokenizer: &mut Tokenizer) -> State {
         Some(b'*' | b'_') if tokenizer.parse_state.constructs.attention => {
             tokenizer.tokenize_state.marker = tokenizer.current.unwrap();
             tokenizer.enter(Token::AttentionSequence);
-            State::Retry(StateName::AttentionInside)
+            State::Retry(Name::AttentionInside)
         }
         _ => State::Nok,
     }
@@ -136,7 +137,7 @@ pub fn inside(tokenizer: &mut Tokenizer) -> State {
     match tokenizer.current {
         Some(b'*' | b'_') if tokenizer.current.unwrap() == tokenizer.tokenize_state.marker => {
             tokenizer.consume();
-            State::Next(StateName::AttentionInside)
+            State::Next(Name::AttentionInside)
         }
         _ => {
             tokenizer.exit(Token::AttentionSequence);