<?php


use Puock\Theme\classes\PuockClassLoad;

add_action('after_setup_theme', 'puock_theme_setup');
function puock_theme_setup()
{
    load_theme_textdomain(PUOCK, PUOCK_ABS_DIR . '/languages');
}

if (is_dir(PUOCK_ABS_DIR . '/inc/puock')) {
    if (file_exists(PUOCK_ABS_DIR . '/inc/puock/fun.php')) {
        require_once PUOCK_ABS_DIR . '/inc/puock/fun.php';
    }
}

function pk_ajax_resp($data = null, $msg = '', $code = 0)
{
    header('Content-Type: application/json; charset=utf-8');
    return json_encode(array('data' => $data, 'msg' => $msg, 'code' => $code));
}

function pk_ajax_resp_error($msg = 'fail', $data = null)
{
    return pk_ajax_resp($data, $msg, -1);
}

require_once PUOCK_ABS_DIR . '/inc/fun/cache.php';
require_once PUOCK_ABS_DIR . '/inc/setting/index.php';
require_once PUOCK_ABS_DIR . '/inc/ext/init.php';
require_once PUOCK_ABS_DIR . '/inc/fun/ajax.php';
require_once PUOCK_ABS_DIR . '/inc/fun/mail.php';
require_once PUOCK_ABS_DIR . '/inc/oauth/oauth.php';
require_once PUOCK_ABS_DIR . '/inc/fun/security.php';
require_once PUOCK_ABS_DIR . '/inc/fun/comment-ajax.php';
require_once PUOCK_ABS_DIR . '/inc/fun/widget.php';
require_once PUOCK_ABS_DIR . '/inc/init.php';
require_once PUOCK_ABS_DIR . '/inc/category-seo.php';
require_once PUOCK_ABS_DIR . '/inc/fun/comment.php';
require_once PUOCK_ABS_DIR . '/inc/fun/short-code.php';
require_once PUOCK_ABS_DIR . '/inc/fun/opt.php';
require_once PUOCK_ABS_DIR . '/inc/fun/post-meta.php';
require_once PUOCK_ABS_DIR . '/inc/fun/sidebar.php';
require_once PUOCK_ABS_DIR . '/inc/fun/post-tags.php';
require_once PUOCK_ABS_DIR . '/inc/fun/comment-notify.php';
require_once PUOCK_ABS_DIR . '/inc/user-agent-parse.php';
require_once PUOCK_ABS_DIR . '/inc/phpqrcode.php';
require_once PUOCK_ABS_DIR . '/inc/ajax/index.php';
if (pk_is_checked('no_category')) {
    require_once PUOCK_ABS_DIR . '/inc/no-category.php';
}
$puock_class_load = new PuockClassLoad();

/*Auth-Domains*/

//钩子添加集合
if (pk_is_checked('html_page_permalink')) {
    add_action('init', 'html_page_permalink', -1);
}
add_filter('user_trailingslashit', 'add_init_trailingslashit', 10, 2);


function pk_open_session()
{
    session_start();
}

function pk_wclose_session()
{
    session_write_close();
}

function pk_session_call($function)
{
    pk_open_session();
    try {
        $function();
    } finally {
        session_write_close();
    }
}

function pk_get_theme_option_url($to = '')
{
    return admin_url() . 'admin.php?page=puock-options&to=' . $to;
}


// 顶部添加自定义菜单
function pk_toolbar_link(WP_Admin_Bar $bar)
{
    $menu_id = 'theme-quick-start';
    $bar->add_node(array(
        'id' => $menu_id,
        'title' => '<i class="czs-paper-plane"></i>&nbsp;' . __('Puock Theme 快捷入口', PUOCK),
        'href' => '#'
    ));
    $bar->add_node(array(
        'id' => 'theme-setting',
        'parent' => $menu_id,
        'title' => '<i class="czs-setting" style="color:#9627e3"></i>&nbsp;' . __('主题设置', PUOCK),
        'href' => pk_get_theme_option_url()
    ));
    $bar->add_node(array(
        'id' => 'theme-docs',
        'parent' => $menu_id,
        'title' => '<i class="czs-doc-file" style="color:#496cf9"></i>&nbsp;' . __('主题文档', PUOCK),
        'href' => 'https://licoy.cn/puock-doc.html',
        'meta' => array(
            'target' => 'blank'
        )
    ));
    $bar->add_node(array(
        'id' => 'theme-sponsor',
        'parent' => $menu_id,
        'title' => '<i class="czs-heart" style="color:#f54747"></i>&nbsp;' . __('赞助主题', PUOCK),
        'href' => 'https://licoy.cn/puock-theme-sponsor.html',
        'meta' => array(
            'target' => 'blank'
        )
    ));
    $bar->add_node(array(
        'id' => 'theme-group',
        'parent' => $menu_id,
        'title' => '<i class="czs-weixin" style="color:#177b17"></i>&nbsp;' . __('主题交流群', PUOCK),
        'href' => 'https://licoy.cn/go/puock-update.php?r=qq_qun',
        'meta' => array(
            'target' => 'blank'
        )
    ));
    $bar->add_node(array(
        'id' => 'theme-github',
        'parent' => $menu_id,
        'title' => '<i class="czs-github-logo"></i>&nbsp;' . __('Github 开源主页', PUOCK),
        'href' => 'https://github.com/Licoy/wordpress-theme-puock',
        'meta' => array(
            'target' => 'blank'
        )
    ));
}

if (is_user_logged_in() && current_user_can('manage_options')) {
    add_action('admin_bar_menu', 'pk_toolbar_link', 999);
}

function pk_admin_scripts()
{
    wp_enqueue_script('puock-admin', get_stylesheet_directory_uri() . '/assets/dist/admin.min.js',
        array(), PUOCK_CUR_VER_STR, true);
}

