# Copyright 2017 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""A binary for constructing images from a source context."""

import sys
import argparse

from containerregistry.tools import patched

from ftl.common import args
from ftl.common import logger
from ftl.common import context
from ftl.common import ftl_util
from ftl.common import ftl_error

from ftl.node import builder as node_builder

version = args.version_parser()
parser = args.base_parser()
node_parser = argparse.ArgumentParser(
    add_help=False,
    parents=[parser],
    description='Construct node images from source.')
args.extra_args(node_parser, args.node_flgs)


def main(cli_args):
    try:
        version.parse_known_args(cli_args)
        builder_args = node_parser.parse_args(cli_args)
        logger.setup_logging(builder_args)
        logger.preamble("node", builder_args)
        with ftl_util.Timing("full build"):
            with ftl_util.Timing("builder initialization"):
                node_ftl = node_builder.Node(
                    context.Workspace(builder_args.directory), builder_args)
            with ftl_util.Timing("build process for FTL image"):
                node_ftl.Build()
    except ftl_error.UserError as e:
        ftl_error.UserErrorHandler(
            e, builder_args.builder_output_path, builder_args.fail_on_error)
    except ftl_error.InternalError as e:
        ftl_error.InternalErrorHandler(
            e, builder_args.builder_output_path, builder_args.fail_on_error)


if __name__ == '__main__':
    with patched.Httplib2():
        main(sys.argv[1:])