#!/usr/bin/python3
#nuda: a minimal, command-line managed, ascii database for organizing pictures with dates and tags
#Copyright 2018 Randall Evan McClellan

#This file is part of nuda.
#
#    nuda is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License as published by
#    the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.
#
#    nuda is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with nuda.  If not, see <http://www.gnu.org/licenses/>.

import sys, os
import tkinter as tk
import pickle
import importClass as imp 
import slideShowClass as ssc

if sys.argv[1] == "init":
    print("Initializing nuda into "+os.getcwd())
    os.system("mkdir ./nudaDir/")
    os.system("mkdir ./inbox/")
    os.system("mkdir ./search/")
    os.system("mkdir ./inbox/imported/")
    os.system("mkdir ./inbox/skipped/")
    os.system("mkdir ./flask/")
    os.system("mkdir ./flask/static/")
    os.system("mkdir ./flask/templates/")
    installDir = os.path.join(os.path.split(os.path.realpath("/bin/nuda"))[0])+'/'
    os.system("ln -s "+installDir+"flask/app.py ./flask/app.py")
    os.system("ln -s ../../nudaDir/ ./flask/static/nudaDir")
    os.system("ln -s ../tags.pickle ./flask/tags.pickle")
    os.system("ln -s ../tags.pickledhash ./flask/tags.pickledhash")
    os.system("ln -s "+installDir+"flask/static/search.css ./flask/static/search.css")
    os.system("ln -s "+installDir+"flask/static/home.css ./flask/static/home.css")
    os.system("ln -s "+installDir+"flask/static/nuda-icon.svg ./flask/static/nuda-icon.svg")
    os.system("ln -s "+installDir+"flask/templates/search.html ./flask/templates/search.html")
    os.system("ln -s "+installDir+"flask/templates/home.html ./flask/templates/home.html")
    if not os.path.exists(imp.NUDATABLE):
        print("Creating "+imp.NUDATABLE)
        with open(imp.NUDATABLE, 'w') as table:
            table.write("#filename\tpath\tdate\ttime\ttags\n")

if sys.argv[1] == "install":
    print("Installing nuda into "+os.getcwd())
    os.system("mkdir ./nudaDir/")
    os.system("mkdir ./inbox/")
    os.system("mkdir ./search/")
    os.system("mkdir ./inbox/imported/")
    os.system("mkdir ./inbox/skipped/")
    if not os.path.exists(imp.NUDATABLE):
        print("Creating "+imp.NUDATABLE)
        with open(imp.NUDATABLE, 'w') as table:
            table.write("#filename\tpath\tdate\ttime\ttags\n")
    if len(sys.argv) > 2:
        if sys.argv[2] == "-f":
            print("Forcing")
            os.system("sudo ln -sf "+os.getcwd()+"/nuda.py /bin/nuda")
    else:
        os.system("sudo ln -s "+os.getcwd()+"/nuda.py /bin/nuda")

else:
    if os.path.exists(imp.NUDADIR) and os.path.exists(imp.NUDATABLE):
        pass
    else:
        print("Current directory, "+os.getcwd()+", is not initialized as a nuda home directory.")
        sys.exit()


if sys.argv[1] == "tags":
    if len(sys.argv) > 2:
        selectList = sys.argv[2:]
        prefix = '-'.join(selectList)
        #check if subnuda flask exists, create it if not
        subnudaFlaskName = os.getcwd()+'/'+prefix+'.flask'
        if os.path.exists(subnudaFlaskName):
            os.system("rm -r "+subnudaFlaskName+'/static/nudaDir/*')
        else:
            os.system("mkdir "+subnudaFlaskName)
            os.system("mkdir "+subnudaFlaskName+'/static')
            os.system("mkdir "+subnudaFlaskName+'/static/nudaDir')
            os.system("mkdir "+subnudaFlaskName+'/templates')
            installDir = os.path.join(os.path.split(os.path.realpath("/bin/nuda"))[0])+'/'
            os.system("ln -s ../"+prefix+".pickle "+subnudaFlaskName+"/tags.pickle")
            os.system("ln -s ../"+prefix+".pickledhash "+subnudaFlaskName+"/tags.pickledhash")
            os.system("ln -s "+installDir+"flask/app.py "+subnudaFlaskName+"/app.py")
            os.system("ln -s "+installDir+"flask/static/search.css "+subnudaFlaskName+"/static/search.css")
            os.system("ln -s "+installDir+"flask/static/home.css "+subnudaFlaskName+"/static/home.css")
            os.system("ln -s "+installDir+"flask/static/nuda-icon.svg "+subnudaFlaskName+"/static/nuda-icon.svg")
            os.system("ln -s "+installDir+"flask/templates/search.html "+subnudaFlaskName+"/templates/search.html")
            os.system("ln -s "+installDir+"flask/templates/home.html "+subnudaFlaskName+"/templates/home.html")
    else:
        selectList = []
        prefix = 'tags'
    tagDict = {}
    hashDict = {}

    with open(imp.NUDATABLE, 'r') as table:
        for line in table:
            keepBool = True
            if line[0] == '#':
                continue
            else:
                fname, path, date, time, tags = line.split('\t')
                tagList = tags.rstrip().split(',')
                for sel in selectList:
                    if sel not in tagList:
                       keepBool = False 
                if "nsfw" not in tagList and keepBool:
                    if len(sys.argv) > 2:
                        if not os.path.exists(subnudaFlaskName+"/static/nudaDir/"+date[:-3]):
                            os.system("mkdir "+subnudaFlaskName+"/static/nudaDir/"+date[:-3])
                        os.system("ln -s ../../../../nudaDir/"+date[:-3]+"/"+fname+" "+subnudaFlaskName+"/static/nudaDir/"+date[:-3]+"/"+fname)
                    for tag in tagList:
                        tagDict.setdefault(tag, []).append(path+fname)
                    hashDict[path+fname] = tags[:-1]
    print(f"Pickling dictionary of {len(tagDict)} tags.")
    with open(f"{prefix}.pickle","wb") as pickleFile:
        pickle.dump(tagDict, pickleFile, protocol=pickle.HIGHEST_PROTOCOL)
    with open(f"{prefix}.pickledhash","wb") as hashFile:
        pickle.dump(hashDict, hashFile, protocol=pickle.HIGHEST_PROTOCOL)

