<?php
/**
 * 各页面重写规则.
 */

//页面伪静态规则
function gdk_page_permalink()
{
    global $wp_rewrite;
    if (!strpos($wp_rewrite->get_page_permastruct(), '.html')) {
        $wp_rewrite->page_structure = $wp_rewrite->page_structure.'.html';
    }
}
add_action('init', 'gdk_page_permalink', -1);

// 更新重写规则
function gdk_rewrite_flush_rules()
{
    $rules = get_option('rewrite_rules');
    if (!isset($rules['^sitemap(.*?)\.xml$'], $rules['^sitemap(.*?)\.html$'], $rules['^daohang(.*?)\.html$'])) {
        global $wp_rewrite;
        $wp_rewrite->flush_rules();
    }
}
add_action('wp_loaded', 'gdk_rewrite_flush_rules');

// 添加自定义URL重写
function gdk_add_custom_rewrite_rule()
{
    add_rewrite_rule('^sitemap(.*?)\.xml$', 'index.php?sitemap=gdk$matches[1]', 'top'); //xml地图规则
    add_rewrite_rule('^sitemap(.*?)\.html$', 'index.php?sitemap=gdkk$matches[1]', 'top'); //html地图规则
    add_rewrite_rule('^daohang(.*?)\.html$', 'index.php?daohang=gdkk$matches[1]', 'top'); //导航页面规则
    add_rewrite_rule('^weauth', 'index.php?user=$matches[1]&sk=$matches[2]', 'top'); //WeAuth微信登陆二维码规则
    add_rewrite_rule('^goauth', 'index.php?userinfo=$matches[1]&sk=$matches[2]', 'top'); //GoAuth微信登陆二维码规则
}
add_action('init', 'gdk_add_custom_rewrite_rule', 10, 0);

//自定义链接跳转
function gdk_custom_cancel_redirect($redirect_url)
{
    $api_sitemap = get_query_var('sitemap');
    $api_daohang = get_query_var('daohang');
    if (!empty($api_sitemap) || !empty($api_daohang)) {
        return false;
    }

    return $redirect_url;
}
add_filter('redirect_canonical', 'gdk_custom_cancel_redirect');

//自定义路由参数,GO跳转和dl下载
function gdk_public_query_vars($public_query_vars)
{
    $public_query_vars[] = 'go';
    $public_query_vars[] = 'dl';
    $public_query_vars[] = 'sitemap';
    $public_query_vars[] = 'daohang';

    return $public_query_vars;
}
add_action('query_vars', 'gdk_public_query_vars');