mirror of
https://github.com/splunk/security_content
synced 2026-06-08 17:32:49 +00:00
peter-cg
80 lines
2.9 KiB
Python
80 lines
2.9 KiB
Python
|
|
import git
|
|
import os
|
|
import logging
|
|
import glob
|
|
|
|
|
|
# Logger
|
|
logging.basicConfig(level=os.environ.get("LOGLEVEL", "INFO"))
|
|
LOGGER = logging.getLogger(__name__)
|
|
|
|
SECURITY_CONTENT_URL = "https://github.com/splunk/security_content"
|
|
|
|
|
|
class GithubService:
|
|
|
|
def __init__(self, security_content_branch):
|
|
self.security_content_branch = security_content_branch
|
|
if os.path.exists('security_content'):
|
|
LOGGER.warning(f"Found Existing Security Content Project")
|
|
self.created_repo = False
|
|
else:
|
|
self.security_content_repo_obj = self.clone_project(SECURITY_CONTENT_URL, f"security_content", f"develop")
|
|
self.security_content_repo_obj.git.checkout(security_content_branch)
|
|
self.created_repo = True
|
|
|
|
|
|
def clone_project(self, url, project, branch):
|
|
LOGGER.info(f"Clone Security Content Project")
|
|
repo_obj = git.Repo.clone_from(url, project, branch=branch)
|
|
return repo_obj
|
|
|
|
def get_changed_test_files_ssa(self):
|
|
branch1 = self.security_content_branch
|
|
branch2 = 'develop'
|
|
g = git.Git('security_content')
|
|
changed_ssa_test_files = []
|
|
ci_changes = False
|
|
|
|
if branch1 != 'develop':
|
|
differ = g.diff('--name-only', branch1, branch2)
|
|
changed_files = differ.splitlines()
|
|
|
|
for file_path in changed_files:
|
|
# added or changed test files
|
|
if file_path.startswith('tests'):
|
|
if os.path.basename(file_path).startswith('ssa'):
|
|
if file_path not in changed_ssa_test_files:
|
|
changed_ssa_test_files.append(file_path)
|
|
|
|
# changed detections
|
|
if file_path.startswith('detections'):
|
|
if os.path.basename(file_path).startswith('ssa'):
|
|
file_path_base = os.path.splitext(file_path)[0].replace('detections', 'tests') + '.test'
|
|
file_path_new = file_path_base + '.yml'
|
|
if file_path_new not in changed_ssa_test_files:
|
|
changed_ssa_test_files.append(file_path_new)
|
|
|
|
# changed CI code
|
|
if file_path == '.gitlab-ci.yml' or file_path.startswith('bin/ssa-end-to-end-testing'):
|
|
ci_changes = True
|
|
|
|
# all SSA test files for nightly build
|
|
else:
|
|
changed_files = sorted(glob.glob('security_content/tests/*/*.yml'))
|
|
|
|
for file_path in changed_files:
|
|
file_path = file_path.replace('security_content/','')
|
|
if os.path.basename(file_path).startswith('ssa'):
|
|
changed_ssa_test_files.append(file_path)
|
|
|
|
# changed CI code
|
|
if file_path == '.gitlab-ci.yml' or file_path.startswith('bin/ssa-end-to-end-testing'):
|
|
ci_changes = True
|
|
|
|
return changed_ssa_test_files, ci_changes
|
|
|
|
|
|
|