# Copyright (C) Ivan Kravets <me@ikravets.com>
# See LICENSE for details.

import click
import requests

from platformio import __version__
from platformio.exception import GetLatestVersionError
from platformio.util import exec_command


@click.command("upgrade",
               short_help="Upgrade PlatformIO to the latest version")
def cli():
    try:
        last = get_latest_version()
    except:
        raise GetLatestVersionError()

    if __version__ == last:
        return click.secho(
            "You're up-to-date!\nPlatformIO %s is currently the "
            "newest version available." % __version__, fg="green"
        )
    else:
        result = exec_command(["pip", "install", "--upgrade", "platformio"])
        click.secho(result['out'], fg="green")
        click.secho(result['err'], fg="red")


def get_latest_version():
    pkgdata = requests.get(
        "https://pypi.python.org/pypi/platformio/json").json()
    return pkgdata['info']['version']