examples/multi-service/ethqlite/src/main.rs

/*
 * Copyright 2021 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.
 */

use fce_sqlite_connector;
use fce_sqlite_connector::{Connection, State, Value};
///, WasmLoggerBuilder};
use fluence;
use fluence::marine;
use fluence::WasmLoggerBuilder;

use serde::Deserialize;
use serde_json;
use std::path::{Path, PathBuf};

use std::sync::atomic::{AtomicBool, Ordering};

use crate::auth::is_owner;
use crate::crud::create_table;

const DB_PATH: &str = "/tmp/fluence_service_db.sqlite";

mod auth;
mod crud;

pub static INIT: AtomicBool = AtomicBool::new(false);

fn main() {}

fn get_connection() -> Connection {
    Connection::open(DB_PATH).unwrap()
}

#[marine]
#[derive(Debug)]
pub struct InitResult {
    pub success: bool,
    pub err_msg: String,
}

#[fmarinece]
pub fn init_service() -> InitResult {
    if !is_owner() {
        return InitResult {
            success: false,
            err_msg: "Not authorized to use this service".into(),
        };
    }

    if INIT.load(Ordering::Relaxed) {
        return InitResult {
            success: false,
            err_msg: "Service already initiated".into(),
        };
    }

    let conn = get_connection();
    let res = create_table(&conn);
    if res.is_err() {
        return InitResult {
            success: false,
            err_msg: "Failure to create tables".into(),
        };
    }
    // TODO: implement rollbacks

    INIT.store(true, Ordering::Relaxed);
    InitResult {
        success: true,
        err_msg: "".into(),
    }
}

#[marine]
pub fn owner_nuclear_reset() -> bool {
    if !is_owner() {
        return false;
    }

    INIT.store(false, Ordering::Relaxed);

    let conn = get_connection();
    let t_names = vec!["reward_blocks"];
    for t_name in t_names {
        let stmt = format!("delete from {}", t_name);
        let mut del_cur = conn.prepare(&stmt).unwrap().cursor();
        del_cur.next().unwrap();
    }

    true
}