Improved new console command

src/cpl_cli/command/new.py

@@ -1,7 +1,7 @@
 import json
 import os
 import sys
-from distutils.dir_util import copy_tree
+from typing import Optional
 
 from cpl.application.application_runtime_abc import ApplicationRuntimeABC
 from cpl.configuration.configuration_abc import ConfigurationABC
@@ -12,6 +12,13 @@ from cpl_cli.configuration.build_settings_name import BuildSettingsName
 from cpl_cli.configuration.project_settings import ProjectSettings
 from cpl_cli.configuration.project_settings_name import ProjectSettingsName
 from cpl_cli.configuration.version_settings_name import VersionSettingsName
+from cpl_cli.templates.new.console.license import LicenseTemplate
+from cpl_cli.templates.new.console.readme_py import ReadmeTemplate
+from cpl_cli.templates.new.console.src.application import ApplicationTemplate
+from cpl_cli.templates.new.console.src.main import MainWithApplicationHostAndStartupTemplate, MainWithoutApplicationHostTemplate, MainWithApplicationHostTemplate
+from cpl_cli.templates.new.console.src.startup import StartupTemplate
+from cpl_cli.templates.new.console.src.tests.init import TestsInitTemplate
+from cpl_cli.templates.template_file_abc import TemplateFileABC
 
 
 class New(CommandABC):
@@ -29,6 +36,9 @@ class New(CommandABC):
         self._project_json = {}
 
         self._command: str = ''
+        self._use_application_api: bool = False
+        self._use_startup: bool = False
+        self._use_service_providing: bool = False
 
     def _create_project_settings(self, name: str):
         self._project_dict = {
@@ -76,12 +86,29 @@ class New(CommandABC):
             BuildSettings.__name__: self._build_dict
         }
 
-    def _create_project_dir(self):
+    def _get_project_path(self) -> Optional[str]:
         project_path = os.path.join(self._runtime.working_directory, self._project.name)
         if os.path.isdir(project_path) and len(os.listdir(project_path)) > 0:
             Console.error('Project path is not empty\n')
-            exit()
+            return None
 
+        return project_path
+
+    def _get_project_informations(self):
+        result = Console.read('Do you want to use application host? (y/n) ')
+        if result.lower() == 'y':
+            self._use_application_api = True
+
+        if self._use_application_api:
+            result = Console.read('Do you want to use startup? (y/n) ')
+            if result.lower() == 'y':
+                self._use_startup = True
+
+        # result = Console.read('Do you want to use service providing? (y/n) ')
+        # if result.lower() == 'y':
+        #     self._use_service_providing = True
+
+    def _build_project_dir(self, project_path: str):
         if not os.path.isdir(project_path):
             os.makedirs(project_path)
 
@@ -89,19 +116,52 @@ class New(CommandABC):
             project_json.write(json.dumps(self._project_json, indent=4))
             project_json.close()
 
-        template_path = os.path.join(self._runtime.runtime_directory, f'templates/new/{self._command}')
-        if not os.path.isdir(template_path):
-            Console.error(template_path, '\n\nTemplate path not found\n')
-            exit()
+        templates: list[TemplateFileABC] = [
+            LicenseTemplate(),
+            ReadmeTemplate(),
+            TestsInitTemplate()
+        ]
+        if self._use_application_api:
+            templates.append(ApplicationTemplate())
+            if self._use_startup:
+                templates.append(StartupTemplate())
+                templates.append(MainWithApplicationHostAndStartupTemplate())
+            else:
+                templates.append(MainWithApplicationHostTemplate())
+        else:
+            templates.append(MainWithoutApplicationHostTemplate())
 
-        copy_tree(template_path, project_path)
+        for template in templates:
+            Console.spinner(f'Creating {template.path}{template.name}', self._create_template, project_path, template)
+
+    @staticmethod
+    def _create_template(project_path: str, template: TemplateFileABC):
+        file_path = os.path.join(project_path, template.path, template.name)
+        file_rel_path = os.path.join(project_path, template.path)
+
+        if not os.path.isdir(file_rel_path):
+            os.makedirs(file_rel_path)
+
+        with open(file_path, 'w') as license_file:
+            license_file.write(template.value)
+            license_file.close()
 
     def _console(self, args: list[str]):
         name = self._config.get_configuration(self._command)
-        self._create_project_settings(name)
-        self._create_build_settings()
-        self._create_project_json()
-        self._create_project_dir()
+
+        Console.spinner('Creating project settings', self._create_project_settings, name)
+        Console.spinner('Creating build settings', self._create_build_settings)
+        path = self._get_project_path()
+        if path is None:
+            return
+
+        Console.spinner('Creating project file', self._create_project_json)
+        Console.write_line('Building project directory:')
+        self._get_project_informations()
+        try:
+            self._build_project_dir(path)
+        except Exception as e:
+            Console.error('Could not create project', str(e))
 
     def run(self, args: list[str]):
         self._command = args[0]