#!/usr/bin/env python3

# Mark layers that are Yocto Project Compatible (2.0) on the AutoBuilder
#
# Copyright (C) 2017 Intel Corporation
# Author: Paul Eggleton <paul.eggleton@linux.intel.com>
# Copyright (C) 2023 Konsulko Group
# Author: Tim Orling <tim.orling@konsulko.com>
#
# Licensed under the MIT license, see COPYING.MIT for details
#
# SPDX-License-Identifier: MIT
#
# Input json file can be for instance generated by:
# https://git.yoctoproject.org/yocto-autobuilder-helper/tree/scripts/list-yp-compatible-layers.py

import sys
import os

sys.path.insert(0, os.path.realpath(os.path.join(os.path.dirname(__file__), '..')))

import argparse
from argparse import ArgumentError
from layerindex import utils
import logging
import json
from json import JSONDecodeError

class DryRunRollbackException(Exception):
    pass

class LayerNotFoundException(Exception):
    pass

class BranchNotFoundException(Exception):
    pass

class YPCompatibleVersionNotFoundException(Exception):
    pass

logger = utils.logger_create('LayerIndexYoctoProjectCompatible')



def main():
    parser = argparse.ArgumentParser(description='Mark layers that are Yocto Project Compatible (2.0)')

    layer_branch_arg_group = parser.add_argument_group()
    layer_branch_arg_group.add_argument("-l", "--layer",
            help = "Layer to mark as Yocto Project Compatible",
            action="store", dest="layer")
    layer_branch_arg_group.add_argument("-b", "--branch",
            help = "Branch to mark as Yocto Project Compatible",
            action="store", dest="branch")
    file_or_layer_branch_arg_group = layer_branch_arg_group.add_mutually_exclusive_group()
    file_or_layer_branch_arg_group.add_argument("-f", "--from-file",
            help = "(Recommended) JSON file compatible with 'list-yp-compatible-layers' script from 'yocto-autobuilder-helper' to use for input",
            action="store", dest="from_file")
    parser.add_argument("-v", "--compatible-version",
            help = "Yocto Project Compatible version",
            action="store", dest="yp_compatible_version", default="2.0")
    parser.add_argument("-n", "--dry-run",
            help = "Don't write any data back to the database",
            action="store_true", dest="dryrun")
    parser.add_argument("-d", "--debug",
            help = "Enable debug output",
            action="store_const", const=logging.DEBUG, dest="loglevel", default=logging.INFO)
    parser.add_argument("-q", "--quiet",
            help = "Hide all output except error messages",
            action="store_const", const=logging.ERROR, dest="loglevel")

    args = parser.parse_args()

    utils.setup_django()
    import settings
    from layerindex.models import Branch, LayerBranch, LayerItem, YPCompatibleVersion
    from django.db import transaction

    logger.setLevel(args.loglevel)

    def update_from_layer_branch(layer, branch, yp_compatible_version, dryrun):
        try:
            with transaction.atomic():
                try:
                    layer_id=LayerItem.objects.filter(name__exact=layer)[0].id
                except IndexError:
                    raise LayerNotFoundException( f'Layer {layer} not found in layerindex ddatabase')
                try:
                    branch_id=Branch.objects.filter(name__exact=branch)[0].id
                except IndexError:
                    raise BranchNotFoundException( f'Branch {branch} not found in layerindex database')
                try:
                    yp_compatible_version_instance=YPCompatibleVersion.objects.filter(name__exact=yp_compatible_version)[0]
                except IndexError:
                    raise YPCompatibleVersionNotFoundException( f' Yocto Project Compatible Version {yp_compatible_version} not found in layerindex database')
                for layerbranch in LayerBranch.objects.filter(layer=layer_id,branch=branch_id):
                    logger.debug(' BEFORE: Yocto Project Compatible %s   %s:%s' % (layerbranch.yp_compatible_version, layerbranch.layer, layerbranch.branch))
                    layerbranch.yp_compatible_version = yp_compatible_version_instance
                    layerbranch.save()
                    logger.info('Yocto Project Compatible %s   %s:%s' % (layerbranch.yp_compatible_version, layerbranch.layer, layerbranch.branch))

                if dryrun:
                    raise DryRunRollbackException()
        except DryRunRollbackException:
            pass

    def update_from_json_file( from_file, yp_compatible_version, dryrun ):
        try:
            with open(from_file) as fp:
                input_dict = json.load(fp)
                for job in input_dict:
                    for branch in input_dict[job]:
                        try:
                            for layer in input_dict[job][branch]:
                                try:
                                    update_from_layer_branch( layer, branch, yp_compatible_version, dryrun )
                                except LayerNotFoundException as err:
                                    logger.warning( f"{err}" )
                                    pass
                        except BranchNotFoundException as err:
                            logger.warning( f"{err}" )
                            pass
        except DryRunRollbackException:
            pass
        except FileNotFoundError as err:
            logger.critical( f"{err}: File {from_file} not found" )
            sys.exit(1)
        except JSONDecodeError as err:
            logger.critical( f"{err}: JSONDecodeError loading file {from_file}" )
            sys.exit(1)

    if args.layer and args.branch and not args.from_file:
        update_from_layer_branch( layer=args.layer, branch=args.branch, yp_compatible_version=args.yp_compatible_version, dryrun=args.dryrun )
    elif args.from_file and not ( args.branch or args.layer ):
        update_from_json_file( from_file=args.from_file, yp_compatible_version=args.yp_compatible_version, dryrun=args.dryrun )

    sys.exit(0)


if __name__ == "__main__":
    main()