From cae6d59e193312c3817068553be88aa5db5f346a Mon Sep 17 00:00:00 2001
From: xenolf <xenolf@users.noreply.github.com>
Date: Wed, 18 Nov 2015 21:06:45 +0100
Subject: [PATCH] Move back to square/go-jose

---
 acme/jws.go      | 25 ++++++++++++++++---------
 acme/messages.go |  2 +-
 2 files changed, 17 insertions(+), 10 deletions(-)

diff --git a/acme/jws.go b/acme/jws.go
index 96f7ae9e..c5c496a1 100644
--- a/acme/jws.go
+++ b/acme/jws.go
@@ -4,10 +4,11 @@ import (
 	"bytes"
 	"crypto/ecdsa"
 	"crypto/rsa"
+	"errors"
 	"fmt"
 	"net/http"
 
-	"github.com/letsencrypt/go-jose"
+	"github.com/square/go-jose"
 )
 
 type jws struct {
@@ -15,10 +16,15 @@ type jws struct {
 	nonces  []string
 }
 
-func keyAsJWK(key *ecdsa.PublicKey) jose.JsonWebKey {
-	return jose.JsonWebKey{
-		Key:       key,
-		Algorithm: "EC",
+func keyAsJWK(key interface{}) *jose.JsonWebKey {
+	switch k := key.(type) {
+	case *ecdsa.PublicKey:
+		return &jose.JsonWebKey{Key: k, Algorithm: "EC"}
+	case *rsa.PublicKey:
+		return &jose.JsonWebKey{Key: k, Algorithm: "RSA"}
+
+	default:
+		return nil
 	}
 }
 
@@ -52,8 +58,9 @@ func (j *jws) signContent(content []byte) (*jose.JsonWebSignature, error) {
 	if err != nil {
 		return nil, err
 	}
+	signer.SetNonceSource(j)
 
-	signed, err := signer.Sign(content, j.consumeNonce())
+	signed, err := signer.Sign(content)
 	if err != nil {
 		return nil, err
 	}
@@ -79,12 +86,12 @@ func (j *jws) getNonce(url string) error {
 	return j.getNonceFromResponse(resp)
 }
 
-func (j *jws) consumeNonce() string {
+func (j *jws) Nonce() (string, error) {
 	nonce := ""
 	if len(j.nonces) == 0 {
-		return nonce
+		return nonce, errors.New("No nonce available.")
 	}
 
 	nonce, j.nonces = j.nonces[len(j.nonces)-1], j.nonces[:len(j.nonces)-1]
-	return nonce
+	return nonce, nil
 }
diff --git a/acme/messages.go b/acme/messages.go
index 948a86cc..857ec32a 100644
--- a/acme/messages.go
+++ b/acme/messages.go
@@ -3,7 +3,7 @@ package acme
 import (
 	"time"
 
-	"github.com/letsencrypt/go-jose"
+	"github.com/square/go-jose"
 )
 
 type directory struct {