src/MQTTHandler.h
author František Kučera <franta-hg@frantovo.cz>
Sun, 05 Jun 2022 00:08:52 +0200
branchv_0
changeset 3 62ced7c41c67
parent 2 1a0fbd17db13
permissions -rw-r--r--
parse connection string, credentials, check return values
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     1
/**
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     2
 * Relational pipes
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     3
 * Copyright © 2022 František Kučera (Frantovo.cz, GlobalCode.info)
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     4
 *
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     5
 * This program is free software: you can redistribute it and/or modify
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     6
 * it under the terms of the GNU General Public License as published by
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     7
 * the Free Software Foundation, version 3 of the License.
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     8
 *
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     9
 * This program is distributed in the hope that it will be useful,
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    12
 * GNU General Public License for more details.
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    13
 *
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    14
 * You should have received a copy of the GNU General Public License
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    15
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    16
 */
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    17
#pragma once
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    18
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    19
#include <memory>
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    20
#include <string>
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    21
#include <vector>
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    22
#include <iostream>
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    23
#include <sstream>
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    24
#include <locale>
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    25
#include <codecvt>
2
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    26
#include <random>
3
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    27
#include <regex>
0
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    28
1
cb9577442d3b link to mosquittopp library
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    29
#include <mosquittopp.h>
cb9577442d3b link to mosquittopp library
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    30
0
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    31
#include <relpipe/common/type/typedefs.h>
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    32
#include <relpipe/reader/TypeId.h>
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    33
#include <relpipe/reader/handlers/RelationalReaderStringHandler.h>
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    34
#include <relpipe/reader/handlers/AttributeMetadata.h>
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    35
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    36
#include "Configuration.h"
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    37
#include "Hex.h"
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    38
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    39
namespace relpipe {
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    40
namespace out {
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    41
namespace mqtt {
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    42
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    43
class MQTTHandler : public relpipe::reader::handlers::RelationalReaderStringHandler {
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    44
private:
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    45
	std::wstring_convert<codecvt_utf8<wchar_t>> convertor; // TODO: support also other encodings.
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    46
	Configuration configuration;
2
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    47
	std::shared_ptr<mosqpp::mosquittopp> mq = std::make_shared<mosqpp::mosquittopp>(generateClientID().c_str());
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    48
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    49
	/**
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    50
	 * @return unique (random) client ID for MQTT to allow multiple simultaneous connections
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    51
	 */
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    52
	static std::string generateClientID() {
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    53
		std::stringstream result;
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    54
		std::string symbols("0123456789abcdef");
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    55
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    56
		std::random_device dev;
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    57
		std::mt19937 rng(dev());
3
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    58
		std::uniform_int_distribution<std::mt19937::result_type> dist(0, symbols.size());
2
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    59
3
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    60
		for (int i = 0; i < 8; i++) result << symbols[dist(rng)];
2
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    61
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    62
		return result.str();
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    63
	}
0
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    64
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    65
	struct CurrentRelation {
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    66
		relpipe::common::type::StringX name;
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    67
		std::vector<relpipe::reader::handlers::AttributeMetadata> attributes;
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    68
		relpipe::common::type::Integer attributeIndex = 0;
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    69
		std::string currentValue;
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    70
	} currentRelation;
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    71
3
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    72
	static void parseConnectionString(const std::string& connectionString, std::string& hostname, int& port) {
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    73
		std::regex pattern("mqtt:(//)?([^:]+)(:([0-9]+))?");
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    74
		std::smatch match;
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    75
		if (std::regex_match(connectionString, match, pattern)) {
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    76
			hostname = match[2];
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    77
			port = stoi(match[4]);
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    78
		} else {
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    79
			throw std::invalid_argument("Invalid connection string format. Expecting something like: mqtt://localhost:1883");
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    80
		}
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    81
	}
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    82
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    83
	static void check(std::string operation, int result) {
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    84
		if (result) throw std::logic_error("mosquitto operation failed: " + operation + " = " + std::to_string(result));
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    85
	}
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    86
0
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    87
public:
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    88
3
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    89
	static MQTTHandler* create(Configuration configuration) {
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    90
		MQTTHandler* h = new MQTTHandler();
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    91
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    92
		std::string connectionString = h->convertor.to_bytes(configuration.connectionString);
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    93
		std::string username;
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    94
		std::string password;
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    95
		std::string hostname;
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    96
		int port;
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    97
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    98
		parseConnectionString(connectionString, hostname, port);
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
    99
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   100
		for (auto o : configuration.connectionOptions) {
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   101
			if (o.name == L"username") username = h->convertor.to_bytes(o.value);
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   102
			else if (o.name == L"password") password = h->convertor.to_bytes(o.value);
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   103
			else throw std::invalid_argument("Unsupported connection option: " + h->convertor.to_bytes(o.name));
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   104
		}
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   105
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   106
		if (username.size()) check("set credentials", h->mq->username_pw_set(username.c_str(), password.c_str()));
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   107
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   108
		check("connect", h->mq->connect(hostname.c_str(), port));
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   109
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   110
		return h;
0
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   111
	}
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   112
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   113
	void startRelation(relpipe::common::type::StringX name, std::vector<relpipe::reader::handlers::AttributeMetadata> attributes) override {
2
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   114
		// TODO: check relation name according to the configuration
0
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   115
		currentRelation = CurrentRelation{name, attributes};
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   116
	}
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   117
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   118
	void attribute(const relpipe::common::type::StringX& value) override {
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   119
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   120
		auto attributeName = currentRelation.attributes[currentRelation.attributeIndex].getAttributeName();
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   121
		if (attributeName == L"text" && value.size()) currentRelation.currentValue = convertor.to_bytes(value);
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   122
		else if (attributeName == L"data" && value.size()) currentRelation.currentValue = Hex::fromHex(value).str();
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   123
		else if (attributeName == L"text"); // keep empty or value from 'data'
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   124
		else if (attributeName == L"data"); // keep empty or value from 'text'
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   125
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   126
		currentRelation.attributeIndex++;
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   127
		if (currentRelation.attributeIndex == currentRelation.attributes.size()) {
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   128
			currentRelation.attributeIndex = 0;
2
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   129
			int mid = -1;
3
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   130
			check("publish", mq->publish(&mid, convertor.to_bytes(configuration.stream).c_str(), currentRelation.currentValue.size(), currentRelation.currentValue.c_str()));
2
1a0fbd17db13 first version
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   131
			// std::cerr << "MQTT message enqueued: " << mid << std::endl;
0
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   132
		}
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   133
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   134
	}
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   135
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   136
	void endOfPipe() {
3
62ced7c41c67 parse connection string, credentials, check return values
František Kučera <franta-hg@frantovo.cz>
parents: 2
diff changeset
   137
		check("disconnect", mq->disconnect());
0
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   138
	}
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   139
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   140
};
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   141
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   142
}
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   143
}
7ef5ce9477c8 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   144
}