Bootstrap gem symbols from Rubydex graph

Signed-off-by: Alexandre Terrasa <alexandre.terrasa@shopify.com>

Author: Morriar

lib/tapioca/gem/pipeline.rb

@@ -31,8 +31,10 @@ def initialize(
         @events = [] #: Array[Gem::Event]
 
         @payload_symbols = Static::SymbolLoader.payload_symbols #: Set[String]
-        @bootstrap_symbols = load_bootstrap_symbols(@gem) #: Set[String]
-        gem_graph = Static::SymbolLoader.graph_from_paths(@gem.files) if include_doc
+        gem_graph = Static::SymbolLoader.graph_from_paths(@gem.files) #: Rubydex::Graph
+        gem_symbols = gem_graph.declarations.map(&:name).to_set
+        engine_symbols = Static::SymbolLoader.engine_symbols(@gem)
+        @bootstrap_symbols = gem_symbols.union(engine_symbols) #: Set[String]
 
         @bootstrap_symbols.each { |symbol| push_symbol(symbol) }
 
@@ -191,14 +193,6 @@ def name_of(constant)
 
       private
 
-      #: (Gemfile::GemSpec gem) -> Set[String]
-      def load_bootstrap_symbols(gem)
-        engine_symbols = Static::SymbolLoader.engine_symbols(gem)
-        gem_symbols = Static::SymbolLoader.gem_symbols(gem)
-
-        gem_symbols.union(engine_symbols)
-      end
-
       # Events handling
 
       #: -> Gem::Event

lib/tapioca/static/symbol_loader.rb

@@ -26,11 +26,6 @@ def graph_from_paths(paths)
           graph
         end
 
-        #: (Gemfile::GemSpec gem) -> Set[String]
-        def gem_symbols(gem)
-          symbols_from_paths(gem.files)
-        end
-
         #: (Gemfile::GemSpec gem) -> Set[String]
         def engine_symbols(gem)
           gem_engine = engines.find do |engine|
@@ -51,7 +46,8 @@ def engine_symbols(gem)
             Pathname.glob("#{load_path}/**/*.rb")
           end
 
-          symbols_from_paths(paths)
+          engine_graph = graph_from_paths(paths)
+          engine_graph.declarations.map(&:name).to_set
         rescue
           Set.new
         end