# -*- coding: iso8859-1 -*-
#
# Copyright (C) 2005 Christopher Lenz <cmlenz@gmx.de>
#
# Bitten 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.
#
# Trac 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., 675 Mass Ave, Cambridge, MA 02139, USA.
#
# Author: Christopher Lenz <cmlenz@gmx.de>

import logging
import os
import shlex

from bitten.util import xmlio
from bitten.util.cmdline import Commandline

log = logging.getLogger('bitten.build.shtools')

def exec_(ctxt, file_=None, output=None, args=None):
    """Execute a shell script."""
    assert file_, 'Missing required attribute "file"'

    if args:
        args = shlex.split(args)
    else:
        args = []

    output_file = None
    if output:
        output = ctxt.resolve(output)
        output_file = file(output, 'w')

    try:
        cmdline = Commandline(file_, args, cwd=ctxt.basedir)
        log_elem = xmlio.Fragment()
        for out, err in cmdline.execute():
            if out is not None:
                log.info(out)
                xmlio.SubElement(log_elem, 'message', level='info')[out]
                if output:
                    output_file.write(out + os.linesep)
            if err is not None:
                log.error(err)
                xmlio.SubElement(log_elem, 'message', level='error')[err]
                if output:
                    output_file.write(err + os.linesep)
        ctxt.log(log_elem)
    finally:
        if output:
            output_file.close()

    if cmdline.returncode != 0:
        ctxt.error('exec failed (%s)' % cmdline.returncode)