ug_fess.py

import sys

sys.dont_write_bytecode = True


import os
import re

import django
from django.apps import apps
from django.conf import settings
from django.utils import timezone
import filetype
import streamlit as st
from streamlit.delta_generator import DeltaGenerator
from streamlit.runtime.uploaded_file_manager import UploadedFile

from auth import authenticate
from content_moderation import (
    has_disallowed_entities,
    has_inappropriate_content,
    has_inappropriate_image,
)
from x import create_tweet, is_valid_tweet_url, upload_images


if not settings.configured or not apps.ready:
    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "django_settings")
    django.setup()


from db.models import Menfess, User


MAX_MENFESS_PER_USER_PER_DAY = 3

MENFESS_SIGNATURE = "yuji!"

X_MAX_IMAGE_ATTACHMENTS = 4

MAX_IMAGE_SIZE_BYTES = 5 * 1024 * 1024

ALLOWED_IMAGE_EXTS = ["jpeg", "jpg", "png", "webp"]

ALLOWED_IMAGE_MIME_TYPES = {"image/jpeg", "image/png", "image/webp"}

MAX_QRT_URL_LENGTH = 54


@st.dialog("Status")
def show_menfess_creation_status(status_type: str, message: str) -> None:
    if status_type == "success":
        st.success(message)
    elif status_type == "error":
        st.error(message)


def can_create_menfess_today(user_id: int) -> bool:
    today = timezone.localdate()
    today_menfess_count = Menfess.objects.filter(
        user_id=user_id, created_at__date=today
    ).count()
    return today_menfess_count < MAX_MENFESS_PER_USER_PER_DAY


def has_invalid_image(images: list[UploadedFile]) -> bool:
    for image in images:
        if image.size > MAX_IMAGE_SIZE_BYTES:
            return True

        _, image_ext = os.path.splitext(image.name)
        image_ext = image_ext[1:]

        if image_ext not in ALLOWED_IMAGE_EXTS:
            return True

        image.seek(0)

        kind = filetype.guess(image)

        if (
            kind is None
            or kind.extension not in ALLOWED_IMAGE_EXTS
            or kind.mime != image.type
            or image.type not in ALLOWED_IMAGE_MIME_TYPES
        ):
            return True

    return False


def sign_in(username: str, password: str, error_placeholder: DeltaGenerator) -> None:
    try:
        if authenticate(username, password):
            user = User.objects.get_or_create(username=username)[0]

            if user.is_banned:
                error_placeholder.error(
                    "Sorry, lo di-ban dari UG FESS karena ngelanggar rules!"
                )
                return

            st.session_state.user_id = user.user_id
            st.session_state.is_authenticated = True
            st.rerun()
        else:
            error_placeholder.error(
                "Username atau password yang lo masukin salah nih. Coba cek lagi ya!"
            )
    except Exception as e:
        print(e)
        error_placeholder.error(
            "Sign in lagi bermasalah nih :disappointed:. Coba lagi nanti ya!"
        )


