#!/usr/bin/env python3
from os import listdir, remove, walk
from os.path import expanduser


def delete(_file_path, _target_database):  # deletes an entry or folder
    from shutil import rmtree
    try:
        if _file_path.endswith('/'):
            rmtree(f"{expanduser('~/.local/share/sshyp/')}{_file_path}")
        else:
            remove(f"{expanduser('~/.local/share/sshyp/')}{_file_path}.gpg")
    except FileNotFoundError:
        print(f"location does not exist {_target_database}")
    if _target_database == 'remotely':
        for _device_name in listdir(expanduser('~/.config/sshyp/devices')):
            open(expanduser('~/.config/sshyp/deleted/') + _file_path.replace('/', '\x1e') + '\x1f' + _device_name, 'w')


def deletion_check(_client_device_name):  # creates a list of entries and folders to be deleted from a client device
    for _file in listdir(expanduser('~/.config/sshyp/deleted')):
        _file_path, _sep, _device = _file.partition('\x1f')
        _file_path = _file_path.replace('\x1e', '/')
        if _device == _client_device_name:
            print(_file_path)
            try:
                remove(f"{expanduser('~/.config/sshyp/deleted/')}{_file}")
            except FileNotFoundError:
                pass


def folder_check():  # creates a list of folders to be compared with those of a client device
    for _root, _directories, _files in walk(expanduser('~/.local/share/sshyp')):
        for _dir in _directories:
            print(f"{_root.replace(expanduser('~'), '')}/{_dir}")