Add support for arithmetic assignment statements

1 files changed, 62 insertions, 1 deletions

diff --git a/expression.go b/expression.go
index 9752730..27f4ddb 100644
--- a/expression.go
+++ b/expression.go
@@ -134,6 +134,11 @@ func (m *mutStmt) markDecl() {}
 
 func (m *mutStmt) line() int { return m.ln }
 
+type assignmentStmt interface {
+	statementExpr
+	markAssign()
+}
+
 type assignStmt struct {
 	name  string
 	value exprExpr
@@ -144,10 +149,66 @@ func (a *assignStmt) markExpr() {}
 
 func (a *assignStmt) markStmt() {}
 
-func (a *assignStmt) markDecl() {}
+func (a *assignStmt) markAssign() {}
 
 func (a *assignStmt) line() int { return a.ln }
 
+type addAssignStmt struct {
+	name  string
+	value exprExpr
+	ln    int
+}
+
+func (a *addAssignStmt) markExpr() {}
+
+func (a *addAssignStmt) markStmt() {}
+
+func (a *addAssignStmt) markAssign() {}
+
+func (a *addAssignStmt) line() int { return a.ln }
+
+type subAssignStmt struct {
+	name  string
+	value exprExpr
+	ln    int
+}
+
+func (s *subAssignStmt) markExpr() {}
+
+func (s *subAssignStmt) markStmt() {}
+
+func (s *subAssignStmt) markAssign() {}
+
+func (s *subAssignStmt) line() int { return s.ln }
+
+type mulAssignStmt struct {
+	name  string
+	value exprExpr
+	ln    int
+}
+
+func (m *mulAssignStmt) markExpr() {}
+
+func (m *mulAssignStmt) markStmt() {}
+
+func (m *mulAssignStmt) markAssign() {}
+
+func (m *mulAssignStmt) line() int { return m.ln }
+
+type divAssignStmt struct {
+	name  string
+	value exprExpr
+	ln    int
+}
+
+func (d *divAssignStmt) markExpr() {}
+
+func (d *divAssignStmt) markStmt() {}
+
+func (d *divAssignStmt) markAssign() {}
+
+func (d *divAssignStmt) line() int { return d.ln }
+
 type exprExpr interface {
 	expression
 	markExprExpr()