update_018_elm_package.py

#! /usr/bin/env python
from __future__ import print_function

import elm_deps_upgrade as upgrader
from collections import OrderedDict
import json
import requests
import argparse
import sys
import re
import os
import glob


KNOWN_MOVES = {
    'evancz/elm-http' : 'elm-lang/http'
}


def upgrade_elm_version(version):
    return "0.18.0 <= v < 0.19.0"

def new_packages():
    r = requests.get("http://package.elm-lang.org/new-packages")
    return r.json()

def update_elm_package(root_folder, dry=False):
    with open(root_folder + '/elm-package.json') as f:
        package_data = json.load(f, object_pairs_hook=OrderedDict)

    package_data['elm-version'] = upgrade_elm_version(package_data['elm-version'])

    packages = package_data['dependencies']
    upgraded_packages = new_packages()

    notes = []
    errors = []
    upgradable_packages = {}


    for (package, version) in list(packages.items())[:]:
        if package in KNOWN_MOVES:
            new_name = KNOWN_MOVES[package]

            if new_name == '':
                errors.append('Package {name} has been removed'.format(name=package))
            else:
                notes.append('Package {name} renamed to {new_name}'.format(name=package, new_name=new_name))

                version = '1.0.0 <= v < 2.0.0'
                upgradable_packages[new_name] = version
                del package_data['dependencies'][package]

            continue

        if package not in upgraded_packages:
            errors.append('Package {name} not upgraded yet!'.format(name=package))
            continue


        upgradable_packages[package] = version

    local = upgradable_packages
    remote = upgrader.load_all_packages("0.18", "http://package.elm-lang.org/all-packages?elm-package-version=")


    upgrade_suggestions = upgrader.find_newer_versions(local, remote)

    for (dep, suggestions) in upgrade_suggestions.items():
        try:
            update_to = upgrader.newest_version(suggestions)
        except Exception as e:
            notes.append('package {} already updated'.format(dep))
            package_data['dependencies'][dep] = upgradable_packages[dep]
            continue

        notes.append('updating {} to {}'.format(dep, update_to))

        package_data['dependencies'][dep] = "{} <= v <= {}".format(update_to, update_to)


    print('=======================')
    print('Notes:')
    print('\n'.join(notes))

    print('=======================')
    print('Warnings: ')
    print('\n'.join(errors))

    print('========================\n\n')
    print(json.dumps(package_data, sort_keys=False, indent=4))

    if not dry:
        with open(root_folder + '/elm-package.json', 'w') as f:
            json.dump(package_data, f, sort_keys=False, indent=4)

    if errors:
        print('There were errors that need to be handled manually!')
        sys.exit(1)


def upgrade_elm_files(root_folder):
    with open(root_folder + '/elm-package.json') as f:
        package_data = json.load(f, object_pairs_hook=OrderedDict)

    dirs = package_data['source-directories']

    for folder in dirs:
        if folder[0] != '.':
            folder = root_folder + '/' + folder
        run_elm_format(folder)


def run_elm_format(root_folder):
    from subprocess import call
    os.chdir(root_folder)
    call(["elm-format", ".", "--upgrade", "--yes", "--elm-version=0.18"])


def run_elm_make(root_folder):
    from subprocess import call
    os.chdir(root_folder)

    call(["elm-package", "install", "--yes"])
    call(["elm-make"])


def main():

    parser = argparse.ArgumentParser(description='Automatically upgrade your package to 0.18')
    parser.add_argument('--dry', '-d', action='store_true', help='only print possible changes', default=False)

    parser.add_argument('package_dir')
    args = parser.parse_args()

    package_dir = os.path.realpath(args.package_dir)

    update_elm_package(package_dir, dry=args.dry)
    upgrade_elm_files(package_dir)
    run_elm_make(package_dir)
    run_elm_format(package_dir)


if __name__ == '__main__':
    main()