heyujie
2021-05-24 4885600ecc369aa2e30a65de8dd7a410f13c34df
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
#!/usr/bin/env node
 
var mqtt = require('../')
var path = require('path')
var fs = require('fs')
var helpMe = require('help-me')({
  dir: path.join(__dirname, '..', 'doc')
})
var minimist = require('minimist')
 
function start (args) {
  args = minimist(args, {
    string: ['hostname', 'username', 'password', 'key', 'cert', 'ca', 'clientId', 'i', 'id'],
    boolean: ['stdin', 'help', 'clean', 'insecure'],
    alias: {
      port: 'p',
      hostname: ['h', 'host'],
      topic: 't',
      qos: 'q',
      clean: 'c',
      keepalive: 'k',
      clientId: ['i', 'id'],
      username: 'u',
      password: 'P',
      protocol: ['C', 'l'],
      verbose: 'v',
      help: '-H',
      ca: 'cafile'
    },
    default: {
      host: 'localhost',
      qos: 0,
      retain: false,
      clean: true,
      keepAlive: 30 // 30 sec
    }
  })
 
  if (args.help) {
    return helpMe.toStdout('subscribe')
  }
 
  args.topic = args.topic || args._.shift()
 
  if (!args.topic) {
    console.error('missing topic\n')
    return helpMe.toStdout('subscribe')
  }
 
  if (args.key) {
    args.key = fs.readFileSync(args.key)
  }
 
  if (args.cert) {
    args.cert = fs.readFileSync(args.cert)
  }
 
  if (args.ca) {
    args.ca = fs.readFileSync(args.ca)
  }
 
  if (args.key && args.cert && !args.protocol) {
    args.protocol = 'mqtts'
  }
 
  if (args.insecure) {
    args.rejectUnauthorized = false
  }
 
  if (args.port) {
    if (typeof args.port !== 'number') {
      console.warn('# Port: number expected, \'%s\' was given.', typeof args.port)
      return
    }
  }
 
  if (args['will-topic']) {
    args.will = {}
    args.will.topic = args['will-topic']
    args.will.payload = args['will-message']
    args.will.qos = args['will-qos']
    args.will.retain = args['will-retain']
  }
 
  args.keepAlive = args['keep-alive']
 
  var client = mqtt.connect(args)
 
  client.on('connect', function () {
    client.subscribe(args.topic, { qos: args.qos }, function (err, result) {
      if (err) {
        console.error(err)
        process.exit(1)
      }
 
      result.forEach(function (sub) {
        if (sub.qos > 2) {
          console.error('subscription negated to', sub.topic, 'with code', sub.qos)
          process.exit(1)
        }
      })
    })
  })
 
  client.on('message', function (topic, payload) {
    if (args.verbose) {
      console.log(topic, payload.toString())
    } else {
      console.log(payload.toString())
    }
  })
 
  client.on('error', function (err) {
    console.warn(err)
    client.end()
  })
}
 
module.exports = start
 
if (require.main === module) {
  start(process.argv.slice(2))
}