From 627120b31f22f2f286fc2552f4586f29e7e1762c Mon Sep 17 00:00:00 2001
From: vin <git@vineetk.net>
Date: Tue, 16 Jan 2024 14:50:19 -0500
Subject: [PATCH] add timetable function

---
 aplus.go | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 main.go  | 15 ++++++++++++++-
 2 files changed, 68 insertions(+), 1 deletion(-)

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))
+	}
+}
diff --git a/main.go b/main.go
index fa99806..cd110ad 100644
--- a/main.go
+++ b/main.go
@@ -16,11 +16,12 @@ var client http.Client
 var verbose bool
 
 func usage() {
-	fmt.Println("Usage: aplus --code attendance_code --course course_code [--list] [--listfav] [--help]")
+	fmt.Println("Usage: aplus --code attendance_code --course course_code [--list] [--listfav] [--timetable] [--help]")
 	fmt.Println("  -c, --code\tfive character attendance code")
 	fmt.Println("  -C, --course\tcanvas course ID")
 	fmt.Println("  -L, --list\tlists canvas courses with name and ID")
 	fmt.Println("  -F, --listfav\tlists favourited canvas courses with name and ID")
+	fmt.Println("  -t, --timetable\tlists the class sessions in a course")
 	fmt.Println("  -v, --verbose\tprints with verbose output")
 	fmt.Println("  -h, --help\tdisplays this help message")
 	fmt.Println("  -p, --profile\twrites cpu profile to a file")
@@ -33,6 +34,7 @@ func main() {
 	var course int
 	var listall bool
 	var listfav bool
+	var timetable_ bool
 	var help bool
 	var cpuprofile string
 
@@ -44,6 +46,8 @@ func main() {
 	flag.BoolVar(&listall, "L", false, "list all canvas courses")
 	flag.BoolVar(&listfav, "listfav", false, "list favorited canvas courses")
 	flag.BoolVar(&listfav, "F", false, "list favorited canvas courses")
+	flag.BoolVar(&timetable_, "t", false, "lists the class sessions in a course")
+	flag.BoolVar(&timetable_, "timetable", false, "lists the class sessions in a course")
 	flag.BoolVar(&verbose, "verbose", false, "verbose output")
 	flag.BoolVar(&verbose, "v", false, "verbose output")
 	flag.BoolVar(&help, "help", false, "view usage message")
@@ -84,6 +88,15 @@ func main() {
 		os.Exit(0)
 	}
 
+	if course != -1 && timetable_ {
+		timetable(course)
+		os.Exit(0)
+	} else if course == -1 && timetable_ {
+		fmt.Println("Course code required for timetable")
+		usage()
+		os.Exit(1)
+	}
+
 	if help {
 		usage()
 		os.Exit(0)