uhist

#!/usr/bin/env python3
import argparse
from collections import Counter
from os.path import expanduser

# Commands like 'git' where 'git add' and 'git rebase' should be treated as
# distinctly different.
MULTI_COMMAND_COMMANDS = {
    "cargo",
    "docker",
    "git",
}

COMMAND_WRAPPERS = {
    "notify",
    "sudo",
    "time",
}

BASH_HISTORY = expanduser("~/.bash_history")


def get_subcommands(command):
    """Takes the root, or root + subcommand if indicated by MULTI_COMMAND_COMMANDS."""
    if not command:
        return ""

    # Strip off the wrapper
    if command[0] in COMMAND_WRAPPERS and len(command) >= 2:
        command = command[1:]

    root = command[0]

    # Concatenate any subcommands
    if root in MULTI_COMMAND_COMMANDS and len(command) >= 2:
        root = root + " " + command[1]
    return root


def main(n):
    """Analyze and print statistics about ~/.bash_history"""
    with open(BASH_HISTORY, "r") as history:
        history = Counter(get_subcommands(c.split()) for c in history)
        total = sum(history.values())

        for command, count in history.most_common(n):
            print("\t{}\t{:05.2f}%\t{}".format(count, (count / total) * 100, command))


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-n",
        "--number",
        default=10,
        type=int,
        help="The number of most-common items to show",
    )

    args = parser.parse_args()
    main(args.number)