add_action('admin_enqueue_scripts', 'pk_admin_scripts');
function pk_admin_print_scripts()
{
    $settings = json_encode(array(
        'compatible' => [
            'githubermd' => defined('GITHUBER_PLUGIN_NAME')
        ]
    ));
    echo "<script type='text/javascript'>var puock_admin_setting = $settings</script>";
}

add_action('admin_print_footer_scripts', 'pk_admin_print_scripts', 1);


function pk_is_pjax()
{
    return pk_is_checked('page_ajax_load', false);
}

function pk_view_dedupe_seconds(): int
{
    $hours = intval(pk_get_option('view_dedupe_hours', 24));
    if ($hours <= 0) {
        return 0;
    }
    return $hours * 3600;
}

function pk_view_dedupe_cookie_key($post_ID): string
{
    return 'pk_viewed_' . $post_ID;
}

function pk_has_view_dedupe_cookie($post_ID): bool
{
    $seconds = pk_view_dedupe_seconds();
    if ($seconds <= 0) {
        return false;
    }
    $key = pk_view_dedupe_cookie_key($post_ID);
    return !empty($_COOKIE[$key]);
}

function pk_set_view_dedupe_cookie($post_ID): void
{
    $seconds = pk_view_dedupe_seconds();
    if ($seconds <= 0) {
        return;
    }
    $key = pk_view_dedupe_cookie_key($post_ID);
    setcookie($key, '1', time() + $seconds, COOKIEPATH, COOKIE_DOMAIN, is_ssl(), true);
    $_COOKIE[$key] = '1';
}

function pk_is_instantclick_prefetch(): bool
{
    $instant = $_SERVER['HTTP_X_INSTANTCLICK'] ?? $_SERVER['HTTP_X_INSTANT_CLICK'] ?? '';
    if (!empty($instant) && strtolower($instant) === 'prefetch') {
        return true;
    }
    $purpose = strtolower($_SERVER['HTTP_X_PURPOSE'] ?? ($_SERVER['HTTP_PURPOSE'] ?? ($_SERVER['HTTP_SEC_PURPOSE'] ?? '')));
    return strpos($purpose, 'prefetch') !== false;
}

//判断阅读数量是否需要增加并进行操作
if (!function_exists('the_views_add')) {
    function the_views_add($post_ID, $count, $key, $ajax = false)
    {
        if (is_single() || is_page() || $ajax) {
            if (pk_is_instantclick_prefetch()) {
                return $count;
            }
            if (pk_has_view_dedupe_cookie($post_ID)) {
                return $count;
            }
            if ($count == '') {
                add_post_meta($post_ID, $key, '0');
            } else {
                update_post_meta($post_ID, $key, $count + 1);
                $count++;
            }
            pk_set_view_dedupe_cookie($post_ID);
        }
        return $count;
    }
}

//获取当前的阅读数量与自增
if (!function_exists('the_views')) {
    function the_views($post_id = null, $echo = true, $ajax = false)
    {
        global $post;
        if ($post_id == null) {
            $post_id = $post->ID;
        }
        $key = 'views';
        $count = get_post_meta($post_id, $key, true);
        if ($count == '') {
            $count = 0;
        }
        $count = the_views_add($post_id, $count, $key, $ajax);
        $count_view = number_format_i18n($count);
        if (!$echo) {
            return $count_view;
        }
        echo $count_view;
    }
}
//异步请求浏览量
function async_pk_views()
{
    $postId = $_POST['id'];
    if (empty($postId)) {
        echo pk_ajax_resp(0);
        return;
    }
    echo pk_ajax_resp(the_views($postId, false, true));
    wp_die();
}

add_action('wp_ajax_nopriv_async_pk_views', 'async_pk_views');
add_action('wp_ajax_async_pk_views', 'async_pk_views');

//设置文章发布的时候进行字段添加
function set_views($post_ID)
{
    $key = 'views';
    $count = get_post_meta($post_ID, $key, true);
    if ($count == '') {
        add_post_meta($post_ID, $key, '0');
    }
}

add_action('publish_post', 'set_views');

function get_total_views()
{
    global $wpdb;
    $views = pk_cache_get(PKC_TOTAL_VIEWS);
    if (!$views) {
        $views = $wpdb->get_var("SELECT SUM(meta_value) FROM $wpdb->postmeta where meta_key='views'");
        pk_cache_set(PKC_TOTAL_VIEWS, $views);
    }
    return $views;
}


function pk_check_right_md5()
{
    global $pk_right_slug;
    if (empty($pk_right_slug) || md5($pk_right_slug) != 'd8775b419146efc39dea7b0ab50c56d0') {
        wp_die(base64_decode('UGxlYXNlIGZvbGxvdyB0aGUgb3BlbiBzb3VyY2UgcnVsZXMgYW5kIGRvIG5vdCBjaGFuZ2Ugb3IgZGVsZXRlIHRoZSBib3R0b20gY29weXJpZ2h0IQ=='));
    }
}

add_action('init', 'pk_check_right_md5');


/**
 * 获取查看最多的文章
 * @param $days N天内
 * @param $nums 数量
 * @return array|object|null
 */
function get_views_most_post($days, $nums)
{
    global $wpdb;
    $sql = "select posts.*, meta.meta_value as views
            from {$wpdb->posts} as posts INNER JOIN (select post_id,(meta_value+0) as meta_value from 
            {$wpdb->postmeta} where meta_key='views' order by (meta_value+0) DESC) as meta
            on meta.post_id = posts.ID
            where posts.post_type = 'post'
            AND posts.post_status = 'publish' AND TO_DAYS(now()) - TO_DAYS(posts.post_date) < {$days}
            ORDER BY meta.meta_value DESC limit 0, {$nums}";
    return $wpdb->get_results($sql);
}

//是否隐藏侧边栏
function pk_hide_sidebar($post_id = null)
{
    global $post;
    if (pk_is_checked("hide_global_sidebar")) {
        return true;
    }
    if (is_single() || is_page()) {
        if ($post_id == null) {
            $post_id = $post->ID;
        }
        return get_post_meta($post_id, 'hide_side', true) == "true";
    }
    return false;
}

