from flask.helpers import send_file
from flask import render_template, abort, flash, Blueprint
from Web.main.forms import UploadJavaForm
from random import randint
import shutil
import secrets
import os
import logging

from gui.utils import nassi
from interpreter.NassiShneidermann import NassiShneidermanDiagram, OB

main = Blueprint('main', __name__)

def deleteFile(path):
    file_list = os.listdir(path)
    for f in file_list:
        try:
            os.remove(path + '/' +f)
            print("remove " + f)
        except:
            print("fail to remove " + f)

def deleteFolder(path):
    try:
        shutil.rmtree(path)
        print("remove " + path)
    except:
        print("fail to remove " + path)

def javaDatei(form_file):
    try:
        random_hex = secrets.token_hex(8)
        _, f_ext = os.path.splitext(form_file.filename)
        file_fname = random_hex + f_ext
        dirctory_path = os.path.abspath(os.path.join('Web', os.pardir))
        file_path = os.path.join(dirctory_path, './tmp/input', file_fname)
        form_file.save(file_path)
        return file_path
    except:
        flash('Hier ist was falsch gelaufen!')


@main.route('/', methods=['POST', 'GET'])
@main.route('/generator', methods=['POST','GET'])
def generator():
    form = UploadJavaForm()

    if form.validate_on_submit():
        if form.java.data:
            input_path = javaDatei(form.java.data)
            output_path = os.path.join(os.path.abspath(os.path.join('Web', os.pardir)), './tmp/input')
            outputname = str(randint(0, 100) )
            remove_tags = None 
            comments = None 
            behaviour = OB.RANDOM_NAME 
            types = None

            NSD = NassiShneidermanDiagram(True)
            output_directory = output_path + '/' + outputname
            

            try:
                if not os.path.exists(output_directory):
                    os.makedirs(output_directory)
            except OSError:
                logging.error('Error: Creating directory. ' +  output_directory)


            custom_tags = {"comments" : comments, "ignore" : remove_tags, "types" : types}
            
            NSD.load_from_file(input_path, custom_tags)
            NSD.convert_to_image(output_directory, on_conflict=behaviour)

            zip_path = os.path.join(os.path.abspath(os.path.join('Web', os.pardir)), f'../tmp/output/{outputname}')
            shutil.make_archive(zip_path, 'zip', output_directory) 

            deleteFolder(output_path)
            
            return send_file(zip_path + '.zip', as_attachment=True)

    return render_template('upload.html', title='Upload', legend='Upload', form=form )