// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

#include "HumanData.h"

#include "DroidData.h"

namespace graphql::learn {

Human::Human(std::string&& id, std::optional<std::string>&& name, std::vector<Episode>&& appearsIn,
	std::optional<std::string>&& homePlanet) noexcept
	: id_ { std::move(id) }
	, name_ { std::move(name) }
	, appearsIn_ { std::move(appearsIn) }
	, homePlanet_ { std::move(homePlanet) }
{
}

void Human::addFriends(std::vector<SharedHero> friends) noexcept
{
	friends_.resize(friends.size());

	std::transform(friends.begin(),
		friends.end(),
		friends_.begin(),
		[](const auto& spFriend) noexcept {
			return std::visit(
				[](const auto& hero) noexcept {
					return WeakHero {
						std::weak_ptr<typename std::decay_t<decltype(hero)>::element_type> { hero }
					};
				},
				spFriend);
		});
}

const response::IdType& Human::getId() const noexcept
{
	return id_;
}

const std::optional<std::string>& Human::getName() const noexcept
{
	return name_;
}

std::optional<std::vector<std::shared_ptr<object::Character>>> Human::getFriends() const noexcept
{
	std::vector<std::shared_ptr<object::Character>> result(friends_.size());

	std::transform(friends_.begin(),
		friends_.end(),
		result.begin(),
		[](const auto& wpFriend) noexcept {
			return make_hero(wpFriend);
		});
	result.erase(std::remove(result.begin(), result.end(), std::shared_ptr<object::Character> {}),
		result.end());

	return result.empty() ? std::nullopt : std::make_optional(std::move(result));
}

std::optional<std::vector<std::optional<Episode>>> Human::getAppearsIn() const noexcept
{
	std::vector<std::optional<Episode>> result(appearsIn_.size());

	std::transform(appearsIn_.begin(),
		appearsIn_.end(),
		result.begin(),
		[](const auto& entry) noexcept {
			return std::make_optional(entry);
		});

	return result.empty() ? std::nullopt : std::make_optional(std::move(result));
}

const std::optional<std::string>& Human::getHomePlanet() const noexcept
{
	return homePlanet_;
}

} // namespace graphql::learn