local M = {}

local cbi = require("luci.cbi")
local i18n = require("luci.i18n")
local sys = require("luci.sys")
local translate = i18n.translate

local Flag = cbi.Flag
local Value = cbi.Value
local ListValue = cbi.ListValue
local DummyValue = cbi.DummyValue
local TextValue = cbi.TextValue

-- Status Section Fields
function M.add_status_fields(section)
    -- Enabled Flag
    section:option(Flag, "enabled", translate("Enabled"))

    -- Running Status Display
    local running = section:option(DummyValue, "running", translate("Status"))
    running.rawhtml = true
    running.cfgvalue = function(self, section)
        local pid = sys.exec("pidof ua3f")
        if pid == "" then
            return "<input disabled type='button' style='opacity: 1;' class='btn cbi-button cbi-button-reset' value='" ..
                translate("Stop") .. "'/>"
        else
            return "<input disabled type='button' style='opacity: 1;' class='btn cbi-button cbi-button-add' value='" ..
                translate("Running") .. "'/>"
        end
    end
end

-- General Tab Fields
function M.add_general_fields(section)
    -- Server Mode
    local server_mode = section:taboption("general", ListValue, "server_mode", translate("Server Mode"))
    server_mode:value("HTTP", "HTTP")
    server_mode:value("SOCKS5", "SOCKS5")
    server_mode:value("TPROXY", "TPROXY")
    server_mode:value("REDIRECT", "REDIRECT")
    server_mode:value("NFQUEUE", "NFQUEUE")
    server_mode.default = "SOCKS5"

    -- Bind Address
    local bind = section:taboption("general", Value, "bind", translate("Bind Address"))
    bind:value("127.0.0.1")
    bind:value("0.0.0.0")
    bind:depends("server_mode", "HTTP")
    bind:depends("server_mode", "SOCKS5")

    -- Port
    local port = section:taboption("general", Value, "port", translate("Port"))
    port.placeholder = "1080"
    port:depends("server_mode", "HTTP")
    port:depends("server_mode", "SOCKS5")
    port:depends("server_mode", "TPROXY")
    port:depends("server_mode", "REDIRECT")

    -- Rewrite Mode
    local rewrite_mode = section:taboption("general", ListValue, "rewrite_mode", translate("Rewrite Mode"))
    rewrite_mode:value("DIRECT", translate("Direct Forward"))
    rewrite_mode:value("GLOBAL", translate("Global Rewrite"))
    rewrite_mode:value("RULES", translate("Rule Based"))
    rewrite_mode.default = "GLOBAL"
    rewrite_mode.description = translate(
        "Direct Forward: No rewriting. Global Rewrite: Rewrite all User-Agents. Rule Based: Use rewrite rules to determine behavior.")

    -- User-Agent (for Global Rewrite)
    local ua = section:taboption("general", Value, "ua", translate("User-Agent"))
    ua.placeholder = "FFF"
    ua.description = translate("User-Agent after rewrite")
    ua:depends("rewrite_mode", "GLOBAL")
    ua:depends("server_mode", "NFQUEUE")

    -- User-Agent Regex
    local regex = section:taboption("general", Value, "ua_regex", translate("User-Agent Regex"))
    regex.description = translate("Regular expression pattern for matching User-Agent")
    regex:depends("rewrite_mode", "GLOBAL")
    regex:depends("server_mode", "NFQUEUE")

    -- Partial Replace
    local partialReplace = section:taboption("general", Flag, "partial_replace", translate("Partial Replace"))
    partialReplace.description =
        translate(
            "Replace only the matched part of the User-Agent, only works when User-Agent Regex is not empty")
    partialReplace.default = "0"
    partialReplace:depends("rewrite_mode", "GLOBAL")
    partialReplace:depends("server_mode", "NFQUEUE")
end

-- Rewrite Rules Tab Fields
function M.add_rewrite_fields(section)
    local rules = section:taboption("rewrite", DummyValue, "")
    rules.template = "ua3f/rules"
end

-- Statistics Tab Fields
function M.add_stats_fields(section)
    local stats = section:taboption("stats", DummyValue, "")
    stats.template = "ua3f/statistics"
end

