obs = obslua

-- Versionssprung
local script_version = "v1.1"

-- Benutzereinstellungen
settings = {
    text_source_name = "",
    prefix = "",
    suffix = "",
    seconds_option = "show",
    test_mode = false,
    layout_option = "single"  -- Optionen: "single", "two", "three"
}

start_time = nil
update_interval = 1 -- Sekunden
timer_active = false
test_mode_active = false

-- Hilfsfunktion: Fügt alle OBS-Textquellen zur Property-Liste hinzu (unter Verwendung von unversioned IDs)
local function add_text_sources(p)
    obs.obs_property_list_add_string(p, "Keine Textquelle ausgewählt", "")
    local all_sources = obs.obs_enum_sources()
    if all_sources then
        for _, src in ipairs(all_sources) do
            local sid = obs.obs_source_get_unversioned_id(src)
            if sid == "text_gdiplus" or sid == "text_ft2_source" then
                local name = obs.obs_source_get_name(src)
                obs.obs_property_list_add_string(p, name, name)
            end
        end
        obs.source_list_release(all_sources)
    end
end

function script_description()
    return "Skriptbeschreibung:\n" ..
           "Dieses Skript zeigt die Streaming-Uptime in einer OBS-Textquelle an. Es kann genutzt werden, um die Dauer eines Live-Streams übersichtlich darzustellen.\n\n" ..
           "Anleitung zur Nutzung:\n" ..
           "1. Wähle eine Textquelle aus der Dropdown-Liste aus.\n" ..
           "2. Gib ein Präfix und ein Suffix ein (diese werden automatisch formatiert).\n" ..
           "3. Wähle das gewünschte Layout (einzeilig, zweizeilig oder dreizeilig).\n" ..
           "4. Stelle die Sekundenanzeige ein und aktiviere den Testmodus, falls gewünscht.\n\n" ..
           "Update-Benachrichtigung (Wesentliche Änderungen seit v1.0):\n" ..
           "• Dropdown für die Auswahl der Textquelle hinzugefügt.\n" ..
           "• Automatische Formatierung für Präfix und Suffix.\n" ..
           "• Mehrzeilige Layout-Optionen für flexible Anzeige."
end

function script_properties()
    local props = obs.obs_properties_create()

    -- Dropdown für Textquelle mithilfe der Hilfsfunktion
    local text_sources = obs.obs_properties_add_list(props, "text_source_name", "Textquelle", obs.OBS_COMBO_TYPE_LIST, obs.OBS_COMBO_FORMAT_STRING)
    add_text_sources(text_sources)

    -- Dropdown für das Text-Layout
    local layout_options = obs.obs_properties_add_list(props, "layout_option", "Text-Layout", obs.OBS_COMBO_TYPE_LIST, obs.OBS_COMBO_FORMAT_STRING)
    obs.obs_property_list_add_string(layout_options, "Einzeilig: Präfix Uptime Suffix", "single")
    obs.obs_property_list_add_string(layout_options, "Zweizeilig: Präfix Uptime / Suffix", "two")
    obs.obs_property_list_add_string(layout_options, "Dreizeilig: Präfix / Uptime / Suffix", "three")

    -- Eingabefelder für Präfix und Suffix
    obs.obs_properties_add_text(props, "prefix", "Präfix", obs.OBS_TEXT_DEFAULT)
    obs.obs_properties_add_text(props, "suffix", "Suffix", obs.OBS_TEXT_DEFAULT)
    
    -- Dropdown für die Sekundenanzeige
    local seconds_options = obs.obs_properties_add_list(props, "seconds_option", "Sekundenanzeige", obs.OBS_COMBO_TYPE_LIST, obs.OBS_COMBO_FORMAT_STRING)
    obs.obs_property_list_add_string(seconds_options, "Anzeigen", "show")
    obs.obs_property_list_add_string(seconds_options, "Verbergen", "hide")
    obs.obs_property_list_add_string(seconds_options, "Nach 1 Minute verbergen", "hide_after_first_minute")
    
    -- Kontrollkästchen für den Testmodus
    obs.obs_properties_add_bool(props, "test_mode", "Testmodus")

    return props
end

