#!/usr/bin/python3
'''
Mountall apport hook.
'''

import os
from apport.hookutils import *
import apport.packaging
import glob

msg = \
"""
The contents of your /etc/fstab file and relevant log files may help
developers diagnose your bug more quickly. However, if you have modified
/etc/fstab, these files may contain sensitive information.

Do you want to include the files in your bug report? (you will be able
to review the data before it is sent)
"""

def add_root_file(map, file):
    """
    Add @file to @map such that @map can be passed to
    attach_root_command_outputs() to have @file attached to the report
    event if it is unreadable by the current user.
    """
    if not os.path.exists(file):
        return

    key = path_to_key(file)

    map[key] = 'cat %s' % file


def add_info(report, ui):
    attach_files = False

    # files that are not readable by current user
    unreadable_files = {}

    problem_type = report.get('ProblemType', '')

    if problem_type == 'Bug' and ui:
        if ui.yesno(msg) == None:
            raise StopIteration
        attach_files = True
    elif problem_type == 'Crash':
        # crash bugs are private by default
        attach_files = True

    if attach_files == False:
        return

    attach_file(report, '/etc/fstab')
    attach_file(report, '/etc/mtab')
    attach_file(report, '/proc/cmdline', 'ProcKernelCmdline')
    attach_file(report, '/proc/mounts', 'ProcMounts')
    attach_file(report, '/proc/self/mountinfo', 'ProcSelfMountinfo')
    attach_file_if_exists(report, '/run/mount/utab')

    report['Mounts'] = apport.hookutils.command_output(['mount'])

    # mountall logs to the console so until Upstart can log to both the
    # console and a file, grab the plymouth log which captures the
    # console.
    add_root_file(unreadable_files, '/var/log/boot.log')

    # in case admin changed the console stanza
    add_root_file(unreadable_files, '/var/log/upstart/mountall.log')

    attach_root_command_outputs(report, unreadable_files)