#!/usr/bin/env python
# -*- coding: utf-8 -*-

'''
Install PythonTeX

This script installs one set of PythonTeX scripts for use under Python 2.6-2.7 
(names end in "2") and another set for user under Python 3 (names end in "3").
You will need to launch the correct script depending on the default version of
Python on your system.

This script should work with most TeX distributions.  It is primarily written 
for TeX Live.  It should work with other TeX distributions that use the 
Kpathsea library (such as MiKTeX), though with reduced functionality in some 
cases.  It should work with additional distributions as well, but will require 
manual input when Kpathsea is not present.  When the script cannot find TeX 
Live-style functionality, it alerts the user and attempts to proceed.

The script will overwrite (and thus update) all previously installed PythonTeX 
files.  During automated installation (Kpathsea), files are NOT installed in 
the LOCAL texmf tree, because we want auto-updating once PythonTeX is added to 
CTAN.  The texhash command (if present) is executed to make the system aware 
of any new files.

Copyright (c) 2012, Geoffrey M. Poore
All rights reserved.
Licensed under the BSD 3-Clause License:
    http://www.opensource.org/licenses/BSD-3-Clause

'''


# Imports
import sys
import platform
from os import path, mkdir
from subprocess import check_call, check_output, CalledProcessError
from shutil import copy


# We need a version of input that works under both Python 2 and 3
try:
    input = raw_input
except:
    pass


# Make sure all necessary files are present
needed_files = ['pythontex2.py', 'pythontex_types2.py', 'pythontex_utils2.py', 
                'pythontex3.py', 'pythontex_types3.py', 'pythontex_utils3.py', 
                'pythontex.sty', 'pythontex.ins', 'pythontex.dtx', 
                'pythontex.pdf', 'README.rst']
missing_files = False
# Print a list of all files that are missing, and exit if any are
for eachfile in needed_files:
    if not path.exists(eachfile):
        print('Could not find file ' + eachfile)
        missing_files = True
if missing_files:
    print('Exiting.')
    sys.exit(1)


# Print starting message
print('\nInstalling PythonTeX...')


# Retrieve the location of a valid TeX tree
# Attempt to use kpsewhich; otherwise, resort to manual input 
try:
    if sys.version_info[0] == 2:
        texmf_path = check_output(['kpsewhich', '-var-value', 'TEXMFDIST']).rstrip('\r\n')
    else:
        texmf_path = check_output(['kpsewhich', '-var-value', 'TEXMFDIST']).decode('utf-8').rstrip('\r\n')
except OSError:
    print('\nYour system appears to lack kpsewhich.')
    print('Cannot automatically find a valid texmf path.')
    texmf_path = input('Please enter a valid texmf path: ').rstrip('\r\n')
except CalledProcessError:
    print('\nkpsewhich is not happy with its arguments.')
    print('Cannot automatically find a valid texmf path.')
    texmf_path = input('Please enter a valid texmf path: ').rstrip('\r\n')
# Make sure path slashes are compatible with the operating system 
# This is only needed for Windows
texmf_path = path.normcase(texmf_path)
# Check to make sure the path is valid 
# This is only really needed for manual input 
# The '' check is for empty manual input
if texmf_path == '' or not path.exists(texmf_path):
    print('Invalid texmf path.  Exiting.')
    sys.exit(1)
# Now check that all other needed paths are present
doc_path = path.join(texmf_path, 'doc', 'latex')
package_path = path.join(texmf_path, 'tex', 'latex')
scripts_path = path.join(texmf_path, 'scripts')
source_path = path.join(texmf_path, 'source', 'latex')
missing_paths = False
for eachpath in [doc_path, package_path, scripts_path, source_path]:
    if not path.exists(eachpath):
        print('Could not find path ' + eachpath)
        missing_paths = True
if missing_paths:
    print('Exiting.')
    sys.exit(1)
# Modify the paths by adding the pythontex directory, which will be created
doc_path = path.join(doc_path, 'pythontex')
package_path = path.join(package_path, 'pythontex')
scripts_path = path.join(scripts_path, 'pythontex')
source_path = path.join(source_path, 'pythontex')


#Install docs
if not path.exists(doc_path):
    mkdir(doc_path)
copy('pythontex.pdf', doc_path)
copy('README.rst', doc_path)
# Install package
if not path.exists(package_path):
    mkdir(package_path)
copy('pythontex.sty', package_path)
# Install scripts
if not path.exists(scripts_path):
    mkdir(scripts_path)
copy('pythontex2.py', scripts_path)
copy('pythontex_types2.py', scripts_path)
copy('pythontex_utils2.py', scripts_path)
copy('pythontex3.py', scripts_path)
copy('pythontex_types3.py', scripts_path)
copy('pythontex_utils3.py', scripts_path)
# Install source
if not path.exists(source_path):
    mkdir(source_path)
copy('pythontex.ins', source_path)
copy('pythontex.dtx', source_path)


# Install "binaries" or suggest the cretaion of binaries/batch files/symlinks
# This part is operating-system dependent
#
# If under Windows, we create a binary wrapper if under TeX Live and otherwise
# alert the user regarding the need for a wrapper or batch file
if platform.system() == 'Windows':
    # Assembly the binary path, assuming TeX Live
    # The directory bin/ should be at the same level as texmf
    bin_path = path.join(path.split(texmf_path)[0], 'bin', 'win32') 
    if path.exists(path.join(bin_path, 'runscript.exe')):
        print('\nCreating binary wrapper...')        
        copy(path.join(bin_path, 'runscript.exe'), path.join(bin_path, 'pythontex2.exe'))
        copy(path.join(bin_path, 'runscript.exe'), path.join(bin_path, 'pythontex3.exe'))
    else:
        print('\nCould not create a wrapper for launching pythontex.py.')
        print('You will need to create a wrapper manually, or use a batch file.')
        print('The wrapper or batch file should be placed somewhere on the Windows PATH.')
        print('The bin/ directory in your TeX distribution may be a good location.')
        print('The script pythontex.py is located in the following directory:')
        print('    ' + scripts_path)
# If not under Windows, we alert the user regarding what is necessary to launch
# pythontex.py
else:
    print('\nYou may wish to create a symlink to pythontex.py.')
    print('You may also want to make it executable via chmod.')
    print('The script pythontex.py is located in the following directory:')
    print('    ' + scripts_path)


# Alert TeX to the existence of the package via texhash
print('\nRunning texhash...')
try:
    check_call(['texhash'])
except OSError:
    print('Could not run texhash.  Your system appears to lack texhash.')
    print('Your system may not be aware of newly installed files.')


# Alert the user to the need to choose a version
print('\n\n* * *')
print('PythonTeX contains separate scripts for Python 2 and Python 3.')
print('Choose the correct scripts based on your Python installation.')
print('See the documentation for more information.')
print('* * *\n')

# Pause so that the user can see any errors or other messages
input('\n[Press ENTER to exit]')