Lark/lark

#!/usr/bin/python3
"""
lark
Verify and sort game ROM images.

Intended features:
    DAT downloading
    File validation
    File renaming/moving
    Nice Beets-inspired UI.
    Release grouping (maybe, this might require another large external database)

UI notes

# Key terms
    - hash          Unique identifier for each ROM image.
    - image         ROM image, ripped from physical media.
    - dat           List of hashes, with associated filenames.
    - platform      The original hardware on which the image was intended to run.

# Verbs
    - list [hash, dat, platform, image]
        List items in the database.

    - import [datfile, imagefile]
        Process and add external items to the database.

    - add [platform, hash]
        Manually add items to the database.

    - remove [hash, dat, platform]
        Delete items from the database.
"""
# TODO: Write decent UI
import hashlib
import sys
import os
import uuid
import xdg.BaseDirectory

#import dat
import metadatadb

HASH_CHUNK_SIZE = 10485760 # 10mb
SQLITE_FILENAME = 'metadata.db'

data_path = os.path.join(xdg.BaseDirectory.xdg_data_home, 'lark')

def get_sha1sum(filename):
    sha1sum = hashlib.sha1()
    with open(filename, 'rb') as file_contents:
        while True:
            chunk = file_contents.read(HASH_CHUNK_SIZE)
            if not chunk:
                break
            sha1sum.update(chunk)

    return sha1sum.hexdigest()


'''
smd_dat = dat(SMD_DAT_PATH)
# TODO: Default to '.'
# TODO: Use a proper arg parser.
search_dir = sys.argv[1]
for filename in os.listdir(search_dir):
    # TODO: Ignore or descend into directories.
    # TODO: Compare hashes
    file_path = os.path.abspath(os.path.join(search_dir, filename))
    file_sha1 = get_sha1sum(file_path)
    search_result = smd_dat.search_by_sha1(file_sha1)
    if search_result:
        rom_data = search_result[0]
        print('File %s matches database entry for %s.' % (filename, rom_data.filename))
    else:
        print('File %s is not in database.' % filename)
'''
# Test code! :D
# TODO: Write test code that doesn't depend on external resources.
SMD_DAT_PATH = '/home/lumia/Downloads/Sega - Mega Drive - Genesis (20200303-035539).dat'
TEST_HASH = 'cfbf98c36c776677290a872547ac47c53d2761d6'

def _kwargs_parse(kwargs_list):
    kwargs = {}
    for kwarg_string in kwargs_list:
        key, value = kwarg_string.split('=')
        kwargs[key] = value

    return kwargs

action_object = sys.argv[1]
action = sys.argv[2]

metadatadb.configure(os.path.join(data_path, SQLITE_FILENAME))

# TODO: Use a real UI library. This mess is just intended for development.
if action_object == 'platform':
    if action == 'add':
        print('add a platform')
        platform_shortcode = sys.argv[3]
        platform_name = sys.argv[4]

        platform_data = metadatadb.Platform(shortcode=platform_shortcode,
                                            fullname=platform_name)
        with metadatadb.get_db_session() as session:
            session.add(platform_data)

    elif action == 'list':
        # TODO: Filter support is exclusively limited to SQLAlchemy's filter.ilike function. Figure
        # out a good way to include other filters.
        filters = _kwargs_parse(sys.argv[3:])
        with metadatadb.get_db_session() as session:
            print(metadatadb.search(session, metadatadb.Platform, **filters))

    elif action == 'remove':
        constraints = sys.argv[3:]
        filters = _kwargs_parse(sys.argv[3:])

        with metadatadb.get_db_session() as session:
            platforms = metadatadb.search(session, metadatadb.Platform, **filters)
            for platform in platforms:
                print('Removing %s.' % platform.fullname)
                session.delete(platform)

    elif action == 'test':
        # TODO: Delete this action before merging into dev. It's just for ugly testing.
        platform_shortcode = sys.argv[3]
        platform_name = sys.argv[4]

        platform_data = metadatadb.Platform(shortcode=platform_shortcode,
                                            fullname=platform_name)

        with metadatadb.get_db_session() as session:
            print(metadatadb.search(session, metadatadb.Platform))

elif action_object == 'release-group':
    if action == 'add':
        properties = _kwargs_parse(sys.argv[3:])
        release_group = metadatadb.ReleaseGroup(name=properties['name'])

        with metadatadb.get_db_session() as session:
            if properties['platform']:
                platform = metadatadb.search(session, metadatadb.Platform,
                                             shortcode=properties['platform'])[0]
                release_group.platform = platform
            session.add(release_group)

    if action == 'list':
        # TODO: Filter support is exclusively limited to SQLAlchemy's filter.ilike function. Figure
        # out a good way to include other filters.
        print('Listing release groups.')
        filters = _kwargs_parse(sys.argv[3:])
        with metadatadb.get_db_session() as session:
            print(metadatadb.search(session, metadatadb.ReleaseGroup, **filters))

    elif action == 'remove':
        constraints = sys.argv[3:]
        filters = _kwargs_parse(sys.argv[3:])

        with metadatadb.get_db_session() as session:
            release_groups = metadatadb.search(session, metadatadb.ReleaseGroup, **filters)
            for release_group in release_groups:
                print('Removing %s.' % release_group.name)
                session.delete(release_group)

elif action_object == 'image':
    if action == 'add':
        properties = _kwargs_parse(sys.argv[3:])
        image = metadatadb.Image(filename=properties['filename'],
                                 sha1sum=properties['sha1sum'])

        with metadatadb.get_db_session() as session:
            if properties['release-group']:
                release_group = metadatadb.search(session, metadatadb.ReleaseGroup,
                                                  name=properties['release-group'])[0]
                image.release_group = release_group
            session.add(image)

    if action == 'list':
        # TODO: Filter support is exclusively limited to SQLAlchemy's filter.ilike function. Figure
        # out a good way to include other filters.
        print('Listing release groups.')
        filters = _kwargs_parse(sys.argv[3:])
        with metadatadb.get_db_session() as session:
            print(metadatadb.search(session, metadatadb.Image, **filters))

    elif action == 'remove':
        constraints = sys.argv[3:]
        filters = _kwargs_parse(sys.argv[3:])

        with metadatadb.get_db_session() as session:
            release_groups = metadatadb.search(session, metadatadb.Image, **filters)
            for image in release_groups:
                print('Removing %s.' % image.name)
                session.delete(image)
else:
    print('Unknown object.')