lib/librarian/lockfile/compiler.rb in librarian-0.0.9 vs lib/librarian/lockfile/compiler.rb in librarian-0.0.10

@@ -4,14 +4,15 @@
   class Lockfile
     class Compiler
 
       include Helpers::Debug
 
-      attr_reader :root_module
+      attr_accessor :environment
+      private :environment=
 
-      def initialize(root_module)
-        @root_module = root_module
+      def initialize(environment)
+        self.environment = environment
       end
 
       def compile(resolution)
         out = StringIO.new
         save_sources(out, resolution.manifests)
@@ -59,10 +60,10 @@
         end
         out.puts ""
       end
 
       def dsl_class
-        root_module.dsl_class
+        environment.dsl_class
       end
 
     end
   end
 end