lib/bosh/stemcell/archive_filename.rb in bosh-stemcell-1.2732.0 vs lib/bosh/stemcell/archive_filename.rb in bosh-stemcell-1.2739.0

@@ -2,34 +2,33 @@
 
 module Bosh::Stemcell
   class ArchiveFilename
     extend Forwardable
 
-    def initialize(version, definition, base_name, light)
+    def initialize(version, definition, base_name)
       @version = version
       @definition = definition
       @base_name = base_name
-      @light = light
     end
 
     def to_s
       stemcell_filename_parts = [
         name,
         version,
-        @definition.stemcell_name
+        definition.stemcell_name
       ]
       "#{stemcell_filename_parts.join('-')}.tgz"
     end
 
     private
 
     def name
-      light ? "light-#{base_name}" : base_name
+      definition.light? ? "light-#{base_name}" : base_name
     end
 
     attr_reader(
       :base_name,
       :version,
-      :light,
+      :definition,
     )
   end
 end