From 5206b9ca55ce9e74cadeca777b36ae2385edcbcf Mon Sep 17 00:00:00 2001
From: Luca Piccone <lpiccone19@gmail.com>
Date: Mon, 6 Oct 2025 12:18:50 -0400
Subject: [PATCH] Completed Task 3

---
 src/main/java/api/MongoGradeDataBase.java     | 33 +++++++++++++------
 .../java/usecase/GetAverageGradeUseCase.java  | 14 +++++---
 2 files changed, 33 insertions(+), 14 deletions(-)

diff --git a/src/main/java/api/MongoGradeDataBase.java b/src/main/java/api/MongoGradeDataBase.java
index 77e76548..1a12c8aa 100644
--- a/src/main/java/api/MongoGradeDataBase.java
+++ b/src/main/java/api/MongoGradeDataBase.java
@@ -64,8 +64,9 @@ public Grade getGrade(String username, String course) {
                         .build();
             }
             else {
-                throw new RuntimeException("Grade could not be found for course: " + course
-                                           + " and username: " + username);
+                return null;
+//                throw new RuntimeException("Grade could not be found for course: " + course
+//                                           + " and username: " + username);
             }
         }
         catch (IOException | JSONException event) {
@@ -247,10 +248,6 @@ public void leaveTeam() throws JSONException {
     }
 
     @Override
-    // TODO Task 3b: Implement this method
-    //       Hint: Read the Grade API documentation for getMyTeam (link below) and refer to the above similar
-    //             methods to help you write this code (copy-and-paste + edit as needed).
-    //             https://www.postman.com/cloudy-astronaut-813156/csc207-grade-apis-demo/folder/isr2ymn/get-my-team
     public Team getMyTeam() {
         final OkHttpClient client = new OkHttpClient().newBuilder()
                 .build();
@@ -263,11 +260,27 @@ public Team getMyTeam() {
 
         final Response response;
         final JSONObject responseBody;
+        try {
+            response = client.newCall(request).execute();
+            responseBody = new JSONObject(response.body().string());
+            if (responseBody.getInt(STATUS_CODE) == SUCCESS_CODE) {
+                final JSONObject team = responseBody.getJSONObject("team");
+                final JSONArray membersArray = team.getJSONArray("members");
+                final String[] members = new String[membersArray.length()];
+                for (int i = 0; i < membersArray.length(); i++) {
+                    members[i] = membersArray.getString(i);
+                }
 
-        // TODO Task 3b: Implement the logic to get the team information
-        // HINT 1: Look at the formTeam method to get an idea on how to parse the response
-        // HINT 2: You may find it useful to just initially print the contents of the JSON
-        //         then work on the details of how to parse it.
+                return Team.builder()
+                        .name(team.getString(NAME))
+                        .members(members)
+                        .build();
+            }
+
+
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
         return null;
     }
 }
diff --git a/src/main/java/usecase/GetAverageGradeUseCase.java b/src/main/java/usecase/GetAverageGradeUseCase.java
index 25140b18..8e202b04 100644
--- a/src/main/java/usecase/GetAverageGradeUseCase.java
+++ b/src/main/java/usecase/GetAverageGradeUseCase.java
@@ -23,11 +23,17 @@ public float getAverageGrade(String course) {
         // Call the API to get usernames of all your team members
         float sum = 0;
         int count = 0;
-        // TODO Task 3b: Go to the MongoGradeDataBase class and implement getMyTeam.
+
         final Team team = gradeDataBase.getMyTeam();
-        // Call the API to get all the grades for the course for all your team members
-        // TODO Task 3a: Complete the logic of calculating the average course grade for
-        //              your team members. Hint: the getGrades method might be useful.
+        String[] members = team.getMembers();
+        GetGradeUseCase getGrade = new GetGradeUseCase(gradeDataBase);
+        for (String member : members) {
+            Grade grade = gradeDataBase.getGrade(member, course);
+            if (grade != null) {
+                sum += grade.getGrade();
+                count++;
+            }
+        }
 
         if (count == 0) {
             return 0;