module dtm2

import os
import time

const test_root_name = 'dtm2_test'

fn dtm2_test_root() string {
    return os.join_path(os.vtmp_dir(), '${test_root_name}_${os.getpid()}')
}

fn dtm2_outside_template_path() string {
    return os.join_path(os.vtmp_dir(), '${test_root_name}_${os.getpid()}_outside.html')
}

fn testsuite_begin() {
    root := dtm2_test_root()
    os.rmdir_all(root) or {}
    os.mkdir_all(os.join_path(root, 'partials'))!
    outside_path := dtm2_outside_template_path()
    os.rm(outside_path) or {}
    os.write_file(outside_path, '<p>@title outside</p>')!
    os.write_file(os.join_path(root, 'page.html'), '<main>@title @body @missing</main>')!
    os.write_file(os.join_path(root, 'invalid.css'), 'body { color: red; }')!
    os.write_file(os.join_path(root, 'raw.txt'), 'Raw: @body')!
    os.write_file(os.join_path(root, 'with_include.html'),
        '<header>@include "partials/nav"</header><main>@title</main>')!
    os.write_file(os.join_path(root, 'partials', 'nav.html'), '<nav>@title</nav>')!
    os.write_file(os.join_path(root, 'include_outside_relative.html'),
        '@include "../${os.base(outside_path)}"')!
    os.write_file(os.join_path(root, 'include_outside_absolute.html'), '@include "${outside_path}"')!
    os.write_file(os.join_path(root, 'reload.html'), '<p>@title</p>')!
    os.write_file(os.join_path(root, 'pinned.html'), '<em>@title</em>')!
    os.write_file(os.join_path(root, 'path_alias.html'), '<section>@title</section>')!
    os.write_file(os.join_path(root, 'prefix.html'), '<p>@unknown</p>')!
    os.write_file(os.join_path(root, 'recursive.html'), '@include "recursive"')!
    os.write_file(os.join_path(root, 'compress.html'), '<div>\n <span>@title</span>\n</div>')!
    os.write_file(os.join_path(root, 'empty.html'), '')!
    os.write_file(os.join_path(root, 'feed.xml'), '<feed><title>@title</title></feed>')!
    os.write_file(os.join_path(root, 'custom.page'), '<article>@title</article>')!
    os.write_file(os.join_path(root, 'custom.note'), 'Note: @body')!
    os.write_file(os.join_path(root, 'override.html'), '<div>\n @title\n</div>')!
    os.write_file(os.join_path(root, 'custom.view'), '<section>@title</section>')!
    os.write_file(os.join_path(root, 'custom.mail'), 'Subject: @title')!
    os.write_file(os.join_path(root, 'auto.fragment'), '<aside>@title</aside>')!
    os.write_file(os.join_path(root, 'auto.message'), 'Auto: @title')!
    os.write_file(os.join_path(root, 'bad.path'), '<p>@title</p>')!
    os.write_file(os.join_path(root, 'extensions.json'), '{"html":[".view"],"text":["mail"]}')!
    os.write_file(os.join_path(root, default_extension_config_filename),
        '{"html":[".fragment"],"text":[".message"]}')!
    os.write_file(os.join_path(root, 'invalid_extensions.json'),
        '{"html":["../bad"],"text":["bad/value",".message"]}')!
}

fn testsuite_end() {
    os.rmdir_all(dtm2_test_root()) or {}
    os.rm(dtm2_outside_template_path()) or {}
}

fn new_test_manager(compress_html bool) &Manager {
    return initialize(
        template_dir:  dtm2_test_root()
        compress_html: compress_html
    )
}

fn test_render_placeholders_escape_and_preserve_missing() {
    mut manager := new_test_manager(false)
    placeholders := {
        'title': 'Hello'
        'body':  '<script>alert(1)</script>'
    }
    rendered := manager.expand('page.html', placeholders: &placeholders)
    assert rendered == '<main>Hello <script>alert(1)</script> @missing</main>\n'
    assert manager.compiled_template_count() == 1
}

fn test_include_and_compiled_template_registry() {
    mut manager := new_test_manager(true)
    placeholders := {
        'title': 'Home'
    }
    rendered := manager.expand('with_include.html', placeholders: &placeholders)
    assert rendered == '<header><nav>Home</nav></header><main>Home</main>'
    assert manager.compiled_template_count() == 1
    _ = manager.expand('with_include.html', placeholders: &placeholders)
    assert manager.compiled_template_count() == 1
}

