From b427b2f11970cae8959d4a8b3a91674c89c9d948 Mon Sep 17 00:00:00 2001
From: Vlad Frolov <frolvlad@gmail.com>
Date: Sun, 4 Feb 2024 13:54:03 +0100
Subject: [PATCH] feat: Updated examples to support NEAR ABI by default (#1137)

---
 examples/mission-control/src/account.rs |  6 +++++-
 examples/mission-control/src/asset.rs   | 19 ++++++++++++++++++-
 2 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/examples/mission-control/src/account.rs b/examples/mission-control/src/account.rs
index ad7c16950..4c17f3a7c 100644
--- a/examples/mission-control/src/account.rs
+++ b/examples/mission-control/src/account.rs
@@ -1,6 +1,7 @@
 use crate::asset::*;
 use crate::rate::*;
 use near_sdk::borsh::{BorshDeserialize, BorshSerialize};
+use near_sdk::schemars::JsonSchema;
 use near_sdk::serde::{Deserialize, Serialize};
 use std::collections::HashMap;
 use std::ops;
@@ -17,14 +18,17 @@ use std::ops;
     Debug,
     BorshDeserialize,
     BorshSerialize,
+    JsonSchema,
 )]
 #[serde(crate = "near_sdk::serde")]
 #[borsh(crate = "near_sdk::borsh")]
+#[schemars(crate = "near_sdk::schemars")]
 pub struct Quantity(pub i32);
 
-#[derive(Clone, Serialize, Deserialize, BorshDeserialize, BorshSerialize)]
+#[derive(Clone, Serialize, Deserialize, BorshDeserialize, BorshSerialize, JsonSchema)]
 #[serde(crate = "near_sdk::serde")]
 #[borsh(crate = "near_sdk::borsh")]
+#[schemars(crate = "near_sdk::schemars")]
 pub struct Account(pub HashMap<Asset, Quantity>);
 
 pub enum Tranx {
diff --git a/examples/mission-control/src/asset.rs b/examples/mission-control/src/asset.rs
index 6d082a14c..35890214f 100644
--- a/examples/mission-control/src/asset.rs
+++ b/examples/mission-control/src/asset.rs
@@ -1,4 +1,5 @@
 use near_sdk::borsh::{BorshDeserialize, BorshSerialize};
+use near_sdk::schemars::JsonSchema;
 use near_sdk::serde::{Deserialize, Serialize};
 
 #[derive(
@@ -13,9 +14,11 @@ use near_sdk::serde::{Deserialize, Serialize};
     Deserialize,
     BorshDeserialize,
     BorshSerialize,
+    JsonSchema,
 )]
 #[serde(crate = "near_sdk::serde")]
 #[borsh(crate = "near_sdk::borsh")]
+#[schemars(crate = "near_sdk::schemars")]
 pub enum Resource {
     Battery,
     RgbSensor,
@@ -35,9 +38,11 @@ pub enum Resource {
     Deserialize,
     BorshDeserialize,
     BorshSerialize,
+    JsonSchema,
 )]
 #[serde(crate = "near_sdk::serde")]
 #[borsh(crate = "near_sdk::borsh")]
+#[schemars(crate = "near_sdk::schemars")]
 pub enum Reward {
     Score,
     Token,
@@ -58,9 +63,11 @@ pub enum Reward {
     Deserialize,
     BorshDeserialize,
     BorshSerialize,
+    JsonSchema,
 )]
 #[serde(crate = "near_sdk::serde")]
 #[borsh(crate = "near_sdk::borsh")]
+#[schemars(crate = "near_sdk::schemars")]
 pub enum Asset {
     Resource(Resource),
     Reward(Reward),
@@ -69,10 +76,20 @@ pub enum Asset {
 }
 
 #[derive(
-    PartialEq, Eq, Hash, Serialize, Deserialize, PartialOrd, Ord, BorshDeserialize, BorshSerialize,
+    PartialEq,
+    Eq,
+    Hash,
+    Serialize,
+    Deserialize,
+    PartialOrd,
+    Ord,
+    BorshDeserialize,
+    BorshSerialize,
+    JsonSchema,
 )]
 #[serde(crate = "near_sdk::serde")]
 #[borsh(crate = "near_sdk::borsh")]
+#[schemars(crate = "near_sdk::schemars")]
 pub enum Exchange {
     MissionTimeWithResource,
     MissionTimeWithTrust,