/*
 * Copyright (c) 2023 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 "rdb_global_params_helper.h"

#include "data_storage_errors.h"
#include "data_storage_log_wrapper.h"
#include "parser_util.h"
#include "global_params_data.h"
#include "rdb_errno.h"
#include "rdb_global_params_callback.h"
#include "rdb_store_config.h"
#include "values_bucket.h"
#include "vector"

namespace OHOS {
namespace Telephony {
RdbGlobalParamsHelper::RdbGlobalParamsHelper() {}

int RdbGlobalParamsHelper::Init()
{
    int errCode = NativeRdb::E_OK;
    NativeRdb::RdbStoreConfig config(dbPath_);
    config.SetJournalMode(NativeRdb::JournalMode::MODE_TRUNCATE);
    std::string numMatchTableStr;
    CreateGlobalParamsTableStr(numMatchTableStr, TABLE_NUMBER_MATCH);
    std::string numMatchIndexStr;
    CreateNumMatchIndexStr(numMatchIndexStr);
    std::string eccDataTableStr;
    CreateGlobalParamsTableStr(eccDataTableStr, TABLE_ECC_DATA);
    std::vector<std::string> createTableVec;
    createTableVec.push_back(numMatchTableStr);
    createTableVec.push_back(numMatchIndexStr);
    createTableVec.push_back(eccDataTableStr);
    RdbGlobalParamsCallback callback(createTableVec);
    CreateRdbStore(config, VERSION, callback, errCode);
    return errCode;
}

void RdbGlobalParamsHelper::UpdateDbPath(const std::string &path)
{
    dbPath_ = path + DB_NAME;
}

void RdbGlobalParamsHelper::CreateNumMatchTableStr(std::string &createTableStr)
{
    DATA_STORAGE_LOGD("RdbGlobalParamsHelper::CreateNumMatchTableStr start");
    createTableStr.append("CREATE TABLE IF NOT EXISTS ").append(TABLE_NUMBER_MATCH).append("(");
    createTableStr.append(NumMatchData::ID).append(" INTEGER PRIMARY KEY AUTOINCREMENT, ");
    createTableStr.append(NumMatchData::NAME).append(" TEXT DEFAULT '', ");
    createTableStr.append(NumMatchData::MCC).append(" TEXT DEFAULT '', ");
    createTableStr.append(NumMatchData::MNC).append(" TEXT DEFAULT '', ");
    createTableStr.append(NumMatchData::MCCMNC).append(" TEXT NOT NULL , ");
    createTableStr.append(NumMatchData::NUM_MATCH).append(" INTEGER DEFAULT 0, ");
    createTableStr.append(NumMatchData::NUM_MATCH_SHORT).append(" INTEGER DEFAULT 0, ");
    createTableStr.append("UNIQUE (").append(NumMatchData::MCCMNC).append(", ");
    createTableStr.append(NumMatchData::NAME).append("))");
    DATA_STORAGE_LOGD("RdbGlobalParamsHelper::CreateNumMatchTableStr end: %s", createTableStr.c_str());
}

void RdbGlobalParamsHelper::CreateNumMatchIndexStr(std::string &createIndexStr)
{
    DATA_STORAGE_LOGD("RdbGlobalParamsHelper::CreateNumMatchIndexStr start");
    createIndexStr.append("CREATE INDEX IF NOT EXISTS [").append(NUMERIC_INDEX).append("]");
    createIndexStr.append("ON [").append(TABLE_NUMBER_MATCH).append("]");
    createIndexStr.append("([").append(NumMatchData::MCCMNC).append("])");
    DATA_STORAGE_LOGD("RdbGlobalParamsHelper::CreateNumMatchIndexStr end: %s", createIndexStr.c_str());
}

void RdbGlobalParamsHelper::CreateEccDataTableStr(std::string &createTableStr)
{
    createTableStr.append("CREATE TABLE IF NOT EXISTS ").append(TABLE_ECC_DATA).append("(");
    createTableStr.append(EccData::ID).append(" INTEGER PRIMARY KEY AUTOINCREMENT, ");
    createTableStr.append(EccData::NAME).append(" TEXT DEFAULT '', ");
    createTableStr.append(EccData::MCC).append(" TEXT DEFAULT '', ");
    createTableStr.append(EccData::MNC).append(" TEXT DEFAULT '', ");
    createTableStr.append(EccData::NUMERIC).append(" TEXT DEFAULT '', ");
    createTableStr.append(EccData::ECC_WITH_CARD).append(" TEXT DEFAULT '', ");
    createTableStr.append(EccData::ECC_NO_CARD).append(" TEXT DEFAULT '', ");
    createTableStr.append(EccData::ECC_FAKE).append(" TEXT DEFAULT '', ");
    createTableStr.append("UNIQUE (").append(EccData::NUMERIC).append("))");
}
void RdbGlobalParamsHelper::CreateGlobalParamsTableStr(std::string &createTableStr, const std::string &tableName)
{
    if (tableName == TABLE_ECC_DATA) {
        return CreateEccDataTableStr(createTableStr);
    } else if (tableName == TABLE_NUMBER_MATCH) {
        return CreateNumMatchTableStr(createTableStr);
    } else {
        DATA_STORAGE_LOGI("TableName is not TABLE_ECC_DATA or TABLE_NUMBER_MATCH");
    }
    return;
}
int RdbGlobalParamsHelper::CommitTransactionAction()
{
    int result = Commit();
    if (result != NativeRdb::E_OK) {
        RollBack();
    }
    return result;
}
} // namespace Telephony
} // namespace OHOS