fix returned references

1 files changed, 11 insertions, 0 deletions

diff --git a/clang/rename.cxx b/clang/rename.cxx
index 8cbdecf..4b87cb1 100644
--- a/clang/rename.cxx
+++ b/clang/rename.cxx
@@ -330,6 +330,17 @@ public:
         {
             clang::QualType pType = pDecl->getReturnType();
             const clang::RecordDecl* pRecordDecl = pType->getAsCXXRecordDecl();
+
+            /*
+             * inline C& C::bar() { return *this; }
+             *        ^ Handle this.
+             */
+            if (!pRecordDecl)
+            {
+                if (const clang::ReferenceType* pReferenceType = pType->getAs<clang::ReferenceType>())
+                    pRecordDecl = pReferenceType->getPointeeType()->getAsCXXRecordDecl();
+            }
+
             if (pRecordDecl)
                 RewriteText(pDecl->getReturnTypeSourceRange().getBegin(), pRecordDecl->getNameAsString().length(), pRecordDecl->getNameAsString());
         }