add timetable function

1 files changed, 54 insertions, 0 deletions

diff --git a/aplus.go b/aplus.go
index 2e2ab0a..e546b6c 100644
--- a/aplus.go
+++ b/aplus.go
@@ -7,7 +7,10 @@ import (
         "io"
         "net/http"
         "net/url"
+        "regexp"
         "strings"
+        "strconv"
+        "time"
 )
 
 func submit_code(course_code int, attendance_code string) bool {
@@ -176,3 +179,54 @@ func submit_code_sans_course(attendance_code string) {
 
         fmt.Println("Could not submit code", attendance_code, "to any course")
 }
+
+func timetable(course_code int) {
+        cur_body := launch_aplus(course_code)
+        if cur_body == "" {
+                return
+        }
+
+        if verbose {
+                fmt.Printf("%s", cur_body)
+        }
+
+        if strings.Contains(cur_body, "Student not known") {
+                fmt.Println("Student not known. This course must first be visited by an instructor for students to have access.")
+                return
+        }
+
+        current_url := fmt.Sprintf("%s/student/", base_aplus_link)
+        tt_start := strings.Index(cur_body, "<div class=\"stv_tt_container\"")
+
+        link_start := tt_start
+        link_start += strings.Index(cur_body[link_start:], "<a href=\"./?canvasCourse")
+        link_start += len("<a href=\"")
+        link_end := link_start
+        link_end += strings.Index(cur_body[link_end:], "\"")
+
+        link_url := current_url + cur_body[link_start:link_end]
+        link_url = html.UnescapeString(link_url)
+
+        resp, _ := client.Get(link_url)
+        body, _ := io.ReadAll(resp.Body)
+
+        tt_full := string(body)[(strings.Index(string(body), "<div class=\"stv_tt_container\"")):]
+        tt_full = tt_full[:strings.Index(tt_full, "\n")]
+
+        re_date := regexp.MustCompile("Date\\.UTC\\([^\\)]+")
+        re_status := regexp.MustCompile("title=\"\">[^<]*")
+
+        dates, statuses := re_date.FindAllString(tt_full, -1), re_status.FindAllString(tt_full, -1)
+        for i := range dates {
+                jsdate := strings.Split(dates[i][9:], ",")
+                year, _ := strconv.Atoi(jsdate[0])
+                month, _ := strconv.Atoi(jsdate[1])
+                day, _ := strconv.Atoi(jsdate[2])
+                hour, _ := strconv.Atoi(jsdate[3])
+                min, _ := strconv.Atoi(jsdate[4])
+                sec, _ := strconv.Atoi(jsdate[5])
+                date := time.Date(year, time.Month(month + 1), day, hour, min, sec, 0, time.UTC).Local()
+
+                fmt.Println(date.Format("Mon Jan 02 15:04:05 MST 2006") + ": " + strings.Replace(statuses[i], "title=\"\">", "", 1))
+        }
+}