index.php

<?php
/**
 * @file index.php
 * @brief 主页处理脚本
 * 
 * 该脚本处理来自客户端的请求，根据查询参数获取指定日期和频道的节目信息，
 * 并从 MySQL 数据库中提取或返回默认数据。
 * 此处经过测试，反而不加Redis效果好于加入Redis，就让他这样吧。
 * 
 * 作者: Tak
 * GitHub: https://github.com/TakcC/PHP-EPG-Docker-Server
 * 修改: mxdabc
 * Github: https://github.com/mxdabc/epgphp
 */

// 引入公共脚本
require_once 'public.php';

// 禁止输出错误提示
error_reporting(0);

// 初始化响应头信息
$init = [
    'status' => 200,
    'headers' => [
        'content-type' => 'application/json'
    ]
];

// 生成响应
function makeRes($body, $status = 200, $headers = []) {
    $headers['Access-Control-Allow-Origin'] = '*';
    http_response_code($status);
    foreach ($headers as $key => $value) {
        header("$key: $value");
    }
    echo $body;
}

// 获取当前日期
function getNowDate() {
    return date('Y-m-d');
}

// 格式化时间
function getFormatTime($time) {
    if (strlen($time) < 8) {
        return ['date' => getNowDate(), 'time' => ''];
    }

    $date = substr($time, 0, 4) . '-' . substr($time, 4, 2) . '-' . substr($time, 6, 2);
    $time = strlen($time) >= 12 ? substr($time, 8, 2) . ':' . substr($time, 10, 2) : '';

    return ['date' => $date, 'time' => $time];
}

// 从数据库读取 diyp、lovetv 数据
function readEPGData($date, $channel, $db, $type) {
    $stmt = $db->prepare("SELECT epg_diyp FROM epg_data WHERE date = :date AND channel LIKE :channel");
    $stmt->bindParam(':date', $date);
    $channelPattern = '%' . $channel . '%';
    $stmt->bindParam(':channel', $channelPattern);
    $stmt->execute();
    $row = $stmt->fetch(PDO::FETCH_ASSOC);
    
    if (!$row) {
        return false;
    }

    if ($type === 'diyp') {
        return $row['epg_diyp'];
    }

    if ($type === 'lovetv') {
        $diyp_data = json_decode($row['epg_diyp'], true);
        $program = array_map(function($epg) {
            $start_time = strtotime($epg['start']);
            $end_time = strtotime($epg['end']);
            return [
                'st' => $start_time,
                'et' => $end_time,
                'eventType' => '',
                'eventId' => '',
                't' => $epg['title'],
                'showTime' => date('H:i', $start_time),
                'duration' => $end_time - $start_time
            ];
        }, $diyp_data['epg_data']);

        // 查找当前节目
        $current_programme = findCurrentProgramme($program);

        // 生成 lovetv 数据
        $lovetv_data = [
            $channel => [
                'isLive' => $current_programme ? $current_programme['t'] : '',
                'liveSt' => $current_programme ? $current_programme['st'] : 0,
                'channelName' => $diyp_data['channel_name'],
                'lvUrl' => $diyp_data['url'],
                'program' => array_map(function($epg) {
                    $start_time = strtotime($epg['start']);
                    $end_time = strtotime($epg['end']);
                    return [
                        'st' => $start_time,
                        'et' => $end_time,
                        'eventType' => '',
                        'eventId' => '',
                        't' => $epg['title'],
                        'showTime' => date('H:i', $start_time),
                        'duration' => $end_time - $start_time
                    ];
                }, $diyp_data['epg_data'])
            ]
        ];

        return json_encode($lovetv_data, JSON_UNESCAPED_UNICODE);
    }

    return false;
}

// 获取当前时间戳
function getNowTimestamp() {
    return time();
}

// 查找当前节目
function findCurrentProgramme($programmes) {
    $now = getNowTimestamp();
    foreach ($programmes as $programme) {
        if ($programme['st'] <= $now && $programme['et'] >= $now) {
            return $programme;
        }
    }
    return null;
}

// 处理请求
function fetchHandler() {
    global $init, $db, $Config;

    $uri = parse_url($_SERVER['REQUEST_URI']);
    $query_params = [];
    if (isset($uri['query'])) {
        parse_str($uri['query'], $query_params);
    }

    // 获取并清理频道名称，繁体转换成简体
    $channel = cleanChannelName($query_params['ch'] ?? $query_params['channel'] ?? '', $t2s = true);

    $date = isset($query_params['date']) ? getFormatTime(preg_replace('/\D+/', '', $query_params['date']))['date'] : getNowDate();

    // 频道参数为空时，直接重定向到 t.xml 或 t.xml.gz 文件
    if (empty($channel)) {
        if ($Config['gen_xml'] == 1 || $Config['gen_xml'] == 3) {
            header('Location: ./t.xml.gz');
        } else if ($Config['gen_xml'] == 2) {
            header('Location: ./t.xml');
        } else {
            // 输出消息并设置404状态码
            echo "404 Not Found. <br>未生成 xmltv 文件<br>请检查你的配置是否有误";
            http_response_code(404);
        }
        exit;
    }

    // 返回 diyp、lovetv 数据
    if (isset($query_params['ch']) || isset($query_params['channel'])) {
        $type = isset($query_params['ch']) ? 'diyp' : 'lovetv';
        $response = readEPGData($date, $channel, $db, $type);
    
        if ($response) {
            makeRes($response, $init['status'], $init['headers']);
        } else if(!isset($Config['ret_default']) || $Config['ret_default']) {
            if ($type === 'diyp') {
                // 无法获取到数据时返回默认 diyp 数据
                $default_diyp_program_info = [
                    'date' => $date,
                    'channel_name' => $channel,
                    'url' => "https://github.com/mxdabc/epgphp",
                    'epg_data' => array_map(function($hour) {
                        return [
                            'start' => sprintf('%02d:00', $hour),
                            'end' => sprintf('%02d:00', ($hour + 1) % 24),
                            'title' => '暂无数据或配置错误',
                            'desc' => ''
                        ];
                    }, range(0, 23, 1))
                ];
                $response = json_encode($default_diyp_program_info, JSON_UNESCAPED_UNICODE);
            } else {
                // 无法获取到数据时返回默认 lovetv 数据
                $default_lovetv_program_info = [
                    $channel => [
                        'isLive' => '',
                        'liveSt' => 0,
                        'channelName' => $channel,
                        'lvUrl' => 'https://github.com/mxdabc/epgphp',
                        'program' => array_map(function($hour) {
                            return [
                                'st' => strtotime(sprintf('%02d:00', $hour)),
                                'et' => strtotime(sprintf('%02d:00', ($hour + 1) % 24)),
                                't' => '暂无数据或配置错误',
                                'd' => ''
                            ];
                        }, range(0, 23, 1))
                    ]
                ];
                $response = json_encode($default_lovetv_program_info, JSON_UNESCAPED_UNICODE);
            }
            makeRes($response, $init['status'], $init['headers']);
        }
    }

    // 默认响应
    makeRes('', $init['status'], $init['headers']);
}

// 执行请求处理
fetchHandler();

?>