From a4e9d92680d4701d6c6b3706651dedbf073ab022 Mon Sep 17 00:00:00 2001
From: Revertron <r@revertron.com>
Date: Wed, 17 Mar 2021 20:05:05 +0100
Subject: [PATCH] Fixed getting zone difficulty.

---
 alfis.toml             | 2 +-
 src/web_ui.rs          | 9 +++++++--
 src/webview/scripts.js | 2 +-
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/alfis.toml b/alfis.toml
index e188bec..b41e298 100644
--- a/alfis.toml
+++ b/alfis.toml
@@ -1,5 +1,5 @@
 # Settings
-origin = "00000087A1BA6F94781C77BBDA96CC548E1E7F502DB3EA3F103ACD9301813B43"
+origin = "0000000D48E05DAB6B26C58094A2AF65563D1DB4EECAF18663B01CE77152E56B"
 key_file = "default.key"
 listen = "[::]:4244"
 public = false
diff --git a/src/web_ui.rs b/src/web_ui.rs
index 33d4a66..aba30d7 100644
--- a/src/web_ui.rs
+++ b/src/web_ui.rs
@@ -19,7 +19,7 @@ use alfis::event::Event;
 use alfis::dns::protocol::DnsRecord;
 use alfis::blockchain::{ZONE_MAX_LENGTH, ZONE_DIFFICULTY};
 use Cmd::*;
-use alfis::blockchain::transaction::DomainData;
+use alfis::blockchain::transaction::{DomainData, ZoneData};
 
 pub fn run_interface(context: Arc<Mutex<Context>>, miner: Arc<Mutex<Miner>>) {
     let file_content = include_str!("webview/index.html");
@@ -166,7 +166,7 @@ pub fn run_interface(context: Arc<Mutex<Context>>, miner: Arc<Mutex<Miner>>) {
                             let data = DomainData::new(zone.clone(), records);
                             let (keystore, transaction, difficulty) = {
                                 let context = context.lock().unwrap();
-                                (context.get_keystore(), context.chain.get_domain_transaction(&name), context.chain.get_zone_difficulty(&name))
+                                (context.get_keystore(), context.chain.get_domain_transaction(&name), context.chain.get_zone_difficulty(&zone))
                             };
                             let data = serde_json::to_string(&data).unwrap();
                             match transaction {
@@ -210,6 +210,11 @@ pub fn run_interface(context: Arc<Mutex<Context>>, miner: Arc<Mutex<Miner>>) {
                         return Ok(());
                     }
                     let data = data.to_lowercase();
+                    if serde_json::from_str::<ZoneData>(&data).is_err() {
+                        warn!("Something wrong with zone data!");
+                        let _ = web_view.eval(&format!("showWarning('{}');", "Something wrong with zone data!"));
+                        return Ok(());
+                    }
                     let (keystore, transaction) = {
                         let context = context.lock().unwrap();
                         (context.get_keystore(), context.chain.get_domain_transaction(&name))
diff --git a/src/webview/scripts.js b/src/webview/scripts.js
index 798054f..c69f559 100644
--- a/src/webview/scripts.js
+++ b/src/webview/scripts.js
@@ -136,7 +136,7 @@ function createZone() {
     difficulty = document.getElementById("new_zone_difficulty").value;
     obj = {};
     obj.name = new_zone;
-    obj.difficulty = difficulty;
+    obj.difficulty = parseInt(difficulty);
     data = JSON.stringify(obj);
     external.invoke(JSON.stringify({cmd: 'createZone', name: new_zone, data: data}));
 }