import pytz
from pkg.tool import get_tweets_since_last_friday, format_time_str, get_time_since_last_tweet
from pkg.dash.app_init import app
from dash.dependencies import Input, Output
from datetime import timedelta
from datetime import datetime
from dash import html


@app.callback(
    [Output('info-tooltip', 'children')],
    [Input('clock-interval', 'n_intervals')]
)
def update_info(n):
    # 获取所有指标
    pace = calculate_tweet_pace()
    decline_rates = calculate_pace_decline_rate()
    pace_increases = calculate_pace_increase_in_hour()
    _, days_to_next_friday = get_pace_params()
    table_style_border = {'textAlign': 'center', 'border': '1px solid white'}
    table_style_c = {'textAlign': 'center'}
    table_style_l = {'textAlign': 'left'}

    # First table for Pace
    pace_table_rows = [
        html.Tr([
            html.Th('Pace', style=table_style_border),
            html.Th('DCL', style=table_style_border),
            html.Th('0(1h)', style=table_style_border),
            html.Th('1(1h)', style=table_style_border),
            html.Th('5(1h)', style=table_style_border),
            html.Th('10(1h)', style=table_style_border),
            html.Th('20(1h)', style=table_style_border),
            html.Th('30(1h)', style=table_style_border)
        ]),
        html.Tr([
            html.Td(f"{pace:.2f}", style=table_style_border),
            html.Td(decline_rates, style=table_style_border),
            html.Td(f"{pace_increases['increase_0']:.2f}", style=table_style_border),
            html.Td(f"{pace_increases['increase_1']:.2f}", style=table_style_border),
            html.Td(f"{pace_increases['increase_5']:.2f}", style=table_style_border),
            html.Td(f"{pace_increases['increase_10']:.2f}", style=table_style_border),
            html.Td(f"{pace_increases['increase_20']:.2f}", style=table_style_border),
            html.Td(f"{pace_increases['increase_30']:.2f}", style=table_style_border)
        ]),
        html.Tr([
            html.Td(f"Remain",
                    colSpan=4,
                    style=table_style_c),
            html.Td(format_time_str(days_to_next_friday),
                    colSpan=4,
                    style=table_style_l)
        ]),
        html.Tr([
            html.Td(f"Last Tweet",
                    colSpan=4,
                    style=table_style_c),
            html.Td(format_time_str(get_time_since_last_tweet()),
                    colSpan=4,
                    style=table_style_l)
        ]),
        html.Tr([
            html.Td(f"Warning:",
                    colSpan=1,
                    style=table_style_c),
            html.Td("1.Check Reply",
                    colSpan=7,
                    style=table_style_l)
        ])
    ]
    pace_table = html.Table(pace_table_rows, style={
        'width': '100%',
        'textAlign': 'left',
        'borderCollapse': 'collapse'
    })
    return [pace_table]


def get_pace_params():
    est = pytz.timezone('US/Eastern')
    now = datetime.now(est)
    today = now.date()
    days_to_next_friday = (4 - today.weekday()) % 7
    next_friday = (now.replace(hour=12, minute=0, second=0, microsecond=0) +
                   timedelta(days=days_to_next_friday))
    if now > next_friday:
        next_friday += timedelta(days=7)
    days_to_next_friday = (next_friday - now).total_seconds() / (24 * 60 * 60)
    tweet_count = get_tweets_since_last_friday()
    return tweet_count, days_to_next_friday


def calculate_tweet_pace():
    tweet_count, days_to_next_friday = get_pace_params()
    pace = (tweet_count / (7 - days_to_next_friday)) * days_to_next_friday + tweet_count
    return round(pace, 6) if pace > 0 else float(tweet_count)


def calculate_pace_decline_rate():
    tweet_count, days_to_next_friday = get_pace_params()
    T = 7
    decline_per_day = -(tweet_count * T) / ((T - days_to_next_friday) ** 2)
    decline_per_hour = decline_per_day / 24
    return round(decline_per_hour, 2)


def calculate_pace_increase_in_hour():
    tweet_count, days_to_next_friday = get_pace_params()
    current_pace = (tweet_count / (7 - days_to_next_friday)) * days_to_next_friday + tweet_count
    future_days = days_to_next_friday - (1 / 24)
    increments = [0, 1, 5, 10, 20, 30]
    pace_increases = {}
    for increment in increments:
        new_tweet_count = tweet_count + increment
        new_pace = (new_tweet_count / (7 - future_days)) * future_days + new_tweet_count
        pace_increase = new_pace - current_pace
        # Add current pace to the increase value
        pace_increases[f'increase_{increment}'] = round(current_pace + pace_increase, 2)
    return pace_increases