Refactor uses of cast to dyn_cast

Author: justin-s-kang

lib/ConstExtract/ConstExtract.cpp

@@ -302,10 +302,8 @@ extractCompileTimeValue(Expr *expr, const DeclContext *declContext) {
 
     case ExprKind::Call: {
       auto callExpr = cast<CallExpr>(expr);
-      auto functionKind = callExpr->getFn()->getKind();
 
-      if (functionKind == ExprKind::DeclRef) {
-        auto declRefExpr = cast<DeclRefExpr>(callExpr->getFn());
+      if (auto declRefExpr = dyn_cast<DeclRefExpr>(callExpr->getFn())) {
         auto identifier =
             declRefExpr->getDecl()->getName().getBaseIdentifier().str().str();
 
@@ -314,17 +312,15 @@ extractCompileTimeValue(Expr *expr, const DeclContext *declContext) {
         return std::make_shared<FunctionCallValue>(identifier, parameters);
       }
 
-      if (functionKind == ExprKind::ConstructorRefCall) {
+      if (auto constructorRefCall = dyn_cast<ConstructorRefCallExpr>(callExpr->getFn())) {
         std::vector<FunctionParameter> parameters =
             extractFunctionArguments(callExpr->getArgs(), declContext);
         return std::make_shared<InitCallValue>(callExpr->getType(), parameters);
       }
 
-      if (functionKind == ExprKind::DotSyntaxCall) {
-        auto dotSyntaxCallExpr = cast<DotSyntaxCallExpr>(callExpr->getFn());
+      if (auto dotSyntaxCallExpr = dyn_cast<DotSyntaxCallExpr>(callExpr->getFn())) {
         auto fn = dotSyntaxCallExpr->getFn();
-        if (fn->getKind() == ExprKind::DeclRef) {
-          auto declRefExpr = cast<DeclRefExpr>(fn);
+        if (auto declRefExpr = dyn_cast<DeclRefExpr>(fn)) {
           auto baseIdentifierName =
               declRefExpr->getDecl()->getName().getBaseIdentifier().str().str();
 
@@ -355,10 +351,8 @@ extractCompileTimeValue(Expr *expr, const DeclContext *declContext) {
         }
       }
 
-      if (functionKind == ExprKind::FunctionConversion) {
-        auto functionConversionExpr = cast<FunctionConversionExpr>(callExpr->getFn());
-        if (functionConversionExpr->getSubExpr()->getKind() == ExprKind::DeclRef) {
-          auto declRefExpr = cast<DeclRefExpr>(functionConversionExpr->getSubExpr());
+      if (auto functionConversionExpr = dyn_cast<FunctionConversionExpr>(callExpr->getFn())) {
+        if (auto declRefExpr = dyn_cast<DeclRefExpr>(functionConversionExpr->getSubExpr())) {
           auto identifier =
               declRefExpr->getDecl()->getName().getBaseIdentifier().str().str();
 
@@ -374,8 +368,7 @@ extractCompileTimeValue(Expr *expr, const DeclContext *declContext) {
     case ExprKind::DotSyntaxCall: {
       auto dotSyntaxCallExpr = cast<DotSyntaxCallExpr>(expr);
       auto fn = dotSyntaxCallExpr->getFn();
-      if (fn->getKind() == ExprKind::DeclRef) {
-        auto declRefExpr = cast<DeclRefExpr>(fn);
+      if (auto declRefExpr = dyn_cast<DeclRefExpr>(fn)) {
         auto caseName =
             declRefExpr->getDecl()->getName().getBaseIdentifier().str().str();
         return std::make_shared<EnumValue>(caseName, std::nullopt);