#!/usr/bin/python
#
# adt-virt-chroot is part of autopkgtest
# autopkgtest is a tool for testing Debian binary packages
#
# autopkgtest is Copyright (C) 2006-2007 Canonical Ltd.
#
# 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., 675 Mass Ave, Cambridge, MA 02139, USA.
#
# See the file CREDITS for a full list of credits information (often
# installed as /usr/share/doc/autopkgtest/CREDITS).

import sys
import os
import subprocess
from optparse import OptionParser

try:
    our_base = os.environ['AUTOPKGTEST_BASE'] + '/lib'
except KeyError:
    our_base = '/usr/share/autopkgtest/python'
sys.path.insert(1, our_base)

import VirtSubproc
capabilities = ['downtmp-shared-fifo']
chroot_dir = None


def parse_args():
    global chroot_dir

    usage = "%prog [<options>] =<dchroot>|/path/to/chroot"
    parser = OptionParser(usage=usage)

    parser.add_option('-r', '--gain-root')
    parser.add_option('-d', '--debug', action='store_true')

    (opts, args) = parser.parse_args()
    if len(args) != 1:
        parser.error("need exactly one arg, chroot specification")

    VirtSubproc.debuglevel = opts.debug

    chroot_arg = args[0]
    if not chroot_arg:
        parser.error("chroot specification may not be empty")
    if chroot_arg == '=':
        down = ['dchroot', '-q', '--']
    elif chroot_arg == '=':
        down = ['dchroot', '-q', '--']
    elif chroot_arg[0] == '=':
        down = ['dchroot', '-q', '-c', chroot_arg[1:], '--']
    elif chroot_arg[0] == '/':
        down = ['chroot', chroot_arg]
    else:
        parser.error("chroot spec must be =[DCHROOT] or /PATH/TO/CHROOT")

    if down[0] == 'chroot':
        VirtSubproc.downkind = 'auxverb'
        chroot_dir = chroot_arg
    elif down[0] == 'dchroot':
        # FIXME: figure out chroot_dir
        VirtSubproc.downkind = 'shstring'
    else:
        print >> sys.stderr, 'down[0]=%s' % str(down[0])
        assert(False)

    if opts.gain_root:
        down = opts.gain_root.split() + down

    if opts.gain_root or os.getuid() == 0:
        capabilities.append('root-on-testbed')

    VirtSubproc.down = down


def hook_open():
    pass


def hook_downtmp():
    global capabilities
    d = VirtSubproc.downtmp_mktemp()
    if chroot_dir:
        capabilities.append('downtmp-host=%s/%s' % (chroot_dir, d))
    return d


def hook_cleanup():
    global capabilities
    VirtSubproc.downtmp_remove()
    capabilities = [c for c in capabilities if not c.startswith('downtmp-host')]


def hook_forked_inchild():
    pass


def hook_capabilities():
    return capabilities


def hook_shell(stdin, stdout, stderr, dir):
    with open(stdin, 'rb') as sin:
        with open(stdout, 'wb') as sout:
            with open(stderr, 'wb') as serr:
                subprocess.call(VirtSubproc.down + ['sh', '-c', 'cd %s; exec bash -i' % dir],
                                stdin=sin, stdout=sout, stderr=serr)


parse_args()
VirtSubproc.main()
