epicyon/status.py

__filename__ = "status.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.6.0"
__maintainer__ = "Bob Mottram"
__email__ = "bob@libreserver.org"
__status__ = "Production"
__module_group__ = "Core"
__accounts_data_path__ = None
__accounts_data_path_tests__ = False

from timeFunctions import date_utcnow
from timeFunctions import date_from_string_format
from timeFunctions import date_epoch
from utils import remove_html
from unicodetext import standardize_text

MAX_STATUS_LENGTH = 100


def get_actor_status(profile_json: {}) -> str:
    """returns the actor status if it exists
    https://codeberg.org/fediverse/fep/src/branch/main/fep/82f6/fep-82f6.md
    """
    if not profile_json.get('sm:status'):
        return ''
    status = ''
    if isinstance(profile_json['sm:status'], str):
        status = profile_json['sm:status']
    elif isinstance(profile_json['sm:status'], dict):
        if profile_json['sm:status'].get('content'):
            possible_status = profile_json['sm:status']['content']
            if isinstance(status, str):
                status = possible_status
    if status:
        status = remove_html(status)
        if len(status) > MAX_STATUS_LENGTH:
            status = status[:MAX_STATUS_LENGTH]
        status = standardize_text(status)
    return status


def actor_status_expired(actor_status_json: {}) -> bool:
    """Has the given actor status expired?
    """
    if not actor_status_json.get('endTime'):
        return False
    if not isinstance(actor_status_json['endTime'], str):
        return False
    if 'T' not in actor_status_json['endTime'] or \
       ':' not in actor_status_json['endTime']:
        return False
    date_formats = (
        "%Y-%m-%dT%H:%M:%S%z",
        "%Y-%m-%dT%H:%M:%S%Z"
    )
    status_end_time = None
    try:
        status_end_time = \
            date_from_string_format(actor_status_json['endTime'],
                                    date_formats)
    except BaseException:
        return False
    if not status_end_time:
        return False
    curr_time = date_utcnow()
    if status_end_time < curr_time:
        return True
    return False


def get_status_number(published_str: str = None) -> (str, str):
    """Returns the status number and published date
    """
    if not published_str:
        curr_time = date_utcnow()
    else:
        curr_time = \
            date_from_string_format(published_str, ['%Y-%m-%dT%H:%M:%S%z'])
    days_since_epoch = (curr_time - date_epoch()).days
    # status is the number of seconds since epoch
    status_number = \
        str(((days_since_epoch * 24 * 60 * 60) +
             (curr_time.hour * 60 * 60) +
             (curr_time.minute * 60) +
             curr_time.second) * 1000 +
            int(curr_time.microsecond / 1000))
    # See https://github.com/tootsuite/mastodon/blob/
    # 995f8b389a66ab76ec92d9a240de376f1fc13a38/lib/mastodon/snowflake.rb
    # use the leftover microseconds as the sequence number
    sequence_id = curr_time.microsecond % 1000
    # shift by 16bits "sequence data"
    status_number = str((int(status_number) << 16) + sequence_id)
    published = curr_time.strftime("%Y-%m-%dT%H:%M:%SZ")
    return status_number, published