Merge pull request #48 from intelops/feat/argocd_cronworkflow

added argocd cronworkflow integration and option to add multiple docker file path to rebuild the images.

Author: jegathintelops

argo-cron.yml

@@ -0,0 +1,15 @@
+apiVersion: argoproj.io/v1alpha1
+kind: CronWorkflow
+metadata:
+  name: scsctl-cronworkflow
+spec:
+  schedule: "* * * * *"
+  concurrencyPolicy: "Replace"
+  startingDeadlineSeconds: 0
+  workflowSpec:
+    entrypoint: scsctl-server-test
+    templates:
+    - name: scsctl-server-test
+      container:
+        image: ghcr.io/jegathintelops/scsctl:latest
+

manifest.yaml

@@ -0,0 +1,31 @@
+# Create a manifest for scsctl with a command to run the container
+apiVersion: v1
+kind: Pod
+metadata:
+  name: scsctl-server
+  labels:
+    app: scsctl-server
+spec:
+  containers:
+  - name: scsctl-server
+    image: ghcr.io/jegathintelops/scsctl:latest
+    imagePullPolicy: Always
+    args: ["server"]
+    ports:
+     - containerPort: 5000
+  restartPolicy: Never
+
+# Create a service to expose the scsctl container
+---
+apiVersion: v1
+kind: Service
+metadata:
+  name: scsctl-loadbalancer
+spec:
+  selector:
+    app: scsctl-server
+  ports:
+    - protocol: TCP
+      port: 5000
+      targetPort: 5000
+  type: LoadBalancer

src/scsctl/app.py

@@ -13,7 +13,7 @@
     save_pyroscope_data,
     compare_and_find_pyroscope_extra_packages,
 )
-from scsctl.helper.common import AppDetails, generate_final_report, modify_and_build_docker_image, custom_style_fancy
+from scsctl.helper.common import AppDetails, generate_final_report, modify_and_build_docker_image,modify_and_build_docker_images, custom_style_fancy
 from scsctl.helper.trivy import get_sbom_report, print_sbom_report, save_sbom_data
 from scsctl.helper.renovate import (check_if_node_and_npm_is_installed,check_if_renovate_is_installed_globally,run_renovate_on_a_repository)
 
@@ -54,7 +54,7 @@ def cli():
 @click.option("--renovate_repo_name", help="Repo name for renovate", default=None, is_flag=False, flag_value=None)
 @click.option("--non_interactive", help="Run scsctl in non interactive mode", default= False, is_flag=True, flag_value=True)
 @click.option(
-    "--docker_file_folder_path", help="Path of the docker file to rebuild", default=None, is_flag=False, flag_value=None
+    "--docker_file_folder_path", help="Path of the docker file to rebuild", default=None, is_flag=False, flag_value=None, multiple=True
 )
 @click.option("--config_file", help="Path of the configuration file", default=None, is_flag=False, flag_value=None)
 
@@ -220,7 +220,8 @@ def scan(
                 if choice == "Rebuild the image":
                     if docker_file_folder_path == None:
                         docker_file_folder_path = click.prompt("Enter docker file folder path", type=str)
-                    modify_and_build_docker_image(docker_file_folder_path, pyroscope_found_extra_packages, batch_id)
+                    # modify_and_build_docker_image(docker_file_folder_path, pyroscope_found_extra_packages, batch_id)
+                    modify_and_build_docker_images(file_paths=docker_file_folder_path,package_names=pyroscope_found_extra_packages,batch_id=batch_id)
 
 
 cli.add_command(scan)

src/scsctl/helper/common.py

@@ -35,6 +35,37 @@ class AppDetails:
     pyroscope_app_name: str
     pyroscope_url: str
 
+def modify_and_build_docker_images(file_paths: list, package_names: list, batch_id: str):
+    counter = 0
+    for file_path in file_paths:
+        if os.path.exists("./temp"):
+            shutil.rmtree("./temp/")
+        #create folder
+        os.makedirs("./temp")
+        shutil.copyfile(file_path, "./temp/Dockerfile")
+
+        # Create a new file which contains all the packages names to uninstall
+        with open("./temp/packages.txt", "w") as f:
+            f.write("\n".join(package_names))
+        # Add the uninstall commands at the end of the file
+        with open("./temp/Dockerfile", "a") as f:
+            f.write("\nCOPY packages.txt /tmp/packages.txt")
+            f.write(
+                '\nRUN while read -r package; do \\\n   if dpkg-query -W --showformat=\'${Essential}\' "$package" | grep -q \'^no$\'; then \\\n   apt-get remove -y "$package"; \\\n    else \\\n   echo "Skipping essential package: $package"; \\\n   fi; \\\ndone < /tmp/packages.txt'
+            )
+
+        # Build the docker image with the modified Dockerfile and tag it with the batch id
+        click.echo("Building the docker image...")
+        try:
+            subprocess.check_output(["docker", "build", "-t", f"{batch_id}_{counter}", "./temp/"])
+        except subprocess.CalledProcessError as e:
+            click.echo(f"Error building the docker image: {e}")
+            return False
+
+        # Remove the temp folder
+        if os.path.exists("./temp"):
+            shutil.rmtree("./temp/")
+        counter += 1
 
 def modify_and_build_docker_image(folder_path: str, package_nammes: list, bacth_id: str):
     # Make a copy of folder in a ./temp folder, create folder if it doesn't exist