# -*- coding: utf-8 -*-
# vim:fenc=utf-8
#
# This file is part of Supysonic.
# Supysonic is a Python implementation of the Subsonic server API.
#
# Copyright (C) 2017 Alban 'spl0k' Féron
#
# Distributed under terms of the GNU AGPLv3 license.

# Converts ids from hex-encoded strings to binary data

import argparse
import sqlite3

from uuid import UUID

parser = argparse.ArgumentParser()
parser.add_argument('dbfile', help = 'Path to the SQLite database file')
args = parser.parse_args()

def process_table(connection, table, fields):
    to_update = { field: set() for field in fields }

    c = connection.cursor()
    for row in c.execute('SELECT {1} FROM {0}'.format(table, ','.join(fields))):
        for field, value in zip(fields, row):
            if value is None or not isinstance(value, basestring):
                continue
            to_update[field].add(value)

    for field, values in to_update.iteritems():
        sql = 'UPDATE {0} SET {1}=? WHERE {1}=?'.format(table, field)
        c.executemany(sql, map(lambda v: (buffer(UUID(v).bytes), v), values))

    connection.commit()

with sqlite3.connect(args.dbfile) as conn:
    conn.cursor().execute('PRAGMA foreign_keys = OFF')

    process_table(conn, 'folder', ('id', 'parent_id'))
    process_table(conn, 'artist', ('id',))
    process_table(conn, 'album', ('id', 'artist_id'))
    process_table(conn, 'track', ('id', 'album_id', 'artist_id', 'root_folder_id', 'folder_id'))
    process_table(conn, 'user', ('id', 'last_play_id'))
    process_table(conn, 'client_prefs', ('user_id',))
    process_table(conn, 'starred_folder', ('user_id', 'starred_id'))
    process_table(conn, 'starred_artist', ('user_id', 'starred_id'))
    process_table(conn, 'starred_album', ('user_id', 'starred_id'))
    process_table(conn, 'starred_track', ('user_id', 'starred_id'))
    process_table(conn, 'rating_folder', ('user_id', 'rated_id'))
    process_table(conn, 'rating_track', ('user_id', 'rated_id'))
    process_table(conn, 'chat_message', ('id', 'user_id'))
    process_table(conn, 'playlist', ('id', 'user_id'))