/*
 * Copyright (C) 2021 Huawei Device Co., Ltd.
 * Licensed 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 <iostream>
#include <string>
#include "hilog/log.h"
#include "dbinder_test_service.h"
#include "ipc_skeleton.h"
#include "distributed_agent.h"
#include "dbinder_service_test_helper.h"
#include "log_tags.h"
#include "softbus_bus_center.h"

using namespace testing;
using namespace OHOS;
using namespace OHOS::DistributeSystemTest;
using namespace OHOS::HiviewDFX;

#ifndef TITLE
#define TITLE __PRETTY_FUNCTION__
#endif

static constexpr OHOS::HiviewDFX::HiLogLabel LOG_LABEL = { LOG_CORE, LOG_ID_RPC, "DbinderTestAgent" };
#define DBINDER_LOGE(fmt, args...) \
    (void)OHOS::HiviewDFX::HiLog::Error(LOG_LABEL, "%{public}s %{public}d: " fmt, TITLE, __LINE__, ##args)
#define DBINDER_LOGI(fmt, args...) \
    (void)OHOS::HiviewDFX::HiLog::Info(LOG_LABEL, "%{public}s %{public}d: " fmt, TITLE, __LINE__, ##args)

class DbinderTestAgent : public DistributedAgent {
public:
    DbinderTestAgent();
    ~DbinderTestAgent();
    virtual bool SetUp();
    virtual bool TearDown();
    virtual int OnProcessMsg(const std::string &strMsg, int len, std::string &strReturnValue, int returnBufLen);
    virtual int OnProcessCmd(const std::string &strCommand, int cmdLen, const std::string &strArgs, int argsLen,
        const std::string &strExpectValue, int expectValueLen);

private:
    std::string localUdid;
    void KillService() const;
    void RestartService() const;
};

DbinderTestAgent::DbinderTestAgent()
{
    std::string pkgName = "dbinderService";
    NodeBasicInfo nodeBasicInfo;
    if (GetLocalNodeDeviceInfo(pkgName.c_str(), &nodeBasicInfo) != 0) {
        DBINDER_LOGE("Get local node device info failed");
        return;
    }
    std::string networkId(nodeBasicInfo.networkId);
    localUdid = networkId;
}

DbinderTestAgent::~DbinderTestAgent() {}

bool DbinderTestAgent::SetUp()
{
    DBINDER_LOGI("enter SetUp");
    StartDBinderServiceTestService();
    return true;
}

bool DbinderTestAgent::TearDown()
{
    DBINDER_LOGI("enter TearDown");
    KillService();
    return true;
}

// from test framework
int DbinderTestAgent::OnProcessMsg(const std::string &strMsg, int len, std::string &strReturnValue, int returnValueLen)
{
    std::string msg = "Ask Device ID";
    if (strncmp(msg.c_str(), strMsg.c_str(), len) == 0) {
        strReturnValue = localUdid;
        returnValueLen = strlen(localUdid.c_str());
        return returnValueLen;
    } else {
        return DistributedAgent::OnProcessMsg(strMsg, len, strReturnValue, returnValueLen);
    }
}

// from test framework
int DbinderTestAgent::OnProcessCmd(const std::string &strCommand, int cmdLen, const std::string &strArgs, int argsLen,
    const std::string &strExpectValue, int expectValueLen)
{
    DBINDER_LOGI("enter OnProcessCmd");
    if (strCommand == "KILL") {
        DBINDER_LOGI("strCommand = %{public}s, strArgs = %{public}s", strCommand.c_str(), strArgs.c_str());
        KillService();
    } else if (strCommand == "RESTART") {
        DBINDER_LOGI("strCommand = %{public}s, strArgs = %{public}s", strCommand.c_str(), strArgs.c_str());
        RestartService();
    } else {
        return DistributedAgent::OnProcessCmd(strCommand, cmdLen, strArgs, argsLen, strExpectValue, expectValueLen);
    }

    return 0;
}

void DbinderTestAgent::KillService() const
{
    DBINDER_LOGI("enter KillService");
    StopDBinderServiceTestService();
}

void DbinderTestAgent::RestartService() const
{
    DBINDER_LOGI("enter RestartService");
    StartDBinderServiceTestService();
}

int main()
{
    // Test agent main function
    DbinderTestAgent obj;
    if (obj.SetUp()) {
        obj.Start("agent.desc");
        obj.Join();
    } else {
        DBINDER_LOGE("Init environment failed.");
    }
    if (obj.TearDown()) {
        return 0;
    } else {
        DBINDER_LOGE("Clear environment failed.");
        return -1;
    }
}