//隐藏/显示侧边栏的输出字符
function pk_hide_sidebar_out($hide = '', $show = '', $post_id = null, $echo = true)
{
    $out = $show;
    if (pk_hide_sidebar()) {
        $out = $hide;
    }
    if (!$echo) {
        return $out;
    }
    echo $out;
}

//侧边栏检测数据
function pk_sidebar_check_has($name)
{
    if (!dynamic_sidebar($name)) {
        dynamic_sidebar('sidebar_not');
    }
}

//获取链接对象，用于书籍推荐及其他页面使用
function pk_get_wp_links($link_cats = '', $auto_all = false)
{
    global $wpdb;
    if (empty($link_cats)) {
        if (!$auto_all) {
            return null;
        }
        $all_cats = get_all_category_id_row('link_category');
        if (empty($all_cats)) {
            return null;
        }
        $link_cats = array_map(function ($item) {
            return $item['value'];
        }, $all_cats);
    }
    if (is_array($link_cats)) {
        $link_cats = implode(',', $link_cats);
    }
    $sql = "select links.*,terms.term_id,terms.name  from {$wpdb->links} as links
            LEFT JOIN (select * from {$wpdb->term_relationships} where term_taxonomy_id in ({$link_cats})) as relat on links.link_id = relat.object_id
            LEFT JOIN (selecT * from {$wpdb->terms} where term_id in ({$link_cats})) as terms on terms.term_id = relat.term_taxonomy_id
             where links.link_id in (relat.object_id) and links.link_visible='Y'";
    $orderby_map = [
        'link_id' => 'links.link_id',
        'url' => 'links.link_url',
        'name' => 'links.link_name',
        'rating' => 'links.link_rating',
        'length' => 'LENGTH(links.link_name)',
        'rand' => 'RAND()',
    ];
    $orderby_key = pk_get_option('index_link_order_by', 'link_id');
    $orderby_col = $orderby_map[$orderby_key] ?? 'links.link_id';
    $order = strtoupper(pk_get_option('index_link_order', 'ASC')) === 'DESC' ? 'DESC' : 'ASC';
    $sql .= " ORDER BY terms.term_id ASC, {$orderby_col} {$order}";
    return $wpdb->get_results($sql);
}

//获取懒加载图片信息
function pk_get_lazy_pl_img()
{
    $custom = trim(pk_get_option('lazy_placeholder_url', ''));
    if (!empty($custom)) {
        if (strpos($custom, 'http://') === 0 || strpos($custom, 'https://') === 0) {
            return $custom;
        }
        if (substr($custom, 0, 1) !== '/') {
            $custom = '/' . $custom;
        }
        return home_url() . $custom;
    }
    return pk_get_static_url() . "/assets/img/z/load.svg";
}

function pk_get_lazy_img_info($origin, $class = '', $width = null, $height = null, $thumbnail = true)
{
    if (!pk_is_checked('basic_img_lazy_s')) {
        if ($thumbnail) {
            $out = "src='" . pk_get_img_thumbnail_src($origin, $width, $height) . "' ";
            $out .= "class='" . $class . "' ";
        } else {
            $out = "src='{$origin}' ";
            $out .= "class='{$class}' ";
        }
    } else {
        $out = "src='" . pk_get_lazy_pl_img() . "' ";
        $out .= "class='lazy " . $class . "' ";
        if ($thumbnail) {
            $out .= "data-src='" . pk_get_img_thumbnail_src($origin, $width, $height) . "'";
        } else {
            $out .= "data-src='" . $origin . "'";
        }
    }
    return $out;
}

function pk_content_img_lazy($content)
{
    return preg_replace('/<img(.+)src=[\'"]([^\'"]+)[\'"](.*)>/i', "<img\$1data-src=\"\$2\" data-lazy=\"true\" src=\"" . pk_get_lazy_pl_img() . "\"\$3/>", $content);
}

if (pk_is_checked('basic_img_lazy_z')) {
    add_filter('the_content', 'pk_content_img_lazy');
}
//获取图片缩略图链接
function pk_get_img_thumbnail_src($src, $width, $height, $args = array())
{
    if ($width == null || $height == null) {
        return $src;
    }
    if (pk_is_checked('disable_timthumb')) {
        return $src;
    }
    $src_path = wp_parse_url($src, PHP_URL_PATH);
    $src_ext = strtolower(pathinfo($src_path ?: $src, PATHINFO_EXTENSION));
    if ($src_ext === 'webp' && (!function_exists('imagecreatefromwebp') || !function_exists('imagewebp'))) {
        return $src;
    }
    if ($src_ext === 'avif' && (!function_exists('imagecreatefromavif') || !function_exists('imageavif'))) {
        return $src;
    }
    if (pk_is_checked('thumbnail_rewrite_open')) {
        return home_url() . "/timthumb/w_{$width}/h_{$height}/q_90/zc_1/a_c/" . pk_safe_base64_encode($src) . ".png";
    }
    return PUOCK_ABS_URI . "/timthumb.php?w={$width}&h={$height}&a=c&zc=1&q=90&src=" . $src;
}

//获取文章样式是否是卡片式
function pk_post_style_list()
{
    return pk_get_option('post_style', 'list') == 'list';
}

function pk_cms_card_columns(): int
{
    $cols = (int)pk_get_option('cms_card_columns', 2);
    if ($cols < 2 || $cols > 4) {
        $cols = 2;
    }

    // 仅在「文章列表」且「卡片风格」下生效（首页/分类/标签/作者/搜索/日期归档等）
    if (pk_post_style_list()) {
        return 2;
    }

    $is_post_list = is_home() || is_archive() || is_search();
    if (!$is_post_list) {
        return 2;
    }

    // 有侧边栏时最多 2 列（仅当全局侧边栏未被禁用）
    if (!pk_is_checked('hide_global_sidebar') && $cols > 2) {
        $cols = 2;
    }

    return $cols;
}

