Switch to getopt library from flag library for flag parsing.

1 files changed, 14 insertions, 8 deletions

diff --git a/main.go b/main.go
index 61bf20f..b46d0c7 100644
--- a/main.go
+++ b/main.go
@@ -1,9 +1,11 @@
 package main
 
 import (
-        "flag"
         "fmt"
         "net/http"
+        "os"
+
+        "github.com/pborman/getopt"
 )
 
 var base_link string
@@ -15,12 +17,18 @@ var client http.Client
 func main() {
         initialize()
 
-        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")
+        code := getopt.StringLong("code", 'c', "", "Code to enter")
+        course := getopt.IntLong("course", 'C', -1, "Canvas course number")
+        list_all := getopt.BoolLong("list-all", 0, "List all courses")
+        list_favorites := getopt.BoolLong("list-favorites", 0, "List favorite courses")
+        help := getopt.BoolLong("help", 0, "Help")
+
+        getopt.Parse()
 
-        flag.Parse()
+        if *help {
+                getopt.Usage()
+                os.Exit(0)
+        }
 
         if *list_all {
                 list_all_courses()
@@ -34,6 +42,4 @@ func main() {
                 fmt.Printf("%d %s", *course, *code)
                 submit_code(*course, *code)
         }
-
-        submit_code(7329, "asdkl")
 }