view cmd/tokenserver/main.go @ 15:05d828374256

Reverted to previous setup of /api prefixing only the token route At present it seems best only to prefix the token route if we are planning to serve static files from a web folder
author Thomas Junk <thomas.junk@intevation.de>
date Thu, 21 Jun 2018 09:20:39 +0200
parents afc0de09231f
children a98a282f00e1
line wrap: on
line source

package main

import (
	"flag"
	"fmt"
	"log"
	"net/http"
	"path/filepath"
	"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 main() {
	port := flag.Int("port", 8000, "port to listen at.")
	host := flag.String("host", "localhost", "host to listen at.")
	flag.Parse()
	p, _ := filepath.Abs("./web")
	mux := http.NewServeMux()
	mux.Handle("/", http.StripPrefix("/", http.FileServer(http.Dir(p))))
	mux.HandleFunc("/api/token", token)

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