//评论添加@功能
if (pk_is_checked('comment_has_at')) {
    add_filter('comment_text', 'pk_comment_add_at', 10, 2);
}
//GrAvatar头像源切换
if (pk_get_option('gravatar_url', 'wp') != 'wp') {
    $type = pk_get_option('gravatar_url', 'wp');
    if ($type == 'cravatar') {
        add_filter('get_avatar', 'cr_avatar');
        add_filter('get_avatar_url', 'cr_avatar');
    } else if ($type == 'loli') {
        add_filter('get_avatar', 'loli_avatar');
        add_filter('get_avatar_url', 'loli_avatar');
    } else if ($type == 'cn') {
        add_filter('get_avatar', 'cn_avatar');
        add_filter('get_avatar_url', 'cn_avatar');
    } else if ($type == 'cn_ssl') {
        add_filter('get_avatar', 'cn_ssl_avatar');
        add_filter('get_avatar_url', 'cn_ssl_avatar');
    } else if ($type == 'loli_ssl') {
        add_filter('get_avatar', 'loli_ssl_avatar');
        add_filter('get_avatar_url', 'loli_ssl_avatar');
    } else if ($type == 'v2ex') {
        add_filter('get_avatar', 'v2ex_ssl_avatar');
        add_filter('get_avatar_url', 'v2ex_ssl_avatar');
    } else if ($type == 'custom') {
        add_filter('get_avatar', 'pk_custom_avatar');
        add_filter('get_avatar_url', 'pk_custom_avatar');
    }
}
//评论者链接
function pk_get_comment_author_url_html($comment_ID = 0)
{
    global $comment;
    $comment_obj = $comment_ID ? get_comment($comment_ID) : $comment;
    $author = esc_html(get_comment_author($comment_ID));

    if (empty($comment_obj)) {
        return $author;
    }

    if (!empty($comment_obj->user_id)) {
        $url = get_author_posts_url((int)$comment_obj->user_id);
        $attr = '';
    } else {
        $url = get_comment_author_url($comment_ID);
        $attr = ' target="_blank" rel="external nofollow"';
        if (!empty($url) && pk_is_checked('comment_author_link_go', true)) {
            $url = pk_go_link($url);
        }
    }

    if (empty($url)) {
        return $author;
    }

    return '<a' . $attr . ' href="' . esc_url($url) . '" class="url">' . $author . '</a>';
}

function pk_comment_author_url($comment_ID = 0)
{
    echo pk_get_comment_author_url_html($comment_ID);
}

//评论回复通知
if (pk_is_checked('comment_mail_notify')) {
    add_action('comment_unapproved_to_approved', 'comment_mail_notify');
    add_action('comment_post', 'comment_mail_notify');
}
//覆盖邮件配置
if (pk_is_checked('smtp_open')) {
    function mail_smtp_set($phpmailer)
    {
        pk_smtp_apply_config($phpmailer, pk_smtp_get_option_config());
    }

    add_action('phpmailer_init', 'mail_smtp_set');
}
//检测是否默认的第三方生成邮箱
function pk_email_change_email($email_change_email, $user = null, $userdata = null)
{
    if (pk_check_email_is_sysgen($email_change_email['to'])) {
        return null;
    }
    return $email_change_email;
}

add_filter('email_change_email', 'pk_email_change_email');
//检测邮箱是否系统生成
function pk_check_email_is_sysgen($email)
{
    return preg_match("/^_p_[\w].+@null.null/", $email);
}

//后台登录保护
function login_protection()
{
    if (!is_user_logged_in()) {
        if ($_GET[pk_get_option('lp_user', 'admin')] != pk_get_option('lp_pass', 'admin')) {
            header("Location: " . home_url());
        }
    }
}

if (pk_is_checked('login_protection')) {
    add_action('login_enqueue_scripts', 'login_protection');
}

//后台注册验证码
function pk_register_captcha_form()
{
    $vd_type = pk_get_option('vd_type', 'img');
    if ($vd_type === 'img') {
        ?>
        <p>
            <label for="register_captcha"><?php _e('验证码', PUOCK) ?></label>
            <span style="display:flex;gap:5px;align-items:center">
                <input type="text" name="register_captcha" id="register_captcha" class="input"
                       maxlength="4" autocomplete="off" style="flex:1" />
                <img style="cursor:pointer;height:30px"
                     src="<?php echo esc_url(pk_captcha_url('register', 100, 30)) ?>"
                     onclick="this.src=this.src.replace(/&t=\d+/,'')+'&t='+Date.now()"
                     alt="<?php esc_attr_e('验证码', PUOCK) ?>" />
            </span>
        </p>
        <?php
    } elseif ($vd_type === 'turnstile') {
        $site_key = pk_get_option('vd_turnstile_site_key', '');
        if ($site_key) {
            ?>
            <div style="margin-bottom:16px;">
                <div class="cf-turnstile" data-sitekey="<?php echo esc_attr($site_key) ?>"></div>
            </div>
            <?php
        }
    }
}

