/*
 * Copyright (C) 2016 The Android Open Source Project
 *
 * 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 "utils/InterfaceSpecUtil.h"

#include <fstream>
#include <sstream>
#include <string>

#include <android-base/logging.h>
#include <assert.h>
#include <google/protobuf/message.h>
#include <google/protobuf/text_format.h>

#include "utils/StringUtil.h"
#include "test/vts/proto/ComponentSpecificationMessage.pb.h"

using namespace std;

namespace android {
namespace vts {

bool ParseInterfaceSpec(const char* file_path,
                        ComponentSpecificationMessage* message) {
  ifstream in_file(file_path);
  stringstream str_stream;
  if (!in_file.is_open()) {
    LOG(ERROR) << "Unable to open file. " << file_path;
    return false;
  }
  str_stream << in_file.rdbuf();
  in_file.close();
  const string data = str_stream.str();

  message->Clear();
  if (!google::protobuf::TextFormat::MergeFromString(data, message)) {
    LOG(ERROR) << "Can't parse a given proto file " << file_path;
    return false;
  }
  return true;
}

string GetFunctionNamePrefix(const ComponentSpecificationMessage& message) {
  stringstream prefix_ss;
  if (message.component_class() != HAL_HIDL) {
    prefix_ss << VTS_INTERFACE_SPECIFICATION_FUNCTION_NAME_PREFIX
              << message.component_class() << "_" << message.component_type()
              << "_" << GetVersionString(message.component_type_version(), true)
              << "_";
  } else {
    string package_as_function_name(message.package());
    ReplaceSubString(package_as_function_name, ".", "_");
    prefix_ss << VTS_INTERFACE_SPECIFICATION_FUNCTION_NAME_PREFIX
        << message.component_class() << "_" << package_as_function_name << "_"
        << GetVersionString(message.component_type_version(), true) << "_"
        << message.component_name() << "_";
  }
  return prefix_ss.str();
}

#define DEFAULT_FACTOR 10000

string GetVersionString(float version, bool for_macro) {
  std::ostringstream out;
  if (for_macro) {
    out << "V";
  }
  long version_long = version * DEFAULT_FACTOR;
  out << (version_long / DEFAULT_FACTOR);
  if (!for_macro) {
    out << ".";
  } else {
    out << "_";
  }
  version_long -= (version_long / DEFAULT_FACTOR) * DEFAULT_FACTOR;
  bool first = true;
  long factor = DEFAULT_FACTOR / 10;
  while (first || (version_long > 0 && factor > 1)) {
    out << (version_long / factor);
    version_long -= (version_long / factor) * factor;
    factor /= 10;
    first = false;
  }
  return out.str();
}

string GetHidlHalDriverLibName(const string& package_name,
                               const float version) {
  return package_name + "@" + GetVersionString(version) + "-vts.driver.so";
}

string GetInterfaceFQName(const string& package_name, const float version,
                          const string& interface_name) {
  return package_name + "@" + GetVersionString(version) + "::" + interface_name;
}

string GetPackageName(const string& type_name) {
  string str = type_name.substr(0, type_name.find('V') - strlen("::"));
  if (str.find("::") == 0) {
    str = str.substr(strlen("::"));
  }
  ReplaceSubString(str, "::", ".");
  return str;
}

float GetVersion(const string& type_name) {
  string str = type_name.substr(type_name.find('V') + 1);
  string version_str = str.substr(0, str.find("::"));
  string major_version = version_str.substr(0, version_str.find("_"));
  string minor_version = version_str.substr(version_str.find("_") + 1);
  // TODO(zhuoyao): handle the case when minor_version >= 10
  assert(std::stof(minor_version) < 10.0);
  return std::stof(major_version) + 0.1 * (std::stof(minor_version));
}

string GetComponentName(const string& type_name) {
  string str = type_name.substr(type_name.find('V'));
  return str.substr(str.find("::") + strlen("::"));
}

}  // namespace vts
}  // namespace android