<?php
/**
 * Theme Functions
 *
 * Dynamically generates /llms.txt
 */

/**
 * Register the rewrite rule.
 */
function ps4u_llms_rewrite_rule() {
    add_rewrite_rule('^llms\.txt$', 'index.php?llms_file=1', 'top');
}
add_action('init', 'ps4u_llms_rewrite_rule');

/**
 * Register custom query variable.
 */
function ps4u_llms_query_var($vars) {
    $vars[] = 'llms_file';
    return $vars;
}
add_filter('query_vars', 'ps4u_llms_query_var');

/**
 * Generate llms.txt dynamically.
 */
function ps4u_generate_llms_file() {

    if (!get_query_var('llms_file')) {
        return;
    }

    header('Content-Type: text/plain; charset=UTF-8');

    echo "# Psychiatry Services 4U\n\n";
    echo "> Psychiatry Services 4U provides comprehensive psychiatric care, psychotherapy, child & adolescent psychiatry, and telepsychiatry services in Dubai, UAE.\n\n";

    echo "## Main Pages\n\n";

    $pages = array(
        ''                    => 'Home',
        'about-us'            => 'About Us',
        'services'            => 'Services',
        'our-doctors'         => 'Our Doctors',
        'conditions-we-treat' => 'Conditions We Treat',
        'blog'                => 'Blog',
        'help-faqs'           => 'FAQs',
        'contact-us'          => 'Contact',
    );

    foreach ($pages as $slug => $title) {
        echo "- {$title}: " . home_url('/' . ltrim($slug, '/')) . "\n";
    }

    echo "\n## Latest Blog Articles\n\n";

    $posts = get_posts(array(
        'post_type'      => 'post',
        'posts_per_page' => 20,
        'post_status'    => 'publish',
        'orderby'        => 'date',
        'order'          => 'DESC',
    ));

    foreach ($posts as $post) {
        echo "- {$post->post_title}: " . get_permalink($post->ID) . "\n";
    }

    echo "\n## Sitemap\n";
    echo home_url('/sitemap.xml') . "\n";

    echo "\n## Robots\n";
    echo home_url('/robots.txt') . "\n";

    exit;
}
add_action('template_redirect', 'ps4u_generate_llms_file');