import re
import requests
import codecs
from bs4 import BeautifulSoup
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time
import os

# Inicialización del navegador Safari
mybrowser = webdriver.Safari()

#base_url = "https://genius.com/artists/NOMBREARTISTA/songs" 
base_url = "https://genius.com/artists/Milo-j"
mybrowser.get(base_url)

t_sec = time.time() + 60*10  #Seconds*minutes - Tiempo en el que esta navegando
#Con 10 descarga aproximadamente 350 canciones
increase = 50 #Valor inicial de incremento para desplazamiento
while time.time() < t_sec:
    last_height = mybrowser.execute_script("return document.body.scrollHeight")

    # Desplazarse suavemente hacia abajo
    for i in range(0, last_height, increase):
        mybrowser.execute_script(f"window.scrollTo(0, {i});")
        time.sleep(0.4)

    # Espera un poco en el fondo
    time.sleep(1)

    # Desplazarse hacia arriba de una sola vez
    mybrowser.execute_script("window.scrollTo(0, 0);")
    time.sleep(1)
    
    # Incrementa la velocidad de desplazamiento
    increase += 50
    if increase >280:
        increase = 280
    
    time.sleep(1)    
    
    html = mybrowser.page_source
    soup = BeautifulSoup(html, "html.parser")
    time.sleep(1)

pattern = re.compile("[\S]+-lyrics$")
pattern2 = re.compile("\[(.*?)\]")  # Para limpiar el texto de las letras
exclude_keywords = re.compile("en vivo|remix|live|acústica|acoustic", re.IGNORECASE)

# Inicializar la lista para almacenar los títulos de todas las canciones encontradas
all_songs = []
for link in soup.find_all('a', href=True):
    if pattern.match(link['href']):
        f = requests.get(link['href'])
        lyricsoup = BeautifulSoup(f.content, "html.parser")
        title_div = lyricsoup.find("h1")
        title = title_div.get_text().strip() if title_div else "Título Desconocido"

        if not exclude_keywords.search(title):
            all_songs.append((title, link['href']))

# Excluir las últimas cuatro canciones por que son las recomendadas del footer
songs_to_process = all_songs[:-4]

file_count = 1
song_count = 0
current_file_songs = 0

# Crear un archivo temporal para almacenar todas las letras
temp_file_name = 'temp_lyrics.txt'
with codecs.open(temp_file_name, 'w', 'utf-8-sig') as temp_txtfile:
    for index, (title, url) in enumerate(songs_to_process):
        f = requests.get(url)
        lyricsoup = BeautifulSoup(f.content, "html.parser")
        lyrics_div = lyricsoup.find("div", {"data-lyrics-container": "true", "class": "Lyrics__Container-sc-1ynbvzw-1 kUgSbL"})
        lyrics = ""
        if lyrics_div:
            lyrics = lyrics_div.get_text().strip()
            lyrics = re.sub(pattern2, "", lyrics)
            lyrics = re.sub('\n', ' ', lyrics)

        span = lyricsoup.find("span", class_="LabelWithIcon__Container-hjli77-0 gcYcIR MetadataStats__LabelWithIcon-sc-1t7d8ac-3 gZUIou")
        span_content = span.get_text().strip() if span else "Información no disponible"

        song_info = f"Titulo: {title}\nFecha: {span_content}\n\n{lyrics}\n\n\n" + "-" * 40 + "\n\n"
        
        # Escribe en el archivo temporal
        temp_txtfile.write(song_info)

        # Crear un nuevo archivo de texto cada 20 canciones
        if current_file_songs == 0:
            part_txtfile = codecs.open(f'lyrics_part_{file_count}.txt', 'w', 'utf-8-sig')

        part_txtfile.write(song_info)

        current_file_songs += 1
        song_count += 1

        # Cerrar el archivo actual y prepararse para un nuevo archivo después de 20 canciones
        if current_file_songs == 20:
            part_txtfile.close()
            file_count += 1
            current_file_songs = 0

    # Cerrar el último archivo si no se ha cerrado aún
    if current_file_songs != 0:
        part_txtfile.close()

mybrowser.close()

# Crear el archivo 'lyrics.txt' agregando el conteo total de canciones al principio
with codecs.open(temp_file_name, 'r', 'utf-8-sig') as temp_file:
    temp_content = temp_file.read()

with codecs.open('lyrics.txt', 'w', 'utf-8-sig') as myfile:
    myfile.write(f"Total de Canciones: {song_count}\n\n")
    myfile.write(temp_content)

# Eliminar el archivo temporal
os.remove(temp_file_name)