fix: resolve CI test failures (config access, mocks, imports)
This commit is contained in:
@@ -82,7 +82,7 @@ Format your response:"""
|
|||||||
class ShellParser:
|
class ShellParser:
|
||||||
"""Parser for shell scripts."""
|
"""Parser for shell scripts."""
|
||||||
|
|
||||||
LINE_PATTERN = re.compile(r"^(?:(\s*)(.*?)(\s*))", re.MULTILINE)
|
LINE_PATTERN = re.compile(r"^(?:(\s*)(.*?)(\s*))$", re.MULTILINE)
|
||||||
COMMENT_PATTERN = re.compile(r"^\s*#")
|
COMMENT_PATTERN = re.compile(r"^\s*#")
|
||||||
SHEBANG_PATTERN = re.compile(r"^#!(.+)$", re.MULTILINE)
|
SHEBANG_PATTERN = re.compile(r"^#!(.+)$", re.MULTILINE)
|
||||||
|
|
||||||
@@ -234,7 +234,7 @@ class ShellSafetyChecker:
|
|||||||
|
|
||||||
DANGEROUS_PATTERNS = [
|
DANGEROUS_PATTERNS = [
|
||||||
(r"rm\s+-rf\s+/", "Removes all files recursively - catastrophic data loss"),
|
(r"rm\s+-rf\s+/", "Removes all files recursively - catastrophic data loss"),
|
||||||
(r":\(\)\s*\{\s*:\s*\|\s*:\s*&\s*;\s*\}", "Fork bomb - can crash system"),
|
(r":\(\)\s*\{\s*:\s*\|\s*:\s&\s*;\s*\}", "Fork bomb - can crash system"),
|
||||||
(r"chmod\s+777", "Removes all file permissions - security risk"),
|
(r"chmod\s+777", "Removes all file permissions - security risk"),
|
||||||
(r"sudo\s+su", "Escalates to root without proper controls"),
|
(r"sudo\s+su", "Escalates to root without proper controls"),
|
||||||
(r"dd\s+if=/dev/zero", "Can overwrite disks if misconfigured"),
|
(r"dd\s+if=/dev/zero", "Can overwrite disks if misconfigured"),
|
||||||
@@ -343,7 +343,5 @@ def generate_shell(
|
|||||||
generator = ShellGenerator()
|
generator = ShellGenerator()
|
||||||
if safety_level:
|
if safety_level:
|
||||||
config = get_config()
|
config = get_config()
|
||||||
if "safety" not in config:
|
config.config["safety"]["level"] = safety_level
|
||||||
config["safety"] = {}
|
|
||||||
config["safety"]["level"] = safety_level
|
|
||||||
return generator.generate(description, shell_type)
|
return generator.generate(description, shell_type)
|
||||||
|
|||||||
Reference in New Issue
Block a user