#include "./stream_manager_ui.hpp"

#include "./content/sdl_video_frame_stream2.hpp"
#include "./content/audio_stream.hpp"

#include <solanaceae/object_store/object_store.hpp>

#include <imgui/imgui.h>

#include <string>

StreamManagerUI::StreamManagerUI(ObjectStore2& os, StreamManager& sm) : _os(os), _sm(sm) {
}

void StreamManagerUI::render(void) {
	if (ImGui::Begin("StreamManagerUI")) {
		// TODO: node canvas?

		// by fametype ??

		ImGui::SeparatorText("Sources");

		// list sources
		if (ImGui::BeginTable("sources_and_sinks", 4, ImGuiTableFlags_SizingFixedFit | ImGuiTableFlags_BordersInnerV)) {
			ImGui::TableSetupColumn("id");
			ImGui::TableSetupColumn("name");
			ImGui::TableSetupColumn("##conn");
			ImGui::TableSetupColumn("type");

			ImGui::TableHeadersRow();

			for (const auto& [oc, ss] : _os.registry().view<Components::StreamSource>().each()) {
				//ImGui::Text("src  %d (%s)[%s]", entt::to_integral(entt::to_entity(oc)), ss.name.c_str(), ss.frame_type_name.c_str());
				ImGui::PushID(entt::to_integral(oc));

				ImGui::TableNextColumn();
				ImGui::Text("%d", entt::to_integral(entt::to_entity(oc)));

				const auto *ssrc = _os.registry().try_get<Components::StreamSource>(oc);
				ImGui::TableNextColumn();
				ImGui::TextUnformatted(ssrc!=nullptr?ssrc->name.c_str():"none");

				ImGui::TableNextColumn();
				if (ImGui::SmallButton("->")) {
					// TODO: list type sinks
				}

				ImGui::TableNextColumn();
				ImGui::TextUnformatted(ssrc!=nullptr?ssrc->frame_type_name.c_str():"???");

				ImGui::PopID();
			}

			ImGui::EndTable();
		}

		ImGui::SeparatorText("Sinks");

		// list sinks
		if (ImGui::BeginTable("sources_and_sinks", 4, ImGuiTableFlags_SizingFixedFit | ImGuiTableFlags_BordersInnerV)) {
			ImGui::TableSetupColumn("id");
			ImGui::TableSetupColumn("name");
			ImGui::TableSetupColumn("##conn");
			ImGui::TableSetupColumn("type");

			ImGui::TableHeadersRow();

			for (const auto& [oc, ss] : _os.registry().view<Components::StreamSink>().each()) {
				//ImGui::Text("sink %d (%s)[%s]", entt::to_integral(entt::to_entity(oc)), ss.name.c_str(), ss.frame_type_name.c_str());

				ImGui::PushID(entt::to_integral(oc));

				ImGui::TableNextColumn();
				ImGui::Text("%d", entt::to_integral(entt::to_entity(oc)));

				const auto *ssink = _os.registry().try_get<Components::StreamSink>(oc);
				ImGui::TableNextColumn();
				ImGui::TextUnformatted(ssink!=nullptr?ssink->name.c_str():"none");

				ImGui::TableNextColumn();
				if (ImGui::SmallButton("->")) {
					// TODO: list type sinks
				}
				if (ImGui::BeginPopupContextItem("sink_connect")) {
					if (ImGui::BeginMenu("connect video", ss.frame_type_name == entt::type_name<SDLVideoFrame>::value())) {
						for (const auto& [oc_src, s_src] : _os.registry().view<Components::StreamSource>().each()) {
							if (s_src.frame_type_name != ss.frame_type_name) {
								continue;
							}

							ImGui::PushID(entt::to_integral(oc_src));

							std::string source_label {"src "};
							source_label += std::to_string(entt::to_integral(entt::to_entity(oc_src)));
							source_label += " (";
							source_label += s_src.name;
							source_label += ")[";
							source_label += s_src.frame_type_name;
							source_label += "]";
							if (ImGui::MenuItem(source_label.c_str())) {
								_sm.connect<SDLVideoFrame>(oc_src, oc);
							}

							ImGui::PopID();
						}

						ImGui::EndMenu();
					}

					if (ImGui::BeginMenu("connect audio", ss.frame_type_name == entt::type_name<AudioFrame>::value())) {
						for (const auto& [oc_src, s_src] : _os.registry().view<Components::StreamSource>().each()) {
							if (s_src.frame_type_name != ss.frame_type_name) {
								continue;
							}

							ImGui::PushID(entt::to_integral(oc_src));

							std::string source_label {"src "};
							source_label += std::to_string(entt::to_integral(entt::to_entity(oc_src)));
							source_label += " (";
							source_label += s_src.name;
							source_label += ")[";
							source_label += s_src.frame_type_name;
							source_label += "]";
							if (ImGui::MenuItem(source_label.c_str())) {
								_sm.connect<AudioFrame>(oc_src, oc);
							}

							ImGui::PopID();
						}

						ImGui::EndMenu();
					}
					ImGui::EndPopup();
				}

				ImGui::TableNextColumn();
				ImGui::TextUnformatted(ssink!=nullptr?ssink->frame_type_name.c_str():"???");

				ImGui::PopID();
			}

			ImGui::EndTable();
		}

		ImGui::SeparatorText("Connections");

		// list connections
		if (ImGui::BeginTable("connections", 6, ImGuiTableFlags_SizingFixedFit | ImGuiTableFlags_BordersInnerV)) {
			ImGui::TableSetupColumn("##id"); // TODO: remove?
			ImGui::TableSetupColumn("##disco");
			ImGui::TableSetupColumn("##qdesc");
			ImGui::TableSetupColumn("from");
			ImGui::TableSetupColumn("to");
			ImGui::TableSetupColumn("type");

			ImGui::TableHeadersRow();

			//for (const auto& con : _sm._connections) {
			for (size_t i = 0; i < _sm._connections.size(); i++) {
				const auto& con = _sm._connections[i];
				//ImGui::Text("con %d->%d", entt::to_integral(entt::to_entity(con->src.entity())), entt::to_integral(entt::to_entity(con->sink.entity())));

				ImGui::PushID(i);

				ImGui::TableNextColumn();
				ImGui::Text("%zu", i); // do connections have ids?

				ImGui::TableNextColumn();
				if (ImGui::SmallButton("X")) {
					// TODO: disconnect
				}

				ImGui::TableNextColumn();
				ImGui::Text("%d->%d", entt::to_integral(entt::to_entity(con->src.entity())), entt::to_integral(entt::to_entity(con->sink.entity())));

				const auto *ssrc = con->src.try_get<Components::StreamSource>();
				ImGui::TableNextColumn();
				ImGui::TextUnformatted(ssrc!=nullptr?ssrc->name.c_str():"none");

				const auto *ssink = con->sink.try_get<Components::StreamSink>();
				ImGui::TableNextColumn();
				ImGui::TextUnformatted(ssink!=nullptr?ssink->name.c_str():"none");

				ImGui::TableNextColumn();
				ImGui::TextUnformatted(
					(ssrc!=nullptr)?
						ssrc->frame_type_name.c_str():
						(ssink!=nullptr)?
							ssink->frame_type_name.c_str()
							:"???"
				);

				ImGui::PopID();
			}
			ImGui::EndTable();
		}
	}
	ImGui::End();
}