#!/usr/bin/python3

import signal
import argparse
import os
import requests
from simple_term_menu import TerminalMenu

def getImagesList():
    ImagesURL="https://git.djg.dev/dahjah/Proxmox-Cloud-Image-Importer/raw/branch/main/sources.json"
    return requests.get(ImagesURL).json()

def getStorage():
    return "local"

def getId():
    id = 20000
    output = os.popen('qm list').read()
    if str(id) in output:
        id = int(os.popen('echo $(qm list | tail -| tr -s " " | cut -d " " -f2 | tail -1)').read()) + 1
    return id

def importTemplate(name, storage, id):
    file = name.replace(" ", "")
    file = f"/tmp/{file}"
    print(f"Importing {file} to {storage} with id {id}")
    templateName = os.path.basename(file).split(".")[0]
    os.popen(f'qm create {id} --memory 2048 --net0 virtio,bridge=vmbr0 --scsihw virtio-scsi-pci  --name "{templateName}" --tag template').read()
    os.popen(f'qm set {id} --scsi0 {storage}:0,import-from={file}').read()
    os.popen(f'qm set {id} --ide2 {storage}:cloudinit').read()
    os.popen(f'qm set {id} --boot order=scsi0').read()
    os.popen(f'qm set {id} --serial0 socket --vga serial0').read()
    os.popen(f'qm template {id}').read()
    # add a tag to the vm 
    

def generateMenuOptions():
    ImagesList = getImagesList()
    menuOptions = []
    menuOptions.append("Download All")
    for image in ImagesList:
        menuOptions.append(image['Name'])
    return menuOptions

def downloadImage(image):
    file = image["Name"].replace(" ", "")
    if str(file) in os.popen('qm list').read():
        return False
    print(f"Downloading template '{image['Name']}'... ")
    with open(f"/tmp/{file}", 'wb') as f:
        f.write(requests.get(image["URL"]).content)
    return True

def deleteImage(file):
    file = file.replace(" ","")
    os.remove(f"/tmp/{file}")

def importAllImages():
    ImagesList = getImagesList()
    for image in ImagesList:
        downloadImage(image)
        importTemplate(image['Name'], getStorage(), getId())
        deleteImage(image['Name'])

def importImage(index):
    ImagesList = getImagesList()
    if(downloadImage(ImagesList[index])):
        importTemplate(ImagesList[index]['Name'], getStorage(), getId())
        deleteImage(ImagesList[index]['Name'])
    else:
        print(f"Template '{ImagesList[index]['Name']}' already exists")

def showMenu():
    terminalMenu = TerminalMenu(generateMenuOptions())
    menuEntryIndex = terminalMenu.show()
    if menuEntryIndex == 0:
        importAllImages()
    else:
        importImage(menuEntryIndex-1)



def signalHandler(signum,frame):
    print("Exiting...                                                          ")
    exit(1)

if __name__ == "__main__":
    signal.signal(signal.SIGINT,signalHandler)
    showMenu()