From de5900a278ca077719ca3f5ae2dfa0b38a1f082b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jonathan=20M=C3=BCller?= <git@foonathan.net>
Date: Wed, 12 Apr 2017 15:19:27 +0200
Subject: [PATCH] Augment is_definition() & co to work for templates

---
 src/cpp_forward_declarable.cpp | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/cpp_forward_declarable.cpp b/src/cpp_forward_declarable.cpp
index 1041341..c4976d9 100644
--- a/src/cpp_forward_declarable.cpp
+++ b/src/cpp_forward_declarable.cpp
@@ -8,6 +8,7 @@
 #include <cppast/cpp_entity_kind.hpp>
 #include <cppast/cpp_enum.hpp>
 #include <cppast/cpp_function.hpp>
+#include <cppast/cpp_template.hpp>
 #include <cppast/cpp_variable.hpp>
 
 using namespace cppast;
@@ -30,6 +31,11 @@ namespace
         case cpp_entity_kind::constructor_t:
         case cpp_entity_kind::destructor_t:
             return type_safe::ref(static_cast<const cpp_function_base&>(e));
+        case cpp_entity_kind::function_template_t:
+        case cpp_entity_kind::function_template_specialization_t:
+        case cpp_entity_kind::class_template_t:
+        case cpp_entity_kind::class_template_specialization_t:
+            return get_declarable(*static_cast<const cpp_template&>(e).begin());
 
         case cpp_entity_kind::file_t:
         case cpp_entity_kind::macro_definition_t:
@@ -52,10 +58,6 @@ namespace
         case cpp_entity_kind::template_template_parameter_t:
         case cpp_entity_kind::alias_template_t:
         case cpp_entity_kind::variable_template_t:
-        case cpp_entity_kind::function_template_t:
-        case cpp_entity_kind::function_template_specialization_t:
-        case cpp_entity_kind::class_template_t:
-        case cpp_entity_kind::class_template_specialization_t:
         case cpp_entity_kind::static_assert_t:
         case cpp_entity_kind::unexposed_t:
             return nullptr;