epicyon/petnames.py

__filename__ = "petnames.py"
__author__ = "Bob Mottram"
__license__ = "AGPL3+"
__version__ = "1.5.0"
__maintainer__ = "Bob Mottram"
__email__ = "bob@libreserver.org"
__status__ = "Production"
__module_group__ = "Core"

import os
from utils import acct_dir


def set_pet_name(base_dir: str, nickname: str, domain: str,
                 handle: str, petname: str) -> bool:
    """Adds a new petname
    """
    if '@' not in handle:
        return False
    if ' ' in petname:
        petname = petname.replace(' ', '_')
    if handle.startswith('@'):
        handle = handle[1:]
    if petname.startswith('@'):
        petname = petname[1:]
    petnames_filename = acct_dir(base_dir, nickname, domain) + '/petnames.txt'
    entry = petname + ' ' + handle + '\n'

    # does this entry already exist?
    if os.path.isfile(petnames_filename):
        petnames_str = ''
        try:
            with open(petnames_filename, 'r',
                      encoding='utf-8') as fp_petnames:
                petnames_str = fp_petnames.read()
        except OSError:
            print('EX: set_pet_name unable to read ' + petnames_filename)
        if entry in petnames_str:
            return True
        if ' ' + handle + '\n' in petnames_str:
            petnames_list = petnames_str.split('\n')
            new_petnames_str = ''
            for pet in petnames_list:
                if not pet.endswith(' ' + handle):
                    new_petnames_str += pet + '\n'
                else:
                    new_petnames_str += entry
            # save the updated petnames file
            try:
                with open(petnames_filename, 'w+',
                          encoding='utf-8') as fp_petnames:
                    fp_petnames.write(new_petnames_str)
            except OSError:
                print('EX: set_pet_name unable to save ' + petnames_filename)
                return False
            return True
        # entry does not exist in the petnames file
        try:
            with open(petnames_filename, 'a+',
                      encoding='utf-8') as fp_petnames:
                fp_petnames.write(entry)
        except OSError:
            print('EX: set_pet_name unable to append ' + petnames_filename)
            return False
        return True

    # first entry
    try:
        with open(petnames_filename, 'w+', encoding='utf-8') as fp_petnames:
            fp_petnames.write(entry)
    except OSError:
        print('EX: set_pet_name unable to write ' + petnames_filename)
        return False
    return True


def get_pet_name(base_dir: str, nickname: str, domain: str,
                 handle: str) -> str:
    """Given a handle returns the petname
    """
    if '@' not in handle:
        return ''
    if handle.startswith('@'):
        handle = handle[1:]
    petnames_filename = acct_dir(base_dir, nickname, domain) + '/petnames.txt'

    if not os.path.isfile(petnames_filename):
        return ''
    petnames_str = ''
    try:
        with open(petnames_filename, 'r', encoding='utf-8') as fp_petnames:
            petnames_str = fp_petnames.read()
    except OSError:
        print('EX: get_pet_name unable to read ' + petnames_filename)
    if ' ' + handle + '\n' in petnames_str:
        petnames_list = petnames_str.split('\n')
        for pet in petnames_list:
            if pet.endswith(' ' + handle):
                return pet.replace(' ' + handle, '').strip()
    elif ' ' + handle.lower() + '\n' in petnames_str.lower():
        petnames_list = petnames_str.split('\n')
        handle = handle.lower()
        for pet in petnames_list:
            if pet.lower().endswith(' ' + handle):
                handle2 = pet.split(' ')[-1]
                return pet.replace(' ' + handle2, '').strip()
    return ''


def _get_pet_name_handle(base_dir: str, nickname: str, domain: str,
                         petname: str) -> str:
    """Given a petname returns the handle
    """
    if petname.startswith('@'):
        petname = petname[1:]
    petnames_filename = acct_dir(base_dir, nickname, domain) + '/petnames.txt'

    if not os.path.isfile(petnames_filename):
        return ''
    petnames_str = ''
    try:
        with open(petnames_filename, 'r', encoding='utf-8') as fp_petnames:
            petnames_str = fp_petnames.read()
    except OSError:
        print('EX: _get_pet_name_handle unable to read ' + petnames_filename)
    if petname + ' ' in petnames_str:
        petnames_list = petnames_str.split('\n')
        for pet in petnames_list:
            if pet.startswith(petname + ' '):
                handle = pet.replace(petname + ' ', '').strip()
                return handle
    return ''


def resolve_petnames(base_dir: str, nickname: str, domain: str,
                     content: str) -> str:
    """Replaces petnames with their full handles
    """
    if not content:
        return content
    if ' ' not in content:
        return content
    words = content.strip().split(' ')
    for wrd in words:
        # check initial words beginning with @
        if not wrd.startswith('@'):
            break
        # does a petname handle exist for this?
        handle = _get_pet_name_handle(base_dir, nickname, domain, wrd)
        if not handle:
            continue
        # replace the petname with the handle
        content = content.replace(wrd + ' ', '@' + handle + ' ')
    return content