105
+ − 1 #!/usr/bin/env python2
+ − 2
+ − 3 # This script set account domain in sat.conf if not already set
+ − 4 # if not set, domain is got from prosody container or DOMAIN environment variable
+ − 5
+ − 6 import os , os.path , xmlrpclib , ConfigParser , socket , subprocess
+ − 7 from sat.core.constants import Const as C
+ − 8 from sat.tools import config as sat_config
+ − 9
+ − 10 SECTION = "plugin account"
+ − 11 OPTION = "new_account_domain"
+ − 12 CONFIG_PATH = "/home/sat/.config/sat/sat.conf"
+ − 13
+ − 14 try :
+ − 15 os . makedirs ( os . path . dirname ( CONFIG_PATH ))
+ − 16 except OSError :
+ − 17 pass
+ − 18
+ − 19 config = ConfigParser . SafeConfigParser ()
+ − 20 config . read ( C . CONFIG_FILES )
+ − 21 domain = sat_config . getConfig ( config , SECTION , OPTION )
+ − 22
+ − 23 if domain is None :
+ − 24 os . getenv ( "DOMAIN" )
+ − 25 if domain is None :
+ − 26 proxy = xmlrpclib . ServerProxy ( "http://prosody:9999/" )
+ − 27 try :
+ − 28 if "prosody" not in open ( "/etc/hosts" ) . read ():
+ − 29 raise socket . gaierror # this avoid waiting for timeout if prosody is not linked
+ − 30 domain = proxy . getenv ( "DOMAIN" )
+ − 31 except socket . gaierror :
+ − 32 print "No prosody container connected or known domain, using \" localhost \" for new domains"
+ − 33 domain = "localhost"
+ − 34
+ − 35 config = ConfigParser . SafeConfigParser ()
+ − 36 config . readfp ( open ( CONFIG_PATH , "a+" ))
+ − 37
+ − 38 try :
+ − 39 config . add_section ( SECTION )
+ − 40 except ConfigParser . DuplicateSectionError :
+ − 41 pass
+ − 42
+ − 43 config . set ( SECTION , OPTION , domain )
+ − 44 config . write ( open ( CONFIG_PATH , "w" ))
+ − 45
+ − 46 subprocess . call ([ "add_host" , domain , "prosody" ])
+ − 47
+ − 48 for subdomain in ( "chat" , "proxy" , "upload" , "pubsub" , "salut" ):
+ − 49 subprocess . call ([ "add_host" , "{}.{}" . format ( subdomain , domain ), "prosody" ])
