codecombat/scripts/devSetup/node.py

from __future__ import print_function
__author__ = u'schmatz'
from downloader import Downloader
import tarfile
import errors
from errors import DownloadCorruptionError
import warnings
import os
from configuration import Configuration
from dependency import Dependency
import shutil
from which import which
import subprocess
from stat import S_IRWXU,S_IRWXG,S_IRWXO
import sys

if sys.version_info.major >= 3:
    raw_input = input 

class Node(Dependency):
    def __init__(self,configuration):
        super(self.__class__, self).__init__(configuration)
        operating_system = configuration.system.operating_system
        self.config.directory.create_directory_in_tmp(u"node")
        if operating_system == u"mac":
            self.downloader = MacNodeDownloader(self)
        elif operating_system == u"win":
            self.downloader = WindowsNodeDownloader(self)
            self.config.directory.create_directory_in_bin(u"node") #TODO: Fix windows node installation
        elif operating_system == u"linux":
            self.downloader = LinuxNodeDownloader(self)
    @property
    def tmp_directory(self):
        return self.config.directory.tmp_directory
    @property
    def bin_directory(self):
        return self.config.directory.bin_directory

    def download_dependencies(self):
        install_directory = self.config.directory.bin_directory + os.sep + u"node"
        if os.path.exists(install_directory):
            print(u"Skipping Node download because " + install_directory + " exists.")
        else:
            self.downloader.download()
            self.downloader.decompress()
    def bashrc_string(self):
        return "COCO_NODE_PATH=" + self.config.directory.bin_directory + os.sep + u"node" + os.sep + "bin" + os.sep +"node"
    def install_dependencies(self):
        install_directory = self.config.directory.bin_directory + os.sep + u"node"
        #check for node here
        if self.config.system.operating_system in ["mac","linux"] and not which("node"):
            unzipped_node_path = self.findUnzippedNodePath()
            print("Copying node into /usr/local/bin/...")
            shutil.copy(unzipped_node_path + os.sep + "bin" + os.sep + "node","/usr/local/bin/")
            os.chmod("/usr/local/bin/node",S_IRWXG|S_IRWXO|S_IRWXU)
        if os.path.exists(install_directory):
            print(u"Skipping creation of " + install_directory + " because it exists.")
        else:
            unzipped_node_path = self.findUnzippedNodePath()
            shutil.copytree(self.findUnzippedNodePath(),install_directory)
        wants_to_upgrade = True
        if self.check_if_executable_installed(u"npm"):
            warning_string = u"A previous version of npm has been found. \nYou may experience problems if you have a version of npm that's too old. Would you like to upgrade?(y/n) "
            from distutils.util import strtobool
            print(warning_string)
            #for bash script, you have to somehow redirect stdin to raw_input()
            user_input = raw_input()
            while True:
                try:
                    wants_to_upgrade = strtobool(user_input)
                except:
                    print(u"Please enter y or n. ")
                    user_input = raw_input()
                    continue
                break
        if wants_to_upgrade:
            if sys.version_info.major < 3:
                import urllib2, urllib
            else:
                import urllib.request as urllib
            print(u"Retrieving npm update script...")
            npm_install_script_path  = install_directory + os.sep + u"install.sh"
            urllib.urlretrieve(u"https://npmjs.org/install.sh",filename=npm_install_script_path)
            print(u"Retrieved npm install script. Executing...")
            subprocess.call([u"sh", npm_install_script_path])
            print(u"Updated npm version installed")



    def findUnzippedNodePath(self):
        return self.downloader.download_directory + os.sep + \
               (next(os.walk(self.downloader.download_directory))[1])[0]
    def check_if_executable_installed(self,name):
        executable_path = which(name)
        if executable_path:
            return True
        else:
            return False

    def check_node_version(self):
        version = subprocess.check_output(u"node -v")
        return version
    def check_npm_version(self):
        version = subprocess.check_output(u"npm -v")
        return version


class NodeDownloader(Downloader):
    @property
    def download_url(self):
        raise NotImplementedError
    @property
    def download_directory(self):
        return self.dependency.tmp_directory + os.sep + u"node"
    @property
    def downloaded_file_path(self):
        return self.download_directory + os.sep + u"node.tgz"
    def download(self):
        print(u"Downloading Node from URL " + self.download_url)
        self.download_file(self.download_url,self.downloaded_file_path)
        self.check_download()
    def decompress(self):
        print(u"Decompressing Node...")
        tfile = tarfile.open(self.downloaded_file_path)
        #TODO: make directory handler class
        tfile.extractall(self.download_directory)
        print(u"Decompressed Node into " + self.download_directory)

    def check_download(self):
        isFileValid = tarfile.is_tarfile(self.downloaded_file_path)
        if not isFileValid:
            raise DownloadCorruptionError(u"Node download was corrupted.")



class LinuxNodeDownloader(NodeDownloader):
    @property
    def download_url(self):
        if self.dependency.config.mem_width == 64:
            return u"http://nodejs.org/dist/v0.10.35/node-v0.10.35-linux-x64.tar.gz"
        else:
            return u"http://nodejs.org/dist/v0.10.35/node-v0.10.35-linux-x86.tar.gz"

class WindowsNodeDownloader(NodeDownloader):
    @property
    def download_url(self):
        raise NotImplementedError(u"Needs MSI to be executed to install npm")
        #"http://nodejs.org/dist/v0.10.24/x64/node-v0.10.24-x64.msi"
        if self.dependency.config.mem_width == 64:
            return u"http://nodejs.org/dist/v0.10.35/x64/node.exe"
        else:
            return u"http://nodejs.org/dist/v0.10.35/node.exe"

class MacNodeDownloader(NodeDownloader):
    @property
    def download_url(self):
        if self.dependency.config.mem_width == 64:
            return u"http://nodejs.org/dist/v0.10.35/node-v0.10.35-darwin-x64.tar.gz"
        else:
            return u"http://nodejs.org/dist/v0.10.35/node-v0.10.35-darwin-x86.tar.gz"