export.php

<?php

use Doctrine\DBAL;

require 'vendor/autoload.php';
include './common.inc.php';

$connectionParams = [
    'url' => $dbDSN,
];
$conn             = DBAL\DriverManager::getConnection($connectionParams);
$args             = [];
parse_str($argv[1] ?? '', $args);
$udpxy        = $args['udpxy'] ?? null;
$epgKey       = $args['epg'] ?? null;
$epgKey       = ($epgKey ?: 'rytec') . '_epg_id';
$queryBuilder = $conn->createQueryBuilder();
$lists        = $queryBuilder->select('*')->from('iptv')->orderBy('sort', 'ASC')->fetchAllAssociative();
$data         = new \M3uParser\M3uData();
$data->setAttribute('name', '上海电信IPTV组播节目单');
foreach ($lists as $item) {
    $uri     = null;
    $quality = null;
    foreach (QUALITY_NAME_MAP as $Q => $q) {
        $key = 'uri_' . $q;
        if (!empty($item[$key])) {
            $uri     = $item[$key];
            $quality = '「' . $Q . '」';
            break;
        }
    }
    if (empty($uri) && !empty($item['uri'])) {
        $uri = $item['uri'];
    }
    if (empty($uri)) {
        continue;
    }
    if (!empty($udpxy) && stripos($uri, 'rtp://') === 0) {
        $uri = rtrim($udpxy, ' \t\n\r') . '/' . str_replace('://', '/', $uri);
    }
    $title = $item['display_title'] ?? $item['name'];
    if ($quality) {
        $title .= $quality;
    }
    $entry = new M3uParser\M3uEntry();
    $entry->setPath($uri);
    $extInf     = (new M3uParser\Tag\ExtInf())
        ->setDuration(-1)
        ->setTitle($title);
    $attributes = [];
    if (!empty($item[$epgKey])) {
        $attributes['tvg-id'] = $item[$epgKey];
    }
    $attributes = array_replace($attributes, [
        'tvg-name'    => $item['name'],
        'group-title' => $item['group'],
    ]);
    $extInf->setAttributes($attributes);
    $entry->addExtTag($extInf);
    $data->append($entry);
}

if (!SERVER_MODE) {
    echo $data;
} else {
    return $data;
}