#!/usr/bin/env python2.4
#
# Copyright (c) 2003-2006 Andrea Luzzardi <scox@sig11.org>
#
# This file is part of the pam_usb project. pam_usb is free software;
# you can redistribute it and/or modify it under the terms of the GNU General
# Public License version 2, as published by the Free Software Foundation.
#
# pam_usb is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
# details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Software Foundation, Inc., 59 Temple
# Place, Suite 330, Boston, MA  02111-1307  USA


import dbus
import sys
import os
from xml.dom import minidom

class Device:
    def __init__(self, udi):
        self.__udi = udi
        self.__findStorageDevice()
        deviceObj = bus.get_object('org.freedesktop.Hal',
                                   udi)
        deviceProperties = deviceObj.GetAllProperties(
            dbus_interface = 'org.freedesktop.Hal.Device')
        self.vendor = deviceProperties['usb_device.vendor']
        self.product = deviceProperties['info.product']
        self.serialNumber = deviceProperties['usb_device.serial']

    def __findStorageDevice(self):
        for child in halManager.FindDeviceByCapability('storage'):
            obj = bus.get_object('org.freedesktop.Hal',
                                 child)
            properties = obj.GetAllProperties(
                dbus_interface = 'org.freedesktop.Hal.Device')
            if properties['storage.physical_device'] == self.__udi + '_if0':
                self.__storageUdi = child
                return
        raise Exception, '%s is not a storage device.' % self.__udi

    def __repr__(self):
        return "%s %s (%s)" % (self.vendor, self.product, self.serialNumber)

    def volumes(self):
        vols = []
        for volume in halManager.FindDeviceByCapability('volume'):
            deviceObj = bus.get_object('org.freedesktop.Hal',
                                       volume)
            deviceProperties = deviceObj.GetAllProperties(
                dbus_interface = 'org.freedesktop.Hal.Device')
            if deviceProperties['block.storage_device'] != self.__storageUdi:
                continue
            vols.append({'uuid' : deviceProperties['volume.uuid'],
                   'device' : deviceProperties['block.device']})
        return vols

def listOptions(question, options, force = False):
    if force == False and len(options) == 1:
        return 0
    while True:
        try:
            print question
            for i in range(len(options)):
                print "%d) %s" % (i, options[i])
            print
            sys.stdout.write('[%s-%s]: ' % (0, len(options) - 1))
            optionId = int(sys.stdin.readline())
            print
            if optionId not in range(len(options)):
                raise Exception
            return optionId
        except KeyboardInterrupt: sys.exit()
        except Exception: pass
        else: break

def addDevice(options):
    devices = []

    for udi in halManager.FindDeviceStringMatch('info.bus', 'usb_device'):
        try:
            devices.append(Device(udi))
        except Exception, ex:
            pass

    if len(devices) == 0:
        print 'No devices detected.'
        sys.exit()

    device = devices[listOptions("Please select the device you wish to add.",
                                 devices, force = options['force'])]
    volumes = device.volumes()
    volume = volumes[listOptions("Which volume would you like to use for " \
                                 "storing data ?",
                                 ["%s (UUID: %s)" % (volume['device'],
                                                     volume['uuid'])
                                  for volume in volumes],
                                 force = options['force'])]

    print 'Name\t\t: %s' % options['deviceName']
    print 'Vendor\t\t: %s' % device.vendor
    print 'Model\t\t: %s' % device.product
    print 'Serial\t\t: %s' % device.serialNumber
    print 'Volume UUID\t: %s (%s)' % (volume['uuid'], volume['device'])
    print
    print 'Save device to %s ?' % options['configFile']

    sys.stdout.write('[y/n] ')
    if sys.stdin.readline().strip() != 'y':
        sys.exit(1)

    doc = minidom.parse(options['configFile'])
    devs = doc.getElementsByTagName('devices')
    dev = doc.createElement('device')
    dev.attributes['id'] = options['deviceName']
    devs[0].appendChild(dev)

    for name, value in (('vendor', device.vendor),
                        ('model', device.product),
                        ('serial', device.serialNumber),
                        ('volume_uuid', volume['uuid'])):
        e = doc.createElement(name)
        t = doc.createTextNode(value)
        e.appendChild(t)
        dev.appendChild(e)

    f = open(options['configFile'], 'w')
    f.write(doc.toxml())
    f.close()
    print 'Done.'

def usage():
    print 'Usage: %s [--config file] --add-device <name> [--no-autodetect]' % os.path.basename(__file__)
    sys.exit(1)

import getopt

try:
    opts, args = getopt.getopt(sys.argv[1:], "ha:nc:",
                               ["help", "add-device=", "no-autodetect",
                                "config="])
except getopt.GetoptError:
    usage()

if len(args) != 0:
    usage()

options = { 'force' : False, 'deviceName' : None,
            'configFile' : '/etc/pamusb/pusb.conf' }

for o, a in opts:
    if o in ("-h", "--help"):
        usage()
    if o in ("-a", "--add-device"):
        options['deviceName'] = a
    if o in ("-n", "--no-autodetect"):
        options['force'] = True
    if o in ("-c", "--config"):
        options['configFile'] = a

if options['deviceName'] is None:
    usage()

bus = dbus.SystemBus()
halService = bus.get_object('org.freedesktop.Hal',
                            '/org/freedesktop/Hal/Manager')
halManager = dbus.Interface(halService, 'org.freedesktop.Hal.Manager')

addDevice(options)