Browse Source

Updates to example

Bruce Marriner 8 years ago
parent
commit
68ebab258b
1 changed files with 16 additions and 14 deletions
  1. 16 14
      examples/mytoken/main.go

+ 16 - 14
examples/mytoken/main.go

@@ -1,28 +1,30 @@
-// This is an example of using DiscordGo to obtain the
-// authentication token for a given user account.
 package main
 package main
 
 
 import (
 import (
+	"flag"
 	"fmt"
 	"fmt"
-	"os"
 
 
 	"github.com/bwmarrin/discordgo"
 	"github.com/bwmarrin/discordgo"
 )
 )
 
 
-func main() {
+var (
+	Email    string
+	Password string
+)
 
 
-	// Check for Username and Password CLI arguments.
-	if len(os.Args) != 3 {
-		fmt.Println("You must provide username and password as arguments. See below example.")
-		fmt.Println(os.Args[0], " [email] [password]")
-		return
-	}
+func init() {
+
+	flag.StringVar(&Email, "e", "", "Account Email")
+	flag.StringVar(&Password, "p", "", "Account Password")
+	flag.Parse()
+}
+
+func main() {
 
 
-	// Create a New Discord session and login with the provided
-	// email and password.
-	dg, err := discordgo.New(os.Args[1], os.Args[2])
+	// Create a new Discord session using the provided login information.
+	dg, err := discordgo.New(Email, Password)
 	if err != nil {
 	if err != nil {
-		fmt.Println(err)
+		fmt.Println("error creating Discord session,", err)
 		return
 		return
 	}
 	}