fn test_include_html_suffix_keeps_only_allowed_tags() {
    mut manager := new_test_manager(false)
    placeholders := {
        'body_#includehtml': '<thead><tr><td>safe</td></tr></thead><script>bad</script>'
        'title':             'HTML'
    }
    rendered := manager.expand('page.html', placeholders: &placeholders)
    assert rendered.contains('<thead><tr><td>safe</td></tr></thead>')
    assert rendered.contains('<script>bad</script>')
}

fn test_text_mode_always_escapes_html() {
    mut manager := new_test_manager(false)
    placeholders := {
        'body_#includehtml': '<span>text</span>'
    }
    rendered := manager.expand('raw.txt', placeholders: &placeholders)
    assert rendered == 'Raw: <span>text</span>\n'
}

fn test_reload_modified_template_without_render_cache() {
    mut manager := new_test_manager(false)
    placeholders := {
        'title': 'One'
    }
    first := manager.expand('reload.html', placeholders: &placeholders)
    assert first == '<p>One</p>\n'
    os.write_file(os.join_path(dtm2_test_root(), 'reload.html'), '<strong>@title</strong>')!
    second := manager.expand('reload.html', placeholders: &placeholders)
    assert second == '<strong>One</strong>\n'
    assert manager.compiled_template_count() == 1
}

fn test_reload_same_second_same_size_template_content_change() {
    path := os.join_path(dtm2_test_root(), 'same_size.html')
    fixed_time := i64(2_306_102_495)
    os.write_file(path, '<p>@title A</p>')!
    os.utime(path, fixed_time, fixed_time)!
    mut manager := new_test_manager(false)
    placeholders := {
        'title': 'One'
    }
    first := manager.expand('same_size.html', placeholders: &placeholders)
    assert first == '<p>One A</p>\n'
    os.write_file(path, '<b>@title B</b>')!
    os.utime(path, fixed_time, fixed_time)!
    second := manager.expand('same_size.html', placeholders: &placeholders)
    assert second == '<b>One B</b>\n'
    assert manager.compiled_template_count() == 1
}

fn test_reload_same_second_same_size_include_content_change() {
    parent_path := os.join_path(dtm2_test_root(), 'same_size_include_parent.html')
    include_path := os.join_path(dtm2_test_root(), 'partials', 'same_size_include.html')
    fixed_time := i64(2_306_102_496)
    os.write_file(parent_path, '<main>@include "partials/same_size_include"</main>')!
    os.write_file(include_path, '<p>@title A</p>')!
    os.utime(include_path, fixed_time, fixed_time)!
    mut manager := new_test_manager(false)
    placeholders := {
        'title': 'One'
    }
    first := manager.expand('same_size_include_parent.html', placeholders: &placeholders)
    assert first == '<main><p>One A</p></main>\n'
    os.write_file(include_path, '<b>@title B</b>')!
    os.utime(include_path, fixed_time, fixed_time)!
    second := manager.expand('same_size_include_parent.html', placeholders: &placeholders)
    assert second == '<main><b>One B</b></main>\n'
    assert manager.compiled_template_count() == 1
}

fn test_reload_modified_include_without_render_cache() {
    mut manager := new_test_manager(true)
    placeholders := {
        'title': 'Home'
    }
    first := manager.expand('with_include.html', placeholders: &placeholders)
    assert first == '<header><nav>Home</nav></header><main>Home</main>'
    time.sleep(1100 * time.millisecond)
    os.write_file(os.join_path(dtm2_test_root(), 'partials', 'nav.html'),
        '<nav class="updated">@title</nav>')!
    second := manager.expand('with_include.html', placeholders: &placeholders)
    assert second == '<header><nav class="updated">Home</nav></header><main>Home</main>'
    assert manager.compiled_template_count() == 1
}

fn test_reload_disabled_keeps_cached_template_tree() {
    mut manager := initialize(
        template_dir:              dtm2_test_root()
        compress_html:             false
        reload_modified_templates: false
    )
    placeholders := {
        'title': 'Pinned'
    }
    first := manager.expand('pinned.html', placeholders: &placeholders)
    assert first == '<em>Pinned</em>\n'
    os.write_file(os.join_path(dtm2_test_root(), 'pinned.html'), '<strong>@title</strong>')!
    second := manager.expand('pinned.html', placeholders: &placeholders)
    assert second == '<em>Pinned</em>\n'
    assert manager.compiled_template_count() == 1
}

