Files
marine/examples/ipfs-node/effector/src/main.rs
2020-12-08 18:15:18 +03:00

83 lines
2.3 KiB
Rust

/*
* Copyright 2020 Fluence Labs Limited
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
mod path;
use crate::path::to_full_path;
use fluence::fce;
use fluence::WasmLoggerBuilder;
const RESULT_FILE_PATH: &str = "/tmp/ipfs_rpc_file";
const IPFS_ADDR_ENV_NAME: &str = "IPFS_ADDR";
const TIMEOUT_ENV_NAME: &str = "timeout";
pub fn main() {
WasmLoggerBuilder::new()
.with_log_level(log::Level::Info)
.build()
.unwrap();
}
/// Put file from specified path to IPFS and return its hash.
#[fce]
pub fn put(file_path: String) -> String {
log::info!("put called with file path {}", file_path);
let file_path = to_full_path(file_path);
let timeout = std::env::var(TIMEOUT_ENV_NAME).unwrap_or_else(|_| "1s".to_string());
let cmd = format!("add --timeout {} -Q {}", timeout, file_path);
unsafe { ipfs(cmd) }
}
/// Get file by provided hash from IPFS, saves it to a temporary file and returns a path to it.
#[fce]
pub fn get(hash: String) -> String {
log::info!("get called with hash {}", hash);
let result_file_path = to_full_path(RESULT_FILE_PATH);
let timeout = std::env::var(TIMEOUT_ENV_NAME).unwrap_or_else(|_| "1s".to_string());
let cmd = format!(
"get --timeout {} -o {} {}",
timeout, result_file_path, hash
);
unsafe { ipfs(cmd) };
RESULT_FILE_PATH.to_string()
}
#[fce]
pub fn get_address() -> String {
match std::env::var(IPFS_ADDR_ENV_NAME) {
Ok(addr) => addr,
Err(e) => format!(
"getting {} env variable failed with error {:?}",
IPFS_ADDR_ENV_NAME, e
),
}
}
#[fce]
#[link(wasm_import_module = "host")]
extern "C" {
/// Execute provided cmd as a parameters of ipfs cli, return result.
pub fn ipfs(cmd: String) -> String;
}