diff --git a/app/models/package.rb b/app/models/package.rb index 28b7094..6099b09 100644 --- a/app/models/package.rb +++ b/app/models/package.rb @@ -31,4 +31,22 @@ class Package < ActiveRecord::Base subselect = Screenshot.select(:package_id) where.not(id: subselect) end + + # Return a query of all approved/public screenshots of this package + def self.query_approved + 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) + working on it… + def self.query_visible_to_user + approved_screenshots = Screenshot.find_by() + + subselect = Screenshot.select(:package_id) + where.not(id: subselect) + end + end