DispelEvilBot/sql/R__functions_economy.sql

-- Экономика: баланс, опыт, очищение, яички.
-- Repeatable-миграция: перезапускается при изменении (CREATE OR REPLACE).

CREATE OR REPLACE FUNCTION bot.get_money(p_id_user bigint) RETURNS bigint
    LANGUAGE plpgsql
AS $$
DECLARE
    v_money bigint;
BEGIN
    SELECT sum(amount) INTO v_money FROM bot.money WHERE id_user = p_id_user;
    RETURN v_money;
END;
$$;

CREATE OR REPLACE FUNCTION bot.get_eggs(p_id_user bigint) RETURNS bigint
    LANGUAGE plpgsql
AS $$
DECLARE
    v_result bigint;
BEGIN
    SELECT sum(amount) INTO v_result FROM bot.eggs WHERE id_user = p_id_user;
    RETURN v_result;
END;
$$;

CREATE OR REPLACE PROCEDURE bot.money_action(
    IN p_id_user bigint,
    IN p_money bigint,
    IN p_id_action bigint DEFAULT 1
) LANGUAGE plpgsql
AS $$
BEGIN
    INSERT INTO bot.money (id_user, amount, id_action)
    VALUES (p_id_user, p_money, p_id_action);
END;
$$;

CREATE OR REPLACE FUNCTION bot.purification(p_id_user bigint)
    RETURNS TABLE(res integer, expr_add bigint, mny_add bigint, h integer, m integer, s integer)
    LANGUAGE plpgsql
AS $$
DECLARE
    v_lvl bigint;
    v_expr bigint;
    v_expr_add bigint;
    v_streak bigint;
    v_mny_add bigint;
    v_dt_pur timestamp;
    v_days bigint;
    v_result integer;
    v_h integer;
    v_m integer;
    v_s integer;
BEGIN
    SELECT lvl, expr, expr_streak
    INTO v_lvl, v_expr, v_streak
    FROM bot.de_users
    WHERE id_user = p_id_user;

    BEGIN
        SELECT dt INTO v_dt_pur
        FROM (
            SELECT dt, row_number() OVER (ORDER BY dt DESC) rn
            FROM bot.purifications
            WHERE id_user = p_id_user
        ) t
        WHERE rn = 1;
    EXCEPTION
        WHEN others THEN
            v_dt_pur := null;
    END;

    IF v_dt_pur IS NULL THEN
        v_days := -1;
    ELSE
        v_days = extract(day FROM now() - v_dt_pur);
    END IF;

    IF v_days = 0 THEN
        v_result := 0;
        v_h := 23 - extract(hour FROM now() - v_dt_pur);
        v_m := 59 - extract(minute FROM now() - v_dt_pur);
        v_s := 59 - extract(sec FROM now() - v_dt_pur);
    ELSE
        v_mny_add := floor(random() * 21) + 30;
        v_expr_add := 1;

        IF v_days = 1 THEN
            IF v_streak < 30 THEN
                v_streak := v_streak + 1;
            END IF;
            v_expr_add := v_expr_add + v_streak;
        ELSE
            v_streak := 0;
        END IF;

        v_expr := v_expr + v_expr_add;

        WHILE v_expr >= v_lvl * 10 LOOP
            v_expr := v_expr - v_lvl * 10;
            v_lvl := v_lvl + 1;
        END LOOP;

        UPDATE bot.de_users
        SET lvl = v_lvl, expr = v_expr, expr_streak = v_streak
        WHERE id_user = p_id_user;

        INSERT INTO bot.money (id_user, amount, id_action)
        VALUES (p_id_user, v_mny_add, 4);

        INSERT INTO bot.purifications (id_user) VALUES (p_id_user);

        v_result := 1;
    END IF;

    RETURN QUERY SELECT v_result, v_expr_add, v_mny_add, v_h, v_m, v_s;
END;
$$;

CREATE OR REPLACE FUNCTION bot.add_eggs(p_id_user bigint)
    RETURNS TABLE(res integer, eggs_add bigint, h integer, m integer, s integer)
    LANGUAGE plpgsql
AS $$
DECLARE
    v_dt timestamp;
    v_egg_add bigint;
    v_days bigint;
    v_h integer;
    v_m integer;
    v_s integer;
    v_result integer;
BEGIN
    BEGIN
        SELECT dt_action INTO v_dt
        FROM (
            SELECT dt_action, row_number() OVER (ORDER BY dt_action DESC) rn
            FROM bot.eggs
            WHERE id_user = p_id_user AND id_action = 1
        ) t
        WHERE rn = 1;
    EXCEPTION
        WHEN others THEN
            v_dt := null;
    END;

    IF v_dt IS NULL THEN
        v_days := -1;
    ELSE
        v_days = extract(day FROM now() - v_dt);
    END IF;

    IF v_days = 0 THEN
        v_result := 0;
        v_h := 23 - extract(hour FROM now() - v_dt);
        v_m := 59 - extract(minute FROM now() - v_dt);
        v_s := 59 - extract(sec FROM now() - v_dt);
    ELSE
        v_egg_add := floor(random() * 31) + 10;

        INSERT INTO bot.eggs (id_user, id_action, amount)
        VALUES (p_id_user, 1, v_egg_add);

        v_result := 1;
    END IF;

    RETURN QUERY SELECT v_result, v_egg_add, v_h, v_m, v_s;
END;
$$;