#!/usr/bin/env python
import sys, os, plist, shutil, tempfile, subprocess

def _decode(value, encoding="utf-8", errors="ignore"):
 # Helper method to only decode if bytes type
 if sys.version_info >= (3,0) and isinstance(value, bytes):
 return value.decode(encoding,errors)
 return value

def main():
 # Let's check for an existing app - remove it if it exists,
 # then create and format a new bundle
 temp = None
 print("Getting python 2 path...")
 p = subprocess.Popen(["which","python"], shell=False, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
 c = p.communicate()
 pypath = _decode(c[0]).split("\n")[0]
 if not len(pypath):
 print(" - Python not found in PATH! Aborting...")
 exit(1)
 print(" - {} ".format(pypath))
 print("Checking for existing ProperTree.app...")
 os.chdir(os.path.dirname(os.path.realpath(__file__)))
 os.chdir("../")
 if os.path.exists("ProperTree.app"):
 print(" - Found, removing...")
 if os.path.exists("ProperTree.app/Contents/MacOS/Scripts/settings.json"):
 print(" --> Found settings.json - preserving...")
 temp = tempfile.mkdtemp()
 shutil.copy("ProperTree.app/Contents/MacOS/Scripts/settings.json", os.path.join(temp, "settings.json"))
 shutil.rmtree("ProperTree.app")
 # Make the directory structure
 print("Creating bundle structure...")
 os.makedirs("ProperTree.app/Contents/MacOS/Scripts")
 os.makedirs("ProperTree.app/Contents/Resources")
 print("Copying scripts...")
 print(" - ProperTree.command")
 with open("ProperTree.command","r") as f:
 ptcom = f.read().split("\n")
 if ptcom[0].startswith("#!"):
 # Got a shebang - remove it
 ptcom.pop(0)
 # Set the new shebang
 ptcom.insert(0,"#!{}".format(pypath))
 with open("ProperTree.app/Contents/MacOS/ProperTree.command","w") as f:
 f.write("\n".join(ptcom))
 # chmod +x
 p = subprocess.Popen(["chmod","+x","ProperTree.app/Contents/MacOS/ProperTree.command"], shell=False, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
 c = p.communicate()
 # Copy everything else
 for x in os.listdir("Scripts"):
 if x.startswith(".") or not x.lower().endswith((".py",".plist",".icns")):
 continue
 print(" - "+x)
 target = "ProperTree.app/Contents/Resources" if x.lower().endswith(".icns") else "ProperTree.app/Contents/MacOS/Scripts"
 shutil.copy(os.path.join("Scripts",x),target)
 print("Building Info.plist...")
 info = {
 "CFBundleShortVersionString": "0.0", 
 "CFBundleSignature": "????", 
 "CFBundleInfoDictionaryVersion": "0.0", 
 "CFBundleIconFile": "icon", 
 "NSHumanReadableCopyright": "Copyright 2019 CorpNewt", 
 "CFBundleIconFile": "shortcut.icns",
 "CFBundleIdentifier": "com.corpnewt.ProperTree", 
 "CFBundleDocumentTypes": [
 {
 "CFBundleTypeName": "Property List", 
 "CFBundleTypeRole": "Viewer", 
 "CFBundleTypeIconFile": "plist", 
 "CFBundleTypeExtensions": [
 "plist"
 ]
 }
 ], 
 "CFBundleDevelopmentRegion": "English", 
 "CFBundleExecutable": "ProperTree.command", 
 "CFBundleName": "ProperTree", 
 "LSMinimumSystemVersion": "10.4",
 "LSMultipleInstancesProhibited": True,
 "CFBundlePackageType": "APPL", 
 "CFBundleVersion": "0.0"
 }
 with open("ProperTree.app/Contents/Info.plist","wb") as f:
 plist.dump(info,f)
 if temp:
 print("Restoring settings.json...")
 shutil.copy(os.path.join(temp, "settings.json"),"ProperTree.app/Contents/MacOS/Scripts/settings.json")
 shutil.rmtree(temp,ignore_errors=True)

if __name__ == '__main__':
 if not str(sys.platform) == "darwin":
 print("Can only be run on macOS")
 exit(1)
 try:
 main()
 except Exception as e:
 print("An error occurred!")
 print(str(e))
 exit(1)