import os
import csv
from datetime import datetime
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import time
import configparser
import zipfile
import requests
from io import BytesIO

# Set the path to the Chrome executable
# Define the directory path and URLs for Chrome and Chromedriver
chrome_directory = 'c:/tmp/chrome'
chrome_exe_path = 'c:/tmp/chrome/chrome-win64/chrome.exe'  # Modify this path as needed
chrome_url = 'https://edgedl.me.gvt1.com/edgedl/chrome/chrome-for-testing/117.0.5938.92/win64/chrome-win64.zip'

# Check if the Chrome directory exists, and if not, create it
if not os.path.exists(chrome_directory):
    os.makedirs(chrome_directory)

# Check if Chrome.exe and Chromedriver.exe exist in the directory
if not os.path.exists(os.path.join(chrome_exe_path)):
    # Download and extract Chrome
    chrome_response = requests.get(chrome_url)
    with zipfile.ZipFile(BytesIO(chrome_response.content), 'r') as zip_ref:
        zip_ref.extractall(chrome_directory)

# Set the URL of the website
url = 'https://www.mrcooper.com/servicing/overview'

current_directory = os.getcwd()
csv_file_path = os.path.join(current_directory, 'equity.csv')

# Configure Chrome options
chrome_options = webdriver.ChromeOptions()
chrome_options.binary_location = chrome_exe_path  # Specify Chrome executable path
chrome_options.add_argument('--headless')  # Run Chrome in headless mode (optional)

config_path = os.path.join(os.path.dirname(__file__), 'config.ini')

if not os.path.exists(config_path):
    # If config.ini doesn't exist, prompt the user for credentials
    username_input = input("Enter your username: ")
    password_input = input("Enter your password: ")

    # Create config.ini with user-provided credentials
    config = configparser.ConfigParser()
    config['Credentials'] = {'username': username_input,
                             'password': password_input}
    
    with open(config_path, 'w') as configfile:
        config.write(configfile)

# Load credentials from config file
config = configparser.ConfigParser()
config.read('config.ini')  # Update with your config file path

username_value = config.get('Credentials', 'username')
password_value = config.get('Credentials', 'password')

# Initialize the Chrome WebDriver
driver = webdriver.Chrome(options=chrome_options)

try:
    # Navigate to the website
    driver.get(url)

    # Set your login credentials
    username = WebDriverWait(driver, 20).until(EC.presence_of_element_located((By.ID, "signInName")))
    password = WebDriverWait(driver, 20).until(EC.presence_of_element_located((By.ID, "password")))

    # Enter the username and password
    username.send_keys(username_value)
    password.send_keys(password_value)

    # Submit the login form
    password.send_keys(Keys.RETURN)

    # Wait for a few seconds for the page to load after login (you may need to adjust the timing)
    time.sleep(10)

    # Find the equity with the data you want to scrape (by class name in this case)
    equity = WebDriverWait(driver, 20).until(EC.presence_of_element_located((By.CLASS_NAME, 'estimate-section__equity__content')))
    home = WebDriverWait(driver, 20).until(EC.presence_of_element_located((By.CLASS_NAME, 'estimate-section__home__content')))
    principal = WebDriverWait(driver, 20).until(EC.presence_of_element_located((By.CLASS_NAME, 'principal-balance-value')))
    
    # Get the text content of the equity
    equity = equity.text
    home = home.text
    principal = principal.text

    # Create or update the CSV file
    data = [datetime.now().strftime('%Y-%m-%d %H:%M:%S'), equity, home, principal]

    # Check if the CSV file already exists
    if os.path.exists(csv_file_path):
        # Read the existing data
        with open(csv_file_path, 'r', newline='') as csv_file:
            reader = csv.reader(csv_file)
            rows = list(reader)
        
        # Shift existing rows down and insert the new data at row 2
        rows.insert(1, data)
        
        # Update the existing CSV file with the shifted data
        with open(csv_file_path, 'w', newline='') as csv_file:
            writer = csv.writer(csv_file)
            writer.writerows(rows)
    else:
        # Create a new CSV file and write the data
        with open(csv_file_path, 'w', newline='') as csv_file:
            csv_writer = csv.writer(csv_file)
            # Write column headers
            csv_writer.writerow(['Date', 'Equity', 'Home_Value', 'Principal_Remaining'])
            # Write the data to the CSV file
            csv_writer.writerow(data)

    print(f'Data saved to {csv_file_path}')

finally:
    # Close the browser
    driver.quit()