#!/usr/bin/python
import sys
from ldaptor.protocols.ldap import ldapclient, ldif, distinguishedname, ldapconnector, ldapsyntax
from ldaptor.protocols import pureber, pureldap
from ldaptor import usage, ldapfilter, config
from twisted.internet import protocol, reactor, defer

def _cbSearch(obj):
    a={}
    for attr, vals in obj.items():
        attr=str(attr)
        assert not a.has_key(attr)
        a[attr]=map(str, vals)

    print ':'.join((
        a['uid'][0],
        'x',
        a['uidNumber'][0],
        a['gidNumber'][0],
        a.get('gecos', a.get('cn', ['']))[0],
        a['homeDirectory'][0],
        a.get('loginShell', [''])[0],
        ))

def search(client, baseDN, filt):
    e=ldapsyntax.LDAPEntry(client=client, dn=baseDN)
    d = e.search(filterObject=filt,
                 attributes=['uid',
                             'uidNumber',
                             'gidNumber',
                             'gecos',
                             'cn',
                             'homeDirectory',
                             'loginShell',
                             ],
                 callback=_cbSearch)
    return d

exitStatus=0

def error(fail):
    print >>sys.stderr, 'fail:', fail.getErrorMessage()
    global exitStatus
    exitStatus=1

def main(cfg, filter_text):
    try:
        baseDN = cfg.getBaseDN()
    except config.MissingBaseDNError, e:
        print >>sys.stderr, "%s: %s." % (sys.argv[0], e)
        sys.exit(1)

    filt = ldapfilter.parseFilter('(objectClass=posixAccount)')
    if filter_text is not None:
        filt = pureldap.LDAPFilter_and([
            filt,
            ldapfilter.parseFilter(filter_text)])

    c = ldapconnector.LDAPClientCreator(reactor,
                                        ldapclient.LDAPClient)
    d = c.connectAnonymously(dn=baseDN,
                             overrides=cfg.getServiceLocationOverrides())
    d.addCallback(search, baseDN, filt)
    d.addErrback(error)
    d.addBoth(lambda x: reactor.stop())

    reactor.run()
    sys.exit(exitStatus)

class MyOptions(usage.Options, usage.Options_service_location, usage.Options_base_optional):
    """LDAPtor command line search utility"""

    def parseArgs(self, filter=None):
        self.opts['filter'] = filter

if __name__ == "__main__":
    try:
        opts = MyOptions()
        opts.parseOptions()
    except usage.UsageError, ue:
        sys.stderr.write('%s: %s\n' % (sys.argv[0], ue))
        sys.exit(1)

    cfg = config.LDAPConfig(baseDN=opts['base'],
                            serviceLocationOverrides=opts['service-location'])
    main(cfg, opts['filter'])
