refactor: polish code organization

axonweb3 · Nov 1, 2023 · bbd824f · bbd824f
1 parent 6edeb43
commit bbd824f
Show file tree

Hide file tree

Showing 4 changed files with 81 additions and 78 deletions.
diff --git a/devtools/axon-tools/src/error.rs b/devtools/axon-tools/src/error.rs
@@ -1,3 +1,5 @@
+use derive_more::{Display, From};
+use ethereum_types::H256;
 use std::fmt::{self, Display};
 
 #[allow(dead_code)]
@@ -61,3 +63,59 @@ impl Display for Error {
         }
     }
 }
+
+#[derive(Debug, Display, From)]
+pub enum TypesError {
+    #[display(fmt = "Expect {:?}, get {:?}.", expect, real)]
+    LengthMismatch { expect: usize, real: usize },
+
+    #[display(
+        fmt = "Eip1559Transaction hash mismatch origin {:?}, computed {:?}",
+        origin,
+        calc
+    )]
+    TxHashMismatch { origin: H256, calc: H256 },
+
+    #[display(fmt = "{:?}", _0)]
+    FromHex(faster_hex::Error),
+
+    #[display(fmt = "{:?} is an invalid address", _0)]
+    InvalidAddress(String),
+
+    #[display(fmt = "Hex should start with 0x")]
+    HexPrefix,
+
+    #[display(fmt = "Invalid public key")]
+    InvalidPublicKey,
+
+    #[display(fmt = "Invalid check sum")]
+    InvalidCheckSum,
+
+    #[display(fmt = "Unsigned")]
+    Unsigned,
+
+    // #[display(fmt = "Crypto error {:?}", _0)]
+    // Crypto(CryptoError),
+    #[display(fmt = "Missing signature")]
+    MissingSignature,
+
+    #[display(fmt = "Invalid crosschain direction")]
+    InvalidDirection,
+
+    #[display(fmt = "Signature R is empty")]
+    SignatureRIsEmpty,
+
+    #[display(fmt = "Invalid signature R type")]
+    InvalidSignatureRType,
+
+    #[display(fmt = "Invalid address source type")]
+    InvalidAddressSourceType,
+
+    #[display(fmt = "Missing interoperation sender")]
+    MissingInteroperationSender,
+
+    #[display(fmt = "InvalidBlockVersion {:?}", _0)]
+    InvalidBlockVersion(u8),
+}
+
+impl std::error::Error for TypesError {}
diff --git a/devtools/axon-tools/src/lib.rs b/devtools/axon-tools/src/lib.rs
@@ -10,6 +10,7 @@ pub mod hash;
 pub mod hex;
 #[cfg(feature = "proof")]
 mod proof;
+mod rlp_codec;
 pub mod types;
 
 pub use error::Error;

diff --git a/devtools/axon-tools/src/rlp_codec.rs b/devtools/axon-tools/src/rlp_codec.rs
@@ -0,0 +1,18 @@
+use crate::types::BlockVersion;
+#[cfg(feature = "impl-rlp")]
+use rlp::{Decodable, DecoderError, Encodable, Rlp, RlpStream};
+
+impl Encodable for BlockVersion {
+    fn rlp_append(&self, s: &mut RlpStream) {
+        let ver: u8 = (*self).into();
+        s.begin_list(1).append(&ver);
+    }
+}
+
+impl Decodable for BlockVersion {
+    fn decode(r: &Rlp) -> Result<Self, DecoderError> {
+        let ver: u8 = r.val_at(0)?;
+        ver.try_into()
+            .map_err(|_| DecoderError::Custom("Invalid block version"))
+    }
+}
diff --git a/devtools/axon-tools/src/types.rs b/devtools/axon-tools/src/types.rs
@@ -1,13 +1,10 @@
-use core::cmp::Ordering;
-
+use crate::error::TypesError;
 use alloc::vec::Vec;
 use bytes::{Bytes, BytesMut};
+use core::cmp::Ordering;
 use core::str::FromStr;
-use derive_more::{Display, From};
+use ethereum_types::{Bloom, H160, H256, U256};
 use faster_hex::withpfx_lowercase;
-use rlp::{Decodable, DecoderError, Rlp};
-
-pub use ethereum_types::{Bloom, H160, H256, H64, U256};
 
 #[cfg(feature = "impl-serde")]
 use serde::{Deserialize, Serialize};
@@ -111,62 +108,6 @@ impl<'de> Deserialize<'de> for Hex {
     }
 }
 
-#[derive(Debug, Display, From)]
-pub enum TypesError {
-    #[display(fmt = "Expect {:?}, get {:?}.", expect, real)]
-    LengthMismatch { expect: usize, real: usize },
-
-    #[display(
-        fmt = "Eip1559Transaction hash mismatch origin {:?}, computed {:?}",
-        origin,
-        calc
-    )]
-    TxHashMismatch { origin: H256, calc: H256 },
-
-    #[display(fmt = "{:?}", _0)]
-    FromHex(faster_hex::Error),
-
-    #[display(fmt = "{:?} is an invalid address", _0)]
-    InvalidAddress(String),
-
-    #[display(fmt = "Hex should start with 0x")]
-    HexPrefix,
-
-    #[display(fmt = "Invalid public key")]
-    InvalidPublicKey,
-
-    #[display(fmt = "Invalid check sum")]
-    InvalidCheckSum,
-
-    #[display(fmt = "Unsigned")]
-    Unsigned,
-
-    // #[display(fmt = "Crypto error {:?}", _0)]
-    // Crypto(CryptoError),
-    #[display(fmt = "Missing signature")]
-    MissingSignature,
-
-    #[display(fmt = "Invalid crosschain direction")]
-    InvalidDirection,
-
-    #[display(fmt = "Signature R is empty")]
-    SignatureRIsEmpty,
-
-    #[display(fmt = "Invalid signature R type")]
-    InvalidSignatureRType,
-
-    #[display(fmt = "Invalid address source type")]
-    InvalidAddressSourceType,
-
-    #[display(fmt = "Missing interoperation sender")]
-    MissingInteroperationSender,
-
-    #[display(fmt = "InvalidBlockVersion {:?}", _0)]
-    InvalidBlockVersion(u8),
-}
-
-impl std::error::Error for TypesError {}
-
 #[derive(Default, Copy, Clone, Debug, PartialEq, Eq)]
 #[cfg(feature = "impl-serde")]
 #[derive(Serialize, Deserialize)]
@@ -186,29 +127,14 @@ impl From<BlockVersion> for u8 {
 impl TryFrom<u8> for BlockVersion {
     type Error = TypesError;
 
-    fn try_from(value: u8) -> Result<Self, Self::Error> {
+    fn try_from(value: u8) -> Result<Self, TypesError> {
         match value {
             0 => Ok(BlockVersion::V0),
             _ => Err(TypesError::InvalidBlockVersion(value)),
         }
     }
 }
 
-impl Encodable for BlockVersion {
-    fn rlp_append(&self, s: &mut RlpStream) {
-        let ver: u8 = (*self).into();
-        s.begin_list(1).append(&ver);
-    }
-}
-
-impl Decodable for BlockVersion {
-    fn decode(r: &Rlp) -> Result<Self, DecoderError> {
-        let ver: u8 = r.val_at(0)?;
-        ver.try_into()
-            .map_err(|_| DecoderError::Custom("Invalid block version"))
-    }
-}
-
 pub type Hash = H256;
 pub type MerkleRoot = Hash;
 pub type BlockNumber = u64;