feat: implement fetch jwk

x/zkauth/keeper/keeper.go

@@ -7,6 +7,7 @@
 	"net/http"
 	"os"
 	"path/filepath"
+	"strings"
 	"time"
 
 	"github.com/Finschia/finschia-sdk/codec"
@@ -45,8 +46,8 @@
 	go func() {
 		for {
 			select {
 			case <-ctx.Context().Done():
 				return
 			default:
 				for _, name := range defaultZKAuthOAuthProviders {
 					provider := types.GetConfig(name)
@@ -54,24 +55,23 @@
 
 					resp, err := k.GetJWK(provider.JwkEndpoint)
 					if err != nil {
 						time.Sleep(time.Duration(fetchIntervals) * time.Second)
 						logger.Error(fmt.Sprintf("%s", err))
 						continue
 					}
 
-					file, err := os.Create(filepath.Join(nodeHome, JwkFileName))
+					file, err := os.Create(filepath.Join(nodeHome, k.CreateJWKFileName(name)))
 					if err != nil {
 						time.Sleep(time.Duration(fetchIntervals) * time.Second)
 						logger.Error(fmt.Sprintf("%s", err))
-						fmt.Println(err)
 						continue
 					}
 
 					_, err = io.Copy(file, resp.Body)
 					if err != nil {
 						time.Sleep(time.Duration(fetchIntervals) * time.Second)
 						logger.Error(fmt.Sprintf("%s", err))
 						continue
 					}
 
 					resp.Body.Close()
@@ -85,19 +85,22 @@
 
 func (k Keeper) ParseJWKs(byteArray []byte) (jwks []types.JWK, err error) {
 	var data map[string]interface{}
-	json.Unmarshal(byteArray, &data)
+	err = json.Unmarshal(byteArray, &data)
+	if err != nil {
+		return jwks, err
+	}
 
 	for _, v := range data["keys"].([]interface{}) {
 		var jwk types.JWK
 
 		b, err := json.Marshal(v)
 		if err != nil {
 			return jwks, err
 		}
 
 		if err := json.Unmarshal(b, &jwk); err != nil {
 			return jwks, err
 		}
 
 		jwks = append(jwks, jwk)
 	}
@@ -108,13 +111,18 @@
 func (k Keeper) GetJWK(endpoint string) (*http.Response, error) {
 	req, err := http.NewRequest("GET", endpoint, nil)
 	if err != nil {
 		return nil, err
 	}
 	client := new(http.Client)
 	resp, err := client.Do(req)
 	if err != nil {
 		return nil, err
 	}
 
 	return resp, nil
 }
+
+func (k Keeper) CreateJWKFileName(name types.OidcProvider) string {
+	fileNamePattern := strings.Replace(JwkFileName, ".", "-%s.", 1)
+	return fmt.Sprintf(fileNamePattern, name)
+}

x/zkauth/keeper/keeper_test.go

@@ -3,15 +3,14 @@ package keeper_test
 import (
 	"context"
 	"encoding/json"
-	"io/ioutil"
+	"io"
 	"net/http"
 	"net/http/httptest"
 	"os"
 	"path/filepath"
 	"testing"
 	"time"
 
-	"github.com/Finschia/finschia-sdk/x/zkauth/keeper"
 	"github.com/Finschia/finschia-sdk/x/zkauth/testutil"
 	"github.com/Finschia/finschia-sdk/x/zkauth/types"
 	"github.com/stretchr/testify/require"
@@ -63,7 +62,7 @@ func TestGetJWK(t *testing.T) {
 	require.NoError(t, err)
 
 	expected := testData
-	bodyBytes, err := ioutil.ReadAll(res.Body)
+	bodyBytes, err := io.ReadAll(res.Body)
 	bodyString := string(bodyBytes)
 	require.Equal(t, expected, bodyString)
 }
@@ -77,7 +76,7 @@ func TestParseJWKs(t *testing.T) {
 	defer res.Body.Close()
 	require.NoError(t, err)
 
-	bodyBytes, err := ioutil.ReadAll(res.Body)
+	bodyBytes, err := io.ReadAll(res.Body)
 
 	jwks, err := k.ParseJWKs(bodyBytes)
 	require.NoError(t, err)
@@ -92,14 +91,14 @@ func TestFetchJwk(t *testing.T) {
 	timeoutCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
 	defer cancel()
 
-	tempDir, err := ioutil.TempDir("", types.StoreKey)
+	tempDir, err := os.MkdirTemp("", types.StoreKey)
 	require.NoError(t, err)
 	defer os.RemoveAll(tempDir)
 
 	k.FetchJWK(ctx.WithContext(timeoutCtx), tempDir)
 	<-timeoutCtx.Done()
 
-	content, err := ioutil.ReadFile(filepath.Join(tempDir, keeper.JwkFileName))
+	content, err := os.ReadFile(filepath.Join(tempDir, k.CreateJWKFileName(types.Google)))
 
 	require.NoError(t, err)
 	var expectedObj []types.JWK