# -*- coding: utf-8 -*-
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
from __future__ import print_function
from __future__ import division
from __future__ import absolute_import
import glob
import os
from clselect.clselectexcept import ClSelectExcept
from clselect.utils import check_output, demote
ALT_PATH = '/opt/alt'
INT_BASE = 'ruby'
INT_VERS = '??'
class Interpreter(object):
def __init__(self, prefix=None, binary=None, version=None,
version_full=None, target_user=None):
self.user = target_user
if prefix and binary:
self.prefix = prefix
self.binary = binary
elif prefix:
self.prefix = prefix
ruby_bin = os.path.join(self.prefix, 'bin', 'ruby')
if (os.path.isfile(ruby_bin) and not os.path.islink(ruby_bin)):
binary = ruby_bin
if not binary:
raise ClSelectExcept.InterpreterError(
'Can not find ruby binary')
self.binary = binary
elif binary:
self.binary = binary
self.prefix = os.path.abspath(os.path.join(
os.path.dirname(binary), os.path.pardir))
else:
interpreter = interpreters('version')['2.1']
self.binary = interpreter.binary
self.prefix = interpreter.prefix
self.version = interpreter.version
self.version_full = interpreter.version_full
if version:
self.version = version
else:
self.version = None
if version_full:
self.version_full = version_full
else:
try:
if self.user:
version_raw = check_output(self.binary, '-e', 'puts RUBY_VERSION',
preexec_fn=demote(self.user))
else:
version_raw = check_output(self.binary, '-e', 'puts RUBY_VERSION')
except (OSError, ClSelectExcept.ExternalProgramFailed):
raise ClSelectExcept.InterpreterError(
'Error checking ruby binary version')
self.version_full = version_raw.strip()
if not self.version:
self.version = '.'.join(self.version_full.split('.')[:2])
def __repr__(self):
return (
"%s.%s(prefix='%s', version='%s', binary='%s', "
"version_full='%s')" % (
self.__class__.__module__, self.__class__.__name__,
self.prefix, self.version, self.binary, self.version_full
))
def as_dict(self, key=None):
i = {
'prefix': self.prefix,
'binary': self.binary,
'version': self.version,
'version_full': self.version_full,
}
if key:
del i[key]
return {getattr(self, key): i}
return i
def interpreters(key=None):
search_path = os.path.join(ALT_PATH, INT_BASE + INT_VERS)
interps = []
for path in glob.glob(search_path):
try:
interpreter = Interpreter(path)
except ClSelectExcept.InterpreterError:
continue
interps.append(interpreter)
if key:
return dict((getattr(i, key), i) for i in interps)
else:
return interps
def interpreters_dict(key):
return dict(list(i.as_dict(key).items())[0] for i in interpreters())
|