diff --git a/bin/humbug-send b/bin/humbug-send new file mode 100755 index 0000000..36761fd --- /dev/null +++ b/bin/humbug-send @@ -0,0 +1,123 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- +# humbug-send -- Sends a message to the specified recipients. + +# Copyright © 2012 Humbug, Inc. +# +# Permission is hereby granted, free of charge, to any person obtaining a copy +# of this software and associated documentation files (the "Software"), to deal +# in the Software without restriction, including without limitation the rights +# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +# copies of the Software, and to permit persons to whom the Software is +# furnished to do so, subject to the following conditions: +# +# The above copyright notice and this permission notice shall be included in +# all copies or substantial portions of the Software. +# +# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN +# THE SOFTWARE. + +import sys +import os +import optparse +import logging + + +logging.basicConfig() + +log = logging.getLogger('humbug-send') + +def do_send_message(client, message_data ): + '''Sends a message and optionally prints status about the same.''' + + if message_data['type'] == 'stream': + log.info('Sending message to stream "%s", subject "%s"... ' % \ + (message_data['to'], message_data['subject'])) + else: + log.info('Sending message to %s... ' % message_data['to']) + response = client.send_message(message_data) + if response['result'] == 'success': + log.info('Message sent.') + return True + else: + log.error(response['msg']) + return False + +def main(argv=None): + if argv is None: + argv = sys.argv + + usage = """%prog [options] [recipient...] + + Sends a message specified recipients. + + Examples: %prog --stream denmark --subject castle -m "Something is rotten in the state of Denmark." + %prog hamlet@example.com cordelia@example.com -m "Conscience doth make cowards of us all." + """ + + sys.path.append(os.path.join(os.path.dirname(__file__), os.path.join('..', '..'))) + + import api.humbug + + parser = optparse.OptionParser(usage=usage) + + # Grab parser options from the API common set + parser.add_option_group(api.humbug.generate_option_group(parser)) + + parser.add_option('-m', '--message', + help='Specifies the message to send, prevents interactive prompting.') + + group = optparse.OptionGroup(parser, 'Stream parameters') + group.add_option('-s', '--stream', + dest='stream', + action='store', + help='Allows the user to specify a stream for the message.') + group.add_option('-S', '--subject', + dest='subject', + action='store', + help='Allows the user to specify a subject for the message.') + parser.add_option_group(group) + + + (options, recipients) = parser.parse_args(argv[1:]) + + if options.verbose: + logging.getLogger().setLevel(logging.INFO) + # Sanity check user data + if len(recipients) != 0 and (options.stream or options.subject): + parser.error('You cannot specify both a username and a stream/subject.') + if len(recipients) == 0 and (bool(options.stream) != bool(options.subject)): + parser.error('Stream messages must have a subject') + if len(recipients) == 0 and not (options.stream and options.subject): + parser.error('You must specify a stream/subject or at least one recipient.') + + client = api.humbug.init_from_options(options) + + if not options.message: + options.message = sys.stdin.read() + + if options.stream: + message_data = { + 'type': 'stream', + 'content': options.message, + 'subject': options.subject, + 'to': options.stream, + } + else: + message_data = { + 'type': 'private', + 'content': options.message, + 'to': recipients, + } + + if not do_send_message(client, message_data): + return 1 + + +if __name__ == '__main__': + sys.exit(main())