function pk_register_captcha_validate($errors, $sanitized_user_login, $user_email)
{
    $vd_type = pk_get_option('vd_type', 'img');
    if ($vd_type === 'img') {
        $captcha_val = isset($_POST['register_captcha']) ? sanitize_text_field($_POST['register_captcha']) : '';
        if (empty($captcha_val) || !pk_captcha_validate('register', $captcha_val)) {
            $errors->add('captcha_error', __('<strong>错误</strong>：验证码不正确', PUOCK));
        }
    } elseif ($vd_type === 'turnstile') {
        $cf_token = isset($_POST['cf-turnstile-response']) ? sanitize_text_field($_POST['cf-turnstile-response']) : '';
        if (empty($cf_token)) {
            $errors->add('captcha_error', __('<strong>错误</strong>：请完成 Turnstile 验证', PUOCK));
        } else {
            $secret_key = pk_get_option('vd_turnstile_secret_key', '');
            $response = wp_remote_post('https://challenges.cloudflare.com/turnstile/v0/siteverify', [
                'body' => [
                    'secret' => $secret_key,
                    'response' => $cf_token,
                    'remoteip' => isset($_SERVER['REMOTE_ADDR']) ? $_SERVER['REMOTE_ADDR'] : '',
                ],
            ]);
            if (is_wp_error($response)) {
                $errors->add('captcha_error', __('<strong>错误</strong>：验证服务请求失败', PUOCK));
            } else {
                $result = json_decode(wp_remote_retrieve_body($response), true);
                if (empty($result['success'])) {
                    $errors->add('captcha_error', __('<strong>错误</strong>：Turnstile 验证失败，请重试', PUOCK));
                }
            }
        }
    }
    return $errors;
}

function pk_register_captcha_scripts()
{
    $vd_type = pk_get_option('vd_type', 'img');
    if ($vd_type === 'turnstile') {
        wp_enqueue_script('cf-turnstile', 'https://challenges.cloudflare.com/turnstile/v0/api.js', [], null, true);
    }
}

if (pk_is_checked('vd_register')) {
    add_action('register_form', 'pk_register_captcha_form');
    add_filter('registration_errors', 'pk_register_captcha_validate', 10, 3);
    add_action('login_enqueue_scripts', 'pk_register_captcha_scripts');
}

if (pk_is_checked('compress_html')) {
    add_action('get_header', 'wp_compress_html');
}
//百度主动推送
function pk_baidu_submit($post_ID)
{
    if (get_post_meta($post_ID, 'baidu_submit_url_status', true) == 1) return;
    $post_url = get_permalink($post_ID);
    $api_url = pk_get_option('baidu_submit_url');
    $resp = wp_remote_post($api_url, array('body' => $post_url, 'headers' => 'Content-Type: text/plain'));
    $res = @json_decode($resp['body'], true);
    if (isset($res['success'])) {
        add_post_meta($post_ID, 'baidu_submit_url_status', 1, true);
    }
}

if (pk_is_checked('open_baidu_submit')) {
    add_action('publish_post', 'pk_baidu_submit', 0);
}
//对pre里面的内容进行转义
function pk_tag_pre_encode($content)
{
    preg_match_all("/<pre.*?>(.+?)<\/pre>/is", $content, $matches);
    if (isset($matches[1])) {
        foreach ($matches[1] as $match) {
            $m = trim($match);
            if (!(substr($m, 0, strlen("<code")) === "<code")) {
                $m = "<code class='language-default'>$m</code>";
            }
            if (substr($m, 0, strlen("<code>")) === "<code>") {
                $m = "<code class='language-default'>" . substr($m, strlen("<code>"));
            }
            $content = str_replace($match, $m, $content);
        }
    }
    return $content;
}

add_filter('the_content', 'pk_tag_pre_encode');
add_filter('comment_text', 'pk_tag_pre_encode');
function pk_htmlspecialchars($content)
{
    $content = str_replace("<", "&lt;", $content);
    $content = str_replace(">", "&gt;", $content);
    return $content;
}


function create_taxs($tax_slug, $hook_type, $tax_name)
{
    //自定义分类法标签
    $labels_tax = array(
        'name' => $tax_name,
        'singular_name' => $tax_name,
        /* translators: %s: taxonomy name */
        'search_items' => sprintf(__('搜索%s', PUOCK), $tax_name),
        /* translators: %s: taxonomy name */
        'all_items' => sprintf(__('所有%s', PUOCK), $tax_name),
        /* translators: %s: taxonomy name */
        'parent_item' => sprintf(__('父级%s', PUOCK), $tax_name),
        /* translators: %s: taxonomy name */
        'parent_item_colon' => sprintf(__('父级%s', PUOCK), $tax_name),
        /* translators: %s: taxonomy name */
        'edit_item' => sprintf(__('编辑%s', PUOCK), $tax_name),
        /* translators: %s: taxonomy name */
        'update_item' => sprintf(__('更新%s', PUOCK), $tax_name),
        /* translators: %s: taxonomy name */
        'add_new_item' => sprintf(__('添加新%s', PUOCK), $tax_name),
        /* translators: %s: taxonomy name */
        'new_item_name' => sprintf(__('新%s名称', PUOCK), $tax_name),
        'menu_name' => $tax_name,
    );

    //自定义分类法参数
    $args_tax = array(
        'hierarchical' => true,
        'labels' => $labels_tax,
        'show_ui' => true,
        'show_admin_column' => true,
        'query_var' => true,
        'rewrite' => array('slug' => $tax_slug),
    );

    register_taxonomy($tax_slug, array($hook_type), $args_tax);
}


//注册菜单
register_nav_menus(array(
    'primary' => __('主要菜单', PUOCK),
));

//获取主题配置
function pk_get_option($name, $default = null)
{
    $config = get_option(PUOCK_OPT);
    if ($config && isset($config[$name])) {
        return $config[$name];
    }
    return $default;
}

//配置是否选择
function pk_is_checked($name, $default = false)
{
    $val = pk_get_option($name);
    if ($val === true || $val === 'true' || $val === 1 || $val === '1') {
        return true;
    }
    if ($val === null) {
        return $default;
    }
    return false;
}

//配置选择输出
function pk_checked_out($name, $out = '', $default = 0)
{
    if (pk_is_checked($name, $default)) {
        echo $out;
    }
}

//主题模式
function pk_theme_light()
{
    if (isset($_COOKIE['mode'])) {
        return ($_COOKIE['mode'] == 'light') || $_COOKIE['mode'] == 'auto';
    }
    return pk_get_option('theme_mode', 'light') == 'light';
}

