# HG changeset patch
# User sqwishy <sqwishybon@gmail.com>
# Date 1392672707 28800
#      Mon Feb 17 13:31:47 2014 -0800
# Node ID 8d4983409d85efaa54c1d82a754bd8df9737d518
# Parent  2e9669d50a26f8debcc37a8e918dac79cf1cd574
introducing flags!

diff --git a/main.go b/main.go
--- a/main.go
+++ b/main.go
@@ -1,12 +1,17 @@
 package main
 
+import "flag"
 import "fmt"
 import "log"
+import "os"
 import "./gorcon" // Relative imports! Woo hoo!
 
 var _ = fmt.Println
 var _ = log.Println
 
+var serverFlag = flag.String("server", "", "Address and port of rcon server (ex. 127.0.0.1:27015)")
+var passwdFlag = flag.String("password", "", "Password for rcon authentication")
+
 type response struct {
     Id  int32
     Type gorcon.CmdType
@@ -14,14 +19,24 @@
 }
 
 func main() {
+    flag.Parse()
+
+    if *serverFlag == "" {
+        fmt.Fprintf(os.Stderr, "Invalid value for -server flag %s:\n", os.Args[0])
+        flag.Usage()
+        return
+    }
+
     log.Println("Hello, world!")
 
-    client, err := gorcon.NewClient("74.91.117.120:27017")
+    client, err := gorcon.NewClient(*serverFlag)
     if err != nil {
         log.Panicln(err)
     }
 
-    errch := client.Auth("hunter2")
+    log.Println("Connected to server")
+
+    errch := client.Auth(*passwdFlag)
     select {
     case err := <-errch:
         if err != nil {