app logic and routing
parent
12ea1038f1
commit
6c2dd2dfca
@ -0,0 +1,283 @@
|
||||
#![feature(proc_macro_hygiene, decl_macro)]
|
||||
#[macro_use] extern crate rocket;
|
||||
#[macro_use] extern crate rocket_contrib;
|
||||
#[macro_use] extern crate serde_derive;
|
||||
extern crate reqwest;
|
||||
extern crate qrcode_generator;
|
||||
mod data_types;
|
||||
|
||||
use rocket::http::RawStr;
|
||||
use rocket::response::Redirect;
|
||||
use rocket_contrib::json::{Json,JsonValue};
|
||||
use rocket_contrib::templates::Template;
|
||||
use rocket_contrib::serve::StaticFiles;
|
||||
use reqwest::blocking::{RequestBuilder, Client};
|
||||
use reqwest::Error;
|
||||
use qrcode_generator::QrCodeEcc;
|
||||
use std::env;
|
||||
use std::clone;
|
||||
use data_types::*;
|
||||
|
||||
// fn issue_rpc(method: &str, params: Option<RPCParams>) -> RequestBuilder {
|
||||
// let http_client = Client::new();
|
||||
// let url = format!(
|
||||
// "{}/json_rpc",
|
||||
// env::var("DAEMON_URI").unwrap()
|
||||
// );
|
||||
// let post_data = RPCPayload {
|
||||
// method: Some(method.to_string()),
|
||||
// params: params,
|
||||
// ..Default::default()
|
||||
// };
|
||||
// http_client.post(&url).json(&post_data)
|
||||
// }
|
||||
|
||||
// fn issue_raw_rpc(method: &str, params: JsonValue) -> RequestBuilder {
|
||||
// let http_client = Client::new();
|
||||
// let url = format!(
|
||||
// "{}/{}",
|
||||
// env::var("DAEMON_URI").unwrap(),
|
||||
// &method
|
||||
// );
|
||||
// http_client.post(&url).json(¶ms)
|
||||
// }
|
||||
//
|
||||
fn build_rpc(method: &str, data: Option<JsonValue>, raw: bool) -> RequestBuilder {
|
||||
let http_client = Client::new();
|
||||
let daemon_uri = env::var("DAEMON_URI").unwrap();
|
||||
match raw {
|
||||
true => {
|
||||
let uri = format!("{}/{}", &daemon_uri, &method);
|
||||
if let None = data {
|
||||
http_client.post(&uri)
|
||||
} else {
|
||||
http_client.post(&uri).json(&data)
|
||||
}
|
||||
},
|
||||
false => {
|
||||
let uri = format!("{}/json_rpc", &daemon_uri);
|
||||
let data = RPCPayload {
|
||||
method: Some(method.to_string()),
|
||||
..Default::default()
|
||||
};
|
||||
http_client.post(&uri).json(&data)
|
||||
}
|
||||
}
|
||||
}
|
||||
//
|
||||
// #[get("/block/hash/<block_hash>")]
|
||||
// fn get_block_by_hash(block_hash: String) -> Template {
|
||||
// let params = RPCParams {
|
||||
// hash: Some(block_hash),
|
||||
// ..Default::default()
|
||||
// };
|
||||
// let res: GetBlock = issue_rpc(&"get_block", Some(params))
|
||||
// .send().unwrap().json().unwrap();
|
||||
// Template::render("block", &res.result)
|
||||
// }
|
||||
//
|
||||
// #[get("/block/height/<block_height>")]
|
||||
// fn get_block_by_height(block_height: String) -> Template {
|
||||
// let params = RPCParams {
|
||||
// height: Some(block_height),
|
||||
// ..Default::default()
|
||||
// };
|
||||
// let res: GetBlock = issue_rpc(&"get_block", Some(params))
|
||||
// .send().unwrap().json().unwrap();
|
||||
// Template::render("block", &res.result)
|
||||
// }
|
||||
|
||||
#[get("/transaction/<tx_hash>")]
|
||||
fn get_transaction_by_hash(tx_hash: String) -> Template {
|
||||
let params: JsonValue = json!({
|
||||
"txs_hashes": [&tx_hash],
|
||||
"decode_as_json": true
|
||||
});
|
||||
|
||||
let mut res: GetTransactions = build_rpc(
|
||||
&"get_transactions", Some(params), true
|
||||
).send().unwrap().json().unwrap();
|
||||
|
||||
if res.txs.len() > 0 {
|
||||
for f in &mut res.txs {
|
||||
f.process();
|
||||
};
|
||||
};
|
||||
|
||||
let context = json!({
|
||||
"tx_info": res.txs,
|
||||
"tx_hash": tx_hash
|
||||
});
|
||||
Template::render("transaction", context)
|
||||
}
|
||||
|
||||
// #[get("/address/<wallet_address>?<tx_amount>&<tx_description>&<recipient_name>&<tx_payment_id>")]
|
||||
// fn show_wallet_address(
|
||||
// wallet_address: String,
|
||||
// tx_amount: Option<String>,
|
||||
// tx_description: Option<String>,
|
||||
// recipient_name: Option<String>,
|
||||
// tx_payment_id: Option<String>
|
||||
// ) -> Template {
|
||||
// let address_uri = format!(
|
||||
// "wownero:{}&tx_amount={}&tx_description={}&recipient_name={}&tx_payment_id={}",
|
||||
// wallet_address,
|
||||
// tx_amount.unwrap_or("".to_string()),
|
||||
// tx_description.unwrap_or("".to_string()),
|
||||
// recipient_name.unwrap_or("".to_string()),
|
||||
// tx_payment_id.unwrap_or("".to_string())
|
||||
// );
|
||||
// let qr_code: String = qrcode_generator::to_svg_to_string(address_uri, QrCodeEcc::Low, 256, None)
|
||||
// .unwrap();
|
||||
// let qr_code: String = base64::encode(qr_code);
|
||||
// let context: JsonValue = json!({
|
||||
// "qr_code": qr_code,
|
||||
// "wallet_address": wallet_address
|
||||
// });
|
||||
// Template::render("address", context)
|
||||
// }
|
||||
//
|
||||
// #[get("/search?<value>")]
|
||||
// fn search(value: &RawStr) -> Redirect {
|
||||
// // This search implementation is not ideal but it works.
|
||||
// // We basically check the length of the search value and
|
||||
// // attempt to redirect to the appropriate route.
|
||||
// let sl: usize = value.len();
|
||||
// if sl == 0 {
|
||||
// return Redirect::found(uri!(index));
|
||||
// } else if sl < 8 {
|
||||
// // Less than 8 characters is probably a block height. If it can
|
||||
// // be parsed as valid u32 then redirect to `get_block_by_height`,
|
||||
// // otherwise redirect to the error response.
|
||||
// match value.parse::<u32>() {
|
||||
// Ok(_) => return Redirect::found(uri!(get_block_by_height: value.as_str())),
|
||||
// Err(_) => return Redirect::found(uri!(error))
|
||||
// }
|
||||
// } else if sl == 64 {
|
||||
// // Equal to 64 characters is probably a hash; block or tx.
|
||||
// // For this we attempt to query for a block with
|
||||
// // given hash. If we don't receive a valid/expected
|
||||
// // response then we attempt to query for a transaction hash.
|
||||
// // If neither works then redirect to error response.
|
||||
// let block_hash_params = RPCParams {
|
||||
// hash: Some(value.to_string()),
|
||||
// ..Default::default()
|
||||
// };
|
||||
// let check_valid_block_hash: Result<GetBlock, Error> = issue_rpc(
|
||||
// &"get_block", Some(block_hash_params)
|
||||
// ).send().unwrap().json();
|
||||
//
|
||||
// match check_valid_block_hash {
|
||||
// Ok(_) => return Redirect::found(uri!(get_block_by_hash: value.as_str())),
|
||||
// Err(_) => {
|
||||
// let tx_hash_params: JsonValue = json!({"txs_hashes": [&value.as_str()]});
|
||||
// let check_valid_tx_hash: Result<GetTransactions, Error> = issue_raw_rpc(
|
||||
// &"get_transactions", tx_hash_params
|
||||
// ).send().unwrap().json();
|
||||
//
|
||||
// match check_valid_tx_hash {
|
||||
// Ok(_) => return Redirect::found(uri!(get_transaction_by_hash: value.as_str())),
|
||||
// Err(_) => return Redirect::found(uri!(error))
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
// } else if sl == 95 {
|
||||
// // Equal to 95 characters is probably a wallet address.
|
||||
// // For this let's just redirect to the `show_wallet_address` route.
|
||||
// return Redirect::found(uri!(show_wallet_address: value.as_str(), "", "", "", ""))
|
||||
// } else if sl == 105 {
|
||||
// // Equal to 105 characters is probably an integrated address.
|
||||
// // For this let's just redirect to the `show_wallet_address` route.
|
||||
// return Redirect::found(uri!(show_wallet_address: value.as_str(), "", "", "", ""))
|
||||
// } else {
|
||||
// // Anything else hasn't been implemented yet
|
||||
// // so redirect to error response.
|
||||
// return Redirect::found(uri!(error));
|
||||
// };
|
||||
// }
|
||||
//
|
||||
// #[get("/tx_pool")]
|
||||
// fn show_tx_pool() -> Json<GetTransactionPool> {
|
||||
// let mut tx_pool: GetTransactionPool = build_rpc(
|
||||
// &"get_transaction_pool", None, true
|
||||
// ).send().unwrap().json().unwrap();
|
||||
//
|
||||
// for f in &mut tx_pool.transactions {
|
||||
// f.process();
|
||||
// };
|
||||
//
|
||||
// Json(tx_pool)
|
||||
// }
|
||||
|
||||
#[get("/")]
|
||||
fn index() -> Template {
|
||||
let daemon_info: GetInfo = build_rpc(
|
||||
&"get_info", None, false
|
||||
).send().unwrap().json().unwrap();
|
||||
|
||||
let tx_pool: GetTransactionPool = build_rpc(
|
||||
&"get_transaction_pool", None, true
|
||||
).send().unwrap().json().unwrap();
|
||||
|
||||
// let mut tx_pool_txs = tx_pool.transactions;
|
||||
//
|
||||
// match tx_pool_txs {
|
||||
// Some(s) => {
|
||||
// for mut f in s {
|
||||
// let mut _fp = f.process();
|
||||
// }
|
||||
// },
|
||||
// None => {}
|
||||
// };
|
||||
|
||||
// for f in &mut tx_pool.transactions {
|
||||
// f.process();
|
||||
// };
|
||||
|
||||
let context: JsonValue = json!({
|
||||
"daemon_info": daemon_info.result,
|
||||
"tx_pool_txs": tx_pool.transactions
|
||||
});
|
||||
|
||||
Template::render("index", context)
|
||||
}
|
||||
|
||||
#[get("/error", )]
|
||||
fn error() -> JsonValue {
|
||||
json!({
|
||||
"status": "error",
|
||||
"reason": "There was an error while searching the provided values."
|
||||
})
|
||||
}
|
||||
|
||||
#[catch(404)]
|
||||
fn not_found() -> JsonValue {
|
||||
json!({
|
||||
"status": "error",
|
||||
"reason": "Resource was not found."
|
||||
})
|
||||
}
|
||||
|
||||
fn main() {
|
||||
let env_url = env::var("DAEMON_URI");
|
||||
match env_url {
|
||||
Ok(_) => {
|
||||
rocket::ignite()
|
||||
.mount("/", routes![
|
||||
index,
|
||||
// search,
|
||||
// show_tx_pool,
|
||||
// get_block_by_height,
|
||||
// get_block_by_hash,
|
||||
get_transaction_by_hash,
|
||||
// show_wallet_address,
|
||||
error
|
||||
])
|
||||
.mount("/static", StaticFiles::from("./static"))
|
||||
.register(catchers![not_found])
|
||||
.attach(Template::fairing())
|
||||
.launch();
|
||||
},
|
||||
Err(_) => panic!("Environment variable `DAEMON_URI` not provided.")
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue