from django.conf import settings from django.core.files.uploadedfile import UploadedFile from django.http import HttpRequest, HttpResponse from django.utils.translation import gettext as _ from zerver.actions.realm_emoji import check_add_realm_emoji, do_remove_realm_emoji from zerver.decorator import require_member_or_admin from zerver.lib.emoji import check_remove_custom_emoji, check_valid_emoji_name, name_to_codepoint from zerver.lib.exceptions import JsonableError, ResourceNotFoundError from zerver.lib.response import json_success from zerver.lib.typed_endpoint import PathOnly, typed_endpoint from zerver.lib.upload import get_file_info from zerver.models import RealmEmoji, UserProfile from zerver.models.realm_emoji import get_all_custom_emoji_for_realm def list_emoji(request: HttpRequest, user_profile: UserProfile) -> HttpResponse: # We don't do any checks here because the list of realm # emoji is public. return json_success( request, data=dict(emoji=get_all_custom_emoji_for_realm(user_profile.realm_id)) ) @require_member_or_admin @typed_endpoint def upload_emoji( request: HttpRequest, user_profile: UserProfile, *, emoji_name: PathOnly[str] ) -> HttpResponse: emoji_name = emoji_name.strip().replace(" ", "_") valid_built_in_emoji = name_to_codepoint.keys() check_valid_emoji_name(emoji_name) if not user_profile.can_add_custom_emoji(): raise JsonableError(_("Insufficient permission")) if RealmEmoji.objects.filter( realm=user_profile.realm, name=emoji_name, deactivated=False ).exists(): raise JsonableError(_("A custom emoji with this name already exists.")) if len(request.FILES) != 1: raise JsonableError(_("You must upload exactly one file.")) if emoji_name in valid_built_in_emoji and not user_profile.is_realm_admin: raise JsonableError(_("Only administrators can override default emoji.")) [emoji_file] = request.FILES.values() assert isinstance(emoji_file, UploadedFile) assert emoji_file.size is not None if emoji_file.size > settings.MAX_EMOJI_FILE_SIZE_MIB * 1024 * 1024: raise JsonableError( _("Uploaded file is larger than the allowed limit of {max_size} MiB").format( max_size=settings.MAX_EMOJI_FILE_SIZE_MIB, ) ) _filename, content_type = get_file_info(emoji_file) check_add_realm_emoji(user_profile.realm, emoji_name, user_profile, emoji_file, content_type) return json_success(request) def delete_emoji(request: HttpRequest, user_profile: UserProfile, emoji_name: str) -> HttpResponse: if not RealmEmoji.objects.filter( realm=user_profile.realm, name=emoji_name, deactivated=False ).exists(): raise ResourceNotFoundError( _("Emoji '{emoji_name}' does not exist").format(emoji_name=emoji_name) ) check_remove_custom_emoji(user_profile, emoji_name) do_remove_realm_emoji(user_profile.realm, emoji_name, acting_user=user_profile) return json_success(request)