import os
import re

# Caminho da pasta onde está o script e os HTMLs
pasta_site = os.getcwd()

# Novo link com a nova mensagem
novo_link = "https://wa.me/5561995530473?text=Ol%C3%A1%2C%20encontrei%20vcs%20no%20site%20Ca%C3%A7a%20Vazamento%20em%20S%C3%A3o%20Paulo%2C%20gostaria%20de%20anunciar%20minha%20Empresa"

# Regex para capturar o botão com qualquer link anterior
padrao_botao = re.compile(
    r'(<button\s+class="btn-anunciar"\s+onclick="window\.location\.href=\')(https://wa\.me/[^"]+)(\'[^>]*>[^<]*</button>)',
    re.IGNORECASE
)

# Contador
total_modificados = 0

# Varre todos os arquivos .html em todas as subpastas
for root, dirs, files in os.walk(pasta_site):
    for file in files:
        if file.endswith(".html"):
            caminho_arquivo = os.path.join(root, file)
            with open(caminho_arquivo, 'r', encoding='utf-8') as f:
                conteudo = f.read()

            # Substitui o link do botão
            novo_conteudo = re.sub(padrao_botao, rf'\1{novo_link}\3', conteudo)

            if novo_conteudo != conteudo:
                with open(caminho_arquivo, 'w', encoding='utf-8') as f:
                    f.write(novo_conteudo)
                print(f"✅ Link do botão atualizado: {caminho_arquivo}")
                total_modificados += 1

print(f"\n🚀 Substituição de link concluída em {total_modificados} arquivo(s)!")