function pk_theme_mode()
{
    if (isset($_COOKIE['mode'])) {
        return $_COOKIE['mode'];
    }
    return pk_get_option('theme_mode', 'light');
}

function pk_theme_dark_palette()
{
    $palette = pk_get_option('theme_dark_palette', 'bootstrap');
    return in_array($palette, ['bootstrap', 'blueblack'], true) ? $palette : 'bootstrap';
}

//动画载入
function pk_open_box_animated($class, $echo = true)
{
    $open = pk_is_checked("page_animate") == 1;
    if ($open) {
        if (!$echo) {
            return $class;
        }
        echo $class;
    }
}


//获取所有站点分类id
function get_all_category_id($type = null)
{
    global $wpdb;
    $sql = "SELECT term_id, name FROM $wpdb->terms";
    if ($type != null) {
        $sql .= " where term_id in (select term_id from $wpdb->term_taxonomy where taxonomy='{$type}')";
    }
    $cats = $wpdb->get_results($sql);
    $out = '<span style="line-height: 1.5rem">';
    foreach ($cats as $cat) {
        $out .= '<span>[' . $cat->name . "：<code>" . $cat->term_id . '</code></span>]、';
    }
    if (count($cats) > 0) {
        $out = mb_substr($out, 0, mb_strlen($out) - 1);
    }
    $out .= '</span>';
    return $out;
}

//获取所有站点分类id
function get_all_category_id_row($type = null)
{
    global $wpdb;
    $sql = "SELECT term_id, name FROM $wpdb->terms";
    if ($type != null) {
        $sql .= " where term_id in (select term_id from $wpdb->term_taxonomy where taxonomy='{$type}')";
    }
    $cats = $wpdb->get_results($sql);
    $result = [];
    foreach ($cats as $cat) {
        $result[] = ['label' => $cat->name, 'value' => $cat->term_id];
    }
    return $result;
}


//获取菜单数据
function pk_get_main_menu($mobile = false)
{
    global $wp;
    $menus = get_nav_menu_object('primary');
    $out = $mobile ? "<ul class='puock-links t-md'>" : "<ul>";
    if ($menus && count($menus) > 0) {
        pk_get_menu_obj_to_html($menus, $out, $mobile);
    }
    if (is_user_logged_in()) {
        $user = wp_get_current_user();
        $avatar = get_avatar_url($user->user_email);
        $out .= '<li><a ' . (pk_is_user_center_theme() ? '' : 'data-no-instant') . ' data-bs-toggle="tooltip" title="' . esc_attr__('用户中心', PUOCK) . '" href="' . pk_user_center_url() . '"><img alt="' . esc_attr__('用户中心', PUOCK) . '" src="' . $avatar . '" class="min-avatar"></a></li>';
    } else {
        if (pk_is_checked('open_quick_login')) {
            $url = pk_ajax_url('pk_font_login_page', ['redirect' => home_url($wp->request)]);
            $out .= '<li><a data-no-instant data-bs-toggle="tooltip" title="' . esc_attr__('登入', PUOCK) . '" data-title="' . esc_attr__('登入', PUOCK) . '" href="javascript:void(0)" class="pk-modal-toggle" data-once-load="true" data-url="' . $url . '"><i class="fa fa-right-to-bracket"></i></a></li>';
        }
    }
    if (!$mobile) {
        if (pk_is_checked('theme_mode_s')) {
            $out .= '<li><a class="colorMode" data-bs-toggle="tooltip" title="' . esc_attr__('模式切换', PUOCK) . '" href="javascript:void(0)"><i class="fa fa-' . ((pk_theme_mode() === 'auto' ? 'circle-half-stroke' : (pk_theme_light() ? 'sun' : 'moon'))) . '"></i></a></li>';
        }
        if (pk_is_checked('cn_sc_tc_toggle')) {
            $out .= '<li><a class="sc-tc-toggle" data-bs-toggle="tooltip" title="' . esc_attr__('简繁转换', PUOCK) . '" href="javascript:void(0)"><i class="fa fa-language"></i></a></li>';
        }
        $out .= '<li><a class="search-modal-btn" data-bs-toggle="tooltip" title="' . esc_attr__('搜索', PUOCK) . '" href="javascript:void(0)"><i class="fa fa-search"></i></a></li>';
    }
    $out .= '</ul>';
    return $out;
}

//获取菜单对象数据
function get_nav_menu_object($location)
{
    $locations = get_nav_menu_locations();
    if (!$locations) {
        return null;
    }
    $menu_id = $locations[$location];
    $menu_object = wp_get_nav_menu_object($menu_id);
    if (!$menu_object) {
        return array();
    }
    $menu_items = wp_get_nav_menu_items($menu_object->term_id);
    $menus = array();
    if ($menu_items == null || count($menu_items) == 0) {
        return $menus;
    } else {
        foreach ($menu_items as &$menu_item) {
            if (!isset($menu_item->classes) || $menu_item->classes == null) {
                $menu_item->classes = array();
            }
        }
    }
    _wp_menu_item_classes_by_context($menu_items);
    $submenus = array();
    foreach ($menu_items as $m) {
        $m->children = array();
        if (!$m->menu_item_parent) {
            $menus[$m->ID] = $m;
        } else {
            $submenus[$m->ID] = $m;
            if (isset($menus[$m->menu_item_parent])) {
                $menus[$m->menu_item_parent]->children[$m->ID] = &$submenus[$m->ID];
            } else {
                $submenus[$m->menu_item_parent]->children[$m->ID] = $submenus[$m->ID];
            }
        }
    }
    return $menus;
}

