Add flags, update README

1 files changed, 22 insertions, 3 deletions

diff --git a/main.go b/main.go
index 6e4205e..61bf20f 100644
--- a/main.go
+++ b/main.go
@@ -1,6 +1,8 @@
 package main
 
 import (
+        "flag"
+        "fmt"
         "net/http"
 )
 
@@ -13,8 +15,25 @@ var client http.Client
 func main() {
         initialize()
 
-        course_code := 7326
-        submit_code(course_code, "A1B2C")
+        code := flag.String("code", "", "Code to enter")
+        course := flag.Int("course", -1, "Canvas course number")
+        list_all := flag.Bool("list-all", false, "List all courses")
+        list_favorites := flag.Bool("list-favorites", false, "List favorite courses")
 
-        // enter_code()
+        flag.Parse()
+
+        if *list_all {
+                list_all_courses()
+        }
+
+        if *list_favorites {
+                list_favorite_courses()
+        }
+
+        if *code != "" && *course != -1 {
+                fmt.Printf("%d %s", *course, *code)
+                submit_code(*course, *code)
+        }
+
+        submit_code(7329, "asdkl")
 }