Fix private key base64 decode (#30)

* Decode the vapid private key in URL or Raw base64 encoding

* Decode and encode VAPID public key

README.md

@@ -28,7 +28,7 @@ func main() {
 	json.Unmarshal([]byte("<YOUR_SUBSCRIPTION>"), s)
 
 	// Send Notification
-	_, err := webpush.SendNotification([]byte("Test"), s, &webpush.Options{
+	resp, err := webpush.SendNotification([]byte("Test"), s, &webpush.Options{
 		Subscriber:      "[email protected]",
 		VAPIDPublicKey:  "<YOUR_VAPID_PUBLIC_KEY>",
 		VAPIDPrivateKey: "<YOUR_VAPID_PRIVATE_KEY>",
@@ -37,6 +37,7 @@ func main() {
 	if err != nil {
 		// TODO: Handle error
 	}
+	defer resp.Body.Close()
 }
 ```
 

example/main.go

@@ -18,7 +18,7 @@ func main() {
 	json.Unmarshal([]byte(subscription), s)
 
 	// Send Notification
-	_, err := webpush.SendNotification([]byte("Test"), s, &webpush.Options{
+	resp, err := webpush.SendNotification([]byte("Test"), s, &webpush.Options{
 		Subscriber:      "[email protected]", // Do not include "mailto:"
 		VAPIDPublicKey:  vapidPublicKey,
 		VAPIDPrivateKey: vapidPrivateKey,
@@ -27,4 +27,5 @@ func main() {
 	if err != nil {
 		// TODO: Handle error
 	}
+	defer resp.Body.Close()
 }

go.mod

@@ -4,3 +4,5 @@ require (
 	github.com/dgrijalva/jwt-go v3.2.0+incompatible
 	golang.org/x/crypto v0.0.0-20190131182504-b8fe1690c613
 )
+
+go 1.13

vapid.go

@@ -77,8 +77,8 @@ func getVAPIDAuthorizationHeader(
 		"sub": fmt.Sprintf("mailto:%s", subscriber),
 	})
 
-	// ECDSA
-	decodedVapidPrivateKey, err := base64.RawURLEncoding.DecodeString(vapidPrivateKey)
+	// Decode the VAPID private key
+	decodedVapidPrivateKey, err := decodeVapidKey(vapidPrivateKey)
 	if err != nil {
 		return "", err
 	}
@@ -91,9 +91,26 @@ func getVAPIDAuthorizationHeader(
 		return "", err
 	}
 
+	// Decode the VAPID public key
+	pubKey, err := decodeVapidKey(vapidPublicKey)
+	if err != nil {
+		return "", err
+	}
+
 	return fmt.Sprintf(
 		"vapid t=%s, k=%s",
 		jwtString,
-		vapidPublicKey,
+		base64.RawURLEncoding.EncodeToString(pubKey),
 	), nil
 }
+
+// Need to decode the vapid private key in multiple base64 formats
+// Solution from: https://github.com/SherClockHolmes/webpush-go/issues/29
+func decodeVapidKey(key string) ([]byte, error) {
+	bytes, err := base64.URLEncoding.DecodeString(key)
+	if err == nil {
+		return bytes, nil
+	}
+
+	return base64.RawURLEncoding.DecodeString(key)
+}

webpush.go

@@ -229,6 +229,7 @@ func decodeSubscriptionKey(key string) ([]byte, error) {
 	if err == nil {
 		return bytes, nil
 	}
+
 	return base64.URLEncoding.DecodeString(buf.String())
 }