import pytz
from datetime import datetime, timedelta
from urllib.request import urlopen
import certifi
import json
import ujson
import schedule
import time
import subprocess
from pocketbase import PocketBase  # Client also works the same
import asyncio
import aiohttp
import pytz
import pandas as pd
import numpy as np
from tqdm import tqdm

import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText
import boto3
from botocore.exceptions import NoCredentialsError
from bs4 import BeautifulSoup

from dotenv import load_dotenv
import os

load_dotenv()
api_key = os.getenv('FMP_API_KEY')
pb_admin_email = os.getenv('POCKETBASE_ADMIN_EMAIL')
pb_password = os.getenv('POCKETBASE_PASSWORD')

aws_access_key_id = os.getenv('AWS_ACCESS_KEY_ID')
aws_secret_access_key = os.getenv('AWS_SECRET_ACCESS_KEY')

berlin_tz = pytz.timezone('Europe/Berlin')
pb = PocketBase('http://127.0.0.1:8090')
admin_data = pb.collection('_superusers').auth_with_password(pb_admin_email, pb_password)


now = datetime.today().date()
#one_month_ago = now - timedelta(days=30)
time_threshold = now - timedelta(days=7)


def send_email(recipient):
    # Replace the placeholders with your AWS SES credentials
    region_name = 'eu-north-1' #email-smtp.eu-north-1.amazonaws.com

    # Replace the placeholders with your sender email and password
    sender_email = 'mrahimi@stocknear.com'

    to_email = recipient # user email address
    subject = f'Your Free Trial expired'
    
    # Read the index.html file
    with open('html_template/free_trial.html', 'r') as file:
        html_content = file.read()

  
    # Create a MIMEMultipart object
    message = MIMEMultipart('alternative')
    message['Subject'] = subject
    message['From'] = sender_email
    message['To'] = to_email

    #Preheader text
    preheader = MIMEText("Your free trial ended.", 'plain')
    
    message.attach(MIMEText(html_content, 'html'))

    # Use Amazon SES to send the email
    ses_client = boto3.client(
        'ses',
        aws_access_key_id=aws_access_key_id,
        aws_secret_access_key=aws_secret_access_key,
        region_name=region_name
    )

    try:
        # Send the email
        response = ses_client.send_raw_email(
            Source=message['From'],
            Destinations=[message['To']],
            RawMessage={'Data': message.as_string()},
        )
        print("Email sent successfully!")
    except NoCredentialsError:
        print("AWS credentials not available")
    except Exception as e:
        print(f"Error sending email: {e}")


async def update_free_trial():

    data = pb.collection("users").get_full_list(query_params = {"filter": f'freeTrial = True'})

    for item in data:
        created_date = item.created
        # Check if the created date is more than N days ago
        if created_date < time_threshold:
            # Update the user record
            pb.collection("users").update(item.id, {
                "tier": 'Free',
                "freeTrial": False,
            })
            try:
                send_email(item.email)
            except Exception as e:
                print(e)



async def downgrade_user():

    user_data =  pb.collection('users').get_full_list()
    for item in tqdm(user_data):
        if item.tier != 'Pro' or item.tier != 'Plus':
            stock_screener_data = pb.collection("stockscreener").get_full_list(query_params = {"filter": f"user = '{item.id}'"})
            for screener in stock_screener_data:
                pb.collection('stockscreener').delete(screener.id)

            options_watchlist_data = pb.collection("optionsWatchlist").get_full_list(query_params = {"filter": f"user = '{item.id}'"})
            for watchlist in options_watchlist_data:
                pb.collection('optionsWatchlist').delete(watchlist.id)


            payment_data = pb.collection("payments").get_full_list(query_params = {"filter": f"user = '{item.id}'"})
            for item in payment_data:
                pb.collection('payments').delete(item.id)


async def delete_old_notifications():
    # Get the current datetime
    now = datetime.utcnow()
    # Calculate the threshold datetime
    time_threshold = now - timedelta(days=7)

    # Fetch all notifications
    data = pb.collection("notifications").get_full_list()
    for item in data:
        try:
            # Ensure both are datetime objects before comparison
            if item.created < time_threshold:
                pb.collection('notifications').delete(item.id)
        except:
            pass




async def run():

    await update_free_trial()
    await downgrade_user()
    await delete_old_notifications()

asyncio.run(run())