handle roles not supported canvas error

1 files changed, 14 insertions, 1 deletions

diff --git a/aplus.go b/aplus.go
index e546b6c..a84df8c 100644
--- a/aplus.go
+++ b/aplus.go
@@ -16,6 +16,10 @@ import (
 func submit_code(course_code int, attendance_code string) bool {
         cur_body := launch_aplus(course_code)
         if cur_body == "" {
+                fmt.Println("Unknown error")
+                return false
+        } else if cur_body == "Roles not supported" {
+                fmt.Println("Cannot access course", course_code, "(canvas error \"Roles not supported\")")
                 return false
         }
 
@@ -101,7 +105,12 @@ func launch_aplus(course_code int) string {
 
         resp, _ = client.PostForm(base_aplus_link, form_values)
         body, _ = io.ReadAll(resp.Body)
-        return string(body)
+
+        if strings.Index(string(body), "Roles not supported") != -1 {
+                return "Roles not supported"
+        } else {
+                return string(body)
+        }
 }
 
 func get_aplus(token string, link string, client http.Client) string {
@@ -183,6 +192,10 @@ func submit_code_sans_course(attendance_code string) {
 func timetable(course_code int) {
         cur_body := launch_aplus(course_code)
         if cur_body == "" {
+                fmt.Println("Unknown error")
+                return
+        } else if cur_body == "Roles not supported" {
+                fmt.Println("Cannot access course", course_code, "(canvas error \"Roles not supported\")")
                 return
         }