From ee128a07ccf76b8add03399cdbeb5b43c753386a Mon Sep 17 00:00:00 2001
From: Szymon Sobik <sobik.szymon@gmail.com>
Date: Fri, 30 Nov 2018 15:40:11 +0100
Subject: [PATCH] getting translation unit cursor

---
 src/translation_unit.cc | 4 ++++
 src/translation_unit.h  | 1 +
 2 files changed, 5 insertions(+)

diff --git a/src/translation_unit.cc b/src/translation_unit.cc
index f1ab0db..9b25b7e 100644
--- a/src/translation_unit.cc
+++ b/src/translation_unit.cc
@@ -93,6 +93,10 @@ std::unique_ptr<clangmm::Tokens> clangmm::TranslationUnit::get_tokens(unsigned s
   return std::unique_ptr<Tokens>(new Tokens(cx_tu, range));
 }
 
+clangmm::Cursor clangmm::TranslationUnit::get_cursor() {
+  return Cursor(clang_getTranslationUnitCursor(cx_tu));
+}
+
 clangmm::Cursor clangmm::TranslationUnit::get_cursor(const std::string &path, unsigned offset) {
   SourceLocation location(cx_tu, path, offset);
   return Cursor(clang_getCursor(cx_tu, location.cx_location));
diff --git a/src/translation_unit.h b/src/translation_unit.h
index 33f72ec..78dd75f 100644
--- a/src/translation_unit.h
+++ b/src/translation_unit.h
@@ -37,6 +37,7 @@ namespace clangmm {
     std::unique_ptr<Tokens> get_tokens(unsigned start_line, unsigned start_column,
                                        unsigned end_line, unsigned end_column);
 
+    Cursor get_cursor();
     Cursor get_cursor(const std::string &path, unsigned offset);
     Cursor get_cursor(const std::string &path, unsigned line, unsigned column);
     Cursor get_cursor(const SourceLocation &location);