Time to get back to work - starting with some simple comment fixes.

pull/24/head
unknown 2015-01-28 17:56:56 -06:00
parent 33b9ec2196
commit d567153d2a
2 changed files with 9 additions and 4 deletions

View File

@ -28,7 +28,6 @@ func CreateAdminRouter() http.Handler {
router.HandleFunc("/register", Register) router.HandleFunc("/register", Register)
router.HandleFunc("/", Use(Base, mid.RequireLogin)) router.HandleFunc("/", Use(Base, mid.RequireLogin))
router.HandleFunc("/settings", Use(Settings, mid.RequireLogin)) router.HandleFunc("/settings", Use(Settings, mid.RequireLogin))
// Create the API routes // Create the API routes
api := router.PathPrefix("/api").Subrouter() api := router.PathPrefix("/api").Subrouter()
api = api.StrictSlash(true) api = api.StrictSlash(true)
@ -56,7 +55,7 @@ func CreateAdminRouter() http.Handler {
return Use(csrfHandler.ServeHTTP, mid.GetContext) return Use(csrfHandler.ServeHTTP, mid.GetContext)
} }
//CreateEndpointRouter creates the router that handles phishing connections. // CreatePhishingRouter creates the router that handles phishing connections.
func CreatePhishingRouter() http.Handler { func CreatePhishingRouter() http.Handler {
router := mux.NewRouter() router := mux.NewRouter()
router.PathPrefix("/static").Handler(http.FileServer(http.Dir("./static/endpoint/"))) router.PathPrefix("/static").Handler(http.FileServer(http.Dir("./static/endpoint/")))
@ -96,6 +95,7 @@ func Use(handler http.HandlerFunc, mid ...func(http.Handler) http.HandlerFunc) h
return handler return handler
} }
// Register creates a new user
func Register(w http.ResponseWriter, r *http.Request) { func Register(w http.ResponseWriter, r *http.Request) {
// If it is a post request, attempt to register the account // If it is a post request, attempt to register the account
// Now that we are all registered, we can log the user in // Now that we are all registered, we can log the user in
@ -142,6 +142,7 @@ func Register(w http.ResponseWriter, r *http.Request) {
} }
} }
// Logout destroys the current user session
func Logout(w http.ResponseWriter, r *http.Request) { func Logout(w http.ResponseWriter, r *http.Request) {
// If it is a post request, attempt to register the account // If it is a post request, attempt to register the account
// Now that we are all registered, we can log the user in // Now that we are all registered, we can log the user in
@ -151,6 +152,7 @@ func Logout(w http.ResponseWriter, r *http.Request) {
http.Redirect(w, r, "login", 302) http.Redirect(w, r, "login", 302)
} }
// Base handles the default path and template execution
func Base(w http.ResponseWriter, r *http.Request) { func Base(w http.ResponseWriter, r *http.Request) {
// Example of using session - will be removed. // Example of using session - will be removed.
params := struct { params := struct {
@ -162,6 +164,7 @@ func Base(w http.ResponseWriter, r *http.Request) {
getTemplate(w, "dashboard").ExecuteTemplate(w, "base", params) getTemplate(w, "dashboard").ExecuteTemplate(w, "base", params)
} }
// Settings handles the changing of settings
func Settings(w http.ResponseWriter, r *http.Request) { func Settings(w http.ResponseWriter, r *http.Request) {
switch { switch {
case r.Method == "POST": case r.Method == "POST":
@ -178,6 +181,8 @@ func Settings(w http.ResponseWriter, r *http.Request) {
} }
} }
// Login handles the authentication flow for a user. If credentials are valid,
// a session is created
func Login(w http.ResponseWriter, r *http.Request) { func Login(w http.ResponseWriter, r *http.Request) {
params := struct { params := struct {
User models.User User models.User

View File

@ -54,7 +54,7 @@
{{%template "flashes" .Flashes%}} {{%template "flashes" .Flashes%}}
<input type="text" name="username" class="form-control" placeholder="Username" required autofocus> <input type="text" name="username" class="form-control" placeholder="Username" required autofocus>
<input type="password" name="password" class="form-control" placeholder="Password" required> <input type="password" name="password" class="form-control" placeholder="Password" required>
<input type="hidden" name="csrf_token" value={{%.Token%}}/> <input type="hidden" name="csrf_token" value="{{%.Token%}}"/>
<button class="btn btn-lg btn-primary btn-block" type="submit">Sign in</button> <button class="btn btn-lg btn-primary btn-block" type="submit">Sign in</button>
</form> </form>
</div> </div>
@ -67,4 +67,4 @@
</body> </body>
</html> </html>
{{% end %}} {{% end %}}