Disallow spaces and other troublesome characters in environment names

softpack_core/artifacts.py

@@ -53,13 +53,15 @@
 
     ready = 'ready'
     queued = 'queued'
+    failed = 'failed'
 
 
 class Artifacts:
     """Artifacts repo access class."""
 
     environments_root = "environments"
     environments_file = "softpack.yml"
+    builder_out = "builder.out"
     module_file = "module"
     readme_file = "README.md"
     built_by_softpack_file = ".built_by_softpack"
@@ -131,6 +133,8 @@
 
             if Artifacts.module_file in self.obj:
                 info["state"] = State.ready
+            elif Artifacts.builder_out in self.obj:
+                info["state"] = State.failed
             else:
                 info["state"] = State.queued
 

softpack_core/schemas/environment.py

@@ -5,6 +5,7 @@
 """
 
 import io
+import re
 from dataclasses import dataclass
 from pathlib import Path
 from traceback import format_exception_only
@@ -146,15 +147,24 @@
     description: str
     packages: list[PackageInput]
 
-    def validate(cls) -> Union[None, InvalidInputError]:
-        """Validate that all values have been supplied.
+    def validate(self) -> Union[None, InvalidInputError]:
+        """Validate all values.
+
+        Checks all values have been supplied.
+        Checks that name consists only of alphanumerics, dash, and underscore.
 
         Returns:
             None if good, or InvalidInputError if not all values supplied.
         """
-        if any(len(value) == 0 for value in vars(cls).values()):
+        if any(len(value) == 0 for value in vars(self).values()):
             return InvalidInputError(message="all fields must be filled in")
 
+        if not re.fullmatch(r"^[a-zA-Z0-9_-]+$", self.name):
+            return InvalidInputError(
+                message="name must only contain alphanumerics, "
+                "dash, and underscore"
+            )
+
         return None
 
     @classmethod
@@ -239,9 +249,9 @@
         Returns:
             A message confirming the success or failure of the operation.
         """
-        result = env.validate()
-        if result is not None:
-            return result
+        input_err = env.validate()
+        if input_err is not None:
+            return input_err
 
         name = env.name
         version = 1
@@ -262,9 +272,9 @@
         env.name = name
 
         # Send build request
         try:
             host = app.settings.builder.host
             port = app.settings.builder.port
             r = httpx.post(
                 f"http://{host}:{port}/environments/build",
                 json={
@@ -282,9 +292,9 @@
                     },
                 },
             )
             r.raise_for_status()
         except Exception as e:
             return BuilderError(
                 message="Connection to builder failed: "
                 + "".join(format_exception_only(type(e), e))
             )
@@ -340,9 +350,9 @@
                     for pkg in env.packages
                 ],
             )
             ymlData = yaml.dump(softpack_definition)
 
             tree_oid = cls.artifacts.create_files(
                 new_folder_path,
                 [
                     (env_type, ""),  # e.g. .built_by_softpack
@@ -354,7 +364,7 @@
                 tree_oid, "create environment folder"
             )
         except RuntimeError as e:
             return InvalidInputError(
                 message="".join(format_exception_only(type(e), e))
             )
 
@@ -574,7 +584,7 @@
             )
 
         except Exception as e:
             return InvalidInputError(
                 message="".join(format_exception_only(type(e), e))
             )
 

tests/integration/test_environment.py

@@ -93,12 +93,34 @@ def test_create(httpx_post, testable_env_input: EnvironmentInput) -> None:
     )
     assert file_in_remote(path)
 
-    orig_name = testable_env_input.name
+
+def test_create_name_empty_disallowed(httpx_post, testable_env_input):
     testable_env_input.name = ""
     result = Environment.create(testable_env_input)
     assert isinstance(result, InvalidInputError)
 
-    testable_env_input.name = orig_name
+
+def test_create_name_spaces_disallowed(httpx_post, testable_env_input):
+    testable_env_input.name = "names cannot have spaces"
+    result = Environment.create(testable_env_input)
+    assert isinstance(result, InvalidInputError)
+
+
+def test_create_name_slashes_disallowed(httpx_post, testable_env_input):
+    testable_env_input.name = "names/cannot/have/slashes"
+    result = Environment.create(testable_env_input)
+    assert isinstance(result, InvalidInputError)
+
+
+def test_create_name_dashes_and_number_first_allowed(
+    httpx_post, testable_env_input
+):
+    testable_env_input.name = "7-zip_piz-7"
+    result = Environment.create(testable_env_input)
+    assert isinstance(result, CreateEnvironmentSuccess)
+
+
+def test_create_path_invalid_disallowed(httpx_post, testable_env_input):
     testable_env_input.path = "invalid/path"
     result = Environment.create(testable_env_input)
     assert isinstance(result, InvalidInputError)
@@ -225,6 +247,21 @@ async def test_states(httpx_post, testable_env_input):
     assert env.type == Artifacts.built_by_softpack
     assert env.state == State.queued
 
+    upload = UploadFile(
+        filename=Artifacts.builder_out, file=io.BytesIO(b"some output")
+    )
+
+    result = await Environment.write_artifact(
+        file=upload,
+        folder_path=f"{testable_env_input.path}/{testable_env_input.name}-1",
+        file_name=upload.filename,
+    )
+    assert isinstance(result, WriteArtifactSuccess)
+
+    env = get_env_from_iter(testable_env_input.name + "-1")
+    assert env is not None
+    assert env.state == State.failed
+
     upload = UploadFile(
         filename=Artifacts.module_file, file=io.BytesIO(b"#%Module")
     )