fn test_relative_and_absolute_paths_share_compiled_template() {
    mut manager := new_test_manager(false)
    placeholders := {
        'title': 'Alias'
    }
    relative := manager.expand('path_alias.html', placeholders: &placeholders)
    absolute := manager.expand(os.join_path(dtm2_test_root(), 'path_alias.html'),
        placeholders: &placeholders
    )
    assert relative == '<section>Alias</section>\n'
    assert absolute == relative
    assert manager.compiled_template_count() == 1
}

fn test_template_path_cannot_escape_template_dir() {
    mut manager := new_test_manager(false)
    placeholders := {
        'title': 'Escape'
    }
    relative_escape := manager.expand('../${os.base(dtm2_outside_template_path())}',
        placeholders: &placeholders
    )
    absolute_escape := manager.expand(dtm2_outside_template_path(), placeholders: &placeholders)
    assert relative_escape == internal_server_error
    assert absolute_escape == internal_server_error
    assert manager.compiled_template_count() == 0
}

fn test_cached_template_path_revalidates_symlink_escape() {
    $if windows {
        return
    }
    path := os.join_path(dtm2_test_root(), 'symlink_swap.html')
    os.write_file(path, '<p>@title safe</p>')!
    mut manager := new_test_manager(false)
    placeholders := {
        'title': 'Link'
    }
    first := manager.expand('symlink_swap.html', placeholders: &placeholders)
    assert first == '<p>Link safe</p>\n'
    os.rm(path)!
    os.symlink(dtm2_outside_template_path(), path) or { return }
    escaped := manager.expand('symlink_swap.html', placeholders: &placeholders)
    assert escaped == internal_server_error
}

fn test_cached_include_dependency_revalidates_symlink_escape() {
    $if windows {
        return
    }
    parent_path := os.join_path(dtm2_test_root(), 'include_symlink_swap_parent.html')
    include_path := os.join_path(dtm2_test_root(), 'partials', 'symlink_swap.html')
    os.write_file(parent_path, '<main>@include "partials/symlink_swap"</main>')!
    os.write_file(include_path, '<p>@title safe</p>')!
    mut manager := new_test_manager(false)
    placeholders := {
        'title': 'Link'
    }
    first := manager.expand('include_symlink_swap_parent.html', placeholders: &placeholders)
    assert first == '<main><p>Link safe</p></main>\n'
    os.rm(include_path)!
    os.symlink(dtm2_outside_template_path(), include_path) or { return }
    escaped := manager.expand('include_symlink_swap_parent.html', placeholders: &placeholders)
    assert escaped == internal_server_error
}

fn test_reload_disabled_does_not_reopen_cached_symlink_swap() {
    $if windows {
        return
    }
    path := os.join_path(dtm2_test_root(), 'pinned_symlink_swap.html')
    os.write_file(path, '<p>@title safe</p>')!
    mut manager := initialize(
        template_dir:              dtm2_test_root()
        compress_html:             false
        reload_modified_templates: false
    )
    placeholders := {
        'title': 'Pinned'
    }
    first := manager.expand('pinned_symlink_swap.html', placeholders: &placeholders)
    assert first == '<p>Pinned safe</p>\n'
    os.rm(path)!
    os.symlink(dtm2_outside_template_path(), path) or { return }
    second := manager.expand('pinned_symlink_swap.html', placeholders: &placeholders)
    assert second == first
    assert manager.compiled_template_count() == 1
}

fn test_include_path_cannot_escape_template_dir() {
    mut manager := new_test_manager(false)
    placeholders := {
        'title': 'Escape'
    }
    relative_include := manager.expand('include_outside_relative.html', placeholders: &placeholders)
    absolute_include := manager.expand('include_outside_absolute.html', placeholders: &placeholders)
    assert relative_include == internal_server_error
    assert absolute_include == internal_server_error
    assert manager.compiled_template_count() == 0
}

fn test_custom_missing_placeholder_prefix() {
    mut manager := new_test_manager(false)
    placeholders := map[string]string{}
    rendered := manager.expand('prefix.html',
        placeholders:               &placeholders
        missing_placeholder_prefix: '?'
    )
    assert rendered == '<p>?unknown</p>\n'
}

