import dash
from dash import dcc, html, Input, Output, State, callback
import dash_bootstrap_components as dbc
import uuid
import json
import os

app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])

DATA_FILE = 'links.json'

def load_links():
    if os.path.exists(DATA_FILE):
        with open(DATA_FILE, 'r') as f:
            return json.load(f)
    return []

def save_links(links):
    with open(DATA_FILE, 'w') as f:
        json.dump(links, f)

def generate_url_list(links):
    url_list = [
        dbc.ListGroupItem([
            dbc.Button(
                link['name'], 
                id={'type': 'url-button', 'index': link['id']},
                color="link",
                className="text-left"
            ),
            html.Span(
                "✕",
                id={'type': 'delete-button', 'index': link['id']},
                className="float-right text-danger",
                style={"cursor": "pointer"}
            )
        ], className="d-flex justify-content-between align-items-center")
        for link in links
    ]
    print(f"Generated URL list: {url_list}")  # Debug print
    return url_list

app.layout = dbc.Container([
    dcc.Store(id='add-url-status', data=''),
    dcc.Store(id='trigger-load', data='load'),
    dbc.Modal([
        dbc.ModalHeader("Add New URL"),
        dbc.ModalBody([
            dbc.Input(id='new-url-name', placeholder="URL Name", className="mb-2"),
            dbc.Input(id='new-url-link', placeholder="URL", className="mb-2"),
            html.Div(id='add-url-feedback')
        ]),
        dbc.ModalFooter(
            dbc.Button("Add", id="add-url-button", className="ml-auto")
        ),
    ], id="add-url-modal"),
    dbc.Row([
        dbc.Col([
            html.H2("My URL App", className="mt-3 mb-4"),
            dbc.Button("Add URL", id="open-modal-button", color="primary", className="mb-3"),
            html.Div(id='url-list'),
        ], width=3, className="bg-light p-3"),
        dbc.Col([
            html.Iframe(
                id='content-iframe',
                style={'width': '100%', 'height': '800px'},
                sandbox="allow-scripts allow-same-origin allow-forms allow-popups allow-downloads",
                allow="fullscreen; geolocation; microphone; camera; midi; encrypted-media; autoplay",
                referrerPolicy="no-referrer"
            )
        ], width=9)
    ])
], fluid=True)

@callback(
    [Output("add-url-modal", "is_open"),
     Output('new-url-name', 'value'),
     Output('new-url-link', 'value')],
    [Input("open-modal-button", "n_clicks"), Input("add-url-button", "n_clicks")],
    [State("add-url-modal", "is_open")],
)
def toggle_modal(n1, n2, is_open):
    if n1 or n2:
        if is_open:
            return False, '', ''
        return True, dash.no_update, dash.no_update
    return is_open, dash.no_update, dash.no_update

@callback(
    [Output('url-list', 'children'),
     Output('add-url-feedback', 'children'),
     Output('add-url-status', 'data')],
    [Input('trigger-load', 'data'),
     Input('add-url-button', 'n_clicks'),
     Input({'type': 'delete-button', 'index': dash.ALL}, 'n_clicks')],
    [State('new-url-name', 'value'),
     State('new-url-link', 'value')],
    prevent_initial_call=False
)
def update_url_list(trigger_load, add_clicks, delete_clicks, new_name, new_link):
    ctx = dash.callback_context
    if not ctx.triggered:
        raise dash.exceptions.PreventUpdate

    triggered_id = ctx.triggered[0]['prop_id'].split('.')[0]
    print(f"Triggered by: {triggered_id}")  # Debug print

    feedback = None
    status = ''
    links = load_links()

    if triggered_id == 'trigger-load':
        print("Initial load triggered")
    elif triggered_id == 'add-url-button' and new_name and new_link:
        new_id = str(uuid.uuid4())
        links.append({'id': new_id, 'name': new_name, 'url': new_link})
        save_links(links)
        print(f"Added new URL: {new_name}, {new_link}")  # Debug print
        feedback = html.Div("URL added successfully!", style={'color': 'green'})
        status = 'added'
    elif 'delete-button' in triggered_id:
        delete_id = eval(triggered_id)['index']
        links = [link for link in links if link['id'] != delete_id]
        save_links(links)
        print(f"Deleted URL with id: {delete_id}")  # Debug print
        feedback = html.Div("URL deleted successfully!", style={'color': 'green'})
        status = 'deleted'
    
    url_list = generate_url_list(links)
    return url_list, feedback, status

@callback(
    Output('content-iframe', 'src'),
    Input({'type': 'url-button', 'index': dash.ALL}, 'n_clicks')
)
def update_iframe(n_clicks):
    ctx = dash.callback_context
    if not ctx.triggered:
        return dash.no_update
    button_id = ctx.triggered[0]['prop_id'].split('.')[0]
    clicked_id = eval(button_id)['index']
    links = load_links()
    for link in links:
        if link['id'] == clicked_id:
            print(f"Loading URL: {link['url']}")  # Debug print
            return link['url']
    return dash.no_update

if __name__ == '__main__':
    print("Starting the Dash application...")
    print(f"Data file location: {os.path.abspath(DATA_FILE)}")
    app.run(debug=True, host='0.0.0.0', port=7860)
    print("Dash application has finished running.")