//将匹配的到的菜单数据转换为html
function pk_get_menu_obj_to_html($menus, &$out, $mobile = false, $dpath_cur = 1, $max_dpath = 2)
{
    $child_class = $dpath_cur != 1 ? 'menu-item-child' : '';
    // 全局默认打开方式（例如主题设置为新窗口），但优先使用每个菜单项自身的 target 设置
    $default_target = pk_link_target(false);
    foreach ($menus as $menu) {
        $classes = join(" ", $menu->classes);
        $cur = $menu->current ? 'menu-current' : '';
        $out .= "<li id='menu-item-{$menu->ID}' class='menu-item-{$menu->ID} {$classes} {$child_class} {$cur}'>";
        // 计算当前菜单项应使用的 target 属性：当菜单项有设置时优先生效
        $item_target = '';
        if (!empty($menu->target)) {
            $item_target = 'target="' . $menu->target . '"';
        } else {
            $item_target = $default_target;
        }
        if (!$mobile) {
            $out .= "<a class='ww' data-color='auto' {$item_target} href='{$menu->url}'>{$menu->title}";
        } else {
            $out .= '<span><a ' . $item_target . ' href="' . $menu->url . '">' . $menu->title . '</a>';
        }
        if (count($menu->children) > 0) {
            if ($mobile) {
                $out .= '<a href="#menu-sub-' . $menu->ID . '" data-bs-toggle="collapse"><i class="fa fa-chevron-down t-sm ms-1 menu-sub-icon"></i></a>';
            } else {
                $out .= '<i class="fa fa-chevron-down t-sm ms-1 menu-sub-icon"></i>';
            }
        }
        if ($mobile) {
            $out .= '</span>';
        } else {
            $out .= '</a>';
        }
        if (count($menu->children) > 0 && $dpath_cur < $max_dpath) {
            $out .= '<ul ' . ($mobile ? 'id="menu-sub-' . $menu->ID . '"' : '') . ' class="sub-menu ' . ($mobile
                    ? 'collapse' : '') . '">';
            pk_get_menu_obj_to_html($menu->children, $out, $mobile, $dpath_cur + 1, $max_dpath);
            $out .= '</ul>';
        }
    }
}

//获取分类的子集菜单
function get_category_child($parentId)
{
    $child = get_categories("parent={$parentId}&hide_empty=0");
    $list = array();
    foreach ($child as $child_item) {
        $list[] = array(
            'url' => get_category_link($child_item),
            'item' => $child_item
        );
    }
    return $list;
}

//主查询设置
function pk_pre_post_set($query)
{
    if ($query->is_home() && $query->is_main_query()) {
        if (pk_get_option('index_mode', '') == 'cms') {
            $sort = pk_get_option('cms_new_sort', 'published');
            $query->set('posts_per_page', pk_get_option('cms_show_new_num', 6));
            if ($sort == 'published') {
                $query->set('orderby', 'date');
                $query->set('order', 'DESC');
            } elseif ($sort == 'updated') {
                $query->set('orderby', 'modified');
                $query->set('order', 'DESC');
            }
            $exclude_cats = pk_get_option('cms_new_exclude_cats');
            if (!empty($exclude_cats)) {
                $cat_ids = is_array($exclude_cats) ? array_map('intval', $exclude_cats) : [intval($exclude_cats)];
                $query->set('category__not_in', $cat_ids);
            }
        }
    }
    if ($query->is_category() && $query->is_main_query()) {
        $cat_per_page = intval(pk_get_option('category_posts_per_page', 0));
        if ($cat_per_page > 0) {
            $query->set('posts_per_page', $cat_per_page);
        }
    }
}

add_action('pre_get_posts', 'pk_pre_post_set');

// 分类页置顶文章优先显示
function pk_category_sticky_orderby($clauses, $query)
{
    global $wpdb;
    if ($query->is_category() && $query->is_main_query()) {
        $clauses['join'] .= " LEFT JOIN {$wpdb->postmeta} AS pk_sticky ON ({$wpdb->posts}.ID = pk_sticky.post_id AND pk_sticky.meta_key = 'sticky_in_category' AND pk_sticky.meta_value = 'true')";
        $clauses['orderby'] = "CASE WHEN pk_sticky.meta_value = 'true' THEN 0 ELSE 1 END ASC, " . $clauses['orderby'];
    }
    return $clauses;
}

add_filter('posts_clauses', 'pk_category_sticky_orderby', 10, 2);

//静态资源加载源的链接
function pk_get_static_url()
{
    $type = pk_get_option('static_load_origin', 'self');
    switch ($type) {
        case "jsdelivr":
            $url_pre = "https://cdn.jsdelivr.net/gh/Licoy/wordpress-theme-puock@v" . PUOCK_CUR_VER_STR;
            break;
        case "jsdelivr-fastly":
            $url_pre = "https://fastly.jsdelivr.net/gh/Licoy/wordpress-theme-puock@v" . PUOCK_CUR_VER_STR;
            break;
        case "jsdelivr-testingcf":
            $url_pre = "https://testingcf.jsdelivr.net/gh/Licoy/wordpress-theme-puock@v" . PUOCK_CUR_VER_STR;
            break;
        case "jsdelivr-gcore":
            $url_pre = "https://gcore.jsdelivr.net/gh/Licoy/wordpress-theme-puock@v" . PUOCK_CUR_VER_STR;
            break;
        case 'custom':
            $url_pre = pk_get_option('custom_static_load_origin', '');
            break;
        default:
            $url_pre = PUOCK_ABS_URI;
    }
    return $url_pre;
}

//是否打开讨论-显示头像
function pk_open_show_comment_avatar()
{
    return get_option('show_avatars') == "1";
}

//关闭区块小工具
function pk_off_widgets_block()
{
    add_filter('gutenberg_use_widgets_block_editor', '__return_false');
    add_filter('use_widgets_block_editor', '__return_false');
}

//获取中文格式化的实例
function pk_chinese_format($content)
{
    include_once dirname(__FILE__) . '/../lib/ChineseTypesetting.php';
    $typesetting = new ChineseTypesetting();
    $content = $typesetting->insertSpace($content);
    $content = $typesetting->removeSpace($content);
    return $typesetting->full2Half($content);
}

if (pk_is_checked('chinese_format')) {
    add_filter('the_content', 'pk_chinese_format', 199);
}

