chore: refactor

LimeChain · Jan 27, 2025 · f316737 · f316737
1 parent 4702387
commit f316737
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 9 deletions.
diff --git a/src/main/java/com/limechain/babe/Authorship.java b/src/main/java/com/limechain/babe/Authorship.java
@@ -242,11 +242,8 @@ private static Map<Integer, Schnorrkel.KeyPair> getOwnedKeyPairsFromAuthoritySet
 
         authorities.forEach(a -> keyStore.getKeyPair(KeyType.BABE, a.getPublicKey())
                 .ifPresent(keyPair -> {
-                            indexKeyPairMap.put(
-                                    authorities.indexOf(a),
-                                    keyStore.convertToSchnorrKeypair(keyPair));
-                        }
-                ));
+                    indexKeyPairMap.put(authorities.indexOf(a), keyStore.convertToSchnorrKeypair(keyPair));
+                }));
 
         return indexKeyPairMap;
     }

diff --git a/src/main/java/com/limechain/storage/crypto/KeyStore.java b/src/main/java/com/limechain/storage/crypto/KeyStore.java
@@ -51,20 +51,20 @@ public List<byte[]> getPublicKeysByKeyType(KeyType keyType) {
      *
      * @param type      the algorithm type that the key is used for.
      * @param publicKey the pubKey used as a key to retrieve a privKey from the store.
-     * @return Pair of (privateKey, publicKey)
+     * @return Pair of (publicKey, privateKey)
      */
     public Optional<Pair<byte[], byte[]>> getKeyPair(KeyType type, byte[] publicKey) {
         var privateKey = get(type, publicKey);
         if (privateKey != null) {
-            return Optional.of(new Pair<>(privateKey, publicKey));
+            return Optional.of(new Pair<>(publicKey, privateKey));
         }
 
         return Optional.empty();
     }
 
     public Schnorrkel.KeyPair convertToSchnorrKeypair(Pair<byte[], byte[]> pair) {
-        Schnorrkel.PublicKey pubKey = new Schnorrkel.PublicKey(pair.getValue1());
-        return new Schnorrkel.KeyPair(pubKey, pair.getValue0());
+        Schnorrkel.PublicKey pubKey = new Schnorrkel.PublicKey(pair.getValue0());
+        return new Schnorrkel.KeyPair(pubKey, pair.getValue1());
     }
 
     private byte[] removeKeyTypeFromKey(byte[] key) {