from db import get_db, return_db, get_setting, set_setting, get_site_name
import templates as templates_mod
from templates import esc, DEFAULT_TEMPLATE
from ._helpers import _respond, _csrf_field, _get_bookmark_token
from .subscriptions import _count_shared_pages


def handle_style_form(msg="", gateway_host="", scheme="http"):
    template = get_setting("custom_template") or DEFAULT_TEMPLATE
    name = get_site_name()
    sharing = get_setting("sharing_enabled", "0")
    checked = " checked" if sharing == "1" else ""
    sharing_mode = get_setting("sharing_mode", "exclude_private")
    forum = get_setting("forum_enabled", "0")
    forum_checked = " checked" if forum == "1" else ""
    exclude_checked = " checked" if sharing_mode != "require_public" else ""
    require_checked = " checked" if sharing_mode == "require_public" else ""
    shared_count = _count_shared_pages()
    semantic = get_setting("semantic_search", "0")
    semantic_checked = " checked" if semantic == "1" else ""
    reranker = get_setting("use_reranker", "0")
    reranker_checked = " checked" if reranker == "1" else ""
    disabled = "" if semantic == "1" else " disabled"
    dimmed = ' style="opacity:0.4"' if semantic != "1" else ""
    tcp_enabled = get_setting("tcp_enabled", "1")
    tcp_checked = " checked" if tcp_enabled == "1" else ""
    tcp_disabled = "" if tcp_enabled == "1" else " disabled"
    transport_host = get_setting("transport_host", "rnode.bre.land")
    transport_port = get_setting("transport_port", "4242")
    compress = get_setting("compress_embeddings", "0")
    compress_checked = " checked" if compress == "1" else ""
    lora_enabled = get_setting("lora_enabled", "0")
    lora_checked = " checked" if lora_enabled == "1" else ""
    lora_disabled = "" if lora_enabled == "1" else " disabled"
    lora_dimmed = ' style="opacity:0.4"' if lora_enabled != "1" else ""
    lora_port = get_setting("lora_port", "")
    lora_frequency = get_setting("lora_frequency", "867200000")
    lora_bandwidth = get_setting("lora_bandwidth", "125000")
    lora_txpower = get_setting("lora_txpower", "7")
    lora_sf = get_setting("lora_sf", "8")
    lora_cr = get_setting("lora_cr", "5")
    from handlers import forum_plugin as _fp
    if _fp is not None:
        forum_section = (
            f"<h2>forum</h2>"
            f'<label><input type="checkbox" name="forum_enabled" value="1"{forum_checked}>'
            f" enable forum (shared URL discussion board)</label><br>"
            f"<small>Share URLs and discuss them with other TinyWeb instances. "
            f"Requires <code>tinyweb-forum</code> — "
            f'<a href="https://codeberg.org/tinyweb/tinyweb-forum">more info</a>.</small><br><br>'
        )
    else:
        forum_section = ""
    return _respond(
        f"<h1>customize</h1>"
        f"<h2>name your search engine</h2>"
        f'<form method="post" action="/style">'
        f'{_csrf_field()}'
        f'<input name="site_name" value="{esc(name)}" placeholder="tinyweb" size="30"><br><br>'
        f"<h2>sharing</h2>"
        f'<label><input type="checkbox" name="sharing_enabled" value="1"{checked}>'
        f" share your site list publicly at /api/sites</label><br>"
        f'<div style="margin-top:0.6rem">'
        f"<small>What to share:</small><br>"
        f'<label><input type="radio" name="sharing_mode" value="exclude_private"{exclude_checked}>'
        f' share all pages except those tagged <code>private</code></label><br>'
        f'<label><input type="radio" name="sharing_mode" value="require_public"{require_checked}>'
        f' share only pages tagged <code>public</code></label><br>'
        f'<small>The <code>private</code> tag always excludes a page, even in public-only mode.</small>'
        f'</div>'
        f'<p style="margin-top:0.6rem">'
        f'Currently sharing <b>{shared_count}</b> page(s). '
        f'<a href="/share/preview">preview what subscribers would see</a>'
        f'</p>'
        f"<h2>mesh network</h2>"
        f"<p>Choose how to connect to the mesh. You can enable both for maximum reach.</p>"
        f"<h3>internet</h3>"
        f'<label><input type="checkbox" name="tcp_enabled" value="1"{tcp_checked} '
        f'onchange="var d=!this.checked;'
        f'for(var e of document.querySelectorAll(\'#tcp-fields input\'))e.disabled=d;'
        f'document.getElementById(\'tcp-fields\').style.opacity=d?\'0.4\':\'1\'">'
        f" connect via internet transport node</label><br>"
        f"<small>Reach peers anywhere online.</small><br>"
        f'<div id="tcp-fields" style="margin-top:0.5rem{";opacity:0.4" if tcp_enabled != "1" else ""}">'
        f"<small>Default: rnode.bre.land:4242</small><br>"
        f'<input name="transport_host" value="{esc(transport_host)}" placeholder="hostname" size="30"{tcp_disabled}>'
        f' <input name="transport_port" value="{esc(transport_port)}" placeholder="port" size="6"{tcp_disabled}><br>'
        f'<p><a href="https://rmap.world/" target="_blank" rel="noreferrer noopener">discover more nodes</a></p>'
        f'</div><br>'
        f"<h3>LoRa</h3>"
        f'<label><input type="checkbox" name="lora_enabled" value="1"{lora_checked} '
        f'onchange="var d=!this.checked;document.getElementById(\'lora-port\').disabled=d;'
        f'document.getElementById(\'lora-extras\').style.opacity=d?\'0.4\':\'1\';'
        f'for(var e of document.querySelectorAll(\'#lora-extras input\'))e.disabled=d">'
        f" connect via LoRa radio</label><br>"
        f"<small>Reach nearby peers off-grid with an <a href=\"https://unsigned.io/rnode/\" target=\"_blank\" rel=\"noreferrer noopener\">RNode</a>.</small><br><br>"
        f'<div id="lora-fields" style="{";opacity:0.4" if lora_enabled != "1" else ""}">'
        f'<label>Serial port: <input id="lora-port" name="lora_port" value="{esc(lora_port)}" '
        f'placeholder="/dev/ttyUSB0" size="20"{lora_disabled}></label><br><br>'
        f'<details><summary>advanced radio settings</summary>'
        f'<div id="lora-extras" style="margin-top:0.5rem">'
        f'<label>Frequency (Hz): <input name="lora_frequency" value="{esc(lora_frequency)}" size="12"{lora_disabled}></label><br>'
        f"<small>ISM band frequency. Default: 867200000 (868 MHz EU). US: 915000000.</small><br><br>"
        f'<label>Bandwidth (Hz): <input name="lora_bandwidth" value="{esc(lora_bandwidth)}" size="8"{lora_disabled}></label><br>'
        f"<small>Default: 125000</small><br><br>"
        f'<label>TX Power (dBm): <input name="lora_txpower" value="{esc(lora_txpower)}" size="4"{lora_disabled}></label><br>'
        f"<small>0-17 typical. Check local regulations.</small><br><br>"
        f'<label>Spreading Factor: <input name="lora_sf" value="{esc(lora_sf)}" size="4"{lora_disabled}></label><br>'
        f"<small>5-12. Higher = longer range, slower speed.</small><br><br>"
        f'<label>Coding Rate: <input name="lora_cr" value="{esc(lora_cr)}" size="4"{lora_disabled}></label><br>'
        f"<small>5-8. Higher = more error correction.</small><br>"
        f'</div></details></div><br>'
        f"<h2>search</h2>"
        f"<h3>ai</h3>"
        f'<label><input type="checkbox" name="semantic_search" value="1"{semantic_checked} '
        f'onchange="var d=!this.checked;document.getElementById(\'reranker\').disabled=d;'
        f'document.getElementById(\'ai-extras\').style.opacity=d?\'0.4\':\'1\'">'
        f" semantic search (similarity matching)</label><br>"
        f"<small>Requires onnxruntime, tokenizers, hnswlib. Downloads ~30MB of models on first use.</small><br><br>"
        f'<div id="ai-extras"{dimmed}>'
        f'<label><input type="checkbox" id="reranker" name="use_reranker" value="1"{reranker_checked}{disabled}>'
        f" cross-encoder reranking (more accurate)</label><br>"
        f"<small>Uses a 22MB model. Adds ~50ms per search. Disable for faster results.</small><br><br>"
        f'<label><input type="checkbox" name="compress_embeddings" value="1"{compress_checked}{disabled}>'
        f" compress embeddings (50% storage savings)</label><br>"
        f"<small>Saves ~50% on storage for embeddings. Slight quality reduction at large scale.</small><br><br>"
        f'<a href="/reindex">manage semantic index</a><br><br>'
        f"</div>"
        f"{forum_section}"
        f'<button type="submit">save settings</button>'
        f"</form>"
        f"<h2>custom html</h2>"
        f"<p>Edit the full page template. Use <code>{esc('{{content}}')}</code> "
        f"where page content should appear.</p>"
        f'<form method="post" action="/style/template">'
        f'{_csrf_field()}'
        f'<textarea name="template" rows="20" cols="60">{esc(template)}</textarea><br>'
        f'<button type="submit">save template</button>'
        f"</form>"
        f"<h2>bookmarklet</h2>"
        f"<p>Drag this link to your bookmarks bar. Click it on any page to index it instantly.</p>"
        f'<p><a href="javascript:void(fetch(\'{esc(scheme)}://{esc(gateway_host or "localhost:8080")}/bookmark?url=\'+encodeURIComponent(location.href)+\'&token={_get_bookmark_token()}\').then(r=>r.text()).then(t=>alert(t)).catch(()=>alert(\'tinyweb not running\')))">+ save to {esc(name)}</a></p>'
        f"<h2>reset</h2>"
        f'<form method="post" action="/style/reset" '
        f'onsubmit="return confirm(\'Reset the template to default? Your custom template will be lost.\')">'
        f'{_csrf_field()}'
        f'<button type="submit">reset template to default</button>'
        f"</form>"
        f"<h2>maintenance</h2>"
        f'<form method="post" action="/style/vacuum">'
        f'{_csrf_field()}'
        f'<button type="submit">vacuum database</button>'
        f"</form>"
        f"<p>{msg}</p>"
        f'<a href="/">back</a>',
        use_default=True,
    )


