Returning ready to review count with classes

Author: loiswells97

app/controllers/api/school_classes_controller.rb

@@ -10,7 +10,11 @@ def index
       school_classes = @school.classes.accessible_by(current_ability)
       school_classes = school_classes.joins(:teachers).where(teachers: { teacher_id: current_user.id }) if params[:my_classes] == 'true'
       @school_classes_with_teachers = school_classes.with_teachers
-      render :index, formats: [:json], status: :ok
+      if current_user&.school_teacher?(@school) || current_user&.school_owner?(@school)
+        render :teacher_index, formats: [:json], status: :ok
+      else
+        render :index, formats: [:json], status: :ok
+      end
     end
 
     def show

app/models/lesson.rb

@@ -54,7 +54,6 @@ def unarchive!
 
   def submitted_count
     return 0 unless project
-
     project.remixes.count { |remix| remix.school_project&.submitted? }
   end
 

app/models/school_class.rb

@@ -58,6 +58,12 @@ def assign_class_code
     errors.add(:code, 'could not be generated')
   end
 
+  def submitted_count
+    return 0 if lessons.empty?
+
+    lessons.sum(&:submitted_count)
+  end
+
   private
 
   def school_class_has_at_least_one_teacher

app/views/api/school_classes/_school_class.json.jbuilder

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+json.call(
+  school_class,
+  :id,
+  :description,
+  :school_id,
+  :name,
+  :code,
+  :created_at,
+  :updated_at,
+  :import_origin,
+  :import_id
+)
+
+json.teachers(teachers) do |teacher|
+  json.partial! '/api/school_teachers/school_teacher', teacher:, include_email: false
+end

app/views/api/school_classes/index.json.jbuilder

@@ -2,19 +2,4 @@
 
 school_class, teachers = @school_class_with_teachers
 
-json.call(
-  school_class,
-  :id,
-  :description,
-  :school_id,
-  :name,
-  :code,
-  :created_at,
-  :updated_at,
-  :import_origin,
-  :import_id
-)
-
-json.teachers(teachers) do |teacher|
-  json.partial! '/api/school_teachers/school_teacher', teacher:, include_email: false
-end
+json.partial! 'school_class', school_class: school_class, teachers: teachers

app/views/api/school_classes/show.json.jbuilder

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+json.array!(@school_classes_with_teachers) do |school_class, teachers|
+  json.partial! 'school_class', school_class: school_class, teachers: teachers
+end

app/views/api/school_classes/student_index.json.jbuilder

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+json.array!(@school_classes_with_teachers) do |school_class, teachers|
+  json.partial! 'school_class', school_class: school_class, teachers: teachers
+
+  json.submitted_count(school_class.submitted_count)
+end