#!/usr/bin/env python3

"""
flacinfo

A script analoguous to `mp3info`, allowing one to easily tag their music
collection, but for flac files.
"""

import argparse
import subprocess
import os
import sys


VORBIS_ARG_NAME = {
    'artist': 'ARTIST',
    'comment': 'COMMENT',
    'genre': 'GENRE',
    'album': 'ALBUM',
    'albumnumber': 'DISCNUMBER',
    'track': 'TRACKNUMBER',
    'title': 'TITLE',
    'year': 'DATE',
}


def argparser():
    """ Parses the arguments from sys.argv """
    parser = argparse.ArgumentParser(
        description="Edit flac files' metadata")
    parser.add_argument('-a', '--artist',
                        help="Specify artist name")
    parser.add_argument('-c', '--comment',
                        help="Specify an arbitrary comment")
    parser.add_argument('-g', '--genre',
                        help="Specify genre (in plain text)")
    parser.add_argument('-l', '--album',
                        help="Specify album name")
    parser.add_argument('-m', '--albumnumber',
                        help="Specify album number")
    parser.add_argument('-n', '--track',
                        help="Specify track number")
    parser.add_argument('-t', '--title',
                        help="Specify track title")
    parser.add_argument('-y', '--year',
                        help="Specify album year")
    parser.add_argument('file', nargs='+', metavar='FILE',
                        help="The file(s) to work on")
    return parser.parse_args()


def isFlacFile(path):
    """ Checks whether `path` refers to an existing, writeable flac file """
    if not os.path.isfile(path) or not os.access(path, os.W_OK):
        return False
    flacRun = subprocess.run(['metaflac', '--list', path],
                             stdout=subprocess.DEVNULL,
                             stderr=subprocess.DEVNULL)
    if flacRun.returncode != 0:
        return False  # Metaflac failed to list the files' metadata

    return True


def makeMetaflacArgs(inArgs):
    outArgs = []

    for arg in inArgs:
        argVal = inArgs[arg]
        if arg not in VORBIS_ARG_NAME or argVal is None:
            continue
        argName = VORBIS_ARG_NAME[arg]

        outArgs.append('--remove-tag={}'.format(argName))
        outArgs.append("--set-tag={}={}".format(argName, argVal))

    return outArgs


def main():
    args = argparser()

    hasErrors = False
    for cFile in args.file:
        if not isFlacFile(cFile):
            print(("Error: file {} does not exist, or is not writeable by "
                   "metaflac").format(cFile),
                  file=sys.stderr)
            hasErrors = True
    if hasErrors:
        print("One or more file cannot be manipulated. Aborting.",
              file=sys.stderr)
        sys.exit(1)

    metaflacArgs = makeMetaflacArgs(vars(args))
    metaflacArgs += args.file
    metaflacArgs.insert(0, 'metaflac')
    metaflacRun = subprocess.run(metaflacArgs)

    if metaflacRun.returncode != 0:
        print(('\n\nMetaflac exited with return code {}. There was an error '
               'during the execution, you should look at the output to '
               'investigate it.').format(metaflacRun.returncode),
              file=sys.stderr)
        sys.exit(metaflacRun.returncode)


if __name__ == '__main__':
    main()