use golang flag instead of external getopt library

1 files changed, 47 insertions, 16 deletions

diff --git a/main.go b/main.go
index 9386b46..93defd3 100644
--- a/main.go
+++ b/main.go
@@ -1,43 +1,74 @@
 package main
 
 import (
+        "flag"
         "fmt"
         "net/http"
         "os"
-
-        "github.com/pborman/getopt"
 )
 
 var base_link string
+var base_aplus_link string
+var external_tools_code int
 var token string
 var client http.Client
 
+func usage() {
+        fmt.Println("Usage: aplus [--code attendance_code --course course_code] [--list] [--listfav] [--help]")
+}
+
 func main() {
         initialize()
 
-        code := getopt.StringLong("code", 'c', "", "Code to enter")
-        course := getopt.IntLong("course", 'C', -1, "Canvas course number")
-        list_all := getopt.BoolLong("list-all", 'L', "List all courses")
-        list_favorites := getopt.BoolLong("list-favorites", 'F', "List favorite courses")
-        help := getopt.BoolLong("help", 'h', "Help")
+        var code string
+        var course int
+        var listall bool
+        var listfav bool
+        var help bool
 
-        getopt.Parse()
+        flag.StringVar(&code, "code", "", "5 character attendance code")
+        flag.StringVar(&code, "c", "", "5 character attendance code")
+        flag.IntVar(&course, "course", -1, "canvas course code")
+        flag.IntVar(&course, "C", -1, "canvas course code")
+        flag.BoolVar(&listall, "list", false, "list all canvas courses")
+        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(&help, "help", false, "view usage message")
+        flag.BoolVar(&help, "h", false, "view usage message")
 
-        if *help || (*list_all == false && *list_favorites == false && *code == "" && *course == -1) {
-                getopt.Usage()
-                os.Exit(0)
+        flag.Parse()
+
+        if (code == "" && course == -1 && !listall && !listfav) {
+                usage()
+                os.Exit(1)
         }
 
-        if *list_all {
+        if (listall) {
                 list_all_courses()
+                os.Exit(0)
         }
 
-        if *list_favorites {
+        if (listfav) {
                 list_favorite_courses()
+                os.Exit(0)
+        }
+
+        if (help) {
+                usage()
+                os.Exit(0)
+        }
+
+        if (len(code) != 5) {
+                fmt.Println("Attendance code must be 5 characters long")
+                os.Exit(1)
         }
 
-        if *code != "" && *course != -1 {
-                fmt.Printf("%d %s", *course, *code)
-                submit_code(*course, *code)
+        if code != "" && course != -1 {
+                fmt.Printf("%d %s\n", course, code)
+                submit_code(course, code)
+        } else {
+                usage()
+                os.Exit(1)
         }
 }