add timetable function

1 files changed, 14 insertions, 1 deletions

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)