xref: /test/ostest/wukong/report/src/format_csv.cpp

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

#include <iomanip>
#include <sstream>

namespace OHOS {
namespace WuKong {
FormatCSV::FormatCSV() : Format()
{
}

FormatCSV::~FormatCSV()
{
}

void FormatCSV::FormatDetail(std::shared_ptr<Table> tablePtr, std::string &target)
{
    std::vector<std::string> header = tablePtr->GetHeader();
    std::vector<std::vector<std::string>> record = tablePtr->GetRecord();
    std::vector<int> column_size_ = tablePtr->GetColumnSize();
    std::stringstream ss;

    ss << "name, " << tablePtr->GetName() << ", detail, " << tablePtr->GetDetail() << std::endl;
    for (uint32_t col = 0; col < header.size(); col++) {
        ss << std::setw(column_size_[col]) << std::setiosflags(std::ios::left) << std::setfill(' ') << header[col];
        if (col == (header.size() - 1)) {
            break;
        }
        ss << ',';
    }
    ss << std::endl;

    for (uint32_t row = 0; row < record.size(); row++) {
        for (uint32_t col = 0; col < header.size(); col++) {
            ss << std::setw(column_size_[col]) << std::setiosflags(std::ios::left) << std::setfill(' ');
            ss << record[row][col];
            if (col == (header.size() - 1)) {
                break;
            }
            ss << ',';
        }
        ss << std::endl;
    }
    target += ss.str();
}
}  // namespace WuKong
}  // namespace OHOS