#
# Copyright (c) 2016 Wind River Systems, Inc.
#
# SPDX-License-Identifier: Apache-2.0
#
import json

from nfv_vim import objects

from nfv_vim.database import model

from nfv_vim.database._database import database_get


def database_sw_update_add(sw_update_obj):
    """
    Add a software update object to the database
    """
    db = database_get()
    session = db.session()
    query = session.query(model.SoftwareUpdate)
    sw_update = query.filter(model.SoftwareUpdate.uuid ==
                             sw_update_obj.uuid).first()
    if not sw_update:
        sw_update = model.SoftwareUpdate()
        sw_update.uuid = sw_update_obj.uuid
        sw_update.sw_update_type = sw_update_obj.sw_update_type
        if sw_update_obj.strategy is None:
            sw_update.strategy_data = json.dumps(dict())
        else:
            sw_update.strategy_data = json.dumps(sw_update_obj.strategy.as_dict())
        session.add(sw_update)
    else:
        if sw_update_obj.strategy is None:
            sw_update.strategy_data = json.dumps(dict())
        else:
            sw_update.strategy_data = json.dumps(sw_update_obj.strategy.as_dict())
    db.commit()


def database_sw_update_delete(sw_update_uuid):
    """
    Delete a software update object from the database
    """
    db = database_get()
    session = db.session()
    query = session.query(model.SoftwareUpdate)
    query.filter(model.SoftwareUpdate.uuid == sw_update_uuid).delete()
    session.commit()


def database_sw_update_get_list():
    """
    Fetch all the software update objects from the database
    """
    db = database_get()
    session = db.session()
    query = session.query(model.SoftwareUpdate)

    sw_update_objs = list()
    for sw_update in query.all():
        strategy_data = json.loads(sw_update.strategy_data)
        if objects.SW_UPDATE_TYPE.SW_PATCH == sw_update.sw_update_type:
            sw_patch_obj = objects.SwPatch(sw_update.uuid, strategy_data)
            sw_update_objs.append(sw_patch_obj)
        elif objects.SW_UPDATE_TYPE.SW_UPGRADE == sw_update.sw_update_type:
            sw_upgrade_obj = objects.SwUpgrade(sw_update.uuid, strategy_data)
            sw_update_objs.append(sw_upgrade_obj)
    return sw_update_objs