// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/webui/version/version_util_win.h"

#include "base/strings/stringprintf.h"
#include "base/strings/utf_string_conversions.h"
#include "base/win/windows_version.h"
#include "chrome/install_static/install_details.h"
#include "components/strings/grit/components_strings.h"
#include "ui/base/l10n/l10n_util.h"

namespace version_utils {
namespace win {

std::string GetFullWindowsVersion() {
  std::string version;
  base::win::OSInfo* gi = base::win::OSInfo::GetInstance();
  const uint32_t major = gi->version_number().major;
  const uint32_t minor = gi->version_number().minor;
  const uint32_t build = gi->version_number().build;
  const uint32_t patch = gi->version_number().patch;
  // Server or Desktop
  const bool server =
      gi->version_type() == base::win::VersionType::SUITE_SERVER;

  if (major == 11) {
    version += (server) ? "Server OS" : "11";
  } else if (major == 10) {
    if (build < 20348) {
      version += (server) ? "Server OS" : "10";
    } else if (build < 22000) {
      version += (server) ? "Server 2022" : "10";
    } else {
      version += (server) ? "Server OS" : "11";
    }
  } else if (major == 6) {
    switch (minor) {
      case 0:
        // Windows Vista or Server 2008
        version += (server) ? "Server 2008" : "Vista";
        break;
      case 1:
        // Windows 7 or Server 2008 R2
        version += (server) ? "Server 2008 R2" : "7";
        break;
      case 2:
        // Windows 8 or Server 2012
        version += (server) ? "Server 2012" : "8";
        break;
      case 3:
        // Windows 8.1 or Server 2012 R2
        version += (server) ? "Server 2012 R2" : "8.1";
        break;
      default:
        // unknown version
        return base::StringPrintf("unknown version 6.%u", minor);
    }
  } else if ((major == 5) && (minor > 0)) {
    // Windows XP or Server 2003
    version += (server) ? "Server 2003" : "XP";
  } else {
    // unknown version
    return base::StringPrintf("unknown version %u.%u", major, minor);
  }

  const std::string release_id = gi->release_id();

  if (!release_id.empty()) {
    version += " Version " + release_id;
  }

  if (patch > 0) {
    version += base::StringPrintf(" (Build %u.%u)", build, patch);
  } else {
    version += base::StringPrintf(" (Build %u)", build);
  }
  return version;
}

std::u16string GetCohortVersionInfo() {
  std::wstring update_cohort_name =
      install_static::InstallDetails::Get().update_cohort_name();
  if (!update_cohort_name.empty()) {
    return l10n_util::GetStringFUTF16(IDS_VERSION_UI_COHORT_NAME,
                                      base::WideToUTF16(update_cohort_name));
  }

  return std::u16string();
}

}  // namespace win
}  // namespace version_utils