mirror of
https://github.com/Luzifer/mqttcli.git
synced 2024-09-16 14:38:28 +00:00
110 lines
3.4 KiB
Go
110 lines
3.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
mqtt "github.com/eclipse/paho.mqtt.golang"
|
|
"github.com/gofrs/uuid"
|
|
"github.com/pkg/errors"
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/Luzifer/rconfig/v2"
|
|
)
|
|
|
|
var (
|
|
cfg = struct {
|
|
LogLevel string `flag:"log-level" default:"info" description:"Log level (debug, info, warn, error, fatal)"`
|
|
Message string `flag:"message,m" default:"" description:""`
|
|
MQTTBroker string `flag:"mqtt-broker,b" default:"tcp://localhost:1883" description:"Broker URI to connect to scheme://host:port (scheme is one of 'tcp', 'ssl', or 'ws')"`
|
|
MQTTClientID string `flag:"mqtt-client-id" vardefault:"client-id" description:"Client ID to use when connecting, must be unique"`
|
|
MQTTUser string `flag:"mqtt-user,u" default:"" description:"Username to identify against the broker" validate:"nonzero"`
|
|
MQTTPass string `flag:"mqtt-pass,p" default:"" description:"Password to identify against the broker" validate:"nonzero"`
|
|
MQTTTimeout time.Duration `flag:"mqtt-timeout" default:"10s" description:"How long to wait for the client to complete operations"`
|
|
OutputFormat string `flag:"output-format,o" default:"log" description:"How to ouptut received messages (One of 'log', 'csv', 'jsonl')"`
|
|
QOS int `flag:"qos" default:"1" description:"QOS to use (0 - Only Once, 1 - At Least Once, 2 - Only Once)"`
|
|
Retain bool `flag:"retain" default:"false" description:"Retain message on topic"`
|
|
Topics []string `flag:"topic,t" default:"" description:"Topic to subscribe / publish to"`
|
|
VersionAndExit bool `flag:"version" default:"false" description:"Prints current version and exits"`
|
|
}{}
|
|
|
|
version = "dev"
|
|
)
|
|
|
|
func initApp() error {
|
|
rconfig.AutoEnv(true)
|
|
|
|
rconfig.SetVariableDefaults(map[string]string{
|
|
"client-id": uuid.Must(uuid.NewV4()).String(),
|
|
})
|
|
|
|
if err := rconfig.ParseAndValidate(&cfg); err != nil {
|
|
return errors.Wrap(err, "parsing CLI options")
|
|
}
|
|
|
|
l, err := log.ParseLevel(cfg.LogLevel)
|
|
if err != nil {
|
|
return errors.Wrap(err, "parsing log-level")
|
|
}
|
|
log.SetLevel(l)
|
|
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
var err error
|
|
if err = initApp(); err != nil {
|
|
log.WithError(err).Fatal("initializing app")
|
|
}
|
|
|
|
if cfg.VersionAndExit {
|
|
fmt.Printf("mqttcli %s\n", version) //nolint:forbidigo
|
|
os.Exit(0)
|
|
}
|
|
|
|
var cmd string
|
|
if len(rconfig.Args()) > 1 {
|
|
cmd = rconfig.Args()[1]
|
|
}
|
|
|
|
if len(cfg.Topics) == 0 || (len(cfg.Topics) == 1 && cfg.Topics[0] == "") {
|
|
log.Fatal("No topic(s) specified")
|
|
}
|
|
|
|
client := mqtt.NewClient(
|
|
mqtt.NewClientOptions().
|
|
AddBroker(cfg.MQTTBroker).
|
|
SetClientID(cfg.MQTTClientID).
|
|
SetConnectionLostHandler(func(_ mqtt.Client, err error) {
|
|
log.WithError(err).Fatal("Connection to broker lost")
|
|
}).
|
|
SetKeepAlive(cfg.MQTTTimeout).
|
|
SetPassword(cfg.MQTTPass).
|
|
SetUsername(cfg.MQTTUser),
|
|
)
|
|
|
|
tok := client.Connect()
|
|
if !tok.WaitTimeout(cfg.MQTTTimeout) {
|
|
log.Fatal("Unable to connect within timeout")
|
|
}
|
|
if err := tok.Error(); err != nil {
|
|
log.WithError(err).Fatal("Unable to connect to broker")
|
|
}
|
|
|
|
switch cmd {
|
|
case "pub":
|
|
if err := publish(client); err != nil {
|
|
log.WithError(err).Fatal("Failed to publish message")
|
|
}
|
|
|
|
case "sub":
|
|
if err := subscribe(client); err != nil {
|
|
log.WithError(err).Fatal("Failed to subscribe and listen")
|
|
}
|
|
|
|
default:
|
|
log.Fatal("No command specified. Usage: mqttcli [opts] <pub|sub>")
|
|
}
|
|
}
|