Add ExprStmt::ExprStmtType WITH_BLOCK and WITHOUT_BLOCK to allow switching

1 files changed, 16 insertions, 1 deletions

diff --git a/gcc/rust/ast/rust-stmt.h b/gcc/rust/ast/rust-stmt.h
index 942da7f..9d95c3e 100644
--- a/gcc/rust/ast/rust-stmt.h
+++ b/gcc/rust/ast/rust-stmt.h
@@ -183,8 +183,14 @@ protected:
  * expression) */
 class ExprStmt : public Stmt
 {
-  // TODO: add any useful virtual functions
+public:
+  enum ExprStmtType
+  {
+    WITH_BLOCK,
+    WITHOUT_BLOCK
+  };
 
+protected:
   Location locus;
 
 public:
@@ -192,6 +198,8 @@ public:
 
   bool is_item () const override final { return false; }
 
+  virtual ExprStmtType get_type () const = 0;
+
 protected:
   ExprStmt (Location locus) : locus (locus) {}
 };
@@ -261,6 +269,11 @@ public:
     return expr;
   }
 
+  ExprStmtType get_type () const override
+  {
+    return ExprStmtType::WITHOUT_BLOCK;
+  };
+
 protected:
   /* Use covariance to implement clone function as returning this object rather
    * than base */
@@ -328,6 +341,8 @@ public:
 
   bool is_semicolon_followed () const { return semicolon_followed; }
 
+  ExprStmtType get_type () const override { return ExprStmtType::WITH_BLOCK; };
+
 protected:
   /* Use covariance to implement clone function as returning this object rather
    * than base */