Add jwt create with config.

mobingilabs · Jan 25, 2018 · 269ab47 · 269ab47
1 parent 794248f
commit 269ab47
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 5 deletions.
diff --git a/Gopkg.lock b/Gopkg.lock
diff --git a/pkg/jwt/jwt.go b/pkg/jwt/jwt.go
@@ -97,3 +97,35 @@ func NewCtx(pemdir ...string) (*jwtctx, error) {
 
 	return &ctx, nil
 }
+
+type Config struct {
+	PublicTokenFile  string
+	PrivateTokenFile string
+}
+
+func NewCtxWithConfig(cnf *Config) (*jwtctx, error) {
+	if cnf == nil {
+		return nil, fmt.Errorf("config cannot be nil")
+	}
+
+	pubcache, err := ioutil.ReadFile(cnf.PublicTokenFile)
+	if err != nil {
+		debug.Error(err)
+		return nil, errors.Wrap(err, "pub readfile failed")
+	}
+
+	prvcache, err := ioutil.ReadFile(cnf.PrivateTokenFile)
+	if err != nil {
+		debug.Error(err)
+		return nil, errors.Wrap(err, "prv readfile failed")
+	}
+
+	ctx := jwtctx{
+		PemPub: cnf.PublicTokenFile,
+		PemPrv: cnf.PrivateTokenFile,
+		Pub:    pubcache,
+		Prv:    prvcache,
+	}
+
+	return &ctx, nil
+}