//获取缩略图的白名单
function pk_get_thumbnail_allow_sites()
{
    $sites = [];
    $thumbnail_allows = trim(pk_get_option("thumbnail_allows", ''));
    if (!empty($thumbnail_allows)) {
        foreach (explode("\n", $thumbnail_allows) as $site) {
            $site = trim($site);
            if (!empty($site)) {
                $sites[] = $site;
            }
        }
    }
    return $sites;
}

//生成缩略图白名单文件名称
function pk_get_thumbnail_allow_sites_filepath()
{
    return PUOCK_ABS_DIR . '/.tas.php';
}

//生成缩略图白名单文件
function pk_generate_thumbnail_allow_sites_file()
{
    $sites = pk_get_thumbnail_allow_sites();
    $template = "<?php \$ALLOWED_SITES = [\n";
    if (count($sites) > 0) {
        foreach ($sites as $site) {
            $template .= "\t\"$site\",\n";
        }
    }
    $template .= "];";
    return file_put_contents(pk_get_thumbnail_allow_sites_filepath(), $template);
}

add_action('pk_option_updated', 'pk_generate_thumbnail_allow_sites_file', 10, 0);

// get request model data
function pk_get_req_data(array $model)
{
    $data = [];
    foreach ($model as $key => $item) {
        $val = trim($_REQUEST[$key] ?? '');
        if (empty($val)) {
            if ($item['required']) {
                return ($item['name'] ?? $key) . __('不能为空', PUOCK);
            }
            if (isset($item['default'])) {
                $data[$key] = $item['default'];
            }
            if ($item['empty'] ?? false) {
                $data[$key] = '';
            }
        } else {
            if ($item['remove_html'] ?? false) {
                $val = esc_html($val);
            }
            $data[$key] = $val;
        }
    }
    return $data;
}

function pk_get_ip_region_str($ip)
{
    $ip2_instance = $GLOBALS['ip2_region'] ?? false;
    if (!$ip2_instance) {
        $ip2_instance = new \Ip2Region();
        $GLOBALS['ip2_region'] = $ip2_instance;
    }
    try {
        $s = $ip2_instance->memorySearch($ip);
    } catch (Exception $e) {
        return __('未知', PUOCK);
    }
    if (strpos($s['region'], '内网IP') !== false) {
        return __('内网IP', PUOCK);
    }
    $region = explode('|', $s['region']);
    $res = '';
    foreach ($region as $item) {
        if (strpos($item, '0') === 0) {
            continue;
        }
        $res .= $item;
    }
    return $res;
}

/**
 * 极验验证码校验
 * @throws Exception
 */
function pk_vd_gt_validate(array $args = null)
{
    if ($args == null) {
        $args = [
            'lot_number' => $_REQUEST['lot_number'] ?? '',
            'captcha_output' => $_REQUEST['captcha_output'] ?? '',
            'captcha_id' => $_REQUEST['captcha_id'] ?? '',
            'pass_token' => $_REQUEST['pass_token'] ?? '',
            'gen_time' => $_REQUEST['gen_time'] ?? '',
        ];
    }
    $key = pk_get_option('vd_gt_key');
    $args['sign_token'] = hash_hmac('sha256', $args['lot_number'], $key);
    $result = wp_remote_request('https://gcaptcha4.geetest.com/validate?captcha_id=' . $args['captcha_id'], [
        'method' => 'POST',
        'body' => $args,
        'timeout' => 5
    ]);
    if (is_wp_error($result)) {
        throw new Exception(__('验证行为失败', PUOCK));
    }
    $result = json_decode($result['body'], true);
    if ($result['status'] != 'success' || $result['result'] != 'success') {
        throw new Exception(__('验证行为失败', PUOCK) . ': ' . $result['msg'] ?? $result['reason']);
    }
    return true;
}

function pk_get_user_center_entry(): string
{
    $entry = pk_get_option('user_center_entry', '');
    if (!empty($entry)) {
        return $entry;
    }
    return pk_is_checked('user_center') ? 'theme' : 'wp';
}

function pk_is_user_center_theme(): bool
{
    return pk_get_user_center_entry() === 'theme';
}

function pk_user_center_url(): string
{
    $entry = pk_get_user_center_entry();
    if ($entry === 'theme') {
        return home_url() . '/uc';
    }
    if ($entry === 'erphp') {
        $url = trim(pk_get_option('erphpdown_user_center_url', '/erphpdown/user'));
        if (empty($url)) {
            return get_edit_profile_url();
        }
        if (strpos($url, 'http://') === 0 || strpos($url, 'https://') === 0) {
            return $url;
        }
        if (substr($url, 0, 1) !== '/') {
            $url = '/' . $url;
        }
        return home_url() . $url;
    }
    return get_edit_profile_url();
}

function pk_rewrite_rule()
{
    if (pk_is_user_center_theme()) {
        add_rewrite_rule('^uc/?([0-9A-Za-z_\-]+)?$', 'index.php?pagename=user-center&id=$matches[1]', "top");
    }
}

add_action('init', 'pk_rewrite_rule');

function pk_template_redirect()
{
    if (!pk_is_user_center_theme()) {
        return;
    }
    global $wp_query;
    $page_name = $wp_query->get('pagename');
    if (!empty($page_name)) {
        $template = '';
        switch ($page_name) {
            case 'user-center':
                $template = PUOCK_ABS_DIR . '/inc/page/user-center.php';
                break;
            default:
                break;
        }
        if (!empty($template)) {
            pk_load_template($template);
            exit;
        }
    }
}

add_action('template_redirect', 'pk_template_redirect');

function pk_query_vars($vars)
{
    $vars[] = 'id';
    return $vars;
}

add_filter('query_vars', 'pk_query_vars');

function pk_load_template($_template_file, $require_once = true, $args = array())
{
    status_header(200);
    load_template($_template_file, $require_once, $args);
}
