add file including api

Author: woshiccm

Sources/IndexStoreDB/IndexStoreDB.swift

@@ -239,13 +239,13 @@ public final class IndexStoreDB {
     return result
   }
 
-   @discardableResult
+  @discardableResult
   public func foreachFileIncludedByFile(path: String, body: @escaping (String) -> Bool) -> Bool {
-     return indexstoredb_index_files_included_by_file(impl, path) { targetPath, line in
+    return indexstoredb_index_files_included_by_file(impl, path) { targetPath, line in
         let targetPathStr = String(cString: targetPath)
         return body(targetPathStr)
-       }
-   }
+    }
+  }
 
   public func filesIncludedByFile(path: String) -> [String] {
     var result: [String] = []
@@ -255,6 +255,23 @@ public final class IndexStoreDB {
     }
     return result
   }
+    
+  @discardableResult
+  public func foreachFileIncludingFile(path: String, body: @escaping (String) -> Bool) -> Bool {
+    return indexstoredb_index_files_including_file(impl, path) { sourcePath, line in
+        let sourcePathStr = String(cString: sourcePath)
+        return body(sourcePathStr)
+    }
+  }
+      
+  public func filesIncludingFile(path: String) -> [String] {
+    var result: [String] = []
+    foreachFileIncludingFile(path: path) { targetPath in
+        result.append(targetPath)
+        return true
+      }
+      return result
+    }
 
   /// A recorded header `#include` from a unit file.
   public struct UnitIncludeEntry: Equatable {

include/CIndexStoreDB/CIndexStoreDB.h

@@ -141,6 +141,9 @@ typedef bool(^indexstoredb_unit_info_receiver)(_Nonnull indexstoredb_unit_info_t
 /// Returns true to continue.
 typedef bool(^indexstoredb_files_included_receiver)(const char *_Nonnull targetPath, size_t line);
 
+/// Returns true to continue.
+typedef bool(^indexstoredb_files_including_receiver)(const char *_Nonnull sourcePath, size_t line);
+
 /// Returns true to continue.
 typedef bool(^indexstoredb_unit_includes_receiver)(const char *_Nonnull sourcePath, const char *_Nonnull targetPath, size_t line);
 
@@ -388,7 +391,7 @@ indexstoredb_index_units_containing_file(
   const char *_Nonnull path,
   _Nonnull indexstoredb_unit_info_receiver receiver);
 
-/// Return the file path which included by a given file path..
+/// Return the file path which included by a given file path.
 ///
 /// \param index An IndexStoreDB object which contains the symbols.
 /// \param path The source file to search for.
@@ -400,6 +403,18 @@ indexstoredb_index_files_included_by_file(
   const char *_Nonnull path,
   _Nonnull indexstoredb_files_included_receiver receiver);
 
+/// Return the file path which including a given header.
+///
+/// \param index An IndexStoreDB object which contains the symbols.
+/// \param path The source file to search for.
+/// \param receiver A function to be called for each include file path. The pointers are only valid for
+/// the duration of the call. The function should return a true to continue iterating.
+INDEXSTOREDB_PUBLIC bool
+indexstoredb_index_files_including_file(
+  _Nonnull indexstoredb_index_t index,
+  const char *_Nonnull path,
+  _Nonnull indexstoredb_files_including_receiver receiver);
+
 /// Iterates over recorded `#include`s of a unit.
 ///
 /// \param index An IndexStoreDB object which contains the symbols.

lib/CIndexStoreDB/CIndexStoreDB.cpp

@@ -443,6 +443,18 @@ indexstoredb_index_files_included_by_file(
   });
 }
 
+bool
+indexstoredb_index_files_including_file(
+  indexstoredb_index_t index,
+  const char *path,
+  indexstoredb_files_including_receiver receiver)
+{
+  auto obj = (Object<std::shared_ptr<IndexSystem>> *)index;
+  return obj->value->foreachFileIncludingFile(path, [&](const CanonicalFilePathRef &sourcePath, unsigned line) -> bool {
+    return receiver(sourcePath.getPath().str().c_str(), line);
+  });
+}
+
 bool
 indexstoredb_index_includes_of_unit(
   indexstoredb_index_t index,