Cleaned up client

examples/client/main.cpp

@@ -0,0 +1,110 @@
+/**
+ * Copyright (c) 2019-2020 Paul-Louis Ageneau
+ * Copyright (c) 2019 Murat Dogan
+ * Copyright (c) 2020 Will Munn
+ * Copyright (c) 2020 Lara Mackey
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+
+#include "rtc/rtc.hpp"
+
+#include <nlohmann/json.hpp>
+
+#include <iostream>
+#include <memory>
+
+using namespace rtc;
+using namespace std;
+using namespace std::chrono_literals;
+
+using json = nlohmann::json;
+
+template <class T> weak_ptr<T> make_weak_ptr(shared_ptr<T> ptr) { return ptr; }
+
+int main(int argc, char **argv) {
+	rtc::InitLogger(LogLevel::Warning);
+
+	Configuration config;
+	config.iceServers.emplace_back("stun.l.google.com:19302"); // change to your STUN server
+
+	auto pc = std::make_shared<PeerConnection>(config);
+
+	const std::string url = "ws://localhost:8000/";
+
+	auto ws = std::make_shared<WebSocket>(url);
+	ws->onOpen([]() {
+
+	});
+	ws->onMessage([wpc = make_weak_ptr(pc)](const std::variant<binary, string> &data) {
+		auto pc = wpc.lock();
+		if (pc && holds_alternative<string>(data)) {
+			json message = json::parse(data.get<string>());
+			if (auto it = message.find("type")) {
+				auto type = it->second.get<string>();
+				if (type == "offer" || type == "answer") {
+					auto str = message["description"].get<string>();
+					pc->setRemoteDescription(Description(str, type));
+				} else if (type == "candidate") {
+					auto str = message["candidate"].get<string>();
+					auto mid = message["mid"].get<string>();
+					pc->addRemoteCandidate(Candidate(str, mid));
+				}
+			}
+		}
+	});
+	ws->onClosed([]() {
+
+	});
+
+	pc->onLocalDescription([wws = make_weak_ptr(ws)](const Description &description) {
+		json message = {{"type", description.typeString()},
+		                {"description", std::string(description)}};
+
+		if (auto ws = wws.lock())
+			ws->send(message.dump());
+	});
+	pc->onLocalCandidate([wws = make_weak_ptr(ws)](const Candidate &candidate) {
+		json message = {
+		    {"type", "candidate"}, {"candidate", std::string(candidate)}, {"mid", candidate.mid()}};
+
+		if (auto ws = wws.lock())
+			ws->send(message.dump());
+	});
+	pc->onStateChange([wpc = make_weak_ptr(pc)](PeerConnection::State state) {
+		cout << "State: " << state << endl;
+	});
+	pc->onGatheringStateChange(
+	    [](PeerConnection::GatheringState state) { cout << "Gathering State: " << state << endl; });
+
+	auto dc = pc->createDataChannel("test"); // this is the offerer, so create a data channel
+
+	dc->onOpen([&]() { cout << "DataChannel open: " << dc->label() << endl; });
+
+	dc->onClosed([&]() { cout << "DataChannel closed: " << dc->label() << endl; });
+
+	dc->onMessage([](const variant<binary, string> &message) {
+		if (holds_alternative<string>(message)) {
+			cout << "Received: " << get<string>(message) << endl;
+		}
+	});
+
+	// TODO
+
+	if (dc)
+		dc->close();
+	if (pc)
+		pc->close();
+}