#!/usr/bin/python

# Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

#
# Call the Simple.Connect API.
#

import gobject
import dbus
import dbus.mainloop.glib
import mm
import sys
from optparse import OptionParser

def state_changed(old_state, new_state, reason, path):
	if new_state == 90:
		print "Connection established for", path
	else:
		print "Connection failed for", path
	mainloop.quit()

usage='%prog [options]'

parser = OptionParser(usage=usage)
parser.add_option('--apn', type='string',
		  dest='apn',
		  default=None,
		  help='apn to supply to the connect operation')
parser.add_option('--username', type='string',
		  dest='username',
		  default=None,
		  help='user name to supply to the connect operation')
parser.add_option('--password', type='string',
		  dest='password',
		  default=None,
		  help='password to supply to the connect operation')
parser.add_option('--number', type='string',
		  dest='number',
		  default=None,
		  help='phone number to supply to the connect operation')
parser.add_option('--carrier', type='choice',
		  dest='carrier',
		  choices=['tmobile', 'sprint', 'att', 'verizon'],
		  default=None,
		  help='Specifies the carrier. Provided as a convenience '
		  'to avoid having to supply an APN or phone number.')
parser.add_option('-w', '--wait',
		  action='store_true', dest='wait',
		  default=False,
		  help='Wait until connection completes or fails (default: %default)')

(options, args) = parser.parse_args()
if len(args) > 1:
        parser.error("incorrect number of arguments")

props=dict()

carriers = dict(
	att=dict(number="*99#", apn='broadband'),
	tmobile=dict(number="*99#", apn='epc.tmobile.com'),
	sprint=dict(number="#777"),
	verizon=dict(number="#777"),
)

if len(args) < 1:
	modem_pattern = ''
else:
	modem_pattern = args[0]

if options.carrier:
	try:
		props = carriers[options.carrier]
	except KeyError:
		parser.error("Unknown carrier '%s'." % options.carrier)
		sys.exit(1)

if options.apn:
	props['apn'] = options.apn
if options.username:
	props['username'] = options.username
if options.password:
	props['password'] = options.password
if options.number:
	props['number'] = options.number

dbus.mainloop.glib.DBusGMainLoop(set_as_default=True)

manager, path = mm.PickOneModem(modem_pattern)
modem = manager.CdmaModem(path)

print "Connecting", path

if options.wait:
	bus = dbus.SystemBus()
	bus.add_signal_receiver(state_changed,
			 bus_name=manager.provider,
			 signal_name="StateChanged",
			 path_keyword="path")

modem = manager.SimpleModem(path)
try:
	modem.Connect(props)
	options.wait = False
except dbus.exceptions.DBusException, e:
	if e._dbus_error_name != "org.chromium.ModemManager.Error.OperationInitiated":
		raise e

if options.wait:
	mainloop = gobject.MainLoop()
	mainloop.run()