diff --git a/app/helpers/packages_helper.rb b/app/helpers/packages_helper.rb index 14b425f..c50114c 100644 --- a/app/helpers/packages_helper.rb +++ b/app/helpers/packages_helper.rb @@ -19,6 +19,23 @@ module PackagesHelper # end # end + def screenshots_visible_to_user(package, user_session) + # Show all screenshots to admins + if current_user && current_user.is_admin? + package.screenshots + + # Show screenshots belonging to the user or public/approved + else + package.screenshots.where( + uploaderhash: user_session.id.to_s + ).or( + package.screenshots.where( + approved: true + ) + ) + end + end + def screenshot_uploaded_by_current_user?(screenshot) screenshot.uploaderhash == session.id.to_s end