From 5891a1aca16c6b429553e000596476236a7f88fa Mon Sep 17 00:00:00 2001
From: itiB <is0312vx@ed.ritsumei.ac.jp>
Date: Mon, 4 Apr 2022 01:44:04 +0900
Subject: [PATCH] WIP: Text overwrite failed...

---
 src/options/level_tuning.rs | 26 +++++++++++++++++---------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/src/options/level_tuning.rs b/src/options/level_tuning.rs
index f7ff84e8..988596e2 100644
--- a/src/options/level_tuning.rs
+++ b/src/options/level_tuning.rs
@@ -2,7 +2,7 @@ use crate::detections::{configs, print::AlertMessage, utils};
 use crate::filter;
 use crate::yaml::ParseYaml;
 use std::collections::HashMap;
-use std::fs::File;
+use std::fs::{self, File};
 use std::io::BufWriter;
 use std::io::{Read, Write};
 
@@ -14,7 +14,7 @@ impl LevelTuning {
         if read_result.is_err() {
             AlertMessage::warn(
                 &mut BufWriter::new(std::io::stderr().lock()),
-                &read_result.as_ref().unwrap_err(),
+                read_result.as_ref().unwrap_err(),
             )
             .ok();
             return;
@@ -38,8 +38,8 @@ impl LevelTuning {
                 .unwrap()
                 .args
                 .value_of("rules")
-                .unwrap_or(&"rules"),
-            &"informational",
+                .unwrap_or("rules"),
+            "informational",
             &filter::exclude_ids(),
         );
         if result_readdir.is_err() {
@@ -52,18 +52,16 @@ impl LevelTuning {
             if let Some(new_level) = tuning_map.get(rule["id"].as_str().unwrap()) {
                 println!("path: {}", path);
                 let mut file = match File::options()
-                    .create(true)
-                    .write(true)
                     .read(true)
-                    .append(false)
+                    .write(true)
                     .open(&path)
                 {
                     Err(e) => panic!("Couldn't open {}: {}", path, e),
                     Ok(file) => file,
                 };
 
-                let mut content = String::new();
-                file.read_to_string(&mut content).unwrap();
+                let mut content = fs::read_to_string(&path).unwrap(); // TODO: Error Handling
+                // file.read_to_string(&mut content).unwrap();
                 let past_level = "level: ".to_string() + rule["level"].as_str().unwrap();
 
                 if new_level.starts_with("informational") {
@@ -81,6 +79,16 @@ impl LevelTuning {
                 if new_level.starts_with("critical") {
                     content = content.replace(&past_level, "level: critical");
                 }
+                println!("{:?}", &content);
+
+                let mut file = match File::options()
+                    .write(true)
+                    .open(&path)
+                {
+                    Err(e) => panic!("Couldn't open {}: {}", path, e),
+                    Ok(file) => file,
+                };
+
                 file.write_all(content.as_bytes()).unwrap(); // TODO: use result
                 println!(
                     "level: {} -> {}",