#!/usr/bin/env python3

from argparse import ArgumentParser
from getpass import getuser
from os import getuid, getgid, getlogin
from shutil import which
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()

if command is None:
    print(f"Assuming executable shares the same name as package_name {package_name}")
    command = package_name


dockerfile = f"""FROM archlinux:latest

# Install latest build packages and git
RUN pacman -Sy --noconfirm base-devel git

# Set up new user {user} with passwordless sudo
RUN useradd {user} -u {uid} -m && \\
        passwd -d {user} && \\
        set -o pipefail && \\
        printf '{user} ALL=(ALL) ALL\\n' | tee -a /etc/sudoers

# Set up AUR helper yay
USER {user}
WORKDIR /home/{user}
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}

ENTRYPOINT [\"/usr/bin/env\", \"{command}\"]"
"""

# check if either docker or podman are installed

try:
    if which("docker") is None:
        print("Docker not installed, building with Podman")
        engine = "podman"
    elif which("podman") is None:
        engine = "docker"
    else:
        print("Podman/Docker are not installed! Exiting...")
        exit(1)


    subprocess.run([engine, "build", "-t", command, "-"], input = dockerfile, encoding = 'UTF-8')
except FileNotFoundError:
    print("An Error has occured while attempting to build the image.")
    exit(1)