#!/bin/env python

 # -*- coding: utf-8 -*-

 # ------------------------------------------------------------

 # environment.py

 # 

 # pick some current environment infos

 #

 # Autor: Oliver Maurhart, <oliver.maurhart@ait.ac.at>

 #

 # Copyright (C) 2013 AIT Austrian Institute of Technology

 # AIT Austrian Institute of Technology GmbH

 # Donau-City-Strasse 1 | 1220 Vienna | Austria

 # http://www.ait.ac.at

 #

 # This program 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 version 2.

 # 

 # This program 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 this program; if not, write to the Free Software

 # Foundation, Inc., 51 Franklin Street, Fifth Floor, 

 # Boston, MA  02110-1301, USA.

 # ------------------------------------------------------------

 # ------------------------------------------------------------

 # imports

 import os

 import os.path

 import sys

 # ------------------------------------------------------------

 # code

 class Environment(object):

     """Hold some nifty environment stuff in a dedicated class."""

     _log_warning_shown = False

     _prefix_path = ''

     def __init__(self, application = None):

         # if we ain't got a path to start from, all is valid/lost

         if len(sys.path[0]) == 0:

             self._prefix_path = ''

             self._data_path = ''

             return

         # the prefix path

         #

         # - on Linux: this is ../../ to the current executable

         #   e.g. "/usr/bin/myprogram" --> "/usr"

         #

         # - on Windows (inkl. Cygwin): this is the installation folder

         #   e.g. "C:/Program Files/MyProgram/myprogam" --> "C:/Program Files/MyProgram"

         #

         if sys.platform == 'linux2':

             self._prefix_path = os.path.split(sys.path[0])[0]

         elif sys.platform == 'win32' or sys.platform == 'cygwin':

             for app_path in sys.path:

                 if 'OpenSecurity\\bin' in app_path:

                     self._prefix_path = os.path.normpath(os.path.join(app_path, '..'))

                     break

         if self._prefix_path == '':

             raise OSError()

         # the data path where all data files are stored

         if sys.platform == 'linux2':

             if not application is None:

                 self._data_path = os.path.join(self._prefix_path, os.path.join('share', application))

             else:

                 self._data_path = os.path.join(self._prefix_path, 'share')

         elif sys.platform == 'win32' or sys.platform == 'cygwin':

             self._data_path = self._prefix_path

         else:

             raise OSError()

     def data_path_get(self):

         """dat_path get"""

         return self._data_path

     data_path = property(data_path_get)

     def prefix_path_get(self):

         """prefix_path get"""

         return self._prefix_path

     prefix_path = property(prefix_path_get)

     def log_path_get(self):

         """the path where log files should be stored"""

         user_log_path = os.path.expanduser(os.path.join('~', '.log'))

         if sys.platform == 'linux2':

             if os.access('/var/log', os.W_OK):

                 return '/var/log'

             if not Environment._log_warning_shown:

                 print('no permissions to write log files in /var/log, switching to ~/.log')

                 Environment._log_warning_shown = True

             if not os.path.exists(user_log_path):

                 os.mkdir(user_log_path)

             elif not os.path.isdir(user_log_path):

                 raise IOError(user_log_path + ': not a folder')

             return user_log_path

         elif sys.platform == 'win32' or sys.platform == 'cygwin':

             # in OpenSecurity we expect the log path tp be

             # somewhere like C:\Program Files\OpenSecurity\log

             # having this script residing in 

             # C:\Progam Files\OpenSecurity\bin

             ideal_log_path = os.path.normpath(os.path.join(self.prefix_path, 'log'))

             # check ideal path first

             if not os.path.exists(ideal_log_path):

                 os.mkdir(ideal_log_path)

             elif not os.path.isdir(ideal_log_path):

                 raise IOError(ideal_log_path + ': not a folder')

             return ideal_log_path

         else:

             raise OSError()

     log_path = property(log_path_get)

 # test method

 def test():

     """Test: class Environment"""

     e = Environment('OpenSecurity')

     print('prefix_path: "{0}"'.format(e.prefix_path))

     print('  data_path: "{0}"'.format(e.data_path))

     print('   log_path: "{0}"'.format(e.log_path))

 # test the module			

 if __name__ == '__main__':

     test()