yarrbox/yarrbox.py

#!/usr/bin/env python3

import os
from collections import OrderedDict
import sys

import getpass
import json

# Each entry here is key => (label, hide input)
PROPS = OrderedDict([
    ('net.vpn.host',       ('VPN Provider',         'See https://github.com/sscraggles/docker-deluge-openvpn for options')),
    ('net.vpn.config',     ('VPN Config', 'See README for above, for PIA just say "Netherlands"')),
    ('net.vpn.username',   ('VPN Username',         None)),
    ('net.vpn.password',   ('VPN Password',         None)),
    ('vol.media.movies',   ('Movies Storage Path',  None)),
    ('vol.media.tv',       ('TV Storage Path',      None)),
    ('vol.media.anime',    ('Anime Storage Path',   None)),
    ('vol.conf.torrent',   ('Torrent Data Dir',     'Directory for torrent client(s) to store configs and state')),
    ('vol.conf.bootybot',  ('Bootybot Config Dir',  'Config directory for Bootybot')),
    ('vol.conf.plex',      ('Plex Data Dir',        'Directory for all of Plex\'s stuff (databases, etc.)')),
    ('vol.ingest.torrent', ('Torrent Ingest Dir',   'Directory to drop .torrent files to download through VPN')),
    ('vol.ingest.media',   ('Media Ingest Dir',     'Directory to drop full media files for Bootybot to later process')),
    ('vol.torrent.active', ('Torrent Download Dir', 'Directory for actively downloading torrents to be stored')),
    ('vol.torrent.seed',   ('Torrent Seeding Dir',  'Directory to move torrents after download finished for seeding')),
    ('vol.work.plextrans', ('Plex Transcode Dir',   'Directory for Plex to store certain transcoded files')),
    ('vol.work.bootybot',  ('Bootybot Extract Dir', 'Directory where Bootybot temporarily stores contents of archives, etc'))
])

PROMPT = '> '

def ask_config_opt(config, key):
    label, desc = PROPS[key]
    nodisp = 'password' in key

    # Print titles
    title = '= ' + label + ' ='
    bars = '=' * len(title)
    print(bars + '\n' + title + '\n' + bars)
    if desc is not None:
        print(desc)

    # Current state stuff.
    if key in config:
        if not nodisp:
            print('Current: \'' + config[key] + '\'')

    # Actually ask for input.
    if not nodisp:
        val = input(PROMPT)
        if len(val) > 0:
            config[key] = val
    else:
        val = getpass.getpass('(hidden) ' + PROMPT)
        if len(val) > 0:
            config[key] = val

def apply_template_settings(tmplt, settings):
    cur = tmplt
    for k, v in settings.items():
        cur = cur.replace('${' + k + '}', v)
    return cur

def read_config():
    try:
        with open('config.json', 'r') as f:
            return json.loads(f.read())
    except e:
        print('error loading config, is it missing or corrupt?', e)
        sys.exit(1)

def save_config(cfg):
    try:
        with open('config.json', 'w') as f:
            f.write(json.dumps(cfg, sort_keys=True, indent='  '))
            f.write('\n') # json.dumps doesn't put trailing newline
    except e:
        print('error writing config, do we have write perms?', e)
        sys.exit(1)

USAGE_STR = 'usage: yarrbox.py <init|edit|generate>'

def main():
    if len(sys.argv) != 2:
        print(USAGE_STR)
        sys.exit(1)

    command = sys.argv[1]
    if command == 'init':
        # Check if a config already exists
        if os.path.exists('config.json'):
            print('A config exists here, overwrite? [y/n, default: n] ', end='', flush=True)
            yesno = input()
            if yesno != 'y':
                print('leaving config unchanged')
                sys.exit(1)

        # Make a new config object and read all the properties.
        cfg = {}
        for k in PROPS:
            ask_config_opt(cfg, k)
            print('')

        # Write it to file
        save_config(cfg)

    elif command == 'edit':
        print('Leave options blank to leave unchanged!\n')
        cfg = read_config()

        # Go through all the properties again and reread them.
        for k in PROPS:
            ask_config_opt(cfg, k)
            print('')

        # Write it back.
        save_config(cfg)

    elif command == 'generate':
        # Load the config.
        cfg = read_config()

        # Read the template.
        tmplt = None
        try:
            with open('compose-template.yml', 'r') as f:
                tmplt = f.read()
        except e:
            print('error reading template:', e)
            sys.exit(1)

        # Fill in the template and save it.
        gen = apply_template_settings(tmplt, cfg)
        try:
            with open('docker-compose.yml', 'w') as f:
                f.write(gen)
        except e:
            print('error writing generated file, do we have write perms?', e)
            sys.exit(1)
    else:
        print(USAGE_STR)

if __name__ == '__main__':
    main()