prepare_kit.py

import os
import shutil
import json
import zipfile

from env import APP_KITS_FOLDER, GENERATED_SAMPLES_FOLDER, REPO_URL


def create_next_kit_folder(kit_name, next_folder_number):
    # 2. Crie a pasta no diretório APP_KITS_FOLDER
    new_kit_folder = os.path.join(APP_KITS_FOLDER, str(next_folder_number))
    os.makedirs(new_kit_folder, exist_ok=True)
    print(f"Pasta criada: {new_kit_folder}")

    # 3. Crie a subpasta 'megapianokit' e o arquivo kit.json
    megapianokit_folder = os.path.join(new_kit_folder, "megapianokit")
    os.makedirs(megapianokit_folder, exist_ok=True)
    print(f"Subpasta 'megapianokit' criada em: {megapianokit_folder}")

    kit_json_path = os.path.join(megapianokit_folder, "kit.json")
    with open(kit_json_path, 'w') as kit_file:
        json.dump({"name": kit_name}, kit_file)
    print(f"Arquivo kit.json criado com o conteúdo: {{'name': '{kit_name}'}}")

    # 4. Mova os arquivos do OUTPUT_FOLDER para a pasta 'megapianokit'
    move_files_to_folder(GENERATED_SAMPLES_FOLDER, megapianokit_folder)

    # 5. Compacte a pasta 'megapianokit' e renomeie para 'kit.megapiano'
    create_kit_archive(megapianokit_folder)
    # 6. Apague a pasta 'megapianokit' e todo seu conteúdo
    shutil.rmtree(megapianokit_folder)
    print(f"Subpasta 'megapianokit' removida, mantendo apenas o arquivo 'kit.megapiano'")


def get_next_folder_number():
    # Obtenha o número da próxima pasta com base no maior número existente
    folders = [int(name) for name in os.listdir(APP_KITS_FOLDER) if name.isdigit()]
    return max(folders, default=0) + 1


def move_files_to_folder(source_folder, target_folder):
    # Mova todos os arquivos da pasta source_folder para target_folder
    for filename in os.listdir(source_folder):
        source_file = os.path.join(source_folder, filename)
        target_file = os.path.join(target_folder, filename)
        shutil.move(source_file, target_file)
    print(f"Arquivos movidos de {source_folder} para {target_folder}")


def create_kit_archive(folder_path):
    archive_name = os.path.join(os.path.dirname(folder_path), "kit.zip")

    with zipfile.ZipFile(archive_name, 'w', zipfile.ZIP_DEFLATED) as zipf:
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                file_path = os.path.join(root, file)
                arcname = os.path.relpath(file_path, os.path.dirname(folder_path))
                zipf.write(file_path, arcname)

    # Renomeie para a extensão .megapiano
    renamed_archive = archive_name.replace('.zip', '.megapiano')
    os.rename(archive_name, renamed_archive)
    print(f"Arquivo compactado criado: {renamed_archive}")


def update_index_json(kit_name, folder_number):
    index_file_path = os.path.join(APP_KITS_FOLDER, "index.json")

    # Carregar o arquivo index.json existente
    if os.path.exists(index_file_path):
        with open(index_file_path, 'r') as index_file:
            index_data = json.load(index_file)
    else:
        index_data = []

    # Adicionar o novo kit ao index.json
    new_entry = {
        "name": kit_name,
        "path": f"online_kit_{folder_number}",
        "coverUrl": f"{REPO_URL}/{folder_number}/cover.png",
        "zipUrl": f"{REPO_URL}/{folder_number}/kit.megapiano"
    }
    index_data.append(new_entry)

    # Salvar as alterações no arquivo index.json
    with open(index_file_path, 'w') as index_file:
        json.dump(index_data, index_file, indent=2)
    print(f"Novo kit adicionado ao index.json: {new_entry}")


# Execução do script
def show_notification_info(kit_name, next_folder_number):
    print('\n\n\n\n------------')
    print("Título da notificação:")
    print(f"New kit: {kit_name}")

    print("\nTexto da notificação:")
    print("Click to open!")

    print("Imagem de notificação:")
    print(f"{REPO_URL}/{next_folder_number}/cover.png")

    print("Dados personalizados")
    print("notificationType - newKit")
    print('------------\n\n\n\n')


def run(kit_name):
    next_folder_number = get_next_folder_number()

    # 1. Obtenha o próximo número de pasta
    print(f"Próxima pasta será: {next_folder_number}")

    create_next_kit_folder(kit_name, next_folder_number)

    # 7. Atualize o arquivo index.json com o novo kit
    update_index_json(kit_name, next_folder_number)
    show_notification_info(kit_name, next_folder_number)