function script_update(new_settings)
    settings.text_source_name = obs.obs_data_get_string(new_settings, "text_source_name")
    settings.prefix = obs.obs_data_get_string(new_settings, "prefix")
    settings.suffix = obs.obs_data_get_string(new_settings, "suffix")
    settings.seconds_option = obs.obs_data_get_string(new_settings, "seconds_option")
    settings.test_mode = obs.obs_data_get_bool(new_settings, "test_mode")
    settings.layout_option = obs.obs_data_get_string(new_settings, "layout_option")

    -- Automatische Formatierung: Hänge ein Leerzeichen an das Präfix an, falls nicht vorhanden
    if settings.prefix ~= "" and string.sub(settings.prefix, -1) ~= " " then
        settings.prefix = settings.prefix .. " "
    end

    -- Automatische Formatierung: Füge ein führendes Leerzeichen zum Suffix hinzu, falls nicht vorhanden
    if settings.suffix ~= "" and string.sub(settings.suffix, 1, 1) ~= " " then
        settings.suffix = " " .. settings.suffix
    end

    if timer_active then
        obs.timer_remove(update_text_source)
        obs.timer_add(update_text_source, update_interval * 1000)
    end

    if settings.test_mode and not test_mode_active then
        start_time = os.time()
        test_mode_active = true
        if not timer_active then
            obs.timer_add(update_text_source, update_interval * 1000)
            timer_active = true
        end
    elseif not settings.test_mode and test_mode_active then
        start_time = nil
        test_mode_active = false
        if timer_active then
            obs.timer_remove(update_text_source)
            timer_active = false
        end
    end

    update_text_source()
end

function script_defaults(settings_data)
    obs.obs_data_set_default_string(settings_data, "text_source_name", "")
    obs.obs_data_set_default_string(settings_data, "prefix", "")
    obs.obs_data_set_default_string(settings_data, "suffix", "")
    obs.obs_data_set_default_string(settings_data, "seconds_option", "show")
    obs.obs_data_set_default_bool(settings_data, "test_mode", false)
    obs.obs_data_set_default_string(settings_data, "layout_option", "single")
end

function update_text_source()
    if settings.text_source_name == "" then
        return
    end

    local source = obs.obs_get_source_by_name(settings.text_source_name)
    if source then
        local uptime = get_uptime()
        local final_text = ""
        if settings.layout_option == "single" then
            final_text = settings.prefix .. uptime .. settings.suffix
        elseif settings.layout_option == "two" then
            final_text = settings.prefix .. uptime .. "\n" .. settings.suffix
        elseif settings.layout_option == "three" then
            final_text = settings.prefix .. "\n" .. uptime .. "\n" .. settings.suffix
        end

        local settings_data = obs.obs_data_create()
        obs.obs_data_set_string(settings_data, "text", final_text)
        obs.obs_source_update(source, settings_data)
        obs.obs_data_release(settings_data)
        obs.obs_source_release(source)
    end
end

function get_uptime()
    if not start_time then
        return settings.seconds_option == "hide" and "00:00" or "00:00:00"
    end

    local now = os.time()
    local elapsed = now - start_time
    if settings.test_mode then
        elapsed = os.difftime(now, start_time)
    end

    local hours = math.floor(elapsed / 3600)
    local minutes = math.floor((elapsed % 3600) / 60)
    local seconds = elapsed % 60

    if settings.seconds_option == "hide" or (settings.seconds_option == "hide_after_first_minute" and elapsed >= 60) then
        return string.format("%02d:%02d", hours, minutes)
    else
        return string.format("%02d:%02d:%02d", hours, minutes, seconds)
    end
end

function on_event(event)
    if event == obs.OBS_FRONTEND_EVENT_STREAMING_STARTED then
        start_time = os.time()
        if not timer_active then
            obs.timer_add(update_text_source, update_interval * 1000)
            timer_active = true
        end
    elseif event == obs.OBS_FRONTEND_EVENT_STREAMING_STOPPED then
        start_time = nil
        if timer_active then
            obs.timer_remove(update_text_source)
            timer_active = false
        end
        update_text_source() -- Anzeige zurücksetzen
    end
end

function script_load(settings_data)
    obs.obs_frontend_add_event_callback(on_event)
end

function script_unload()
    if timer_active then
        obs.timer_remove(update_text_source)
    end
end