if sys.argv[1] == "search":
    tagList = sys.argv[2:]
    print('Searching for tag: ',tagList)
    imageList = ssc.getImagesMatchingTags(tagList)
    os.system("rm "+imp.NUDADIR+"../search/*")
    for result in imageList:
        filename = result.split('/')[-1]
        os.system("ln "+imp.NUDADIR+"../"+result+" "+imp.NUDADIR+"../search/"+filename)
        with open(imp.NUDATABLE, 'r') as dbfile:
            for line in dbfile:
                if line[:len(filename)] == filename:
                    print(line.rstrip())

if sys.argv[1] == "slideshow":
    if len(sys.argv) == 2:
        pickleFileName = "./tags.pickle"
    elif len(sys.argv) == 3:
        pickleFileName = sys.argv[2]
    else:
        print("Unknown input for slideshow.")

    tagList = []
    imageList = ssc.getFilesMatchingAllTags(tagList, pickleFileName)

    #initialize tk window
    slideshow = tk.Tk()
    slideshow.attributes('-type','dialog')
    my_slideshow = ssc.slideShowClass(slideshow, imageList, pickleFileName)
    slideshow.mainloop()

if sys.argv[1] == "reset":
    try:
        value = os.environ['ALLOWNUDARESET']
    except:
        value = "False"
    if value == 'True':
        yesorno = input("Really reset the entire DB? ")
        if yesorno in ['yes', 'y', 'Y', 'Yes', 'YES']:
            print("Moving all imported files back to ./inbox/")
            os.system("mv ./inbox/imported/* ./inbox/")
            print("Moving all skipped files back to ./inbox/")
            os.system("mv ./inbox/skipped/* ./inbox/")
            print("Wiping nudaDir")
            os.system("rm ./nudaDir/*/*")
            os.system("rmdir ./nudaDir/*")
            print("Wiping nudaTable.txt")
            os.system("rm "+imp.NUDATABLE)
            if not os.path.exists(imp.NUDATABLE):
                print("Creating "+imp.NUDATABLE)
                with open(imp.NUDATABLE, 'w') as table:
                    table.write("#filename\tpath\tdate\ttime\ttags\n")
    else:
        print("nuda reset disabled!")

if sys.argv[1] == "import":
    if len(sys.argv) == 2:
        inFileNames = ['./inbox/'+f for f in os.listdir(os.getcwd()+'/inbox/') if os.path.isfile(os.getcwd()+'/inbox/'+f)]
    else:
        inFileNames = sys.argv[2:]
    importGUI = tk.Tk()
    importGUI.attributes('-type','dialog')
    my_importGUI = imp.importClass(importGUI, inFileNames)
    importGUI.mainloop()

if sys.argv[1] == "merge":
    if len(sys.argv) == 3:
        thatDir = sys.argv[2]
        thisDir = os.getcwd()
    else:
        print("Must specify target directory to merge FROM.")
        sys.exit()
    print(f"Merging thatDir = {thatDir}")
    print(f"into thisDir = {thisDir}")
    with open(thatDir+'/nudaTable.txt', 'r') as thatTable:
        with open(thisDir+'/nudaTable.txt', 'a') as thisTable:
            for line in thatTable:
                if line[0] == '#':
                    continue
                else:
                    fname, path, date, time, tags = line.split('\t')
                    maybeFile = os.path.abspath(os.getcwd()+path[1:]+fname)
                    if os.path.isfile(maybeFile):
                        #print(f"{path[1:]+fname} already here!")
                        continue
                    else:
                        print(f"{path[1:]+fname} is new! Merging here.")
                        thisTable.write(line)
                        if not os.path.exists(thisDir+path[1:]):
                            os.system(f"mkdir {thisDir+path[1:]}")
                        os.system(f"cp -n {thatDir+path[1:]+fname} {thisDir+path[1:]+fname}")