#    Back In Time
#    Copyright (C) 2008-2017 Oprea Dan, Bart de Koning, Richard Bailey, Germar Reitze
#
#    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; either version 2 of the License, or
#    (at your option) any later version.
#
#    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.


import os
import pluginmanager
import logger
import gettext
from subprocess import Popen, PIPE
from exceptions import StopException

_=gettext.gettext


class UserCallbackPlugin( pluginmanager.Plugin ):
    def __init__( self ):
        return

    def init( self, snapshots ):
        self.config = snapshots.config
        self.callback = self.config.get_take_snapshot_user_callback()
        if not os.path.exists( self.callback ):
            return False
        return True

    def notify_callback(self, *args):
        cmd = [self.callback, self.config.get_current_profile(), self.config.get_profile_name()]
        cmd.extend([str(x) for x in args])
        logger.debug('Call user-callback: %s' %' '.join(cmd), self)
        if self.config.user_callback_no_logging():
            stdout, stderr = None, None
        else:
            stdout, stderr = PIPE, PIPE
        try:
            callback = Popen(cmd,
                             stdout = stdout,
                             stderr = stderr,
                             universal_newlines = True)
            output = callback.communicate()
            if output[0]:
                logger.info('user-callback returned \'%s\'' %output[0].strip('\n'), self)
            if output[1]:
                logger.error('user-callback returned \'%s\'' %output[1].strip('\n'), self)
            if callback.returncode != 0:
                logger.warning('user-callback returncode: %s' %callback.returncode, self)
                raise StopException()
        except OSError as e:
            logger.error("Exception when trying to run user callback: %s" % e.strerror, self)

    def on_process_begins( self ):
        self.notify_callback( '1' )

    def on_process_ends( self ):
        self.notify_callback( '2' )

    def on_error(self, code, message):
        if not message:
            self.notify_callback('4', code)
        else:
            self.notify_callback('4', code, message)

    def on_new_snapshot( self, snapshot_id, snapshot_path ):
        self.notify_callback('3', snapshot_id, snapshot_path)

    def on_app_start(self):
        self.notify_callback('5')

    def on_app_exit(self):
        self.notify_callback('6')

    def do_mount(self):
        self.notify_callback('7')

    def do_unmount(self):
        self.notify_callback('8')
