From 659d1674b344ce797374b07fc79af3855d6ee0db Mon Sep 17 00:00:00 2001
From: Michal Kucharczyk <1728078+michalkucharczyk@users.noreply.github.com>
Date: Mon, 30 Oct 2023 17:58:57 +0100
Subject: [PATCH] Support for no-std (#18)

- add support for no_std,
- bump version to 0.4.5
---
 Cargo.toml |  6 +++---
 src/lib.rs | 40 ++++++++++++++++++++++++++++++++--------
 2 files changed, 35 insertions(+), 11 deletions(-)

diff --git a/Cargo.toml b/Cargo.toml
index a8e067e9a1db..b293816f1a3e 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "substrate-bip39"
-version = "0.4.4"
+version = "0.4.5"
 authors = ["Parity Technologies <admin@parity.io>"]
 license = "Apache-2.0"
 edition = "2021"
@@ -10,9 +10,9 @@ repository = "https://github.com/paritytech/substrate-bip39"
 
 [dependencies]
 pbkdf2 = { version = "0.8.0", default-features = false }
-sha2 = "0.9.5"
+sha2 = { version = "0.9.5", default-features = false }
 hmac = "0.11.0"
-schnorrkel = "0.9.1"
+schnorrkel = { version = "0.9.1", default-features = false }
 zeroize = { version = "1.0.0", default-features = false }
 
 [dev-dependencies]
diff --git a/src/lib.rs b/src/lib.rs
index ac025e4069b0..f813f882648f 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,5 +1,5 @@
 // Copyright 2019-2020 Parity Technologies (UK) Ltd.
-// 
+//
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
 // You may obtain a copy of the License at
@@ -12,10 +12,17 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-use sha2::Sha512;
+#![cfg_attr(not(feature = "std"), no_std)]
+
+#[cfg(not(feature = "std"))]
+extern crate alloc;
+#[cfg(not(feature = "std"))]
+use alloc::string::String;
+
 use hmac::Hmac;
 use pbkdf2::pbkdf2;
 use schnorrkel::keys::MiniSecretKey;
+use sha2::Sha512;
 use zeroize::Zeroize;
 
 #[derive(Clone, Copy, PartialEq, Eq, Debug)]
@@ -63,7 +70,7 @@ pub fn seed_from_entropy(entropy: &[u8], password: &str) -> Result<[u8; 64], Err
 #[cfg(test)]
 mod test {
     use super::*;
-    use bip39::{Mnemonic, Language};
+    use bip39::{Language, Mnemonic};
     use rustc_hex::FromHex;
 
     // phrase, entropy, seed, expanded secret_key
@@ -202,11 +209,28 @@ mod test {
 
             let mnemonic = Mnemonic::from_phrase(phrase, Language::English).unwrap();
             let seed = seed_from_entropy(mnemonic.entropy(), "Substrate").unwrap();
-            let secret = mini_secret_from_entropy(mnemonic.entropy(), "Substrate").unwrap().to_bytes();
-
-            assert_eq!(mnemonic.entropy(), &expected_entropy[..], "Entropy is incorrect for {}", phrase);
-            assert_eq!(&seed[..], &expected_seed[..], "Seed is incorrect for {}", phrase);
-            assert_eq!(&secret[..], &expected_seed[..32], "Secret is incorrect for {}", phrase);
+            let secret = mini_secret_from_entropy(mnemonic.entropy(), "Substrate")
+                .unwrap()
+                .to_bytes();
+
+            assert_eq!(
+                mnemonic.entropy(),
+                &expected_entropy[..],
+                "Entropy is incorrect for {}",
+                phrase
+            );
+            assert_eq!(
+                &seed[..],
+                &expected_seed[..],
+                "Seed is incorrect for {}",
+                phrase
+            );
+            assert_eq!(
+                &secret[..],
+                &expected_seed[..32],
+                "Secret is incorrect for {}",
+                phrase
+            );
         }
     }
 }