diff --git a/app/helpers/courses_helper.rb b/app/helpers/courses_helper.rb index 4631b33..61379aa 100644 --- a/app/helpers/courses_helper.rb +++ b/app/helpers/courses_helper.rb @@ -14,4 +14,17 @@ def enrollment_button(course) link_to "Check price", course_path(course), class: "btn btn-md btn-success" end end + + def review_button(course) + user_course = course.enrollments.where(user: current_user) + if current_user + if user_course.any? + if user_course.pending_review.any? + link_to "Add a review", edit_enrollment_path(user_course.first) + else + link_to "Thanks for reviewing! Your Review", enrollment_path(user_course.first) + end + end + end + end end diff --git a/app/models/enrollment.rb b/app/models/enrollment.rb index 43f14c2..41df6c0 100644 --- a/app/models/enrollment.rb +++ b/app/models/enrollment.rb @@ -10,6 +10,8 @@ class Enrollment < ApplicationRecord validate :cant_subscribe_to_own_course + scope :pending_review, -> { where(rating: [ 0, nil, "" ], review: [ 0, nil, "" ]) } + def to_s user.to_s + " " + course.to_s end diff --git a/app/views/courses/_course.html.haml b/app/views/courses/_course.html.haml index 8c57027..f1f7ac8 100644 --- a/app/views/courses/_course.html.haml +++ b/app/views/courses/_course.html.haml @@ -7,6 +7,8 @@ .card-footer .text-center = enrollment_button(course) + .row + = review_button(course) .card-footer .fa.fa-list-check = course.lessons.count