<?php
include_once $_SERVER['DOCUMENT_ROOT'] . '/config.php';

$db = new PDO(DB_URL, DB_USER, DB_PASS);

$room = $_GET['r'] ?: null;
$user = $_GET['u'] ?: null;
$date = $_GET['d'] ?: null;
$message_query = $_GET['m'] ?: null;

$limit = min(abs(intval($_GET['l'] ?? '500')), 500);
$page = abs(intval($_GET['p'] ?? '1') - 1);
$offset = $limit * $page;

if (isset($user)) {
    $stmt = $db->prepare('SELECT id, `nick`, joined_at, departed_at FROM users WHERE `nick` = ?');
    $stmt->execute([$user]);

    $user = $stmt->fetch(PDO::FETCH_ASSOC) ?: null;
    if (!$user) {
        http_response_code(404);
        exit("User not found.");
    }
}

if (isset($room)) {
    $room = urldecode($room);
    $stmt = $db->prepare('SELECT id, `name`, joined_at, departed_at FROM rooms WHERE `name` = ?');
    $stmt->execute([$room]);

    $room = $stmt->fetch(PDO::FETCH_ASSOC) ?: null;
    if (!$room) {
        http_response_code(404);
        exit("Room not found.");
    }
    $room['encoded'] = urlencode($room['name']);
}

