package com.tencent.tdmq.mqtt.example.paho.v5;

import java.nio.charset.StandardCharsets;
import java.util.List;
import org.eclipse.paho.mqttv5.client.IMqttToken;
import org.eclipse.paho.mqttv5.client.MqttCallback;
import org.eclipse.paho.mqttv5.client.MqttClient;
import org.eclipse.paho.mqttv5.client.MqttConnectionOptions;
import org.eclipse.paho.mqttv5.client.MqttDisconnectResponse;
import org.eclipse.paho.mqttv5.client.persist.MemoryPersistence;
import org.eclipse.paho.mqttv5.common.MqttException;
import org.eclipse.paho.mqttv5.common.MqttMessage;
import org.eclipse.paho.mqttv5.common.packet.MqttProperties;
import org.eclipse.paho.mqttv5.common.packet.UserProperty;

public class PublisherQuickStart {

    public static void main(String[] args) throws MqttException, InterruptedException {
    
        // Get the access point from the MQTT console:
        // For users implementing VPC connectivity via Private Link, use the private network access point.
        // For users accessing over the public network, ensure the public network security policy permits access, and the machine running the program has public network connectivity.
        String serverUri = "tcp://mqtt-xxx.mqtt.tencenttdmq.com:1883";

        // A valid client identifier contains digits 0-9, lowercase letters a-z, and uppercase letters A-Z, with a total length of 1-23 characters
        // See https://docs.oasis-open.org/mqtt/mqtt/v5.0/os/mqtt-v5.0-os.html#_Toc3901059.
        String clientId = "PublisherQuickStart";

        // In the console, on the Authentication tab, create an account and copy the username and password.
        String username = "user0";
        String password = "secret0";

        // MQTT topic
        String topicName = "home/test";

        MqttClient client = new MqttClient(serverUri, clientId, new MemoryPersistence());
        client.setTimeToWait(3000);
        MqttConnectionOptions options = new MqttConnectionOptions();
        options.setUserName(username);
        options.setPassword(password.getBytes(StandardCharsets.UTF_8));
        options.setCleanStart(true);
        options.setAutomaticReconnect(true);

        client.setCallback(new MqttCallback() {
            @Override
            public void disconnected(MqttDisconnectResponse response) {
                System.out.println("Disconnected: " + response.getReasonString());
            }

            @Override
            public void mqttErrorOccurred(MqttException e) {
                e.printStackTrace();
            }

            @Override
            public void messageArrived(String topic, MqttMessage message) {
            }

            @Override
            public void deliveryComplete(IMqttToken token) {
                System.out.println("Delivery completed: packet-id=" + token.getMessageId() +
                    ", reason-code=" + token.getResponse().getReasonCodes()[0]);

                List<UserProperty> userProperties = token.getResponseProperties().getUserProperties();
                printUserProperties(userProperties);
            }

            @Override
            public void connectComplete(boolean reconnect, String serverURI) {
                System.out.println(reconnect ? "Reconnected" : "Connected" + " to " + serverURI);
            }

            @Override
            public void authPacketArrived(int i, MqttProperties properties) {
                System.out.println("Received auth packet with id: " + i);
            }
        });

        client.connect(options);

        int total = 16;
        for (int i = 0; i < total; i++) {
            String msg = "Hello MQTT " + i;
            MqttMessage message = new MqttMessage(msg.getBytes(StandardCharsets.UTF_8));
            message.setQos(1);
            System.out.printf("Prepare to publish message %d%n", i);
            client.publish(topicName, message);
            System.out.printf("Published message %d%n", i);
        }

        client.disconnect();

        client.close();
    }

    static void printUserProperties(List<UserProperty> userProperties) {
        if (null != userProperties) {
            for (UserProperty userProperty : userProperties) {
                System.out.printf("User property: %s = %s%n", userProperty.getKey(), userProperty.getValue());
            }
        }
    }
}