xref: /DragonReach/src/main.rs (revision b40b6b4d2f72c30a382f9f18740fa73d7fc90721)

#![no_std]
#![no_main]
#![feature(slice_pattern)]
#![feature(fn_traits)]

use cfg_if::cfg_if;

cfg_if! {
    if #[cfg(target_os = "dragonos")]{
        extern crate drstd;
        use drstd as std;
    }
}

extern crate hashbrown;

mod contants;
mod error;
mod executor;
mod manager;
mod parse;
mod task;
mod time;
mod unit;

use std::fs;
use std::{eprint, eprintln, print, println};

use std::string::ToString;
use std::vec::Vec;

use error::ErrorFormat;

pub struct FileDescriptor(usize);

const DRAGON_REACH_UNIT_DIR: &'static str = "/etc/reach/system/";

#[cfg(target_os = "dragonos")]
#[no_mangle]
fn main() {
    use manager::timer_manager::TimerManager;
    use parse::UnitParser;

    use crate::{
        executor::Executor,
        manager::{Manager, UnitManager},
        parse::parse_util::UnitParseUtil,
    };

    let mut units_file_name = Vec::new();
    //读取目录里面的unit文件
    if let Ok(entries) = fs::read_dir(DRAGON_REACH_UNIT_DIR) {
        for entry in entries {
            if let Ok(entry) = entry {
                if let Ok(file_type) = entry.file_type() {
                    if file_type.is_file() {
                        let filename = entry.file_name();
                        let filename = filename.to_str().unwrap();
                        units_file_name.push(filename.to_string());
                    }
                }
            }
        }
    }

    //启动服务
    for path in units_file_name {
        let id = match UnitParser::from_path(&path) {
            Ok(id) => id,
            Err(e) => {
                eprintln!("Err:{}", e.error_format());
                0
            }
        };

        if id != 0 {
            if let Err(e) = Executor::exec(id) {
                eprintln!("Err:{}", e.error_format());
            } else {
                println!("Service {} startup success...", id);
            }
        }
    }

    // 启动完服务后进入主循环
    loop {
        // 检查各服务运行状态
        Manager::check_running_status();
        // 检查cmd进程状态
        Manager::check_cmd_proc();
        // 检查计时器任务
        TimerManager::check_timer();
    }
}

#[cfg(not(target_os = "dragonos"))]
fn main() {
    use parse::UnitParser;

    use crate::{
        executor::Executor,
        manager::{timer_manager::TimerManager, Manager, UnitManager},
    };

    let mut units_file_name = Vec::new();
    //读取目录里面的unit文件
    if let Ok(entries) = fs::read_dir("/bin") {
        for entry in entries {
            if let Ok(entry) = entry {
                if let Ok(file_type) = entry.file_type() {
                    if file_type.is_file() {
                        let filename = entry.file_name();
                        let _filename = filename.to_str().unwrap();
                        //units_file_name.push(filename.to_string());
                    }
                }
            }
        }
    }

    units_file_name.push("/home/heyicong/DragonReach/parse_test/test.service".to_string());

    // 完善unit之间依赖关系
    UnitManager::init_units_dependencies();

    //启动服务
    for path in units_file_name {
        let id = match UnitParser::from_path(&path) {
            Ok(id) => id,
            Err(e) => {
                eprintln!("Err:{}", e.error_format());
                0
            }
        };

        if id != 0 {
            let _unit = UnitManager::get_unit_with_id(&id).unwrap();
            if let Err(e) = Executor::exec(id) {
                eprintln!("Err:{}", e.error_format());
            }
        }
    }

    // 启动完服务后进入主循环
    loop {
        // 检查各服务运行状态
        Manager::check_running_status();

        Manager::check_cmd_proc();

        TimerManager::check_timer();
    }
}