refactor file rendering to common method

Author: ysawa0

src/python.rs

@@ -1,12 +1,25 @@
 extern crate regex;
 
 use std::{
-    fs::{self, File}, io::Write, process
+    env, fs::{self, File}, io::Write, process
 };
 
 use askama::Template;
 use regex::Regex;
 
+trait CPATemplate {
+    fn write(&self, prefix: &str, path: &str);
+}
+
+// Implement the new trait for any type that implements `askama::Template`
+impl<T: Template> CPATemplate for T {
+    fn write(&self, prefix: &str, path: &str) {
+        let content = append_eof(self.render().expect("Failed to render file."));
+        let mut f = File::create(format!("{}/{}", prefix, path)).expect("Could not create file");
+        f.write_all(content.as_bytes()).expect("Could not write to file");
+    }
+}
+
 #[derive(Template)]
 #[template(path = ".gitignore", escape = "none")]
 struct GitIgnore {}
@@ -57,6 +70,17 @@ struct Prettier {}
 #[template(path = ".github/workflows/ci.yaml", escape = "none")]
 struct GHWorkflowCI {}
 
+fn append_eof(content: String) -> String {
+    let os = env::consts::OS;
+    let mut s = content.clone();
+    if os == "windows" {
+        s.push_str("\r\n");
+    } else {
+        s.push('\n');
+    }
+    s
+}
+
 pub fn setup_preset(mut preset: String, name: String, create: bool) {
     if preset == "python" {
         preset = "python3.10".to_string();
@@ -66,74 +90,24 @@ pub fn setup_preset(mut preset: String, name: String, create: bool) {
         prefix = format!("./{}", name)
     }
 
+    // Get the current operating system
+
     // Create needed dirs
     let _ = fs::create_dir_all(format!("{}/.cpa", prefix));
     let _ = fs::create_dir_all(format!("{}/.vscode", prefix));
     let _ = fs::create_dir_all(format!("{}/.github/workflows", prefix));
 
-    // Render Github Actions CI
-    File::create(format!("{}/.github/workflows/ci.yaml", prefix))
-        .and_then(|mut file| file.write_all(GHWorkflowCI {}.render().expect("Render fail: ci.yaml").as_bytes()))
-        .expect("Write fail: ci.yaml");
-
-    // Render .vscode/settings.json
-    File::create(format!("{}/.vscode/settings.json", prefix))
-        .and_then(|mut file| file.write_all(VSCodeSettings {}.render().expect("Render fail: .vscode/settings.json").as_bytes()))
-        .expect("Write fail: .vscode/settings.json");
-
-    // Render .vscode/extensions.json
-    File::create(format!("{}/.vscode/extensions.json", prefix))
-        .and_then(|mut file| {
-            file.write_all(
-                VSCodeExtensions {}
-                    .render()
-                    .expect("Render fail: .vscode/extensions.json")
-                    .as_bytes(),
-            )
-        })
-        .expect("Write fail: .vscode/extensions.json");
-
-    // Render .gitignore
-    File::create(format!("{}/.gitignore", prefix))
-        .and_then(|mut file| file.write_all(GitIgnore {}.render().expect("Render fail: .gitignore").as_bytes()))
-        .expect("Write fail: .gitignore");
-
-    // Render Makefile
-    File::create(format!("{}/Makefile", prefix))
-        .and_then(|mut file| file.write_all(Makefile {}.render().expect("Render fail: Makefile").as_bytes()))
-        .expect("Write fail: Makefile");
-
-    // Render Dockerfile
-    File::create(format!("{}/Dockerfile", prefix))
-        .and_then(|mut file| file.write_all(Dockerfile {}.render().expect("Render fail: Dockerfile").as_bytes()))
-        .expect("Write fail: Dockerfile");
-
-    // Render main.py
-    File::create(format!("{}/main.py", prefix))
-        .and_then(|mut file| file.write_all(MainPy {}.render().expect("Render fail").as_bytes()))
-        .expect("Render fail: main.py");
-
-    // Render pre-commit conf
-    File::create(format!("{}/.pre-commit-config.yaml", prefix))
-        .and_then(|mut file| {
-            file.write_all(
-                PreCommitConfig { python: true }
-                    .render()
-                    .expect("Render fail: .pre-commit-config.yaml")
-                    .as_bytes(),
-            )
-        })
-        .expect("Write fail: .pre-commit-config.yaml");
-
-    // Render Flake8 conf
-    File::create(format!("{}/.cpa/flake8.cfg", prefix))
-        .and_then(|mut file| file.write_all(Flake8 {}.render().expect("Render fail: flake8.cfg").as_bytes()))
-        .expect("Write fail: flake8.cfg");
-
-    // Render Prettier conf
-    File::create(format!("{}/.cpa/prettier.json", prefix))
-        .and_then(|mut file| file.write_all(Prettier {}.render().expect("Render fail: prettier.json").as_bytes()))
-        .expect("Write fail: prettier.json");
+    // Render files
+    GHWorkflowCI {}.write(&prefix, ".github/workflows/ci.yaml");
+    VSCodeSettings {}.write(&prefix, ".vscode/settings.json");
+    VSCodeExtensions {}.write(&prefix, ".vscode/extensions.json");
+    GitIgnore {}.write(&prefix, ".gitignore");
+    Makefile {}.write(&prefix, "Makefile");
+    Dockerfile {}.write(&prefix, "Dockerfile");
+    MainPy {}.write(&prefix, "main.py");
+    PreCommitConfig { python: true }.write(&prefix, ".pre-commit-config.yaml");
+    Flake8 {}.write(&prefix, ".cpa/flake8.cfg");
+    Prettier {}.write(&prefix, ".cpa/prettier.json");
 
     // Render Poetry conf
     let re = Regex::new(r"python(3\.\d+|4\.\d+)").unwrap();
@@ -150,8 +124,6 @@ pub fn setup_preset(mut preset: String, name: String, create: bool) {
         python_ver,
         black_target_ver,
     };
-    let out_pyproj: String = pyproj.render().expect("Failed to render");
-    let mut f_pyproj = File::create(format!("{}/pyproject.toml", prefix)).expect("Could not create file");
-    f_pyproj.write_all(out_pyproj.as_bytes()).expect("Could not write to file");
+    pyproj.write(&prefix, "pyproject.toml");
     println!("Project created at: {}", prefix)
 }