view cmd/tokenserver/main.go @ 4:ac5529cb20ba

Added API as routing prefix and change default port to 8000 in order to improve integration with frontend app.
author Thomas Junk <thomas.junk@intevation.de>
date Wed, 13 Jun 2018 11:20:49 +0200
parents 9c6f68a8e8b2
children 1d1236b72e5e
line wrap: on
line source

package main

import (
	"flag"
	"fmt"
	"log"
	"net/http"
	"time"

	jwt "github.com/dgrijalva/jwt-go"
)

func token(rw http.ResponseWriter, req *http.Request) {
	user := req.FormValue("user")
	password := req.FormValue("password")

	_ = password

	eol := time.Now().Add(45 * time.Minute)

	token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{
		"user": user,
		"eol":  eol.Unix(),
	})

	signingKey := []byte("very, very secret!")

	tokenString, err := token.SignedString(signingKey)

	if err != nil {
		http.Error(rw, "Signing failed", http.StatusInternalServerError)
		return
	}

	rw.Header().Set("Content-Type", "text/plain")

	fmt.Fprintf(rw, "%s\n", tokenString)
}

func index(rw http.ResponseWriter, req *http.Request) {
	fmt.Fprintln(rw, "I was here!")
}

func main() {
	port := flag.Int("port", 8000, "port to listen at.")
	host := flag.String("host", "localhost", "host to listen at.")
	flag.Parse()

	mux := http.NewServeMux()
	mux.HandleFunc("/", index)
	mux.HandleFunc("/token", token)
	api := http.NewServeMux()
	api.Handle("/api", mux)

	addr := fmt.Sprintf("%s:%d", *host, *port)
	log.Fatalln(http.ListenAndServe(addr, api))
}