def tweet_menfess(text: str, images: list[UploadedFile], qrt: str) -> None:
    try:
        if not can_create_menfess_today(st.session_state.user_id):
            show_menfess_creation_status(
                "error",
                f"Yah, udah nyentuh limit kirim menfess hari ini. Max {MAX_MENFESS_PER_USER_PER_DAY} menfess aja ya "
                "per hari. Coba lagi besok :smiley:",
            )
            return

        if qrt:
            if len(qrt) > MAX_QRT_URL_LENGTH:
                show_menfess_creation_status(
                    "error",
                    f"QRT-nya ga valid nih, maks {MAX_QRT_URL_LENGTH} karakter aja ya!",
                )
                return

            qrt_match = re.search(r"https://x\.com/ug_fess/status/(\d+)", qrt)

            if qrt_match is None or not is_valid_tweet_url(qrt):
                show_menfess_creation_status(
                    "error",
                    "QRT-nya ga valid nih. Pastiin lo QRT tweet dari @ug_fess ya!",
                )
                return

            qrt_id = qrt_match.group(1)
        else:
            qrt_id = None

        if text:
            if MENFESS_SIGNATURE in text.lower():
                show_menfess_creation_status(
                    "error",
                    f"Menfess-nya jangan ada reserved keyword ***{MENFESS_SIGNATURE}*** ya! Biar sistem aja yang "
                    f"nambahin ***{MENFESS_SIGNATURE}***-nya.",
                )
                return

            if has_disallowed_entities(text):
                show_menfess_creation_status(
                    "error", "Menfess-nya ga boleh ada #, @, atau URL ya!"
                )
                return

            if has_inappropriate_content(text):
                show_menfess_creation_status(
                    "error",
                    "Menfess-nya ga boleh ada konten yang inappropriate ya! Baca lagi rules-nya.",
                )
                return

            text = f"{MENFESS_SIGNATURE} {text}"

        if images:
            if len(images) > X_MAX_IMAGE_ATTACHMENTS:
                show_menfess_creation_status(
                    "error", f"Max {X_MAX_IMAGE_ATTACHMENTS} images aja ya!"
                )
                return

            if has_invalid_image(images):
                show_menfess_creation_status(
                    "error", "Image-nya ada yang ga valid. Coba cek lagi!"
                )
                return

            if has_inappropriate_image(images):
                show_menfess_creation_status(
                    "error", "Ga boleh ada adult, racy, atau gory images ya!"
                )
                return

            media_ids = upload_images(images)
        else:
            media_ids = None

        tweet_or_tweets = create_tweet(text, media_ids, qrt_id)

        if isinstance(tweet_or_tweets, list):
            tweet = tweet_or_tweets[0]
        else:
            tweet = tweet_or_tweets

        Menfess.objects.create(tweet_id=tweet.id, user_id=st.session_state.user_id)

        show_menfess_creation_status(
            "success", "Yay! Menfess lo udah di-tweet :smiley:"
        )
    except Exception as e:
        print(e)
        show_menfess_creation_status(
            "error",
            "Lagi ga bisa kirim menfess nih :disappointed:. Coba lagi nanti ya!",
        )


def sign_out():
    del st.session_state["user_id"]
    st.session_state.is_authenticated = False
    st.rerun()


def sign_in_form():
    st.header("Eitss, sign in dulu!", anchor=False)
    st.write(
        "Sign in pake kredensial Student Site ya! Buat verifikasi kalo lo emang anak Gundar."
    )
    st.caption(
        "Tenang aja, password lo ga bakal disimpen kok. Sistem cuma nyimpen username lo aja. Kalo lo masih ragu, "
        "lo bisa cek codebase UG FESS di [github.com/nxgeo/ug-fess](https://github.com/nxgeo/ug-fess)."
    )
    error_placeholder = st.empty()
    username = st.text_input("Username")
    password = st.text_input("Password", type="password")

    if st.button("Sign in"):
        if username and password:
            sign_in(username, password, error_placeholder)


def main_page():
    st.header("Mau kirim menfess apa?", anchor=False)
    menfess_submission_form = st.form(
        "menfess_submission_form", clear_on_submit=True, enter_to_submit=False
    )
    text = menfess_submission_form.text_area("Ketikin menfess lo di sini:")
    images = menfess_submission_form.file_uploader(
        f"Lo juga bisa upload images (max {X_MAX_IMAGE_ATTACHMENTS}):",
        type=ALLOWED_IMAGE_EXTS,
        accept_multiple_files=True,
    )
    qrt = menfess_submission_form.text_input(
        "QRT (opsional):",
        max_chars=MAX_QRT_URL_LENGTH,
        help="Contoh: https[]()://x.com/ug_fess/status/1845753430381662319",
    )

    if menfess_submission_form.form_submit_button():
        if text or images:
            tweet_menfess(text, images, qrt)

    st.divider()

    st.subheader("Mau sign out?", anchor=False)
    st.write(
        "Kalo lo mau sign out, tinggal refresh (CTRL + R) aja atau klik button :point_down:"
    )

    if st.button("Sign out"):
        sign_out()


if "is_authenticated" not in st.session_state:
    st.session_state.is_authenticated = False

st.set_page_config(page_title="UG FESS", page_icon=":flying_saucer:")

if st.session_state.is_authenticated:
    main_page()
else:
    sign_in_form()