#!/usr/bin/env python
# coding:utf-8
"""
  Author:  Sir Garbagetruck --<truck@notonfire.somewhere>
  Purpose: make editing the list.sahli file easier
  Created: 2020/04/09
"""

import json
import argparse
import os
from sauce import SAUCE
from sahliEditorPython import sahlifile as SF


def getfilesindir(directory):
    """return the files in a directory as an array"""
    for root, dirs, files, rootfd in os.fwalk(directory):
        return files


def getfilenames(filedata):
    """return the file names from a sahli filedata array"""
    f = []
    for i in filedata:
        f.append(i['file'])
    return f


def getdata(filedata, name):
    """get the filedata entry where file = name"""
    for i in filedata:
        if i['file'] == name:
            return i
    return []


def getpicdata(filename):
    """extract picture data from filename"""
    return 'todo.txt'


def getansidata(filename):
    """extract SAUCE data from filename"""
    saucedata = SAUCE(filename)
    ansidata = {
        'author': saucedata.author,
        'group': saucedata.group,
        'title': saucedata.title,
        'filesize': saucedata.filesize,
        'comments': saucedata.comments
    }
    tinfonames = [saucedata.tinfo1_name,
                  saucedata.tinfo2_name,
                  saucedata.tinfo3_name,
                  saucedata.tinfo4_name]
    tinfo = [saucedata.tinfo1,
             saucedata.tinfo2,
             saucedata.tinfo3,
             saucedata.tinfo4]
    for i in range(0, 3):
        if tinfonames[i] == 'width':
            ansidata['width'] = tinfo[i]
        if tinfonames[i] == 'height':
            ansidata['height'] = tinfo[i]
#        print(tinfonames[i])
    return ansidata


def getamigadata(filename):
    """try to get some form of info from file (:"""
    return 'todo.txt'


def main(args):
    """maintain a list.sahli file"""
    if args.new:
        mysahli = SF.sahlifile(None)
    else:
        mysahli = SF.sahlifile(args.filename)
    mysahli.sahli['location'] = args.directory
    files = getfilesindir(args.directory)
    filedata = mysahli.sahli['filedata']
    filedatanames = getfilenames(filedata)
    newdata = []
    for i in files:
        dirfile = '{}/{}'.format(args.directory, i)
        if i in filedatanames:
            print('found! {}'.format(i))
            a = getansidata(dirfile)
            newdata.append(getdata(filedata, i))
        else:
            print('not found! {}'.format(i))
            suf = i.split('.')[-1]
            if suf in ['png', 'jpg', 'jpeg', 'gif',
                       'PNG', 'JPG', 'JPEG', 'GIF']:
                stuff = getpicdata(dirfile)
            elif suf in ['ans', 'ANS', 'BIN', 'bin', 'XB', 'xb']:
                stuff = getansidata(dirfile)
                a = 5
            elif suf in ['TXT', 'ASC', 'txt', 'asc',
                         'NFO', 'nfo', 'diz', 'DIZ']:
                stuff = getamigadata(dirfile)
            else:
                print("dunno what type of file this is...")
    a = 5


if __name__ == '__main__':
    ap = argparse.ArgumentParser()
    ap.add_argument('-f', '--filename', default='list.sahli')
    ap.add_argument('-n', '--new', action='store_true')
    ap.add_argument('-d', '--directory', type=str, required=True,
                    help='directory where compo files are')
    main(ap.parse_args())