qt-migration-legatus/qt6-after/qt6-migrated-legatus/Lts/Services/AssetManagement/object_asset.cpp

/**
  * Asset object
  * $Id$
  * $HeadURL$
  * $Author$
  * Created by Iuliana Ojog, SLSE, 2017-09-20
  */

#include "object_asset.h"
#include "main_handler.h"

#include <LtsUtils/LtsUtilsCore/dds_vector.h>


//------------------------------------------------------------------------------
namespace AssetManagement
{
ObjectAsset::ObjectAsset(MainHandler *mainHandler)
  : m_mainHandler(mainHandler)
{
}

//------------------------------------------------------------------------------
ObjectAsset::~ObjectAsset()
{
  if (isValid())
    m_mainHandler->dds()->dts()->removeAssetByKey(m_data.id());
}

//------------------------------------------------------------------------------
bool ObjectAsset::isValid() const
{
  return m_data.id() > 0;
}

//------------------------------------------------------------------------------
bool ObjectAsset::isUsed() const
{
  return m_data.isUsed();
}

//------------------------------------------------------------------------------
Core::FunctionReplyObject ObjectAsset::createObject(const Asset &data)
{
  m_data = data;

  if (m_data.id() <= 0)
  {
    m_data.isUsed(false);

    Database::AsmAsset dbData(toDbData(data));

    if (!m_mainHandler->db()->insert(dbData))
      return Core::FunctionReplyObject(0, tr_lts("dds", "Error inserting AsmAsset-data into database"));

    m_data.id(dbData.id);
    DEBUG_LOG() << "New assetId: " << m_data.id();
  }

  // Publish Data via DTS
  onPublishData();

  return Core::FunctionReplyObject(m_data.id());
}

//------------------------------------------------------------------------------
Core::FunctionReplyObject ObjectAsset::updateObject(const Asset &data)
{
  if (!isValid())
    return Core::FunctionReplyObject(data.id(), tr_lts("dds", "Object not valid/blocked"));

  Database::AsmAsset dbData;
  dbData = toDbData(data);

  if (!m_mainHandler->db()->update(dbData))
    return Core::FunctionReplyObject(data.id(), tr_lts("dds", "Error updating AsmAsset-data into database"));

  m_data = data;
  DEBUG_LOG() << "Data changed for assetId: " << m_data.id();

  // Publish Data via DTS
  onPublishData();

  return Core::FunctionReplyObject(m_data.id());
}

//------------------------------------------------------------------------------
Core::FunctionReplyObject ObjectAsset::setUsed(bool isUsed)
{
  if (!isValid())
    return Core::FunctionReplyObject(m_data.id(), tr_lts("dds", "Trying to update an already removed AsmAsset object"));

  m_data.isUsed(isUsed);

  // Publish Data via DTS
  onPublishData();

  return Core::FunctionReplyObject(m_data.id());
}

//------------------------------------------------------------------------------
Core::FunctionReplyObject ObjectAsset::removeObject()
{
  if (isUsed())
    return Core::FunctionReplyObject(m_data.id(), tr_lts("dds", "Asset still in use"));

  if (!m_mainHandler->db()->removeAsmAsset(m_data.id()))
    return Core::FunctionReplyObject(m_data.id(), tr_lts("dds", "Asset with id %1 can not be removed from DB")).arg(m_data.id());

  if (!m_mainHandler->dds()->dts()->removeAssetByKey(m_data.id()))
    return Core::FunctionReplyObject(m_data.id(), tr_lts("dds", "Asset with id %1 can not be removed from DDS")).arg(m_data.id());

  return Core::FunctionReplyObject(m_data.id());
}

//------------------------------------------------------------------------------
void ObjectAsset::onPublishData()
{
  INFO_LOG() << "Publish AsmAsset-data to dds (" << m_data.id() << ") ...";

  if (!m_mainHandler->dds()->dts()->publish(m_data))
    ERROR_LOG() << "Error publishing AsmAsset-data to dds";
}

// -------------------------------------------------------------------------------------------------------
// convert methods
// -------------------------------------------------------------------------------------------------------

// AssetManagement::Asset => Database::AsmAsset
Database::AsmAsset ObjectAsset::toDbData(const Asset &ddsData)
{
  Database::AsmAsset dbData;

  dbData.id          = ddsData.id();
  dbData.name        = QString::fromStdString(ddsData.name());
  dbData.assetTypeId = ddsData.assetTypeId();
  dbData.locationId  = ddsData.locationId();
  dbData.subLocation = ddsData.subLocation();
  dbData.certificate = LtsUtils::toQByteArray(ddsData.certificate());
  dbData.properties  = QString::fromStdString(ddsData.properties());

  return dbData;
}

// -------------------------------------------------------------------------------------------------------
// Database::AsmAsset => AssetManagement::Asset
Asset ObjectAsset::toDdsData(const Database::AsmAsset &dbData)
{
  Asset ddsData;

  ddsData.id(dbData.id);
  ddsData.name(dbData.name.toStdString());
  ddsData.assetTypeId(dbData.assetTypeId);
  ddsData.locationId(dbData.locationId);
  ddsData.subLocation(dbData.subLocation);
  ddsData.certificate(LtsUtils::convert<std::vector<uint8_t>>(dbData.certificate));
  ddsData.properties(dbData.properties.toStdString());

  // init data not in db
  ddsData.isDeleted(false);
  ddsData.isUsed(false);

  return ddsData;
}

// -------------------------------------------------------------------------------------------------------

} // Namespace AssetManagement