From 3faed5c980835ac9fe736e2ad38712c64d04dc2d Mon Sep 17 00:00:00 2001
From: vh-zuka <vuhoang.kevin@gmail.com>
Date: Tue, 8 Nov 2022 08:35:58 +0700
Subject: [PATCH] Add new project template

---
 crates/cli/src/cmd/new.rs | 76 ++++++++++++++++++++++++++++++++++++---
 1 file changed, 71 insertions(+), 5 deletions(-)

diff --git a/crates/cli/src/cmd/new.rs b/crates/cli/src/cmd/new.rs
index ab4ce26b..41f5ee52 100644
--- a/crates/cli/src/cmd/new.rs
+++ b/crates/cli/src/cmd/new.rs
@@ -1,5 +1,6 @@
 use miette::IntoDiagnostic;
 use std::fs;
+use std::io::Write;
 use std::path::PathBuf;
 
 #[derive(clap::Args)]
@@ -9,12 +10,77 @@ pub struct Args {
     name: PathBuf,
 }
 
-pub fn exec(Args { name }: Args) -> miette::Result<()> {
-    if !name.exists() {
-        fs::create_dir_all(name.join("lib")).into_diagnostic()?;
-        fs::create_dir_all(name.join("policies")).into_diagnostic()?;
-        fs::create_dir_all(name.join("scripts")).into_diagnostic()?;
+pub struct Creator {
+    root: PathBuf,
+    src: PathBuf,
+    project_name: String,
+}
+
+impl Creator {
+    fn new(args: Args) -> Self {
+        let root = args.name;
+        let src = root.join("src");
+        let project_name = root.clone().into_os_string().into_string().unwrap();
+        Self{
+            root,
+            src,
+            project_name
+        }
     }
 
+    fn run(&self) -> miette::Result<()> {
+        fs::create_dir_all(&self.src).into_diagnostic()?;
+
+        self.aiken_toml()?;
+        self.always_true_script()?;
+
+        Ok(())
+    }
+
+    fn always_true_script(&self) -> miette::Result<()> {
+        write(
+            self.src.join("always_true.ak"), 
+            "
+pub fn spend() -> Bool {
+    true
+}            
+"
+        )
+    }
+
+    fn aiken_toml(&self) -> miette::Result<()> {
+        write(
+            self.root.join("aiken.toml"),
+            &format!(
+                r#"name = "{name}"
+version = "0.1.0"
+licences = ["Apache-2.0"]
+description = "Aiken contracts"
+
+[dependencies]
+aiken = "~> {aiken}"
+
+"#,
+                name = self.project_name,
+                aiken = "0.0.24",
+            ),
+        )
+    }
+}
+
+fn write(path: PathBuf, contents: &str) -> miette::Result<()> {
+    let mut f = fs::File::create(&path).into_diagnostic()?;
+
+    f.write_all(contents.as_bytes()).into_diagnostic()?;
+    
+    Ok(())
+}
+
+pub fn exec(args: Args) -> miette::Result<()> {
+    if !args.name.exists() {
+        let creator = Creator::new(args);
+        creator.run()?;
+    } 
+
     Ok(())
 }