def handle_style_submit(body, gateway_host="", scheme="http"):
    template = body.get("template", [""])[0].replace("\r\n", "\n").replace("\r", "\n")
    name = body.get("site_name", ["tinyweb"])[0].strip()
    sharing = "1" if body.get("sharing_enabled") else "0"
    sharing_mode = body.get("sharing_mode", ["exclude_private"])[0]
    if sharing_mode not in ("exclude_private", "require_public"):
        sharing_mode = "exclude_private"
    set_setting("sharing_mode", sharing_mode)
    semantic = "1" if body.get("semantic_search") else "0"
    reranker = "1" if body.get("use_reranker") else "0"
    compress = "1" if body.get("compress_embeddings") else "0"
    tcp_enabled = "1" if body.get("tcp_enabled") else "0"
    transport_host = body.get("transport_host", [""])[0].strip()
    transport_port = body.get("transport_port", [""])[0].strip()
    set_setting("custom_template", template if template.strip() != DEFAULT_TEMPLATE.strip() else "")
    set_setting("site_name", name or "tinyweb")
    set_setting("sharing_enabled", sharing)
    set_setting("semantic_search", semantic)
    set_setting("use_reranker", reranker)
    set_setting("compress_embeddings", compress)
    set_setting("tcp_enabled", tcp_enabled)
    if transport_host:
        set_setting("transport_host", transport_host)
    if transport_port:
        set_setting("transport_port", transport_port)
    lora_enabled = "1" if body.get("lora_enabled") else "0"
    set_setting("lora_enabled", lora_enabled)
    set_setting("lora_port", body.get("lora_port", [""])[0].strip())
    set_setting("lora_frequency", body.get("lora_frequency", ["867200000"])[0].strip())
    set_setting("lora_bandwidth", body.get("lora_bandwidth", ["125000"])[0].strip())
    set_setting("lora_txpower", body.get("lora_txpower", ["7"])[0].strip())
    set_setting("lora_sf", body.get("lora_sf", ["8"])[0].strip())
    set_setting("lora_cr", body.get("lora_cr", ["5"])[0].strip())
    forum_enabled = "1" if body.get("forum_enabled") else "0"
    current_forum = get_setting("forum_enabled", "0")
    if forum_enabled != current_forum:
        from handlers import forum_plugin
        if forum_enabled == "1" and forum_plugin is None:
            return handle_style_form(
                "Forum plugin not installed. Run: pip install tinyweb-forum",
                gateway_host=gateway_host, scheme=scheme,
            )
        if forum_enabled == "1":
            forum_plugin.enable()
            try:
                forum_plugin.fdb.set_setting("forum_enabled", "1")
            except Exception:
                pass
        else:
            forum_plugin.disable()
            try:
                forum_plugin.fdb.set_setting("forum_enabled", "0")
            except Exception:
                pass
        set_setting("forum_enabled", forum_enabled)
        templates_mod.FORUM_ENABLED = (forum_enabled == "1")
    return handle_style_form("Saved. Restart TinyWeb for mesh network changes to take effect.",
                             gateway_host=gateway_host, scheme=scheme)


