add support for class components

Markos-Th09 · Apr 5, 2022 · 1d9f544 · 1d9f544
1 parent 8f6dea6
commit 1d9f544
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 23 deletions.
diff --git a/src/create_component_files.rs b/src/create_component_files.rs
@@ -5,6 +5,7 @@ pub mod files_creater {
         component_name: String,
         stylesheet_name: String,
         is_css_module: bool,
+        is_class_component: bool,
     ) {
         let index_file_content = format!(
             "export {{ default }} from \"./{}\";",
@@ -14,34 +15,48 @@ pub mod files_creater {
         let component_file_content;
         if is_css_module == true {
             css_file_content = "".to_string();
-            component_file_content = format!(
-                "import styles from './{:}';
-export default function {:}() {{
-    return (
-            <div className={{styles.h1Container}}>
-                <h1>Hello World</h1>
-            </div>
-            );
-}}
-                ",
-                stylesheet_name,
-                component_name.split(".").collect::<Vec<&str>>()[0]
-            );
+            if is_class_component == true {
+                component_file_content = format!(
+                    "import {{ Component }} from \"react\";\nimport styles from \"./{}\";\n\nclass {} extends Component {{}}\n\nexport default {}",
+                    stylesheet_name, component_name.split(".").collect::<Vec<&str>>()[0], component_name.split(".").collect::<Vec<&str>>()[0]
+                )
+            } else {
+                component_file_content = format!(
+                    "import styles from './{:}';
+                    export default function {:}() {{
+                        return (
+                            <div className={{styles.h1Container}}>
+                            <h1>Hello World</h1>
+                            </div>
+                        );
+                    }}
+                    ",
+                    stylesheet_name,
+                    component_name.split(".").collect::<Vec<&str>>()[0]
+                );
+            }
         } else {
             css_file_content = "".to_string();
-            component_file_content = format!(
-                "import './{:}';
-export default function {:}() {{
-    return (
-            <div className=\"h1Container\" >
-                <h1>Hello, world!</h1>
+            if is_class_component == true {
+                component_file_content = format!(
+                    "import {{ Component }} from \"react\";\nimport \"./{}\";\n\nclass {} extends Component {{}}\n\nexport default {}",
+                    stylesheet_name, component_name.split(".").collect::<Vec<&str>>()[0], component_name.split(".").collect::<Vec<&str>>()[0]
+                )
+            } else {
+                component_file_content = format!(
+                    "import './{:}';
+                    export default function {:}() {{
+                        return (
+                            <div className=\"h1Container\" >
+                            <h1>Hello, world!</h1>
             </div>
-            );
+        );
 }}
                 ",
-                stylesheet_name,
-                component_name.split(".").collect::<Vec<&str>>()[0]
-            );
+                    stylesheet_name,
+                    component_name.split(".").collect::<Vec<&str>>()[0]
+                );
+            }
         }
         let result = fs::create_dir(format!(
             "{:}",

diff --git a/src/main.rs b/src/main.rs
@@ -29,6 +29,10 @@ struct Args {
     /// The name of stylesheet you want to create
     #[clap(short = 'S', long, default_value = "css")]
     stylesheet: String,
+
+    /// Create a Class Based Component
+    #[clap(short='C',long)]
+    class: bool,
 }
 
 fn main() {
@@ -39,6 +43,7 @@ fn main() {
         module: css_module,
         hook,
         stylesheet: stylesheet_name,
+        class
     } = Args::parse();
 
     if hook {
@@ -67,5 +72,6 @@ fn main() {
         component_file_name,
         stylesheet.clone(),
         stylesheet.contains("module"),
+        class
     );
 }