import argparse
import subprocess
import os
import socket
import sys
import yaml

import livereload

def info(message):
    print(f"\033[32m{message}\033[0m")


def get_free_port():
    for port in range(8000, 8100):
        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
            if s.connect_ex(("localhost", port)) != 0:
                return port


class ALMOConfig():
    def __init__(self, target, config_path):
        self.target = target
        self.config = yaml.safe_load(open(config_path))

        self.args = {
            'output': '-o',
            'template': '-b',
            'theme':  '-t',
            'css': '-c',
            'editor_theme': '-e',
        }

        self.validate()

    def validate(self):
        for key in self.config.keys():
            if key not in self.args.keys():
                raise Exception(f"Invalid key: {key}")

    def add_config(self, key, value):
        self.config[key] = value

    def get_config(self, key):
        return self.config[key]
            
    def command(self):
        command = ["almo", self.target]
        for key, value in self.config.items():
            command.append(self.args[key])
            command.append(value)
        
        return " ".join(command)


def watch(target, config_path):
    config = ALMOConfig(target, config_path)
    config.add_config("output", "_almo_output/index.html")
    
    os.makedirs("_almo_output/", exist_ok=True)
    
    subprocess.run(config.command(), shell=True)

    port = get_free_port()
    server = livereload.Server()
    server.watch(target, livereload.shell(config.command(), shell=True))
    server.serve(port=port, root="_almo_output/", host="localhost")


def main():
    parser = argparse.ArgumentParser()

    parser.add_argument("target", help="Markdown file to be converted")
    parser.add_argument("config", help="Configuration file")

    args = parser.parse_args()

    watch(args.target, args.config)
    
    
if __name__ == '__main__':
    main()

from setuptools import setup, find_packages

setup(
    name='almo',
    version='0.1',
    packages=find_packages(),
    install_requires=[
        'livereload',
        'PyYAML'
    ],
    entry_points={
        'console_scripts': [
            'almo-cli=almo_cli.almo_cli:main [cli]',
        ],
    },
)