also make near social work on both mainnet and testnet

Author: ailisp

community/src/lib.rs

@@ -1,6 +1,6 @@
 mod social_db;
 
-use crate::social_db::{ext_social_db, SOCIAL_DB};
+use crate::social_db::social_db_contract;
 use near_sdk::borsh::{BorshDeserialize, BorshSerialize};
 use near_sdk::{env, near_bindgen, require, AccountId, NearToken, Promise};
 
@@ -13,7 +13,7 @@ pub struct Contract {}
 impl Contract {
     #[init]
     pub fn new() -> Self {
-        ext_social_db::ext(SOCIAL_DB.into())
+        social_db_contract()
             .with_unused_gas_weight(1)
             .with_attached_deposit(NearToken::from_near(1))
             .grant_write_permission(

community/src/social_db.rs

@@ -1,7 +1,5 @@
 use near_sdk::serde_json::Value;
-use near_sdk::{ext_contract, AccountId, PublicKey};
-
-pub const SOCIAL_DB: &near_sdk::AccountIdRef = near_sdk::AccountIdRef::new_or_panic("social.near");
+use near_sdk::{ext_contract, AccountId, PublicKey, env};
 
 #[ext_contract(ext_social_db)]
 pub trait SocialDB {
@@ -13,3 +11,12 @@ pub trait SocialDB {
         keys: Vec<String>,
     );
 }
+
+pub fn social_db_contract() -> ext_social_db::SocialDBExt {
+    let social_db: AccountId = if env::current_account_id().to_string().ends_with("testnet") {
+        "v1.social08.testnet"
+    } else {
+        "social.near"
+    }.parse().unwrap();
+    ext_social_db::ext(social_db)
+}

src/lib.rs

@@ -15,7 +15,7 @@ use crate::access_control::AccessControl;
 use community::*;
 use post::*;
 
-use crate::social_db::{ext_social_db, SOCIAL_DB};
+use crate::social_db::social_db_contract;
 use near_sdk::borsh::{self, BorshDeserialize, BorshSerialize};
 use near_sdk::collections::{LookupMap, UnorderedMap, Vector};
 use near_sdk::require;
@@ -527,7 +527,7 @@ impl Contract {
         require!(community.handle == handle, "Community handle cannot be changed");
         require!(env::prepaid_gas() >= UPDATE_COMMUNITY_GAS, "Require at least 30 Tgas");
         self.communities.insert(&handle, &community);
-        ext_social_db::ext(SOCIAL_DB.parse().unwrap()).with_unused_gas_weight(1).set(json!({
+        social_db_contract().with_unused_gas_weight(1).set(json!({
             get_devhub_community_account(&community.handle)
             : {
                 "profile": {
@@ -546,9 +546,9 @@ impl Contract {
             .expect("Only community admins and hub moderators can set community Social DB");
 
         require!(env::prepaid_gas() >= ADD_COMMUNITY_ANNOUNCEMENT_GAS, "Require at least 30 Tgas");
-        ext_social_db::ext(SOCIAL_DB.parse().unwrap())
+        social_db_contract()
             .with_unused_gas_weight(1)
-            .set(json!({ get_devhub_community_account(&handle) : data }));
+            .set(json!({ get_devhub_community_account(&handle): data }));
     }
 
     pub fn delete_community(&mut self, handle: CommunityHandle) {
@@ -564,11 +564,9 @@ impl Contract {
         self.communities.remove(&community.handle);
 
         require!(env::prepaid_gas() >= DELETE_COMMUNITY_GAS, "Require at least 30 Tgas");
-        ext_devhub_community::ext(
-            get_devhub_community_account(&community.handle).parse().unwrap()
-        )
-        .with_unused_gas_weight(1)
-        .destroy();
+        ext_devhub_community::ext(get_devhub_community_account(&community.handle).parse().unwrap())
+            .with_unused_gas_weight(1)
+            .destroy();
     }
 
     pub fn set_featured_communities(&mut self, handles: Vec<CommunityHandle>) {

src/notify.rs

@@ -1,4 +1,4 @@
-use crate::social_db::{ext_social_db, SOCIAL_DB};
+use crate::social_db::social_db_contract;
 use crate::PostId;
 use near_sdk::serde_json::json;
 use near_sdk::{env, AccountId, Promise};
@@ -44,7 +44,7 @@ pub fn notify_mentions(text: &str, post_id: PostId) {
             }));
         }
 
-        ext_social_db::ext(SOCIAL_DB.parse().unwrap())
+        social_db_contract()
             .with_static_gas(env::prepaid_gas() / 4)
             .with_attached_deposit(env::attached_deposit())
             .set(json!({
@@ -69,7 +69,7 @@ pub fn notify_edit(post_id: PostId, post_author: AccountId) -> Promise {
 }
 
 fn notify(post_id: PostId, post_author: AccountId, action: &str) -> Promise {
-    ext_social_db::ext(SOCIAL_DB.parse().unwrap())
+    social_db_contract()
         .with_static_gas(env::prepaid_gas() / 4)
         .with_attached_deposit(env::attached_deposit())
         .set(json!({

src/repost.rs

@@ -1,5 +1,5 @@
 use crate::post::{get_post_description, Post, PostBody};
-use crate::social_db::{ext_social_db, SOCIAL_DB};
+use crate::social_db::social_db_contract;
 use near_sdk::serde_json::json;
 use near_sdk::{env, AccountId, Promise};
 
@@ -47,7 +47,7 @@ fn repost_internal(post: Post, contract_address: AccountId) -> near_sdk::serde_j
 }
 
 pub fn repost(post: Post) -> Promise {
-    ext_social_db::ext(SOCIAL_DB.parse().unwrap())
+    social_db_contract()
         .with_static_gas(env::prepaid_gas() / 3)
         .with_attached_deposit(env::attached_deposit())
         .set(repost_internal(post, env::current_account_id()))

src/social_db.rs

@@ -1,9 +1,18 @@
-use near_sdk::ext_contract;
+use near_sdk::{ext_contract, AccountId, env};
 use near_sdk::serde_json::Value;
 
-pub const SOCIAL_DB: &str = "social.near";
-
 #[ext_contract(ext_social_db)]
 pub trait SocialDB {
     fn set(&mut self, data: Value);
 }
+
+pub fn social_db_contract() -> ext_social_db::SocialDBExt {
+    let social_db: AccountId = if env::current_account_id().to_string().ends_with("testnet") {
+        "v1.social08.testnet"
+    } else {
+        "social.near"
+    }
+    .parse()
+    .unwrap();
+    ext_social_db::ext(social_db)
+}