From cfd98a7131fb01278498eb80973ac03421a59907 Mon Sep 17 00:00:00 2001
From: "Stefano C. Coronado" <stefanocorona12@gmail.com>
Date: Mon, 30 Dec 2019 14:13:43 -0500
Subject: [PATCH] added ability to add extra step into generic dockerfile

---
 pkg-install | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/pkg-install b/pkg-install
index bcfbfb2..ea94205 100755
--- a/pkg-install
+++ b/pkg-install
@@ -9,10 +9,12 @@ import subprocess
 parser = ArgumentParser(description="Install AUR packages into a Docker container")
 parser.add_argument("package_name", type=str, help="name of the AUR package")
 parser.add_argument("-x", type=str, metavar='EXECUTABLE', help="name of executable")
+parser.add_argument("-e", type=str, help="edit generic dockerfile")
 args = vars(parser.parse_args())
 
 package_name = args['package_name']
 command = args['x']
+edit_flag = args['e']
 user = getlogin()
 uid = getuid()
 gid = getgid()
@@ -40,6 +42,14 @@ RUN git clone https://aur.archlinux.org/yay.git
 WORKDIR /home/{user}/yay
 RUN makepkg -si --noconfirm
 
+"""
+if edit_flag is not None: #if -e flag is passed then take on the following string
+    extra_run = f"""RUN {edit_flag}"""
+else:
+    extra_run = ""
+
+dockerfile += extra_run + f"""
+
 # Now you can play with whatever package you'd like
 RUN yay -S --noconfirm {package_name}