mictl.py

import click
import os, sys
import requests
import random, string
from requests.models import HTTPBasicAuth
import os.path

# Mictl
MICTL_DIR = os.path.join(os.getenv("HOME"), ".mictl")
MICTL_KEY_FILE_PATH = os.path.join(os.getenv("HOME"), ".mictl", "key.ctl")
RANDOM_PASSWORD_LENGTH = 10
RANDOM_PASSWORD_SEED_TEXT = string.ascii_letters

# Migadu
MIGADU_API_BASE = 'https://api.migadu.com/v1/'

def crit(text):
    click.echo('error: ' + click.style(text, 'black', 'red'))
    sys.exit(1)

def get_auth():
    if os.path.exists(MICTL_KEY_FILE_PATH):
        data = open(MICTL_KEY_FILE_PATH).read()
        parts = data.split(':')
        return HTTPBasicAuth(parts[0], parts[1])
    else:
        crit('api client not configured. run \'mictl setup <email> <key>\'')

def generate_random_password():
    return ''.join(random.choice(RANDOM_PASSWORD_SEED_TEXT) for i in range(RANDOM_PASSWORD_LENGTH))

@click.group(help = "A command-line interface for the Migadu email API.")
def mictl():
    pass

@mictl.command(help = 'Configures mictl.')
@click.argument('email')
@click.argument('key')
def setup(email, key):
    content = email + ':' + key
    if not os.path.exists(MICTL_DIR):
        os.makedirs(MICTL_DIR)
    if os.path.exists(MICTL_KEY_FILE_PATH):
        os.unlink(MICTL_KEY_FILE_PATH)
    file = open(MICTL_KEY_FILE_PATH, 'w')
    file.seek(0)
    file.write(content)
    click.echo('success: stored credentials')

@mictl.group(help = "Create, update, and list domain mailboxes.")
def boxes():
    get_auth()
    pass

@boxes.command(help = 'Lists all mailboxes for a domain.')
@click.argument('domain')
def all(domain):
    result = requests.get(MIGADU_API_BASE + f'domains/{domain}/mailboxes', auth = get_auth()).json()
    try: 
        mailboxes = result['mailboxes']
    except KeyError:
        click.echo(click.style(f'error: {result["error"]}', 'red'))
        return

    click.echo(click.style('Listing all mailboxes on ' + domain, 'green'))
    for box in mailboxes:
        click.echo(click.style(f'- {box["address"]}', 'blue'))

@boxes.command(help = 'Creates a new mailbox on a specified domain.')
@click.argument('address')
@click.option('--name', required=True, is_flag=False)
@click.option('--invite-address', required=False, is_flag=False)
@click.option('--password', required=False, is_flag=False)
@click.option('--random', required=False, is_flag=True)
def create(address, name, invite_address, password, random):
    parts = str.split(address, '@')
    try:
        local = parts[0]
        domain = parts[1]
    except IndexError:
        crit('invalid address to create')

    data = {
        'name': name,
        'local_part': local
    }
    if invite_address is None and password is None and random is False:
        crit('one of invite_address or password is expected, or \'--random\' to generate a random password')
    
    click.echo(click.style(f'creating user \'{local}\' on \'{domain}\'', 'green'))

    if password is None and invite_address is None and random is True:
        password = generate_random_password()
        click.echo(f'generated password: ' + password)

    if invite_address is not None:
        data['password_method'] = 'invitation'
        data['password_recovery_email'] = invite_address
    else:
        data['password'] = password

    response = requests.post(
        MIGADU_API_BASE + f'domains/{domain}/mailboxes',
        json = data,
        auth = get_auth()
    )

    response_data = response.json()
    if response.status_code == 200:
        if invite_address is not None:
            click.echo('account created, signup email sent to \'' + invite_address + '\'')
        else:
            click.echo(f'account created, log into https://webmail.migadu.com as \'{address}\' with password \'{password}\'')
    else:
        click.echo('error: ' + response_data['error'])

@boxes.command(help = 'Deletes a specified mailbox.')
@click.argument('address')
def delete(address):
    parts = str.split(address, '@')
    try:
        local = parts[0]
        domain = parts[1]
    except IndexError:
        crit('invalid address to delete')
        
    response = requests.delete(
        MIGADU_API_BASE + f'domains/{domain}/mailboxes/{local}',
        auth = get_auth()
    )

    response_data = response.json()
    if response.status_code == 200:
        click.echo(f'\'{address}\' deleted.')
    else:
        click.echo('error: ' + response_data['error'])

mictl.add_command(boxes)

@mictl.command('usage')
def usage():
    click.echo(click.style('usage instructions for migadu domains', 'green'))
    click.echo('incoming - IMAP imap.migadu.com - port 993 - TLS - username: full email')
    click.echo('outgoing - SMTP smtp.migadu.com - port 465 - IMPLICIT TLS - username: full email')
    click.echo('outgoing - SMTP smtp.migadu.com - port 587 - STARTTLS - username: full email')

def _start_app():
    mictl()