def handle_style_template_submit(body, gateway_host="", scheme="http"):
    template = body.get("template", [""])[0].replace("\r\n", "\n").replace("\r", "\n")
    set_setting("custom_template", template if template.strip() != DEFAULT_TEMPLATE.strip() else "")
    return handle_style_form("Template saved.", gateway_host=gateway_host, scheme=scheme)


def handle_about():
    name = get_site_name()
    dest_hash = get_setting("dest_hash")
    sharing = get_setting("sharing_enabled", "0") == "1"
    db = get_db()
    try:
        page_count = db.execute("SELECT count(*) FROM pages").fetchone()[0]
        tag_count = db.execute("SELECT count(DISTINCT tag_id) FROM page_tags").fetchone()[0]
        sub_count = db.execute("SELECT count(*) FROM subscriptions").fetchone()[0]
    finally:
        return_db(db)

    sharing_html = (
        '<p>This instance shares its index publicly. Subscribe to join the network.</p>'
        if sharing else
        '<p>This instance is private.</p>'
    )

    hash_html = ""
    if dest_hash:
        hash_html = (
            f'<h2>subscribe</h2>'
            f'<p>To subscribe to this instance, add this destination hash in your TinyWeb:</p>'
            f'<pre>{esc(dest_hash)}</pre>'
        )

    return _respond(
        f'<h1>{esc(name)}</h1>'
        f'<p>A personal, decentralized search engine.</p>'
        f'<p>You save pages you find. They are stored locally and shared over a mesh network '
        f'so other people can find them too.</p>'
        f'<p>Search results come from your index and the indexes of people you are connected to.</p>'
        f'<ul>'
        f'<li><b>{page_count}</b> page(s) indexed</li>'
        f'<li><b>{tag_count}</b> tag(s)</li>'
        f'<li><b>{sub_count}</b> subscription(s)</li>'
        f'</ul>'
        f'{sharing_html}'
        f'{hash_html}'
        f'<h2>your data</h2>'
        f'<p>Everything is stored locally under <code>~/.tinyweb/</code>:</p>'
        f'<ul>'
        f'<li><code>tinyweb_identity</code> — your permanent mesh identity. '
        f'If you lose this file, your destination hash changes and subscribers '
        f'have to re-subscribe to the new one.</li>'
        f'<li><code>index.db</code> — your full reading history: every page, '
        f'note, tag, and synced remote page.</li>'
        f'<li><code>models/</code> — the semantic search model if you enabled it '
        f'(redownloadable, safe to delete).</li>'
        f'</ul>'
        f'<p><b>Back up <code>~/.tinyweb/</code> periodically.</b> '
        f'Copying the whole directory to another device preserves your identity and index together. '
        f'The <a href="/export">export</a> page gives you a JSON dump of pages only — '
        f'it does not preserve your identity or subscription state, so it is a migration aid, '
        f'not a substitute for a full backup.</p>'
        f'<p><a href="/">search</a> | <a href="/pages">browse</a> | <a href="/tags">tags</a></p>'
    )