package main import ( "flag" "fmt" "io/ioutil" "log" "math/rand" "net/http" "net/url" "os" "strconv" "time" "git.coolaj86.com/coolaj86/go-mockid/mockid" "git.coolaj86.com/coolaj86/go-mockid/xkeypairs" "git.rootprojects.org/root/keypairs" _ "github.com/joho/godotenv/autoload" ) func main() { done := make(chan bool) var port int var host string rand.Seed(time.Now().UnixNano()) portFlag := flag.Int("port", 0, "Port on which the HTTP server should run") urlFlag := flag.String("url", "", "Outward-facing address, such as https://example.com") prefixFlag := flag.String("jwkspath", "", "The path to the JWKs storage directory") flag.Parse() if nil != portFlag && *portFlag > 0 { port = *portFlag } else { portStr := os.Getenv("PORT") port, _ = strconv.Atoi(portStr) } if port < 1 { fmt.Fprintf(os.Stderr, "You must specify --port or PORT\n") os.Exit(1) } jwkpath := "./default.jwk.json" jwkb, err := ioutil.ReadFile(jwkpath) if nil != err { panic(fmt.Errorf("read default jwk %v: %w", jwkpath, err)) return } privkey, err := keypairs.ParseJWKPrivateKey(jwkb) if nil != err { // TODO delete the bad file? panic(fmt.Errorf("unmarshal jwk %v: %w", string(jwkb), err)) return } if nil != urlFlag && "" != *urlFlag { host = *urlFlag } else { host = "http://localhost:" + strconv.Itoa(port) } var jwksPrefix string if nil != prefixFlag && "" != *prefixFlag { jwksPrefix = *prefixFlag } else { jwksPrefix = "public-jwks" } err = os.MkdirAll(jwksPrefix, 0755) if nil != err { fmt.Fprintf(os.Stderr, "couldn't write %q: %s", jwksPrefix, err) os.Exit(1) } mux := mockid.Route(jwksPrefix, privkey) fs := http.FileServer(http.Dir("./public")) http.Handle("/", fs) /* http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { log.Printf(r.Method, r.URL.Path) http.Error(w, "Not Found", http.StatusNotFound) }) */ fmt.Printf("Serving on port %d\n", port) go func() { log.Fatal(http.ListenAndServe(":"+strconv.Itoa(port), mux)) done <- true }() // TODO privB := keypairs.MarshalJWKPrivateKey(privkey) privB := xkeypairs.MarshalJWKPrivateKey(privkey) fmt.Printf("Private Key:\n\t%s\n", string(privB)) pubB := keypairs.MarshalJWKPublicKey(keypairs.NewPublicKey(privkey.Public())) fmt.Printf("Public Key:\n\t%s\n", string(pubB)) protected, payload, token := mockid.GenToken(host, privkey, url.Values{}) fmt.Printf("Protected (Header):\n\t%s\n", protected) fmt.Printf("Payload (Claims):\n\t%s\n", payload) fmt.Printf("Access Token:\n\t%s\n", token) <-done }