#!/usr/bin/python3
import json
import configparser
import requests
import typer
import os

app = typer.Typer()


class Servers:
    @staticmethod
    def list():
        response = requests.get("https://api.cloudvps.reg.ru/v1/reglets", headers=reqHeader)
        return response.json()

    @staticmethod
    def get_systems():
        response = requests.get("https://api.cloudvps.reg.ru/v1/images?type=distribution", headers=reqHeader)
        return response.json()

    @staticmethod
    def create(name, tariff, image):
        response = requests.post("https://api.cloudvps.reg.ru/v1/reglets",
                                 headers=reqHeader,
                                 json={"name": name,
                                       "size": tariff,
                                       "image": image
                                       }
                                 )
        return response.json()


def get_balance():
    response = requests.get("https://api.cloudvps.reg.ru/v1/balance_data", headers=reqHeader)
    return response.json()


@app.command()
def get_tariffs():
    response = requests.get("https://api.cloudvps.reg.ru/v1/prices", headers=reqHeader)
    print(json.dumps(response.json(),indent=4))


@app.command()
def balance():
    print(get_balance())


@app.command()
def list_servers():
    print(Servers.list())


@app.command()
def list_os():
    systems_list = Servers.get_systems()['images']
    for i in systems_list:
        print(i['slug'])


@app.command()
def create_server(name: str = typer.Option(...), tariff: str = typer.Option(...), image: str = typer.Option(...)):
    print(Servers.create(name=name, tariff=tariff, image=image))


def get_api_key():
    config = configparser.ConfigParser()
    config.read(os.path.join(os.getenv('XDG_CONFIG_HOME'), 'regructl.ini'))
    result = config.get('api', 'key', fallback=None)
    if result is None:
        result = input("Enter API Key ")
        config.add_section('api')
        config.set('api', 'key', result)

        with open(os.path.join(os.getenv('XDG_CONFIG_HOME'), 'regructl.ini'), 'w') as configfile:
            config.write(configfile)

    return result


if __name__ == '__main__':
    apikey = get_api_key()
    reqHeader = {"Authorization": "Bearer " + apikey}

    app()