# Copyright 2020-2021 Canonical, Ltd.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import argparse
import logging
import os
import sys

from subiquitycore.log import setup_logger

from subiquity.cmd.common import (
    LOGDIR,
    setup_environment,
    )


def make_server_args_parser():
    parser = argparse.ArgumentParser(
        description='System Setup - Initial Boot Setup',
        prog='system_setup')
    parser.add_argument('--dry-run', action='store_true',
                        dest='dry_run',
                        help='menu-only, do not call installer function')
    parser.add_argument('--socket')
    parser.add_argument('--autoinstall', action='store')
    parser.add_argument('--prefill',
                        dest='prefill',
                        help='Prefills UI models with data provided in'
                        ' a prefill.yaml file yet allowing overrides.')
    parser.add_argument('--output-base', action='store', dest='output_base',
                        default='.subiquity',
                        help='in dryrun, control basedir of files')
    parser.add_argument('--tcp-port',
                        dest='tcp_port',
                        type=int,
                        choices=range(49152, 60999),
                        metavar="[49152 to 60999]",
                        help='The TCP port Subiquity must listen to. It means '
                        'TCP will be used instead of Unix domain sockets. '
                        'Only localhost connections are accepted.')
    return parser


def main():
    print('starting server')
    setup_environment()
    # setup_environment sets $APPORT_DATA_DIR which must be set before
    # apport is imported, which is done by this import:
    from system_setup.server.server import SystemSetupServer
    from subiquity.server.server import NOPROBERARG
    parser = make_server_args_parser()
    opts = parser.parse_args(sys.argv[1:])
    logdir = LOGDIR
    opts.snaps_from_examples = False
    opts.kernel_cmdline = {}
    opts.machine_config = NOPROBERARG
    if opts.dry_run:
        logdir = opts.output_base
    if opts.socket is None:
        if opts.dry_run:
            opts.socket = opts.output_base + '/socket'
        else:
            opts.socket = '/run/subiquity/socket'
    os.makedirs(os.path.dirname(opts.socket), exist_ok=True)

    block_log_dir = os.path.join(logdir, "block")
    os.makedirs(block_log_dir, exist_ok=True)
    handler = logging.FileHandler(os.path.join(block_log_dir, 'discover.log'))
    handler.setLevel('DEBUG')
    handler.setFormatter(
        logging.Formatter("%(asctime)s %(name)s:%(lineno)d %(message)s"))

    logfiles = setup_logger(dir=logdir, base='systemsetup-server')

    logger = logging.getLogger('systemsetup-server')
    version = "unknown"
    logger.info("Starting System Setup server revision {}".format(version))
    logger.info("Arguments passed: {}".format(sys.argv))

    prefillFile = opts.prefill
    if prefillFile:
        if not os.path.isfile(prefillFile):
            logger.error('"File {}" is invalid. Option will be ignored.'
                         .format(prefillFile))
            opts.prefill = None

    server = SystemSetupServer(opts, block_log_dir)

    server.note_file_for_apport(
        "InstallerServerLog", logfiles['debug'])
    server.note_file_for_apport(
        "InstallerServerLogInfo", logfiles['info'])

    server.run()


if __name__ == '__main__':
    sys.exit(main())