Moved screenshots_visible_to_user to model

app/helpers/packages_helper.rb

@@ -1,13 +1,2 @@
 module PackagesHelper
-  # Return a query of all screenshots that the current user may see
-  # Consists of:
-  # - approved (public) screenshots
-  # - screenshots uploaded by the user (determined by cookie session)
-  def screenshots_visible_to_user
-    self.screenshots.where(
-      # "approved=true"
-      "approved=true OR uploaderhash=?", session[:token]
-      )
-  end
-
 end

app/models/package.rb

@@ -37,4 +37,14 @@ class Package < ActiveRecord::Base
     self.screenshots.find_by(approved: true)
   end
 
+  # Return a query of all screenshots that the current user may see
+  # Consists of:
+  # - approved (public) screenshots
+  # - screenshots uploaded by the user (determined by cookie session)
+  def screenshots_visible_to_user(token)
+    self.screenshots.where(
+      # "approved=true"
+      "approved=true OR uploaderhash=?", token
+      )
+  end
 end

app/views/packages/details.slim

@@ -4,8 +4,8 @@
 
   .small-7.medium-7.columns.packagepage
     p.subtitle = @package.description
-    - if @package.screenshots_visible_to_user.count > 0
+    - if @package.screenshots_visible_to_user(session[:token]).count > 0
-      - @package.screenshots_visible_to_user.each do |screenshot|
+      - @package.screenshots_visible_to_user(session[:token]).each do |screenshot|
         .row.listview
           .small-12.columns
             .text-center