max size

Author: zhangzijie-pro

command/src/commands/command.rs

@@ -32,6 +32,7 @@ pub fn help() -> String{
     cat     View file only read                            mv          Move file's path
     python  Run code in python                             tar -zxvf:  Compression  
     html    Open html file                                 tar -xvf:   Decompression
+    pd      Check your password                            sudo        Root
     exit    Exit this process\0\x1B[0m\n"
     );
 
@@ -300,18 +301,26 @@ pub fn cat(file: &str) -> Result<(usize,String),Error>{
     if file.is_empty(){
         return Ok(empty_file());
     }
-    let f = fs::File::open(Path::new(file));
     let mut buffer = String::new();
-    let _ = f.unwrap().read_to_string(&mut buffer);
+    let file_size = max_size_file(file);
+    match file_size{
+        Ok(_) =>{
+            let f = fs::File::open(Path::new(file)); 
+            let _ = f.unwrap().read_to_string(&mut buffer);
+        },
+        Err(err) =>{
+            return Ok((ERR_CODE,err.to_string()));
+        }
+    }
     Ok((STATUE_CODE,buffer))
 }
 
 
 use crate::commands::download::{download_package, find_package};
 use crate::priority::get_priority;
-use crate::set::set::file_create_time;
+use crate::set::set::{file_create_time, max_size_file};
 use crate::run::run;
-use crate::state_code::{empty_dir, empty_file, missing_pattern,  STATUE_CODE};
+use crate::state_code::{empty_dir, empty_file, missing_pattern, ERR_CODE, STATUE_CODE};
 use super::download::update;
 use crate::root::SessionContext;
 

command/src/set/set.rs

@@ -134,4 +134,46 @@ pub fn error_log(err: String){
         return;
     }
     // error.log -> $HOME/.Tiks/error.log -> [time]: [Error]
+}
+
+pub const MB:f64 = 1024.0*1024.0;
+
+use std::fs::metadata;
+use std::io;
+
+fn get_file_size(file:&str) -> f64{
+    let mut size = 0;
+    if let Ok(s) = metadata(file){
+        size = s.len();
+    }
+
+    size as f64
+}
+
+type MB = f64;
+
+fn byte_to_mb(size: f64) -> MB{
+    let new_size = size/MB;
+    new_size
+}
+
+pub fn max_size_file(path: &str) -> Result<(),&'static str>{
+    let size = get_file_size(path);
+    let mb = byte_to_mb(size); // MB
+
+    if mb > 128_f64{
+        loop {
+            let mut choice = String::new();
+            println!("This file is more than 256MB, Do you want to read? [y/n]");
+            io::stdout().flush().unwrap();
+            io::stdin().read_line(&mut choice).unwrap();
+            if choice.as_str() == "y"{
+                break;
+            }else{
+                return Err("Canel read");
+            }
+        }
+    }
+
+    Ok(())
 }

command/src/state_code.rs

@@ -1,4 +1,5 @@
 pub const STATUE_CODE:usize=0;
+pub const ERR_CODE:usize=109;
 
 
 pub fn missing_pattern() -> (usize, String) {