From bd3f159e289ea1db4eee16c8966acd7ee219a146 Mon Sep 17 00:00:00 2001
From: "pixeebot[bot]" <104101892+pixeebot[bot]@users.noreply.github.com>
Date: Sat, 15 Jun 2024 13:19:10 +0000
Subject: [PATCH] Sandbox Process Creation

---
 jtd_codebuild/generators/generator.py            | 4 ++--
 jtd_codebuild/generators/typescript_generator.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/jtd_codebuild/generators/generator.py b/jtd_codebuild/generators/generator.py
index e14d902..668f6b5 100644
--- a/jtd_codebuild/generators/generator.py
+++ b/jtd_codebuild/generators/generator.py
@@ -2,6 +2,7 @@
 import subprocess
 from typing import Dict, Any, AnyStr, List
 from ..utils import safe_mkdir
+from security import safe_command
 
 
 class JTDCodeGenerator:
@@ -66,8 +67,7 @@ def generate(
         target_language = target["language"]
         target_path = self.get_target_path(target)
         safe_mkdir(target_path)
-        process = subprocess.Popen(
-            self._codegen_command(self.schema_path, target_path, target_language),
+        process = safe_command.run(subprocess.Popen, self._codegen_command(self.schema_path, target_path, target_language),
             shell=True,
             stdout=subprocess.PIPE,
             stderr=subprocess.PIPE,
diff --git a/jtd_codebuild/generators/typescript_generator.py b/jtd_codebuild/generators/typescript_generator.py
index 1777441..4e6a431 100644
--- a/jtd_codebuild/generators/typescript_generator.py
+++ b/jtd_codebuild/generators/typescript_generator.py
@@ -3,6 +3,7 @@
 from typing import Dict, Any, AnyStr, List
 from ..utils import wait_for_processes
 from .generator import JTDCodeGenerator
+from security import safe_command
 
 
 class JTDCodeGeneratorTypescriptTarget(JTDCodeGenerator):
@@ -19,8 +20,7 @@ def _compile_typescript(self, tsconfig_path: str) -> subprocess.Popen:
         Args:
             tsconfig_path: The path to the tsconfig.json file.
         """
-        return subprocess.run(
-            f"tsc --project {tsconfig_path}",
+        return safe_command.run(subprocess.run, f"tsc --project {tsconfig_path}",
             shell=True,
             stdout=subprocess.PIPE,
             stderr=subprocess.PIPE,