Add InputStore and Enhance Template Renderer with Input Support (#14)

**Overview:**
This PR introduces several changes, primarily aimed at improving input
management during template generation and adding test coverage
enhancements.

**Changes:**
- Added a new `InputStore` class to manage input data storage in JSON
format.
- Updated the `TemplateRenderer` class to integrate with `InputStore`
for loading and saving user inputs.
- Enhanced the `generate.py` module by adding the `--input-store`
argument, enabling users to specify a file path for input storage.
- Refined argument handling in `generate.py` and improved the file
creation process with the new `input_store` field.
- Minor changes to the `FileItem` class to support `input_store`
integration.
- Added `pytest-cov` to the `requirements.dev.txt` for improved test
coverage.

**Justification:**
These updates are part of an effort to streamline the input handling
process for the structure generation tool. By using `InputStore`, the
system can persistently manage user-provided inputs across sessions,
reducing redundant data entry. Additionally, the integration with
`TemplateRenderer` allows for smoother processing of template variables.

**Impact:**
- Enhanced user experience by minimizing repetitive input prompts.
- Improved maintainability and flexibility in handling structured data.
- Additional test coverage through the inclusion of `pytest-cov`.

Author: httpdss

requirements.dev.txt

@@ -2,3 +2,4 @@ pytest
 flake8
 pytest-md-report
 pre-commit
+pytest-cov

struct_module/commands/generate.py

@@ -13,6 +13,7 @@ def __init__(self, parser):
     parser.add_argument('structure_definition', type=str, help='Path to the YAML configuration file')
     parser.add_argument('base_path', type=str, help='Base path where the structure will be created')
     parser.add_argument('-s', '--structures-path', type=str, help='Path to structure definitions')
+    parser.add_argument('-n', '--input-store', type=str, help='Path to the input store', default='/tmp/struct/input.json')
     parser.add_argument('-d', '--dry-run', action='store_true', help='Perform a dry run without creating any files or directories')
     parser.add_argument('-v', '--vars', type=str, help='Template variables in the format KEY1=value1,KEY2=value2')
     parser.add_argument('-b', '--backup', type=str, help='Path to the backup folder')
@@ -65,13 +66,16 @@ def _create_structure(self, args):
           content["name"] = name
           content["global_system_prompt"] = args.global_system_prompt
           content["config_variables"] = config_variables
+          content["input_store"] = args.input_store
           file_item = FileItem(content)
           file_item.fetch_content()
         elif isinstance(content, str):
           file_item = FileItem(
-            {"name": name,
-             "content": content,
-             "config_variables": config_variables,
+            {
+              "name": name,
+              "content": content,
+              "config_variables": config_variables,
+              "input_store": args.input_store,
             }
           )
 
@@ -108,6 +112,7 @@ def _create_structure(self, args):
               'backup': args.backup,
               'file_strategy': args.file_strategy,
               'global_system_prompt': args.global_system_prompt,
+              'input_store': args.input_store,
             })
           elif isinstance(content['struct'], list):
             for struct in content['struct']:
@@ -120,6 +125,7 @@ def _create_structure(self, args):
                 'backup': args.backup,
                 'file_strategy': args.file_strategy,
                 'global_system_prompt': args.global_system_prompt,
+                'input_store': args.input_store,
               })
         else:
           self.logger.warning(f"Unsupported content in folder: {folder}")

struct_module/file_item.py

@@ -22,6 +22,7 @@ def __init__(self, properties):
       self.config_variables = properties.get("config_variables")
       self.content_location = properties.get("file")
       self.permissions = properties.get("permissions")
+      self.input_store = properties.get("input_store")
 
       self.system_prompt = properties.get("system_prompt") or properties.get("global_system_prompt")
       self.user_prompt = properties.get("user_prompt")
@@ -30,7 +31,7 @@ def __init__(self, properties):
       if openai_api_key:
         self._configure_openai()
 
-      self.template_renderer = TemplateRenderer(self.config_variables)
+      self.template_renderer = TemplateRenderer(self.config_variables, self.input_store)
 
     def _configure_openai(self):
       self.openai_client = OpenAI(api_key=openai_api_key)

struct_module/input_store.py

@@ -0,0 +1,35 @@
+import json
+import os
+
+class InputStore:
+
+  def __init__(self, input_file):
+    self.input_file = input_file
+    self.data = None
+
+    # create directory if it doesn't exist
+    directory = os.path.dirname(input_file)
+    if not os.path.exists(directory):
+      os.makedirs(directory)
+
+    # create file if it doesn't exist
+    if not os.path.exists(input_file):
+      with open(input_file, 'w') as f:
+        json.dump({}, f)
+
+  def load(self):
+    with open(self.input_file, 'r') as f:
+      self.data = json.load(f)
+
+  def get_data(self):
+    return self.data
+
+  def get_value(self, key):
+    return self.data[key]
+
+  def set_value(self, key, value):
+    self.data[key] = value
+
+  def save(self):
+    with open(self.input_file, 'w') as f:
+      json.dump(self.data, f, indent=2)

struct_module/template_renderer.py

@@ -2,9 +2,10 @@
 import logging
 from jinja2 import Environment, meta
 from struct_module.filters import get_latest_release, slugify
+from struct_module.input_store import InputStore
 
 class TemplateRenderer:
-    def __init__(self, config_variables):
+    def __init__(self, config_variables, input_store):
       self.config_variables = config_variables
       self.env = Environment(
         trim_blocks=True,
@@ -22,6 +23,9 @@ def __init__(self, config_variables):
       }
       self.env.filters.update(custom_filters)
       self.logger = logging.getLogger(__name__)
+      self.input_store = InputStore(input_store)
+      self.input_store.load()
+      self.input_data = self.input_store.get_data()
 
     # Get the config variables from the list and create a dictionary that has
     # variable name and their default value
@@ -64,9 +68,11 @@ def prompt_for_missing_vars(self, content, vars):
 
       for var in undeclared_variables:
         if var not in vars:
-          default = default_values.get(var, "")
+          default = self.input_data.get(var, default_values.get(var, ""))
           user_input = input(f"Enter value for {var} [{default}]: ")
           if not user_input:
             user_input = default
+          self.input_store.set_value(var, user_input)
           vars[var] = user_input
+      self.input_store.save()
       return vars