fix return type

1 files changed, 12 insertions, 0 deletions

diff --git a/clang/rename.cxx b/clang/rename.cxx
index 882cb0d..8cbdecf 100644
--- a/clang/rename.cxx
+++ b/clang/rename.cxx
@@ -321,6 +321,18 @@ public:
         std::string aClassName = pDecl->getParent()->getNameAsString();
         if (HasNamespace(aClassName) && pDecl->isThisDeclarationADefinition())
             RewriteText(pDecl->getQualifierLoc().getBeginLoc(), aClassName.length(), aClassName);
+
+        /*
+         * inline C C::bar() { return C(); }
+         *        ^ Handle this.
+         */
+        if (HasNamespace(aClassName))
+        {
+            clang::QualType pType = pDecl->getReturnType();
+            const clang::RecordDecl* pRecordDecl = pType->getAsCXXRecordDecl();
+            if (pRecordDecl)
+                RewriteText(pDecl->getReturnTypeSourceRange().getBegin(), pRecordDecl->getNameAsString().length(), pRecordDecl->getNameAsString());
+        }
         return true;
     }