fix: write methods overload

src/lib/client.ts

@@ -956,6 +956,18 @@ export default class MqttClient extends TypedEventEmitter<MqttClientEventCallbac
 	 *     client.publish('topic', 'message', {qos: 1, retain: true, dup: true});
 	 * @example client.publish('topic', 'message', console.log);
 	 */
+	public publish(topic: string, message: string | Buffer): MqttClient
+	public publish(
+		topic: string,
+		message: string | Buffer,
+		callback?: DoneCallback,
+	): MqttClient
+	public publish(
+		topic: string,
+		message: string | Buffer,
+		opts?: IClientPublishOptions,
+		callback?: DoneCallback,
+	): MqttClient
 	public publish(
 		topic: string,
 		message: string | Buffer,
@@ -1058,6 +1070,22 @@ export default class MqttClient extends TypedEventEmitter<MqttClientEventCallbac
 	 * @example client.subscribe({'topic': {qos: 0}, 'topic2': {qos: 1}}, console.log);
 	 * @example client.subscribe('topic', console.log);
 	 */
+	public subscribe(
+		topicObject: string | string[] | ISubscriptionMap,
+	): MqttClient
+	public subscribe(
+		topicObject: string | string[] | ISubscriptionMap,
+		callback?: ClientSubscribeCallback,
+	): MqttClient
+	public subscribe(
+		topicObject: string | string[] | ISubscriptionMap,
+		opts?: IClientSubscribeOptions | IClientSubscribeProperties,
+	): MqttClient
+	public subscribe(
+		topicObject: string | string[] | ISubscriptionMap,
+		opts?: IClientSubscribeOptions | IClientSubscribeProperties,
+		callback?: ClientSubscribeCallback,
+	): MqttClient
 	public subscribe(
 		topicObject: string | string[] | ISubscriptionMap,
 		opts?:
@@ -1259,6 +1287,20 @@ export default class MqttClient extends TypedEventEmitter<MqttClientEventCallbac
 	 * @example client.unsubscribe('topic');
 	 * @example client.unsubscribe('topic', console.log);
 	 */
+	public unsubscribe(topic: string | string[]): MqttClient
+	public unsubscribe(
+		topic: string | string[],
+		opts?: IClientSubscribeOptions,
+	): MqttClient
+	public unsubscribe(
+		topic: string | string[],
+		callback?: PacketCallback,
+	): MqttClient
+	public unsubscribe(
+		topic: string | string[],
+		opts?: IClientSubscribeOptions,
+		callback?: PacketCallback,
+	): MqttClient
 	public unsubscribe(
 		topic: string | string[],
 		opts?: IClientSubscribeOptions | PacketCallback,
@@ -1348,6 +1390,15 @@ export default class MqttClient extends TypedEventEmitter<MqttClientEventCallbac
 	 *
 	 * @api public
 	 */
+	public end(cb?: DoneCallback): MqttClient
+	public end(force?: boolean): MqttClient
+	public end(opts?: Partial<IDisconnectPacket>, cb?: DoneCallback): MqttClient
+	public end(force?: boolean, cb?: DoneCallback): MqttClient
+	public end(
+		force?: boolean,
+		opts?: Partial<IDisconnectPacket>,
+		cb?: DoneCallback,
+	): MqttClient
 	public end(
 		force?: boolean | Partial<IDisconnectPacket> | DoneCallback,
 		opts?: Partial<IDisconnectPacket> | DoneCallback,

src/lib/connect/index.ts

@@ -50,7 +50,10 @@ function parseAuthOptions(opts: IClientOptions) {
 /**
  * connect - connect to an MQTT broker.
  */
-export default function connect(
+function connect(brokerUrl: string): MqttClient
+function connect(opts: IClientOptions): MqttClient
+function connect(brokerUrl: string, opts?: IClientOptions): MqttClient
+function connect(
 	brokerUrl: string | IClientOptions,
 	opts?: IClientOptions,
 ): MqttClient {
@@ -170,9 +173,12 @@ export default function connect(
 	client.on('error', () => {
 		/* Automatically set up client error handling */
 	})
+
 	return client
 }
 
+export default connect
+
 export {
 	Store,
 	MqttClient,

test/abstract_client.ts

@@ -59,7 +59,7 @@ export default function abstractTest(server, config) {
 		if (typeof opts === 'string') {
 			opts = { host: opts }
 		}
-		opts = { ...config, ...opts }
+		opts = { ...config, ...opts } as IClientOptions
 		return mqtt.connect(opts)
 	}