lib/lookbook/preview.rb in lookbook-1.0.8 vs lib/lookbook/preview.rb in lookbook-1.1.0

@@ -5,22 +5,26 @@
     delegate :name, :render_args, to: :@preview
     delegate :position, :group, :notes, :hidden?, :tags, :tag, to: :@preview_inspector
 
     def initialize(preview, code_object)
       @preview = preview
-      @preview_inspector = SourceInspector.new(code_object)
+      @preview_inspector = SourceInspector.new(code_object, eval_scope: preview_class.new)
       super(preview_class_path(@preview.name))
     end
 
     def id
       @preview_inspector&.id || generate_id(lookup_path)
     end
 
-    def preview_class
+    def preview_class_name
       @preview.name
     end
 
+    def preview_class
+      @preview
+    end
+
     def label
       @preview_inspector&.label&.presence || lookup_path.split("/").last.titleize
     end
 
     def type
@@ -141,13 +145,15 @@
             nil
           end.compact
 
           sorted_previews = previews.compact.sort_by { |preview| [preview.position, preview.label] }
           @previews = PreviewCollection.new(sorted_previews)
-        else
+          @previews
+        elsif !@preview_objects.present?
           PreviewCollection.new([])
+        else
+          @previews
         end
-        @previews
       end
 
       def errors
         @errors ||= []
       end