# BIMTester - OpenBIM Auditing Tool
# Copyright (C) 2021 Dion Moult <dion@thinkmoult.com>
#
# This file is part of BIMTester.
#
# BIMTester is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# BIMTester 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 Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with BIMTester.  If not, see <http://www.gnu.org/licenses/>.

import os

from behave.model import Scenario
from bimtester.ifc import IfcStore
from bimtester.lang import switch_locale
from logfile import append_logfile, create_logfile
from zoom_smart_view import add_smartview, create_zoom_set_of_smartviews

this_path = os.path.dirname(os.path.realpath(__file__))


def before_all(context):

    userdata = context.config.userdata
    context.locale_dir = userdata.get("localedir")

    if context.config.lang:
        switch_locale(context.locale_dir, context.config.lang)

    continue_after_failed = userdata.getbool("runner.continue_after_failed_step", True)
    Scenario.continue_after_failed_step = continue_after_failed

    # context.ifc_path = userdata.get("ifc", "")
    context.ifcfile_basename = os.path.basename(os.path.splitext(userdata["ifc"])[0])
    context.outpath = os.path.join(this_path, "..")
    context.create_log = False
    context.create_smartview = False

    if context.create_log is True:
        # set up log file
        context.thelogfile = os.path.join(context.outpath, context.ifcfile_basename + ".log")
        create_logfile(
            context.thelogfile,
            context.ifcfile_basename,
        )


def before_feature(context, feature):

    # https://github.com/IfcOpenShell/IfcOpenShell/issues/1910#issuecomment-989732600
    # messages language, parsed by behaves lang argument
    print("Messages language: {}".format(context.config.lang))
    # features file language, set in feature files first line
    # html report will use this too
    print("Features language: {}".format(context.feature.language))

    # if messages lang is not set use features lang
    if context.config.lang == "" or context.config.lang is None:
        context.config.lang = context.feature.language
        print("Switch messages language to: {}".format(context.config.lang))
        switch_locale(context.locale_dir, context.config.lang)

    # TODO: refactor zoom smart view support into a decoupled module
    if context.create_smartview is True:
        smartview_name = context.ifcfile_basename + "_" + feature.name
        context.smview_file = os.path.join(context.outpath, smartview_name + ".bcsv")
        # print("SmartView file: {}".format(context.smview_file))
        create_zoom_set_of_smartviews(
            context.smview_file,
            smartview_name,
        )


def after_step(context, step):

    if step.status == "failed" and context.create_log is True:
        append_logfile(context, step)

    if step.status == "failed" and context.create_smartview is True and hasattr(context, "falseguids"):
        # print(context.falseguids)
        add_smartview(context.smview_file, step.name, context.falseguids)