if (isset($message_query)) {
    $user_id = $user ? $user['id'] : "m.user_id";
    $room_id = $room ? $room['id'] : "m.room_id";
    $room_column = !$room ? ", r.name AS room" : "";
    $command = "PRIVMSG";

    $stmt = $db->prepare("SELECT m.id, u.nick, m.command, m.params, m.tags, m.sent_at $room_column
            FROM messages m
            JOIN rooms r ON r.id = m.room_id
            JOIN users u ON u.id = m.user_id
            WHERE m.room_id = $room_id AND m.user_id = $user_id
                AND m.params LIKE ? AND m.command = ?
            ORDER BY sent_at DESC
            LIMIT $limit OFFSET $offset
        ");
    $stmt->execute([$message_query, $command]);
    $messages = $stmt->fetchAll(PDO::FETCH_ASSOC);
} else if (isset($date)) {
    $user_id = $user ? $user['id'] : "m.user_id";
    $room_id = $room ? $room['id'] : "m.room_id";
    $room_column = !$room ? ", r.name AS room" : "";

    $stmt = $db->prepare("SELECT m.id, u.nick, m.command, m.params, m.tags, m.sent_at $room_column
            FROM messages m
            JOIN rooms r ON r.id = m.room_id
            JOIN users u ON u.id = m.user_id
            WHERE m.room_id = $room_id AND m.user_id = $user_id
                AND m.sent_at BETWEEN ? AND DATE_ADD(?, INTERVAL 1 DAY)
            ORDER BY sent_at DESC
            LIMIT $limit OFFSET $offset
        ");
    $stmt->execute([$date, $date]);
    $messages = $stmt->fetchAll(PDO::FETCH_ASSOC);
} else if ($room || $user) {
    $room_id = isset($room) ? $room['id'] : 'room_id';
    $user_id = isset($user) ? $user['id'] : 'user_id';
    $stmt = $db->query("SELECT YEAR(sent_at) AS msg_year,
                MONTH(sent_at) AS msg_month,
                DAY(sent_at) AS msg_day,
                COUNT(*) AS msg_count
            FROM messages
            WHERE room_id = $room_id AND user_id = $user_id
            GROUP BY YEAR(sent_at), MONTH(sent_at), DAY(sent_at)
            ORDER BY YEAR(sent_at) DESC, MONTH(sent_at) DESC, DAY(sent_at) DESC
        ");
    $stmt->execute();
    $raw_dates = $stmt->fetchAll(PDO::FETCH_ASSOC);
    $dates = [];

    foreach ($raw_dates as $rm) {
        $y = $rm['msg_year'];
        $m = $rm['msg_month'];
        $d = $rm['msg_day'];
        $c = $rm['msg_count'];
        if (!array_key_exists($y, $dates)) {
            $dates[$y] = [];
        }
        if (!array_key_exists($m, $dates[$y])) {
            $dates[$y][$m] = [];
        }
        $dates[$y][$m][$d] = $c;
    }
}

$page_title = match (true) {
    isset($user, $room, $message_query) => "Searching for '$message_query' by {$user['nick']} in {$room['name']}",
    isset($user, $message_query) => "Searching for '$message_query' by {$user['nick']}",
    isset($room, $message_query) => "Searching for '$message_query' in {$room['name']}",
    isset($message_query) => "Searching for '$message_query'",
    isset($user, $room) => "{$user['nick']}'s logs in {$room['name']}",
    isset($room) => "Logs of {$room['name']}",
    isset($user) => "{$user['nick']}'s logs",
    default => "Index"
};

if (isset($date)) {
    $page_title .= ", $date";
}

$link = match (true) {
    isset($user, $room) => "r={$room['encoded']}&u={$user['nick']}",
    isset($user) => "u={$user['nick']}",
    isset($room) => "r={$room['encoded']}",
    default => ""
};

if (!isset($user) && !isset($room)) {
    $stmt = $db->query('SELECT r.name, r.departed_at, 
    (
        SELECT MAX(m.sent_at)
        FROM messages m
        WHERE m.room_id = r.id
    ) AS last_message_at
    FROM rooms r
    ORDER BY r.joined_at DESC
    ');
    $stmt->execute();
    $rooms = $stmt->fetchAll(PDO::FETCH_ASSOC);

    foreach ($rooms as &$r) {
        $t = $r['last_message_at'];
        if (!$t) {
            $r['last_message_at'] = 'no messages yet';
            continue;
        }
        $diff = time() - strtotime($t);
        $t = match (true) {
            $diff < 60 => "$diff seconds ago",
            $diff < 3600 => floor($diff / 60) . ' minutes ago',
            $diff < 86400 => floor($diff / 3600) . ' hours ago',
            $diff < 2592000 => floor($diff / 86400) . ' days ago',
            $diff < 31536000 => floor($diff / 2592000) . ' months ago',
            default => floor($diff / 31536000) . ' years ago'
        };
        $r['last_message_at'] = $t;
    }
    unset($r);
}
?>
<!DOCTYPE html>
<html>

<head>
    <title><?= "$page_title - " . INSTANCE_NAME ?></title>
    <link rel="stylesheet" href="/static/style.css">
    <link rel="shortcut icon" href="/favicon.ico" type="image/x-icon">
</head>

<body>
    <h1><a href="/"><?= INSTANCE_NAME ?></a> - <?= $page_title ?></h1>
    <?php if (isset($messages)): ?>
        <?php if (empty($messages)): ?>
            <p>No messages.</p>
        <?php else: ?>
            <p>Showing <?= count($messages) ?> messages on Page <?= $page + 1 ?></p>
            <table>
                <tr>
                    <th>Time</th>
                    <?php if (array_key_exists('room', $messages[0])): ?>
                        <th>Room</th>
                    <?php endif; ?>
                    <th>Nick</th>
                    <th>Message</th>
                </tr>
                <?php foreach ($messages as $m): ?>
                    <tr class="message" data-irc-tags="<?= $m['tags'] ?>" id="msgid=<?= $m['id'] ?>">
                        <td class="datetime"><a
                                href="/?<?= $link ?>&d=<?= urlencode($date) ?>#msgid=<?= $m['id'] ?>"><?= date('H:i', strtotime($m['sent_at'])) ?></a>
                        </td>
                        <?php if (array_key_exists('room', $m)): ?>
                            <td class="room"><a href="/?r=<?= urlencode($m['room']) ?>"><?= $m['room'] ?></a></td>
                        <?php endif; ?>
                        <td class="nick">
                            <?= (!isset($prev_msg_nick) || $prev_msg_nick != $m['nick']) ? $m['nick'] : '' ?>
                        </td>
                        <td class="content"><?= $m['params'] ?></td>
                    </tr>
                    <?php $prev_msg_nick = $m['nick']; ?>
                <?php endforeach; ?>
            </table>
        <?php endif; ?>
    <?php elseif (isset($dates)): ?>
        <div class="calendar-wrapper">
            <?php foreach ($dates as $year => $months): ?>
                <?php foreach ($months as $month => $days): ?>
                    <?php
                    $fd = new DateTime("$year-$month-01");
                    $dm = (int) $fd->format('t');
                    $sw = (int) $fd->format('N');
                    $ms = $fd->format('M');
                    ?>
                    <table class="calendar">
                        <caption><?= "{$ms} {$year}" ?></caption>
                        <tr>
                            <th>Mo</th>
                            <th>Tu</th>
                            <th>We</th>
                            <th>Th</th>
                            <th>Fr</th>
                            <th>Sa</th>
                            <th>Su</th>
                        </tr>
                        <tr>
                            <?php
                            for ($i = 1; $i < $sw; $i++)
                                echo '<td></td>';

                            for ($day = 1; $day <= $dm; $day++) {
                                echo '<td>';
                                if (array_key_exists($day, $days) && $days[$day] > 0) {
                                    echo "<a href='/?$link&d=$year-$month-$day'>$day</a>";
                                } else {
                                    echo $day;
                                }
                                echo '</td>';
                                if ((($day + $sw - 1) % 7) == 0 && $day != $dm)
                                    echo '</tr><tr>';
                            }

                            $remainingdays = (7 - (($dm + $sw - 1) % 7)) % 7;
                            for ($i = 0; $i < $remainingdays; $i++)
                                echo '<td></td>';
                            ?>
                        </tr>
                    </table>
                <?php endforeach; ?>
            <?php endforeach; ?>
        </div>
    <?php elseif (isset($rooms)): ?>
        <?php if (empty($rooms)): ?>
            <p>There are no rooms. <a href="/mod/">Add one!</a></p>
        <?php else: ?>
            <table>
                <tr>
                    <th>Room</th>
                    <th>Last message</th>
                </tr>
                <?php foreach ($rooms as $r): ?>
                    <tr>
                        <td><a href="/?r=<?= urlencode($r['name']) ?>"><?= $r['name'] ?></a></td>
                        <td><?= $r['last_message_at'] ?></td>
                    </tr>
                <?php endforeach; ?>
            </table>
        <?php endif; ?>
    <?php endif; ?>
</body>

<?php if (isset($messages)): ?>
    <script>
        const messages = document.querySelectorAll(".message");
        for (const m of messages) {
            let tags = m.getAttribute("data-irc-tags");

            const nickElement = m.querySelector(".nick");
            const contentElement = m.querySelector(".content");
            let color = null;

            for (const tag of tags.split(";")) {
                const p = tag.split("=");
                if (p[0] == "color") {
                    color = p[1];
                    break;
                }
            }

            if (color) {
                nickElement.style.color = color;

                if (contentElement.textContent.startsWith("ACTION")) {
                    contentElement.textContent = contentElement.textContent.substring(7);
                    contentElement.style.color = color;
                    contentElement.style.fontStyle = 'italic';
                }
            }
        }
    </script>
<?php endif; ?>

</html>