64898b2c79
Some checks failed
Code scanning / CodeQL (push) Failing after 15s
Zulip production suite / Ubuntu 22.04 production build (push) Failing after 26s
Zulip production suite / Debian 12 production install with custom db name and user (push) Has been skipped
Zulip production suite / Ubuntu 22.04 production install and PostgreSQL upgrade with pgroonga (push) Has been skipped
Zulip production suite / Ubuntu 24.04 production install (push) Has been skipped
Zulip production suite / Ubuntu 26.04 production install (push) Has been skipped
Zulip production suite / Debian 13 production install (push) Has been skipped
Zulip production suite / 7.0 Version Upgrade (push) Has been skipped
Zulip production suite / 8.0 Version Upgrade (push) Has been skipped
Zulip production suite / 6.0 Version Upgrade (push) Has been skipped
Zulip production suite / 10.0 Version Upgrade (push) Has been skipped
Zulip production suite / 9.0 Version Upgrade (push) Has been skipped
Zulip production suite / 12.0 Version Upgrade (push) Has been skipped
Zulip production suite / 11.0 Version Upgrade (push) Has been skipped
GitHub Actions Security Analysis with zizmor / zizmor (push) Failing after 3s
Zulip CI / Debian 12 (Python 3.11, backend + documentation) (push) Failing after 26s
Zulip CI / Ubuntu 22.04 (Python 3.10, backend + frontend) (push) Failing after 3s
Zulip CI / Ubuntu 24.04 (Python 3.12, backend) (push) Failing after 25s
Zulip CI / Ubuntu 26.04 (Python 3.14, backend) (push) Failing after 25s
Zulip CI / Debian 13 (Python 3.13, backend) (push) Failing after 28s
Zulip production suite / Required jobs (push) Has been cancelled
Zulip CI / Required jobs (push) Has been cancelled
API Documentation Update Check / check-feature-level-updated (push) Has been cancelled
API Documentation Update Check / notify-if-api-docs-changed (push) Has been cancelled
95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
from django.utils.timezone import now as timezone_now
|
|
|
|
from zerver.lib.timestamp import datetime_to_timestamp
|
|
from zerver.models import UserProfile
|
|
from zerver.models.devices import Device
|
|
from zerver.tornado.django_api import send_event_on_commit
|
|
|
|
|
|
def do_register_push_device(
|
|
user_profile: UserProfile,
|
|
device: Device,
|
|
*,
|
|
token_kind: str,
|
|
push_key_bytes: bytes,
|
|
push_key_id: int,
|
|
token_id_int: int,
|
|
token_id_base64: str,
|
|
) -> None:
|
|
registered_at = timezone_now()
|
|
device.push_key = push_key_bytes
|
|
device.push_key_id = push_key_id
|
|
device.pending_push_token_id = token_id_int
|
|
device.push_token_kind = token_kind
|
|
device.push_token_last_updated_timestamp = registered_at
|
|
device.push_registration_error_code = None
|
|
device.save(
|
|
update_fields=[
|
|
"push_key",
|
|
"push_key_id",
|
|
"pending_push_token_id",
|
|
"push_token_kind",
|
|
"push_token_last_updated_timestamp",
|
|
"push_registration_error_code",
|
|
]
|
|
)
|
|
|
|
event = dict(
|
|
type="device",
|
|
op="update",
|
|
device_id=device.id,
|
|
push_key_id=device.push_key_id,
|
|
pending_push_token_id=token_id_base64,
|
|
push_token_last_updated_timestamp=datetime_to_timestamp(registered_at),
|
|
push_registration_error_code=None,
|
|
)
|
|
send_event_on_commit(user_profile.realm, event, [user_profile.id])
|
|
|
|
|
|
def do_rotate_push_key(
|
|
user_profile: UserProfile, device: Device, push_key_bytes: bytes, push_key_id: int
|
|
) -> None:
|
|
if (
|
|
device.push_key_id == push_key_id
|
|
and device.push_key is not None
|
|
and bytes(device.push_key) == push_key_bytes
|
|
):
|
|
return
|
|
|
|
device.push_key = push_key_bytes
|
|
device.push_key_id = push_key_id
|
|
device.save(update_fields=["push_key", "push_key_id"])
|
|
|
|
event = dict(
|
|
type="device",
|
|
op="update",
|
|
device_id=device.id,
|
|
push_key_id=device.push_key_id,
|
|
)
|
|
send_event_on_commit(user_profile.realm, event, [user_profile.id])
|
|
|
|
|
|
def do_rotate_token(
|
|
user_profile: UserProfile, device: Device, token_id_int: int, token_id_base64: str
|
|
) -> None:
|
|
token_updated_at = timezone_now()
|
|
device.pending_push_token_id = token_id_int
|
|
device.push_token_last_updated_timestamp = token_updated_at
|
|
device.push_registration_error_code = None
|
|
device.save(
|
|
update_fields=[
|
|
"pending_push_token_id",
|
|
"push_token_last_updated_timestamp",
|
|
"push_registration_error_code",
|
|
]
|
|
)
|
|
event = dict(
|
|
type="device",
|
|
op="update",
|
|
device_id=device.id,
|
|
pending_push_token_id=token_id_base64,
|
|
push_token_last_updated_timestamp=datetime_to_timestamp(token_updated_at),
|
|
push_registration_error_code=None,
|
|
)
|
|
send_event_on_commit(user_profile.realm, event, [user_profile.id])
|