import os
import json
import yaml  
import subprocess
import xml.etree.ElementTree as ET
import shutil

def parse_mmp_file(file_path):
    try:
        #abrindo o .mmp, navegando até os instrumentos
        tree = ET.parse(file_path)
        root = tree.getroot()
        bpm = root.find('./head').attrib.get('bpm', 'N/A')
        tracks = root.findall('./song/trackcontainer/track')
        track_info = []

        #coletando as informações
        for track in tracks:
            track_name = track.attrib.get('name', 'N/A')
            track_type = track.attrib.get('type')

            if track_name:
                instruments = []

                if track_type == '0':  # Faixa do tipo plugin
                    track_info.append({
                        'track_name': track_name,
                        'type': 'plugin',
                        'instruments': [],
                    })

                elif track_type == '1':  # Faixa do tipo Beat/Bassline
                    bbtrack = track.find('./bbtrack')
                    if bbtrack is not None:
                        trackcontainers = bbtrack.findall('./trackcontainer')
                        for container in trackcontainers:
                            instrument_tracks = container.findall('./track')
                            for instrument_track in instrument_tracks:
                                instrument_name = instrument_track.attrib.get('name', 'N/A')
                                instruments.append({
                                    'instrument_name': instrument_name,
                                })

                        track_info.append({
                            'track_name': track_name,
                            'type': 'bassline',
                            'instruments': instruments,
                        })

                elif track_type == '2':  # Faixa do tipo Sample Track
                    track_info.append({
                        'track_name': track_name,
                        'type': 'sample',
                        'instruments': [],
                    })               

        return {
            'file': file_path,
            'bpm': bpm,
            'tracks': track_info
        }

    except ET.ParseError as e:
        print(f'Erro ao analisar o arquivo XML {file_path}: {e}')
        return None

def process_mmps_in_folder(folder_path):
    #coletando os arquivos .mmpz e/ou .mmp
    mmp_files = [f for f in os.listdir(folder_path) if f.endswith('.mmp') or f.endswith('.mmpz')]
    all_data = []

    #caso não haja, criando a pasta dos arquivos mmpz
    mmpz_folder = os.path.join(folder_path, 'mmpz')
    if not os.path.exists(mmpz_folder):
        os.makedirs(mmpz_folder)

    #caso não haja, criando a pasta dos arquivos wav
    wav_folder = os.path.join(folder_path, 'wav')
    if not os.path.exists(wav_folder):
        os.makedirs(wav_folder)

    #processamento dos arquivos
    for file in mmp_files:
        file_path = os.path.join(folder_path, file)
        print(f'Processando arquivo: {file_path}')

        if file.endswith('.mmpz'):
            destination_path = os.path.join(mmpz_folder, file) 
            shutil.move(file_path, destination_path)
            print(f'Arquivo {file} movido para {destination_path}')           

            file_name = os.path.basename(destination_path)
            file_name = os.path.splitext(file_name)[0] + ".mmp"
            output_file_path = os.path.join(folder_path, file_name)
            lmms_mmpz_convert = f'lmms --dump "{destination_path}" > "{output_file_path}"'

            try:
                #desabilitando os servidores gráficos
                os.environ['QT_DEBUG_PLUGINS'] = '1'
                os.environ['QT_QPA_PLATFORM'] = 'offscreen'
                #rodando o comando para converter os arquivos .mmpz em .mmp
                subprocess.run(lmms_mmpz_convert, shell=True, check=True)
                print("Comando executado com sucesso!")
            except subprocess.CalledProcessError as e:
                print(f"Ocorreu um erro ao executar o comando: {e}")
            wav_path = '/nethome/jotachina/public_html/mmpSearch/mmp/mmpz/'+file_name.split('.')[0]
            file_name = file_name.split('.')
            #"C:\Program Files\LMMS\lmms.exe" -r beatJulioCesardeSousa.mmp -o C:\Users\bacteria\wav -f wav
            lmms_wav_convert = f'lmms -r "{wav_path}.mmpz" -o /nethome/jotachina/public_html/mmpSearch/mmp/wav/{file_name[0]} -f wav'
            try:
                #desabilitando os servidores gráficos
                os.environ['QT_DEBUG_PLUGINS'] = '1'
                os.environ['QT_QPA_PLATFORM'] = 'offscreen'
                #rodando o comando para converter os arquivos .mmpz em .mmp
                subprocess.run(lmms_wav_convert, shell=True, check=True)
                print("Comando executado com sucesso!")
            except subprocess.CalledProcessError as e:
                print(f"Ocorreu um erro ao executar o comando: {e}")
            mmp_data = parse_mmp_file(output_file_path)  
        #caso tenha algum arquivo .mmp perdido na pasta
        elif file.endswith('.mmp'):
            mmp_data = parse_mmp_file(file_path)  

        #guardando as informações dos arquivos no arquivo json
        #/metadata (json) e /_data(yml)
        if mmp_data:
            all_data.append(mmp_data)

            json_file_name = os.path.splitext(file)[0] + ".json"
            json_file_path = os.path.join('metadata', json_file_name)
            save_to_json(mmp_data, json_file_path)

            yaml_file_name = os.path.splitext(file)[0] + ".yml"
            yaml_file_path = os.path.join('_data', yaml_file_name)
            save_to_yaml(mmp_data, yaml_file_path)

    return all_data

def save_to_json(data, output_file):
    with open(output_file, 'w', encoding='utf-8') as f:
        json.dump(data, f, ensure_ascii=False, indent=4)

def save_to_yaml(data, output_file):
    with open(output_file, 'w', encoding='utf-8') as f:
        yaml.dump(data, f)

mmp_folder_path = 'mmp'
processed_data = process_mmps_in_folder(mmp_folder_path)
print("Processamento concluído.")