examples/url-downloader/facade/src/main.rs

/*
 * 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.
 */
#![allow(improper_ctypes)]

use fluence::fce;
use fluence::MountedBinaryResult as Result;
use fluence::WasmLoggerBuilder;

pub fn main() {
    WasmLoggerBuilder::new().build().unwrap();
}

/// Combining of modules: `curl` and `local_storage`.
/// Calls `curl` and stores returned result into a file.
#[fce]
pub fn get_n_save(url: String, file_name: String) -> String {
    let result = unsafe { download(url) };
    if result.is_success() {
        log::info!("saving file {}", file_name);
        unsafe { file_put(file_name, result.stdout) }
    } else {
        log::error!("download failed: {:#?}", result.as_std());
        format!("download failed: {:#?}", result.as_std())
    }
}

#[fce]
/// Loads file from disk and returns its content as base64
pub fn load_file(file_name: String) -> String {
    let bytes = unsafe { file_get(file_name) };
    base64::encode(bytes)
}

/// Import `curl_adapter` module
#[fce]
#[link(wasm_import_module = "curl_adapter")]
extern "C" {
    pub fn download(url: String) -> Result;
}

/// Import `local_storage` module
#[fce]
#[link(wasm_import_module = "local_storage")]
extern "C" {
    #[link_name = "get"]
    pub fn file_get(file_name: String) -> Vec<u8>;

    #[link_name = "put"]
    pub fn file_put(name: String, file_content: Vec<u8>) -> String;
}