114 lines
5.0 KiB
Python
114 lines
5.0 KiB
Python
import os, settings, app
|
|
from sh import du
|
|
from settings import MAIN_REPO
|
|
from utils import download_checksum, validate_checksum
|
|
from utils import get_message, check_equal_sentence as equals, generate_checksum, remove_checksum_files
|
|
class Danix():
|
|
|
|
@staticmethod
|
|
def copy(environent_is_first, filesystem_name, environment_dir, host_directory):
|
|
|
|
if environent_is_first:
|
|
return os.system(f"cp -r {MAIN_REPO}{filesystem_name}/danixfs{environment_dir} {host_directory} >/dev/null 2>&1")
|
|
else:
|
|
return os.system(f"cp -r {host_directory} {MAIN_REPO}{filesystem_name}/danixfs{environment_dir} >/dev/null 2>&1")
|
|
|
|
@staticmethod
|
|
def remove_snapshot(snapshot_name):
|
|
return os.system(f"rm -r {MAIN_REPO}.snapshots/{snapshot_name} >/dev/null 2>&1")
|
|
|
|
@staticmethod
|
|
def rm(filesystem_name):
|
|
|
|
return os.system(f"rm -r {MAIN_REPO}{filesystem_name}")
|
|
|
|
@staticmethod
|
|
def get_size(environment_name, snapshot_name):
|
|
try:
|
|
|
|
if snapshot_name is None:
|
|
return du(
|
|
f'{MAIN_REPO}{environment_name}/',f'-ch',
|
|
f'--exclude={MAIN_REPO}{environment_name}/danixfs/proc/*',
|
|
f'--exclude={MAIN_REPO}{environment_name}/danixfs/dev/*',
|
|
f'--exclude={MAIN_REPO}{environment_name}/danixfs/sys/*'
|
|
).split('\n')[-2].split('\t')[0]
|
|
else:
|
|
return du(f'{MAIN_REPO}.snapshots/{snapshot_name}/{environment_name}.tar.gz','-ch').split('\n')[-2].split('\t')[0]
|
|
except Exception as e:
|
|
print(e)
|
|
return "00M"
|
|
|
|
@staticmethod
|
|
def make_snapshot(filesystem_name, snapshot_name):
|
|
|
|
os.system(f"mkdir {MAIN_REPO}.snapshots/{snapshot_name} >/dev/null 2>&1")
|
|
exclude_proc_dir = f"{MAIN_REPO}{filesystem_name}/danixfs/proc/*"
|
|
exclude_sys_dir = f"{MAIN_REPO}{filesystem_name}/danixfs/sys/*"
|
|
exclude_dev_dir = f"{MAIN_REPO}{filesystem_name}/danixfs/dev/*"
|
|
|
|
resp = os.system(f"tar -czf {MAIN_REPO}.snapshots/{snapshot_name}/{filesystem_name}.tar.gz --exclude={exclude_proc_dir} --exclude={exclude_dev_dir} --exclude={exclude_sys_dir} {MAIN_REPO}{filesystem_name}/ >/dev/null 2>&1")
|
|
|
|
return resp
|
|
|
|
@staticmethod
|
|
def back_snapshot(filesystem_name, snapshot_name):
|
|
|
|
Danix.rm(filesystem_name)
|
|
resp1 = os.system(f"tar -xf {MAIN_REPO}.snapshots/{snapshot_name}/{filesystem_name}.tar.gz -C {MAIN_REPO} >/dev/null 2>&1")
|
|
resp2 = os.system(f"mv {MAIN_REPO}opt/danix/{filesystem_name} {MAIN_REPO}")
|
|
resp3 = os.system(f"rm -r {MAIN_REPO}opt")
|
|
|
|
return 0 if resp1+resp2+resp3 == 0 else 1
|
|
|
|
@staticmethod
|
|
def navigate(filesystem_uuid):
|
|
return os.system(f"{MAIN_REPO}{filesystem_uuid}/danixfs/proot -r {MAIN_REPO}{filesystem_uuid}/danixfs/ -w / -0 -b /dev -b /sys -b /proc -b /run sh {MAIN_REPO}init.d/init.sh")
|
|
|
|
@staticmethod
|
|
def build_environment(packages, config_comands, filesystem_uuid):
|
|
filesystem = filesystem_uuid
|
|
|
|
os.system(f"mkdir /tmp/{filesystem}")
|
|
os.system(f"curl --silent -LO --output-dir /tmp/{filesystem} {settings.REPO_NAME}/{settings.ROOT_FS}")
|
|
|
|
resp_downloaded_checksum = download_checksum()
|
|
resp_generateed_checksum = generate_checksum(f"/tmp/{filesystem}/{settings.ROOT_FS}")
|
|
|
|
is_downloaded_and_generate = equals(resp_downloaded_checksum, 0) and equals(resp_generateed_checksum, 0)
|
|
|
|
is_valid_checksum = validate_checksum(filesystem)
|
|
|
|
remove_checksum_files()
|
|
|
|
if is_downloaded_and_generate and is_valid_checksum:
|
|
|
|
os.system(f"tar -xf /tmp/{filesystem}/{settings.ROOT_FS} -C /tmp/{filesystem}")
|
|
os.system(f"rm /tmp/{filesystem}/{settings.ROOT_FS}")
|
|
os.system(f"mv /tmp/{filesystem} {MAIN_REPO}")
|
|
|
|
print("\nPlease! Wait a moment!!")
|
|
print("Building container:")
|
|
print(f"Installing {len(packages)} packages\n")
|
|
|
|
for command in config_comands:
|
|
os.system(f"chroot {MAIN_REPO}{filesystem}/danixfs {command}")
|
|
|
|
for package in packages:
|
|
os.system(f"chroot {MAIN_REPO}{filesystem}/danixfs apk add {package}")
|
|
|
|
os.system(f"chroot {MAIN_REPO}{filesystem}/danixfs apk add fish")
|
|
os.system(f"chroot {MAIN_REPO}{filesystem}/danixfs apk add ruby")
|
|
os.system(f"chroot {MAIN_REPO}{filesystem}/danixfs gem install lolcat")
|
|
|
|
os.system(f"rm -r {MAIN_REPO}{filesystem}/danixfs/dev >/dev/null 2>&1")
|
|
os.system(f"rm -r {MAIN_REPO}{filesystem}/danixfs/proc >/dev/null 2>&1")
|
|
os.system(f"rm -r {MAIN_REPO}{filesystem}/danixfs/sys >/dev/null 2>&1")
|
|
|
|
print(f"Environment builded succesfully!")
|
|
print("0 erros reported!")
|
|
|
|
Danix.navigate(filesystem)
|
|
else:
|
|
get_message("🔴 Failed integrity base image verification!", True, 1)
|