Merge pull request #30 from okx/fix/support_media

Fix/support media
okx · May 11, 2023 · 7cdbb30 · 7cdbb30
2 parents 302c5da + 1d2ed18
commit 7cdbb30
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/src/brc20/error.rs b/src/brc20/error.rs
@@ -13,6 +13,12 @@ pub enum Error<L: LedgerRead> {
 
 #[derive(Debug, PartialEq, thiserror::Error)]
 pub enum JSONError {
+  #[error("invalid content type")]
+  InvalidContentType,
+
+  #[error("unsupport content type")]
+  UnSupportContentType,
+
   #[error("invalid json string")]
   InvalidJson,
 

diff --git a/src/brc20/mod.rs b/src/brc20/mod.rs
@@ -19,8 +19,17 @@ pub use self::{
 
 use ledger::{LedgerRead, LedgerReadWrite};
 
+const MEDIA_TYPE_TEXT: &str = "text/plain;charset=utf-8";
+const MEDIA_TYPE_JSON: &str = "application/json";
+
 pub fn deserialize_brc20_operation(inscription: Inscription) -> Result<Operation> {
-  Ok(deserialize_brc20(std::str::from_utf8(
-    inscription.body().ok_or(JSONError::InvalidJson)?,
-  )?)?)
+  match inscription
+    .content_type()
+    .ok_or(JSONError::InvalidContentType)?
+  {
+    MEDIA_TYPE_TEXT | MEDIA_TYPE_JSON => Ok(deserialize_brc20(std::str::from_utf8(
+      inscription.body().ok_or(JSONError::InvalidJson)?,
+    )?)?),
+    &_ => Err(JSONError::UnSupportContentType.into()),
+  }
 }