gha.py

#!/usr/bin/env python

from os import environ as env, makedirs
from os.path import dirname

import click
from utz import check, process, run, err

import njsp_plots
import parse_njsp_xmls
import refresh_data


def configure_author(name, email):
    run('git', 'config', '--global', 'user.name', name)
    run('git', 'config', '--global', 'user.email', email)


@click.command()
@click.option('-b', '--branch', 'branches', multiple=True)
@click.option('-c', '--configure-author', 'do_configure_author', is_flag=True)
@click.option('-d', '--do-dispatch', is_flag=True)
@click.option('-f', '--force', count=True, help=f'Continue past initial no-op data update')
@click.option('-p', '--push', is_flag=True)
@click.option('-r/-R', '--rebase/--no-rebase', is_flag=True, default=None)
def main(branches, do_configure_author, do_dispatch, force, push, rebase):
    refresh_data.main()
    git_is_clean = check('git', 'diff', '--quiet', 'HEAD')
    if git_is_clean:
        err('No data changes found')
        if force:
            err(f'force={force}; continuing')
        else:
            return

    did_commit = False
    def commit(msg, amend=True):
        nonlocal do_configure_author, did_commit
        if do_configure_author:
            configure_author(
                env.get('GIT_AUTHOR_NAME', 'GitHub Actions'),
                env.get('GIT_AUTHOR_EMAIL', 'ryan-williams@users.noreply.github.com'),
            )
            do_configure_author = False
        if did_commit and amend:
            err('Amending commit')
            run('git', 'commit', '--amend', '-am', msg)
        else:
            run('git', 'commit', '-am', msg)
            did_commit = True

    if not git_is_clean:
        commit('GHA: update data')

    err("Calling parse_njsp_xmls.main()")
    parse_njsp_xmls.main()

    err("Calling njsp_plots.main()")
    njsp_plots.main()

    data_changed = not check('git', 'diff', '--quiet', 'HEAD', '--', 'data')
    www_changed = not check('git', 'diff', '--quiet', 'origin/www', '--', 'www')
    do_commit = data_changed or www_changed
    changed_files = [ line[3:] for line in process.lines('git', 'status', '--porcelain') ]
    if changed_files:
        err(f'{len(changed_files)} changed files:')
        for f in changed_files:
            err(f'\t{f}')

    if do_commit:
        commit('GHA: update data/plots')
    else:
        err('No data/plot change found')

    if push:
        if did_commit:
            if rebase is not False:
                # None -> merge, True -> rebase, False -> skip
                run('git', 'config', 'pull.rebase', str(bool(rebase)).lower())
                for branch in branches:
                    run('git', 'pull', 'origin', branch)

            for branch in branches:
                run('git', 'push', 'origin', f'HEAD:{branch}')

            GITHUB_OUTPUT = env.get('GITHUB_OUTPUT')
            if GITHUB_OUTPUT:
                sha = process.line('git', 'log', '-1', '--format=%h')
                with open(GITHUB_OUTPUT, 'a') as f:
                    f.write(f'sha={sha}\n')
                err(f"Wrote SHA {sha} to $GITHUB_OUTPUT")

                if do_dispatch:
                    workflow = "slack-test.yml"
                    err(f"Dispatching to {workflow}")
                    cmd = [
                        "gh", "workflow",
                        "run", workflow,
                        "-f", f"commits={sha}",
                    ]
                    process.run(cmd)
        else:
            err('Nothing to push')


if __name__ == '__main__':
    main()