From f70978760b309fae9a5529da6e023b2f4582ddbf Mon Sep 17 00:00:00 2001 From: Emile Date: Wed, 9 Oct 2019 01:03:23 +0200 Subject: added an endpoint displaying the user their credentials and redirecting them to their companion --- src/http.go | 68 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++--- 1 file changed, 65 insertions(+), 3 deletions(-) (limited to 'src') diff --git a/src/http.go b/src/http.go index 1b5db1a..0f81f37 100644 --- a/src/http.go +++ b/src/http.go @@ -1,8 +1,10 @@ package main import ( + "encoding/base64" "flag" "fmt" + "html/template" "io/ioutil" "log" "net/http" @@ -19,6 +21,12 @@ var ( companion map[string]string ) +// Credentials stores user credentials +type Credentials struct { + Username string + Accesscode string +} + func registerHTTPFlags() { port = flag.Int("port", 8081, "The port the http server should listen on") } @@ -29,6 +37,7 @@ func setupHTTPServer() http.Server { r.HandleFunc("/", indexHandler) r.HandleFunc("/register", registerGetHandler).Methods("GET") r.HandleFunc("/register", registerPostHandler).Methods("POST") + r.HandleFunc("/credentials", credentialsGetHandler).Methods("GET") return http.Server{ Addr: fmt.Sprintf("0.0.0.0:%d", *port), @@ -64,13 +73,28 @@ func registerPostHandler(w http.ResponseWriter, r *http.Request) { // generate a new accesscode accesscode := newAccessCode() - log.Printf("Generated a new AccessCode for user %s: %s", username, accesscode) + log.Printf("Generated a new AccessCode for user \"%s\": \"%s\"", username, accesscode) // generate a new companion spawnCompanion(username, accesscode) - // redirect the user to the front page - http.Redirect(w, r, "/", http.StatusTemporaryRedirect) + log.Println("---") + log.Println("Done generating the containers, filling the credentials page using a template") + + usernameBytes := []byte(username) + usernameBase64 := base64.StdEncoding.EncodeToString(usernameBytes) + + accesscodeBytes := []byte(accesscode) + accesscodeBase64 := base64.StdEncoding.EncodeToString(accesscodeBytes) + + // insert the username and the accesscode into the http request parameters + r.Form["username"] = []string{usernameBase64} + r.Form["accesscode"] = []string{accesscodeBase64} + + credentialsGetHandler(w, r) + //log.Println("redirecting to /credentials") + //endpoint := fmt.Sprintf("/credentials?username=%s&accesscode=%s", usernameBase64, accesscodeBase64) + //http.Redirect(w, r, endpoint, http.StatusSeeOther) } func usernameTakenGetHandler(w http.ResponseWriter, r *http.Request) { @@ -98,3 +122,41 @@ func readFileToReponse(w http.ResponseWriter, path string) { w.Write([]byte(contents)) } } + +func credentialsGetHandler(w http.ResponseWriter, r *http.Request) { + r.ParseForm() + usernameBase64 := r.Form.Get("username") + username, err := base64.StdEncoding.DecodeString(usernameBase64) + if err != nil { + fmt.Println("error decoding username base64:", err) + } + + accesscodeBase64 := r.Form.Get("accesscode") + accesscode, err := base64.StdEncoding.DecodeString(accesscodeBase64) + if err != nil { + fmt.Println("error decoding accesscode base64:", err) + } + + log.Println("[ ] Credentials GET Handler") + log.Println("%#v", r.Form) + log.Printf("username: %s", username) + log.Printf("accesscode: %s", accesscode) + + // create a new template reading the credentials template file + // the template then gets executed inserting the username and the accesscode + log.Println("creating new template") + t := template.New("") + log.Println("parsing template file") + t, err = t.ParseFiles("./hosted/credentials.html") + if err != nil { + log.Println(err) + } + log.Println("creating a credentials struct") + creds := Credentials{ + Username: string(username), + Accesscode: string(accesscode), + } + log.Println("executing the template") + t.ExecuteTemplate(w, "credentials", creds) + log.Println("done executing the template") +} -- cgit 1.4.1