add multiply/divide assign operators

andogq · Aug 22, 2024 · 592c096 · 592c096
1 parent 621b77f
commit 592c096
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/repr/token.rs b/src/repr/token.rs
@@ -42,6 +42,10 @@ pub enum Token {
     AddAssign,
     #[token("-=")]
     MinusAssign,
+    #[token("*=")]
+    MulAssign,
+    #[token("/=")]
+    DivAssign,
     #[token("->")]
     ThinArrow,
     #[token(":")]
@@ -141,6 +145,8 @@ impl Display for Token {
             Token::Eq => write!(f, "="),
             Token::AddAssign => write!(f, "+="),
             Token::MinusAssign => write!(f, "-="),
+            Token::MulAssign => write!(f, "*="),
+            Token::DivAssign => write!(f, "/="),
             Token::ThinArrow => write!(f, "->"),
             Token::Colon => write!(f, ":"),
             Token::SemiColon => write!(f, ";"),

diff --git a/src/stage/parse/expression/e_assign.rs b/src/stage/parse/expression/e_assign.rs
@@ -55,6 +55,8 @@ pub fn parse_op_assign(
     let operation = match tokens.next_token().unwrap() {
         Token::AddAssign => InfixOperation::Plus,
         Token::MinusAssign => InfixOperation::Minus,
+        Token::MulAssign => InfixOperation::Multiply,
+        Token::DivAssign => InfixOperation::Divide,
         token => {
             return Err(ParseError::UnexpectedToken(token));
         }