From 1f27b199263380d85823cdb95f5c303501b49bc7 Mon Sep 17 00:00:00 2001
From: corydickson <7246942+corydickson@users.noreply.github.com>
Date: Mon, 29 Apr 2024 02:13:18 +0000
Subject: [PATCH] Add type def (#20)

* Add type storage

* Add attest to type

* Remove unused import
---
 src/lib.rs | 23 ++++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 669d2ee..ff6d9e9 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,7 +1,7 @@
 use near_sdk::borsh::{BorshDeserialize, BorshSerialize};
 use near_sdk::env::log_str;
 use near_sdk::serde::{Serialize, Deserialize};
-use near_sdk::collections::LookupMap;
+use near_sdk::collections::{LookupMap, UnorderedMap};
 use near_sdk::near_bindgen;
 use near_sdk::{
     AccountId, BorshStorageKey, PublicKey, require
@@ -52,7 +52,7 @@ pub struct Contract {
     // A signer can submit an attestation for a particular package already in the registry
     pub attestations: LookupMap<AccountId, LookupMap<Namespace, Attestations>>,
     pub compiled_types: LookupMap<Namespace, Types>,
-    pub type_list: LookupMap<String, u8>,
+    pub type_list: UnorderedMap<String, u8>,
 }
 
 impl Default for Contract {
@@ -61,7 +61,7 @@ impl Default for Contract {
             packages: LookupMap::new(PrefixKeys::Package),
             attestations: LookupMap::new(PrefixKeys::Attestation),
             compiled_types: LookupMap::new(PrefixKeys::Types),
-            type_list: LookupMap::new(PrefixKeys::TypeList)
+            type_list: UnorderedMap::new(PrefixKeys::TypeList)
         }
     }
 }
@@ -319,6 +319,23 @@ impl Contract {
 
         return at[index].clone();
     }
+
+    pub fn attest_to_type(
+        &mut self,
+        // An account ID of the author who published the manifest
+        author: AccountId,
+        // Name of a type in a package
+        type_name: String,
+    ) {
+        if self.attestations.contains_key(&author) {
+            let count: u8 = match self.type_list.get(&type_name) {
+                Some(v) => v + 1u8,
+                None => 1u8,
+            };
+
+            self.type_list.insert(&type_name, &count);
+        }
+    }
 }
 
 #[cfg(all(test, not(target_arch = "wasm32")))]