epicyon/notifyOnPost.py

__filename__ = "notifyOnPost.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.4.0"
__maintainer__ = "Bob Mottram"
__email__ = "bob@libreserver.org"
__status__ = "Production"
__module_group__ = "Calendar"

import os
from utils import remove_domain_port
from utils import acct_dir
from utils import text_in_file


def _notify_on_post_arrival(base_dir: str, nickname: str, domain: str,
                            following_nickname: str,
                            following_domain: str,
                            add: bool) -> None:
    """Adds or removes a handle from the following.txt list into a list
    indicating whether to notify when a new post arrives from that account
    """
    # check that a following file exists
    domain = remove_domain_port(domain)
    following_filename = \
        acct_dir(base_dir, nickname, domain) + '/following.txt'
    if not os.path.isfile(following_filename):
        print("WARN: following.txt doesn't exist for " +
              nickname + '@' + domain)
        return
    handle = following_nickname + '@' + following_domain

    # check that you are following this handle
    if not text_in_file(handle + '\n', following_filename, False):
        print('WARN: ' + handle + ' is not in ' + following_filename)
        return

    notify_on_post_filename = \
        acct_dir(base_dir, nickname, domain) + '/notifyOnPost.txt'

    # get the contents of the notifyOnPost file, which is
    # a set of handles
    following_handles = ''
    if os.path.isfile(notify_on_post_filename):
        print('notify file exists')
        with open(notify_on_post_filename, 'r',
                  encoding='utf-8') as calendar_file:
            following_handles = calendar_file.read()
    else:
        # create a new notifyOnPost file from the following file
        print('Creating notifyOnPost file ' + notify_on_post_filename)
        following_handles = ''
        with open(following_filename, 'r',
                  encoding='utf-8') as following_file:
            following_handles = following_file.read()
        if add:
            with open(notify_on_post_filename, 'w+',
                      encoding='utf-8') as fp_notify:
                fp_notify.write(following_handles + handle + '\n')

    # already in the notifyOnPost file?
    if handle + '\n' in following_handles or \
       handle + '\n' in following_handles.lower():
        print(handle + ' exists in notifyOnPost.txt')
        if add:
            # already added
            return
        # remove from calendar file
        new_following_handles = ''
        following_handles_list = following_handles.split('\n')
        handle_lower = handle.lower()
        for followed in following_handles_list:
            if followed.lower() != handle_lower:
                new_following_handles += followed + '\n'
        following_handles = new_following_handles

        with open(notify_on_post_filename, 'w+',
                  encoding='utf-8') as fp_notify:
            fp_notify.write(following_handles)
    else:
        print(handle + ' not in notifyOnPost.txt')
        # not already in the notifyOnPost file
        if add:
            # append to the list of handles
            following_handles += handle + '\n'
            with open(notify_on_post_filename, 'w+',
                      encoding='utf-8') as fp_notify:
                fp_notify.write(following_handles)


def add_notify_on_post(base_dir: str, nickname: str, domain: str,
                       following_nickname: str,
                       following_domain: str) -> None:
    """Add a notification
    """
    _notify_on_post_arrival(base_dir, nickname, domain,
                            following_nickname, following_domain, True)


def remove_notify_on_post(base_dir: str, nickname: str, domain: str,
                          following_nickname: str,
                          following_domain: str) -> None:
    """Remove a notification
    """
    _notify_on_post_arrival(base_dir, nickname, domain,
                            following_nickname, following_domain, False)


def notify_when_person_posts(base_dir: str, nickname: str, domain: str,
                             following_nickname: str,
                             following_domain: str) -> bool:
    """Returns true if receiving notifications when the given publishes a post
    """
    if following_nickname == nickname and following_domain == domain:
        return False
    notify_on_post_filename = \
        acct_dir(base_dir, nickname, domain) + '/notifyOnPost.txt'
    handle = following_nickname + '@' + following_domain
    if not os.path.isfile(notify_on_post_filename):
        # create a new notifyOnPost file
        with open(notify_on_post_filename, 'w+',
                  encoding='utf-8') as fp_notify:
            fp_notify.write('')
    return text_in_file(handle + '\n', notify_on_post_filename, False)