#!/usr/bin/python
# -*- coding: utf-8 -*-

import sys, time
from se3DB        import *
from se3LDAP      import *
from se3Logon     import *
from se3Templates import *
from se3Utils     import *
from se3Reg       import *

nbArgs = len (sys.argv)

if nbArgs != 4:
    print "Usage: logon.py user computer arch"
    sys.exit (1)

# Get cmd line args
user = sys.argv[1]
computer = sys.argv[2]
arch     = sys.argv[3]

#########################
# Get se3 configuration #
#########################
se3db = se3DB (user, arch)
rest = []
slapdIp         = se3db.getValue ("ldap_server")
slapdPort       = se3db.getValue ("ldap_port")
adminRdn        = se3db.getValue ("adminRdn")
adminPw         = se3db.getValue ("adminPw")
baseDn          = se3db.getValue ("ldap_base_dn")
peopleRdn       = se3db.getValue ("peopleRdn")
computersRdn    = se3db.getValue ("computersRdn")
parcsRdn        = se3db.getValue ("parcsRdn")
groupsRdn       = se3db.getValue ("groupsRdn")
path2BatFiles   = se3db.getValue ("path2BatFiles")
path2Templates  = se3db.getValue ("path2Templates")
xpPass          = se3db.getValue ("xppass")

#########################
# Get LDAP informations #
#########################
ldap = se3LDAP (slapdIp, slapdPort, adminRdn, adminPw,\
                baseDn, peopleRdn, computersRdn, groupsRdn, parcsRdn)

groups = []
parcs  = []
# Lowercase parcs/groups  
for group in ldap.getUserGroups (user):
    groups.append (group.lower ())
for parc in ldap.getComputerParcs (computer):
    parcs.append (parc.lower ())

master   = ldap.getSe3Master ()

del ldap


###########################
# Create needed directories
###########################
if not os.access ("%s/machine" % path2BatFiles, os.F_OK):
    os.mkdir ("%s/machine" % path2BatFiles)
if not os.access ("%s/machine/%s" % (path2BatFiles, computer), os.F_OK):
    os.mkdir ("%s/machine/%s" % (path2BatFiles, computer))

templates=["base"] + groups + parcs + [computer] + [user] + ["%s@@%s" % (user, computer)]
for parc in parcs:
    templates += (["%s@@%s" % (user, parc)])

for group in groups:
    templates += (["%s@@%s" % (group, computer)])
    for parc in parcs:
        templates += (["%s@@%s" % (group, parc)])

#######################
# User logon creation #
#######################
logon = se3Logon (path2BatFiles, path2Templates, user, computer, master, arch)

logon.addTemplateslogon (templates)

##############################
# User restrictions creation #
##############################

restrictions = se3db.getRestrictions (templates)

regVBS = se3Reg (path2BatFiles, computer)
regVBS.addRest (restrictions )
del regVBS

##########################
# Desktop and Start Menu #
##########################
template = se3Templates (path2Templates, path2BatFiles, user)

template.cleanDesktop ()
template.cleanStartMenu ()

for group in groups:
    template.createDesktop (["%s@@%s" % (group, computer)])
    template.createStartMenu (["%s@@%s" % (group, computer)])
    for parc in parcs:
        template.createDesktop (["%s@@%s" % (group, parc)])
        template.createStartMenu (["%s@@%s" % (group, parc)])

template.createDesktop (["%s@@%s" % (user, computer)])
template.createStartMenu (["%s@@%s" % (user, computer)])

for parc in parcs:
    template.createDesktop (["%s@@%s" % (user, parc)])
    template.createStartMenu (["%s@@%s" % (user, parc)])

template.createDesktop ([computer])
template.createStartMenu ([computer])

template.createDesktop ([user])
template.createStartMenu ([user])

template.createDesktop (parcs)
template.createStartMenu (parcs)

template.createDesktop (groups)
template.createStartMenu (groups)

template.createDesktop (["base"])
template.createStartMenu (["base"])

del se3db
del template
del parcs
del groups

sys.exit (0)