"""
author: deadc0de6 (https://github.com/deadc0de6)
Copyright (c) 2017, deadc0de6
helpers for the unittests
"""

import os
import shutil
import string
import random
import tempfile

TMPSUFFIX = '.dotdrop'


def clean(path):
    '''Delete file or folder.'''
    if not os.path.exists(path):
        return
    if os.path.isdir(path):
        shutil.rmtree(path)
    else:
        os.remove(path)


def get_string(length):
    '''Get a random string of length "length".'''
    alpha = string.ascii_uppercase + string.digits
    return ''.join(random.choice(alpha) for _ in range(length))


def get_tempfolder():
    '''Get a temporary folder'''
    return tempfile.mkdtemp(suffix=TMPSUFFIX)


def create_random_file(folder, content=None):
    '''Create a new file in folder with random content.'''
    fname = get_string(8)
    if content is None:
        content = get_string(100)
    path = os.path.join(folder, fname)
    with open(path, 'w') as f:
        f.write(content)
    return path, content


def create_dir(path):
    '''Create a folder'''
    os.mkdir(path)
    return path


def create_fake_config(folder, configname='config.yaml',
                       dotpath='dotfiles', backup=True, create=True):
    '''Create a fake config file'''
    path = os.path.join(folder, configname)
    with open(path, 'w') as f:
        f.write('config:\n')
        f.write('  backup: %s\n' % (str(backup)))
        f.write('  create: %s\n' % (str(create)))
        f.write('  dotpath: %s\n' % (dotpath))
        f.write('dotfiles:\n')
        f.write('profiles:\n')
    return path