fn test_recursive_include_returns_internal_server_error() {
    mut manager := new_test_manager(false)
    placeholders := map[string]string{}
    rendered := manager.expand('recursive.html', placeholders: &placeholders)
    assert rendered == internal_server_error
    assert manager.compiled_template_count() == 0
}

fn test_html_compression_is_deterministic() {
    mut manager := new_test_manager(true)
    placeholders := {
        'title': 'Compact'
    }
    rendered := manager.expand('compress.html', placeholders: &placeholders)
    assert rendered == '<div><span>Compact</span></div>'
}

fn test_empty_template_renders_empty() {
    mut manager := new_test_manager(false)
    placeholders := map[string]string{}
    rendered := manager.expand('empty.html', placeholders: &placeholders)
    assert rendered == ''
}

fn test_xml_template_uses_html_rendering_mode() {
    mut manager := new_test_manager(false)
    placeholders := {
        'title': '<escaped>'
    }
    rendered := manager.expand('feed.xml', placeholders: &placeholders)
    assert rendered == '<feed><title><escaped></title></feed>\n'
}

fn test_custom_template_extension_map() {
    mut manager := initialize(
        template_dir:        dtm2_test_root()
        template_extensions: {
            '.page': TemplateType.html
            'note':  TemplateType.text
        }
    )
    html_placeholders := {
        'title_#includehtml': '<span>custom</span><script>blocked</script>'
    }
    text_placeholders := {
        'body_#includehtml': '<span>custom</span>'
    }
    html := manager.expand('custom.page', placeholders: &html_placeholders)
    text := manager.expand('custom.note', placeholders: &text_placeholders)
    assert html.contains('<span>custom</span><script>blocked</script>')
    assert text == 'Note: <span>custom</span>\n'
}

fn test_custom_template_extension_map_can_override_builtin_mapping() {
    mut manager := initialize(
        template_dir:        dtm2_test_root()
        compress_html:       true
        template_extensions: {
            '.html': TemplateType.text
        }
    )
    placeholders := {
        'title': 'Text'
    }
    rendered := manager.expand('override.html', placeholders: &placeholders)
    assert rendered == '<div>\n Text\n</div>\n'
}

fn test_json_extension_config_file() {
    mut manager := initialize(
        template_dir:          dtm2_test_root()
        extension_config_file: os.join_path(dtm2_test_root(), 'extensions.json')
    )
    placeholders := {
        'title': '<Config>'
    }
    html := manager.expand('custom.view', placeholders: &placeholders)
    text := manager.expand('custom.mail', placeholders: &placeholders)
    assert html == '<section><Config></section>'
    assert text == 'Subject: <Config>\n'
}

fn test_default_json_extension_config_file_is_loaded_from_template_dir() {
    mut manager := initialize(
        template_dir:  dtm2_test_root()
        compress_html: false
    )
    placeholders := {
        'title': '<Auto>'
    }
    html := manager.expand('auto.fragment', placeholders: &placeholders)
    text := manager.expand('auto.message', placeholders: &placeholders)
    assert html == '<aside><Auto></aside>\n'
    assert text == 'Auto: <Auto>\n'
}

fn test_invalid_json_extension_entries_are_ignored() {
    mut manager := initialize(
        template_dir:          dtm2_test_root()
        extension_config_file: os.join_path(dtm2_test_root(), 'invalid_extensions.json')
        compress_html:         false
    )
    placeholders := {
        'title': '<Safe>'
    }
    ignored := manager.expand('bad.path', placeholders: &placeholders)
    loaded := manager.expand('auto.message', placeholders: &placeholders)
    assert ignored == internal_server_error
    assert loaded == 'Auto: <Safe>\n'
}

fn test_missing_template_returns_internal_server_error() {
    mut manager := new_test_manager(false)
    placeholders := map[string]string{}
    rendered := manager.expand('missing.html', placeholders: &placeholders)
    assert rendered == internal_server_error
    assert manager.compiled_template_count() == 0
}

fn test_invalid_template_extension_returns_internal_server_error() {
    mut manager := new_test_manager(false)
    placeholders := map[string]string{}
    rendered := manager.expand('invalid.css', placeholders: &placeholders)
    assert rendered == internal_server_error
    assert manager.compiled_template_count() == 0
}