#!/usr/bin/python3
#
# autopkgtest-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 argparse

sys.path.insert(0, "/usr/share/autopkgtest/lib")
sys.path.insert(
    0, os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "lib")
)

import VirtSubproc
import adtlog


capabilities = []
chroot_dir = None


def parse_args():
    global chroot_dir

    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-r",
        "--gain-root",
        metavar="COMMAND",
        help="can become root by prefixing commands with COMMAND",
    )
    parser.add_argument(
        "-d", "--debug", action="store_true", help="Enable debugging output"
    )
    parser.add_argument("chroot", metavar="/path/to/chroot")
    args = parser.parse_args()
    if args.debug:
        adtlog.verbosity = 2

    chroot_dir = os.path.abspath(args.chroot)
    down = ["chroot", chroot_dir]
    if args.gain_root:
        down = args.gain_root.split() + down

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

    VirtSubproc.auxverb = down


def hook_open():
    pass


def hook_downtmp(path):
    global capabilities
    d = VirtSubproc.downtmp_mktemp(capabilities, path, chroot_dir)
    return d


def hook_cleanup():
    global capabilities
    VirtSubproc.downtmp_remove(capabilities)


def hook_capabilities():
    return capabilities


parse_args()
VirtSubproc.main()
