/*

 * Copyright 2013-2022 Step Function I/O, LLC

 *

 * Licensed to Green Energy Corp (www.greenenergycorp.com) and Step Function I/O

 * LLC (https://stepfunc.io) under one or more contributor license agreements.

 * See the NOTICE file distributed with this work for additional information

 * regarding copyright ownership. Green Energy Corp and Step Function I/O LLC license

 * this file to you 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.

 */

#include "AssignClassTask.h"

#include "app/HeaderWriter.h"

#include "opendnp3/master/IMasterApplication.h"

namespace opendnp3

{

AssignClassTask::AssignClassTask(const std::shared_ptr<TaskContext>& context,

                                 IMasterApplication& application,

                                 const TaskBehavior& behavior,

                                 const Logger& logger)

    : IMasterTask(context, application, behavior, logger, TaskConfig::Default())

{

}

bool AssignClassTask::BuildRequest(APDURequest& request, uint8_t seq)

{

    request.SetControl(AppControlField(true, true, false, false, seq));

    request.SetFunction(FunctionCode::ASSIGN_CLASS);

    auto writer = request.GetWriter();

    bool success = true;

    auto writeFun = [&](const Header& header) { success &= header.WriteTo(writer); };

    this->application->ConfigureAssignClassRequest(writeFun);

    return success;

}

bool AssignClassTask::IsEnabled() const

{

    return this->application->AssignClassDuringStartup();

}

IMasterTask::ResponseResult AssignClassTask::ProcessResponse(const APDUResponseHeader& header,

                                                             const ser4cpp::rseq_t& objects)

{

    return ValidateNullResponse(header, objects) ? ResponseResult::OK_FINAL : ResponseResult::ERROR_BAD_RESPONSE;

}

} // namespace opendnp3