#!/usr/bin/python3 -u

import argparse
import os
from pathlib import Path
from subprocess import call, DEVNULL
import json
import time
import shutil
import pwd
import distro
import subprocess
import re
import sys


parser=argparse.ArgumentParser(prog='delete_flask.py', description='A tool for delete flask')

parser.add_argument('--home_user', help='Home where flask resides', required=True)
parser.add_argument('--user', help='The user', required=True)

print('Deleting Flask App..')

args=parser.parse_args()

home_user=args.home_user

if home_user=='/':
    print('What are you doing?')
    sys.exit(1)

name_service=os.path.basename(home_user[:-1]).strip()

#d=os.path.basename(p[:-1].replace(f, '')[:-1])

domain=os.path.basename(home_user[:-1].replace(name_service, '')[:-1]).strip()

if name_service=='htdocs':
    name_service='flask'
    home_user=home_user.replace('flask', 'htdocs')

#print(domain)

if subprocess.call('sudo systemctl stop %s-%s.service && sudo rm /etc/systemd/system/%s-%s.service' % (domain, name_service, domain, name_service), shell=True) > 0:
    print('Error: cannot delete app service in systemd...')
    sys.exit(1)

if subprocess.call("sudo su %s -s /bin/bash -c 'rm -f -r %s'" % (args.user, args.home_user),  shell=True) > 0:
    print('Error: cannot delete %s' % args.home_user)
    sys.exit(1)
        
print('Deleted Flask app successfully...')