From e40c63f8a6b6cdf2f7e19fe4b66bde1d6725aef0 Mon Sep 17 00:00:00 2001
From: Gonzalo Martinez <gonzalo.martinez@litermi.com>
Date: Tue, 27 Aug 2024 13:44:24 -0300
Subject: [PATCH] First commit

---
 cloud-import.py  | 87 ++++++++++++++++++++++++++++++++++++++++++++++++
 install.sh       | 17 ++++++++++
 requirements.txt |  1 +
 3 files changed, 105 insertions(+)
 create mode 100644 cloud-import.py
 create mode 100644 install.sh
 create mode 100644 requirements.txt

diff --git a/cloud-import.py b/cloud-import.py
new file mode 100644
index 0000000..e2b3480
--- /dev/null
+++ b/cloud-import.py
@@ -0,0 +1,87 @@
+#!/usr/bin/python3
+
+import signal
+import argparse
+import os
+import requests
+from simple_term_menu import TerminalMenu
+
+def getImagesList():
+    ImagesURL="https://gist.githubusercontent.com/ggMartinez/f20f83d6a7630ab49d782abfe9017bc5/raw/3fb857a37b413403322e21469e155922bbac7d0c/sources.json"
+    return requests.get(ImagesURL).json()
+
+def getStorage():
+    return "local-lvm"
+
+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}" | cut -d"." -f1').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()
+
+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()
diff --git a/install.sh b/install.sh
new file mode 100644
index 0000000..6a30524
--- /dev/null
+++ b/install.sh
@@ -0,0 +1,17 @@
+#!/bin/bash
+
+echo "Installing Python Pip" 
+sudo apt-get install python3-pip
+
+echo "Installing requirements"
+pip3 install -r requirements.txt
+
+echo "Downloading importer"
+git clone https://github.com/ggMartinez/Proxmox-Cloud-Image-Importer /opt/Proxmox-Cloud-Image-Importer
+
+echo "Creating symlink"
+ln -s /opt/Proxmox-Cloud-Image-Importer/cloud-import.py /usr/bin/cloud-import 
+
+
+echo "Installed!! Run with 'cloud-import'"
+echo "If you want to update, run 'cd /opt/Proxmox-Cloud-Image-Importer && git pull'."
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..0e3c531
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+simple_term_menu