misuzu/src/Users/relations.php

<?php
define('MSZ_USER_RELATION_NONE', 0);
define('MSZ_USER_RELATION_FOLLOW', 1);

define('MSZ_USER_RELATION_TYPES', [
    MSZ_USER_RELATION_NONE,
    MSZ_USER_RELATION_FOLLOW,
]);

define('MSZ_USER_RELATION_FOLLOW_PER_PAGE', 21);

function user_relation_is_valid_type(int $type): bool
{
    return in_array($type, MSZ_USER_RELATION_TYPES, true);
}

function user_relation_set(int $userId, int $subjectId, int $type = MSZ_USER_RELATION_FOLLOW): bool
{
    if ($type === MSZ_USER_RELATION_NONE) {
        return user_relation_remove($userId, $subjectId);
    }

    if ($userId < 1 || $subjectId < 1 || !user_relation_is_valid_type($type)) {
        return false;
    }

    $addRelation = db_prepare('
        REPLACE INTO `msz_user_relations`
            (`user_id`, `subject_id`, `relation_type`)
        VALUES
            (:user_id, :subject_id, :type)
    ');
    $addRelation->bindValue('user_id', $userId);
    $addRelation->bindValue('subject_id', $subjectId);
    $addRelation->bindValue('type', $type);
    $addRelation->execute();

    return $addRelation->execute();
}

function user_relation_remove(int $userId, int $subjectId): bool
{
    if ($userId < 1 || $subjectId < 1) {
        return false;
    }

    $removeRelation = db_prepare('
        DELETE FROM `msz_user_relations`
        WHERE `user_id` = :user_id
        AND `subject_id` = :subject_id
    ');
    $removeRelation->bindValue('user_id', $userId);
    $removeRelation->bindValue('subject_id', $subjectId);

    return $removeRelation->execute();
}

function user_relation_info(int $userId, int $subjectId): array
{
    $getRelationInfo = db_prepare('
        SELECT
            :user_id as `user_id_arg`, :subject_id as `subject_id_arg`,
            (
                SELECT `relation_type`
                FROM `msz_user_relations`
                WHERE `user_id` = `user_id_arg`
                AND `subject_id` = `subject_id_arg`
            ) as `user_relation`,
            (
                SELECT `relation_type`
                FROM `msz_user_relations`
                WHERE `subject_id` = `user_id_arg`
                AND `user_id` = `subject_id_arg`
            ) as `subject_relation`,
            (
                SELECT MAX(`relation_created`)
                FROM `msz_user_relations`
                WHERE (`user_id` = `user_id_arg` AND `subject_id` = `subject_id_arg`)
                OR (`user_id` = `subject_id_arg` AND `subject_id` = `user_id_arg`)
            ) as `relation_created`
    ');
    $getRelationInfo->bindValue('user_id', $userId);
    $getRelationInfo->bindValue('subject_id', $subjectId);
    return db_fetch($getRelationInfo);
}

function user_relation_count(int $userId, int $type, bool $from): int
{
    if ($userId < 1 || $type <= MSZ_USER_RELATION_NONE || !user_relation_is_valid_type($type)) {
        return 0;
    }

    static $getCount = [];
    $fetchCount = $getCount[$from] ?? null;

    if (empty($fetchCount)) {
        $getCount[$from] = $fetchCount = db_prepare(sprintf(
            '
                SELECT COUNT(`%1$s`)
                FROM `msz_user_relations`
                WHERE `%2$s` = :user_id
                AND `relation_type` = :type
            ',
            $from ? 'subject_id' : 'user_id',
            $from ? 'user_id' : 'subject_id'
        ));
    }

    $fetchCount->bindValue('user_id', $userId);
    $fetchCount->bindValue('type', $type);

    return (int)($fetchCount->execute() ? $fetchCount->fetchColumn() : 0);
}

function user_relation_count_to(int $userId, int $type): int
{
    return user_relation_count($userId, $type, false);
}

function user_relation_count_from(int $userId, int $type): int
{
    return user_relation_count($userId, $type, true);
}

function user_relation_users(int $userId, int $type, bool $from, int $take = 0, int $offset = 0): array
{
    if ($userId < 1 || $type <= MSZ_USER_RELATION_NONE || !user_relation_is_valid_type($type)) {
        return [];
    }

    $fetchAll = $take < 1;
    $key = sprintf('%s,%s', $from ? 'from' : 'to', $fetchAll ? 'all' : 'page');

    static $prepared = [];
    $fetchUsers = $prepared[$key] ?? null;

    if (empty($fetchUsers)) {
        $prepared[$key] = $fetchUsers = db_prepare(sprintf(
            '
                SELECT
                    ur.`%1$s` AS `user_id`, ur.`relation_created`,
                    u.`username`
                FROM `msz_user_relations` AS ur
                LEFT JOIN `msz_users` AS u
                ON u.`user_id` = ur.`%1$s`
                WHERE ur.`%2$s` = :user_id
                AND ur.`relation_type` = :type
                ORDER BY ur.`relation_created` DESC
                %3$s
            ',
            $from ? 'subject_id' : 'user_id',
            $from ? 'user_id' : 'subject_id',
            !$fetchAll ? 'LIMIT :offset, :take' : ''
        ));
    }

    $fetchUsers->bindValue('user_id', $userId);
    $fetchUsers->bindValue('type', $type);

    if (!$fetchAll) {
        $fetchUsers->bindValue('take', $take);
        $fetchUsers->bindValue('offset', $offset);
    }

    return db_fetch_all($fetchUsers);
}

function user_relation_users_to(int $userId, int $type, int $take = 0, int $offset = 0): array
{
    return user_relation_users($userId, $type, false, $take, $offset);
}

function user_relation_users_from(int $userId, int $type, int $take = 0, int $offset = 0): array
{
    return user_relation_users($userId, $type, true, $take, $offset);
}