-- Log Tab Fields
function M.add_log_fields(section)
    -- Log Display
    local log = section:taboption("log", TextValue, "log")
    log.readonly = true
    log.rows = 30
    log.wrap = "off"
    function log.cfgvalue(self, section)
        local logfile = "/var/log/ua3f/ua3f.log"
        local fs = require("nixio.fs")
        if not fs.access(logfile) then
            return ""
        end
        local n = tonumber(luci.model.uci.cursor():get("ua3f", section, "log_lines")) or 1000
        return luci.sys.exec("tail -n " .. n .. " " .. logfile)
    end

    function log.write(self, section, value) end

    function log.render(self, section, scope)
        TextValue.render(self, section, scope)
        luci.http.write("<script>")
        luci.http.write([[
            var textarea = document.getElementById('cbid.ua3f.main.log');
            if (textarea) {
                textarea.removeAttribute('name');
                textarea.scrollTop = textarea.scrollHeight;
            }
        ]])
        luci.http.write("</script>")
    end

    -- Log Level
    local log_level = section:taboption("log", ListValue, "log_level", translate("Log Level"))
    log_level:value("DEBUG")
    log_level:value("INFO")
    log_level:value("WARN")
    log_level:value("ERROR")
    log_level.default = "WARN"
    log_level.description = translate(
        "Sets the logging level. Do not keep the log level set to DEBUG for an extended period of time.")

    -- Log Lines
    local logLines = section:taboption("log", Value, "log_lines", translate("Display Lines"))
    logLines.default = "1000"
    logLines.datatype = "uinteger"
    logLines.rmempty = false

    -- Button Container (DummyValue to hold all buttons)
    local button_container = section:taboption("log", DummyValue, "_button_container", translate("Log Actions"))
    button_container.rawhtml = true

    function button_container.cfgvalue(self, section)
        return ""
    end

    function button_container.render(self, section, scope)
        luci.http.write([[
            <div class="cbi-value" id="cbi-ua3f-main-_button_container">
                <label class="cbi-value-title">]] .. translate("Log Management") .. [[</label>
                <div class="cbi-value-field" style="display: flex; gap: 10px; flex-wrap: wrap;">
                    <input type="button" class="btn cbi-button cbi-button-reset"
                           value="]] .. translate("Clear Logs") .. [[" id="ua3f-clearlog-btn"/>
                    <input type="button" class="btn cbi-button cbi-button-apply"
                           value="]] .. translate("Download Logs") .. [[" id="ua3f-download-btn"/>
                    <input type="button" class="btn cbi-button cbi-button-save"
                           value="]] .. translate("Issue Report") .. [[" id="ua3f-issue-btn"/>
                </div>
            </div>
            <script>
            (function() {
                // Clear Log Button
                document.getElementById('ua3f-clearlog-btn').addEventListener('click', function(e) {
                    e.preventDefault();
                    fetch(']] .. luci.dispatcher.build_url("admin/services/ua3f/clear_log") .. [[', {method: 'POST'})
                    .then(resp => {
                        if (resp.ok) {
                            var textarea = document.getElementById('cbid.ua3f.main.log');
                            if (textarea) textarea.value = "";
                        }
                    });
                });

                // Download Log Button
                document.getElementById('ua3f-download-btn').addEventListener('click', function(e) {
                    e.preventDefault();
                    window.location.href = ']] .. luci.dispatcher.build_url("admin/services/ua3f/download_log") .. [[';
                });

                // Issue Report Button
                document.getElementById('ua3f-issue-btn').addEventListener('click', function(e) {
                    e.preventDefault();
                    window.open('https://github.com/SunBK201/UA3F/issues/new?template=bug-report.md', '_blank');
                });
            })();
            </script>
        ]])
    end

    return log
end

function M.add_desync_fields(section)
    -- Enable TCP Desync
    local desync_enabled = section:taboption("desync", Flag, "desync_enabled", translate("Enable TCP Desync"))
    desync_enabled.description = translate("Enable TCP Desynchronization to evade DPI")

    -- CT Byte Setting
    local ct_byte = section:taboption("desync", Value, "desync_ct_bytes", translate("Desync Bytes"))
    ct_byte.placeholder = "1500"
    ct_byte.datatype = "uinteger"
    ct_byte.description = translate("Number of bytes for fragmented random emission")
    ct_byte:depends("desync_enabled", "1")

    -- CT Packets Setting
    local ct_packets = section:taboption("desync", Value, "desync_ct_packets", translate("Desync Packets"))
    ct_packets.placeholder = "8"
    ct_packets.datatype = "uinteger"
    ct_packets.description = translate("Number of packets for fragmented random emission")
    ct_packets:depends("desync_enabled", "1")
end

-- Others Tab Fields
function M.add_others_fields(section)
    -- TTL Setting
    local ttl = section:taboption("others", Flag, "set_ttl", translate("Set TTL"))
    ttl.description = translate("Set the TTL 64 for packets")

    -- TCP Timestamp Deletion
    local tcpts = section:taboption("others", Flag, "del_tcpts", translate("Delete TCP Timestamps"))
    tcpts.description = translate("Remove TCP Timestamp option")

    -- TCP Initial Window
    local tcp_init_window = section:taboption("others", Flag, "set_tcp_init_window", translate("Set TCP Initial Window"))
    tcp_init_window.description = translate("Set the TCP Initial Window to 65535 for SYN packets")

    -- IP ID Setting
    local ipid = section:taboption("others", Flag, "set_ipid", translate("Set IP ID"))
    ipid.description = translate("Set the IP ID to 0 for packets")
end

return M