Lamp

Complexity

Total Complexity

36

Size/Duplication

Total Lines	166
Duplicated Lines	0 %

Importance

Changes	5
Bugs	0	Features	1

14 Methods

Rating	Name	Duplication	Size	Complexity
B	display_services_ports()	0	24	6
B	run_services_post_scripts()	0	9	5
A	__init__()	0	19	3
A	stop()	0	3	1
A	start()	0	7	2
B	status()	0	29	3
A	run_php()	0	7	2
A	restart()	0	3	1
A	console()	0	9	3
A	get_vm_item()	0	6	3
A	run_mysql()	0	15	3
A	destroy()	0	2	1
A	fullstart()	0	3	1
A	check_vms_are_running()	0	3	2

import os
import sys
import subprocess

from clint.textui import colored, puts, columns
from lib import docker


class Lamp():
    def __init__(self, base_dir: str):
        self.lamp_base_dir = base_dir
        self.current_dir = os.getcwd()
        # Make sure we are in the right directory
        self.current_dir_relative = self.current_dir[len(self.lamp_base_dir):].lstrip('/')
        os.chdir(self.lamp_base_dir)

        from lib.configreader import Config
        self.default_config_main = Config('conf/compose.ini.tpl').read()['main']
        self.user_config_main = Config('conf/compose.ini').read()['main']

        self.vms = docker.get_vms()

        self.running_vms = 0
        for vm_id, vm_data in self.vms.items():
            if vm_data['running'] is False:
                continue

            self.running_vms += 1


    def run_services_post_scripts(self):
        services = [service for service in self.user_config_main.get('services', '').split(',') if service != '']
        for service in services:
            service_script = 'services/' + service + '.sh'
            vm_name = self.get_vm_item(service, 'name')
            if os.path.isfile(service_script) is False:
                continue

            subprocess.call(['bash', service_script, vm_name])


    def display_services_ports(self):
        puts('To access the {} use : http://{}\n'.format(colored.yellow('web server'), self.get_vm_item('apache', 'ip')))

        mailcatcher_ip = self.get_vm_item('mailcatcher', 'ip')
        if mailcatcher_ip != '':
            puts('For {} use : http://{}'.format(colored.yellow('mailcatcher'), mailcatcher_ip))
            puts(' '*16 + 'and in your VM use the server "{}" with the port 25\n'.format(colored.yellow('mailcatcher')))

        maildev_ip = self.get_vm_item('maildev', 'ip')
        if maildev_ip != '':
            puts('For {} use : http://{}'.format(colored.yellow('maildev'), maildev_ip))
            puts(' '*12 + 'and in your VM use the server "{}" with the port 25\n'.format(colored.yellow('maildev')))

        mongoclient_ip = self.get_vm_item('mongoclient', 'ip')
        if mongoclient_ip != '':
            puts('For {} use : http://{}:3000\n'.format(colored.yellow('mongoclient'), mongoclient_ip))

        pma_ip = self.get_vm_item('phpmyadmin', 'ip')
        if pma_ip != '':
            puts('For {} use : http://{}\n'.format(colored.yellow('phpMyAdmin'), pma_ip))

        xhgui_ip = self.get_vm_item('xhgui', 'ip')
        if xhgui_ip != '':
            puts('For {} use : http://{}\n'.format(colored.yellow('xhgui'), xhgui_ip))


    def start(self, pull: bool):
        if pull is True:
            subprocess.call(['bin/compose', 'pull'])

        subprocess.call(['bin/compose', 'up', '-d', '--remove-orphans'])
        self.vms = docker.get_vms()
        self.run_services_post_scripts()


    def stop(self):
        self.check_vms_are_running()
        subprocess.call(['bin/compose', 'stop'])

    def destroy(self):
        subprocess.call(['bin/compose', 'down'])


    def restart(self):
        self.stop()
        self.start()


    def status(self):
        self.check_vms_are_running()
        puts(columns(
            [(colored.green('VM')), 20],
            [(colored.green('IP')), 15],
            [(colored.green('Ports')), 30],
            [(colored.green('Image')), 30],
            [(colored.green('Docker ID')), 15],
            [(colored.green('Docker Name')), 25]
        ))
        puts(columns(
            ['-'*20, 20],
            ['-'*15, 15],
            ['-'*30, 30],
            ['-'*30, 30],
            ['-'*15, 15],
            ['-'*25, 25]
        ))
        for vm_id, vm_data in self.vms.items():
            if vm_data['ip'] == '':
                continue

            puts(columns(
                [vm_data['compose_name'], 20],
                [vm_data['ip'], 15],
                [', '.join(vm_data['ports']), 30],
                [vm_data['image'], 30],
                [vm_id[:12], 15],
                [vm_data['name'], 25]
            ))


    def fullstart(self):
        subprocess.call(['bin/compose', 'build'])
        self.start()


    def console(self, vm: str, user: str):
        self.check_vms_are_running()

        vm_name = self.get_vm_item(vm, 'name')
        if vm_name == '':
            raise Exception('{} does not seem to be in your services or has crashed'.format(vm))

        tty = 't' if sys.stdin.isatty() else ''
        subprocess.call(['docker', 'exec', '-u', user, '-i' + tty, vm_name, 'env', 'TERM=xterm', 'bash'])


    def run_php(self, user: str, args: str):
        self.check_vms_are_running()

        tty = 't' if sys.stdin.isatty() else ''
        cmd = ['docker', 'exec', '-u', user, '-i' + tty, self.get_vm_item('php', 'name'), 'bash', '-c', '--']
        cmd += ['cd /var/' + self.current_dir_relative + '; exec /usr/bin/php ' + args]
        subprocess.call(cmd, stdin=sys.stdin)


    def run_mysql(self, args: str):
        self.check_vms_are_running()

        vm_name = self.get_vm_item('mysql', 'name')
        if vm_name == '':
            raise Exception('mysql does not seem to be in your services or has crashed')

        tty = 't' if sys.stdin.isatty() else ''
        password = self.user_config_main.get(
            'mysql.root_password',
            self.default_config_main.get('mysql.root_password')
        )
        cmd = ['docker', 'exec', '-u', 'root', '-i' + tty, vm_name]
        cmd += ['mysql', '-u', 'root', '-p' + password, args]
        subprocess.call(cmd, stdin=sys.stdin)


    def get_vm_item(self, compose_name: str, item_name: str):
        for vm_id, vm_data in self.vms.items():
            if vm_data['compose_name'] == compose_name:
                return vm_data[item_name]

        return ''


    def check_vms_are_running(self):
        if self.running_vms == 0:
            raise Exception('Have you started your server with the start or fullstart action ?')