summaryrefslogtreecommitdiff
blob: c763c2169339a87b957bc5f57c2f99af831fb470 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#!/usr/bin/env python
# Copyright 1999-2004 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
# $Header: /var/cvsroot/gentoo-x86/sys-devel/distcc/files/3.0/distcc-config,v 1.2 2008/10/25 06:39:17 matsuu Exp $

import os, re, signal, sys, commands, pwd
from string import rstrip

options=[
	'--get-hosts',
	'--set-hosts',
	'--get-verbose',
	'--set-verbose',
	'--get-log',
	'--set-log',
	'--help',
	'--get-env',
	'--set-env'
]

tmpcmdline=sys.argv[1:]
cmdline=[]
envfile = '/etc/env.d/02distcc'

def exithandler(foo,bar):
	os.kill(0,signal.SIGKILL)
	sys.exit(1)

signal.signal(signal.SIGINT,exithandler)

def isroot(ret=0):
	if os.getuid() != 0:
		if ret == 0:
			print '!!!',sys.argv[:1][0],tmpcmdline[0],'must be run as root'
			sys.exit(1)
		else:
			retval = 0
	else:
		retval = 1
	return retval

def writeenv(var,value):
	isroot()
	distcc_env    = []
	distcc_env    = open(envfile, 'r').readlines()
	distcc_env_new = open(envfile, 'w')
	for i in range(len(distcc_env)):
		if re.compile(var+'="(.*)"').match(distcc_env[i]):
			distcc_env[i] = var+'="'+value+'"\n'
		distcc_env_new.write(distcc_env[i])
	#print 'Set',var,'to:',value
	os.system('/usr/sbin/env-update')
	print 'If you want to use these new settings in an existing shell,'
	print 'you need to "source /etc/profile" to get the changes.'

def readenv(var):
	distcc_env = open(envfile, 'r').read()
	match = re.compile(var+'="(.*)"').search(distcc_env)
	if match:
		print var+'='+match.group(1)
	else:
		print var,'not set.'

def createdistccdir(dir):
	if not os.path.exists(dir):
		os.mkdir(dir)
		os.chmod(dir, 0755)

for x in tmpcmdline:
	if not x:
		continue
	if x[0:2]=="--":
		if not x in options:
			print "!!! Error:",x,"is an invalid option."
			sys.exit(1)
	else:
		cmdline = x

if '--get-hosts' in tmpcmdline:
	HOSTS_ENV  = os.environ.get('DISTCC_HOSTS')
	HOSTS_HOME = os.environ.get('HOME')+'/hosts'
	if HOSTS_ENV:
		print HOSTS_ENV
	elif os.path.isfile(HOSTS_HOME) and os.path.getsize(HOSTS_HOME) != 0:
		print HOSTS_HOME
	elif os.path.exists('/etc/distcc/hosts'):
		print rstrip(open('/etc/distcc/hosts', 'r').read())
	else:
		print 'No configuration file found. Setup your hosts with --set-hosts.'
elif '--set-hosts' in tmpcmdline:
	if isroot(1):
		PATH = '/etc/distcc'
	else:
		PATH = os.environ.get('HOME')
	createdistccdir(PATH)
	open(PATH+'/hosts', 'w').write(cmdline + '\n')
elif '--get-verbose' in tmpcmdline:
	readenv('DISTCC_VERBOSE')
elif '--set-verbose' in tmpcmdline:
	writeenv('DISTCC_VERBOSE',tmpcmdline[1])
elif '--get-log' in tmpcmdline:
	readenv('DISTCC_LOG')
elif '--set-log' in tmpcmdline:
	writeenv('DISTCC_LOG',tmpcmdline[1])
elif '--get-env' in tmpcmdline:
	if len(tmpcmdline) == 1:
		print rstrip(open(envfile, 'r').read())
	elif len(tmpcmdline) == 2:
		readenv(tmpcmdline[1])
	else:
		print '!!! Error: Specify only one variable.'
elif '--set-env' in tmpcmdline:
	if len(tmpcmdline) > 2 and len(tmpcmdline) <= 3:
		isroot()
		writeenv(tmpcmdline[1],tmpcmdline[2])
	else:
		print '!!! Error: Awaiting two parameters.'
else:
	print 'Usage: /usr/bin/distcc-config --set-hosts DISTCC_HOSTS | --get-hosts'
	print '       /usr/bin/distcc-config --set-verbose { 0 | 1 }  | --get-verbose'
	print '       /usr/bin/distcc-config --set-log FILE           | --get-log'
	print '       /usr/bin/distcc-config --set-env VARIABLE VALUE | --get-env [VARIABLE]'