summaryrefslogtreecommitdiffstats
path: root/src/sbin/bcfg2-info
blob: 75e48aa2cfbfa87f704012af622a0642ba07725b (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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
#!/usr/bin/python -i
'''This tool loads the Bcfg2 core into an interactive debugger'''
__revision__ = '$Revision$'

import logging, lxml.etree, sys, time, Bcfg2.Logging, Bcfg2.Server.Core

def printTabular(rows):
    '''print data in tabular format'''
    cmax = tuple([max([len(str(row[index])) for row in rows]) + 1 for index in xrange(len(rows[0]))])
    fstring = (" %%-%ss |" * len(cmax)) % cmax
    fstring = ('|'.join([" %%-%ss "] * len(cmax))) % cmax
    print fstring % rows[0]
    print (sum(cmax)  + (len(cmax) * 2) + (len(cmax) - 1)) * '='
    for row in rows[1:]:
        print fstring % row

def getInput():
    '''read commands from stdin'''
    try:
        return raw_input('> ').split(" ")
    except KeyboardInterrupt:
        return ['']

def doBuild(cmd, core):
    '''build client configuration'''
    if len(cmd) == 3:
        output = open(cmd[2], 'w')
        output.write(lxml.etree.tostring(core.BuildConfiguration(cmd[1])))
        output.close()
    else:
        print 'Usage: build <hostname> <output file>'

def doBundles(_, core):
    '''print out group/bundle info'''
    data = [('Group', 'Bundles')]
    groups = core.metadata.groups.keys()
    groups.sort()
    for group in groups:
        data.append((group, ','.join(core.metadata.groups[group][0])))
    printTabular(data)

def doClients(_, core):
    '''print out client info'''
    data = [('Client', 'Profile')]
    clist = core.metadata.clients.keys()
    clist.sort()
    for client in clist:
        data.append((client, core.metadata.clients[client]))
    printTabular(data)

def doHelp(_, dummy):
    '''print out usage info'''
    print 'Commands:'
    print 'build <hostname> <filename> - build config for hostname, writing to filename'
    print 'bundles - print out group/bundle information'
    print 'clients - print out client/profile information'
    print 'debug - shell out to native python interpreter'
    print 'generators - list current versions of generators'
    print 'groups - list groups'
    print 'help - print this text'
    print 'mappings <type*> <name*>- print generator mappings for optional type and name'
    print 'quit'
    print 'update - process pending file events'
    print 'version - print version of this tool'

def doGenerators(_, core):
    '''print out generator info'''
    for generator in core.generators:
        print generator.__version__

def doGroups(_, core):
    '''print out group info'''
    data = [("Groups", "Profile", "Category", "Contains")]
    grouplist = core.metadata.groups.keys()
    grouplist.sort()
    for group in grouplist:
        if group in core.metadata.profiles:
            prof = 'yes'
        else:
            prof = 'no'
        if core.metadata.categories.has_key(group):
            cat = core.metadata.categories[group]
        else:
            cat = ''
        gdata = [grp for grp in core.metadata.groups[group][1]]
        gdata.remove(group)
        data.append((group, prof, cat, ','.join(gdata)))
    printTabular(data)

def doMappings(cmd, core):
    '''print out mapping info'''
    # dump all mappings unless type specified
    data = [('Plugin', 'Type', 'Name')]
    for generator in core.generators:
        if len(cmd) == 1:
            etypes = generator.Entries.keys()
        elif len(cmd) > 1:
            etypes = [cmd[1]]
        if len(cmd) == 3:
            interested = [(etype, [cmd[2]]) for etype in etypes]
        else:
            interested = [(etype, generator.Entries[etype].keys()) for etype in etypes
                          if generator.Entries.has_key(etype)]
        if [etype for (etype, names) in interested
            if generator.Entries.has_key(etype) and [name for name in names
                                                     if generator.Entries[etype].has_key(name)]]:
            for (etype, names) in interested:
                for name in names:
                    if generator.Entries.has_key(etype) and generator.Entries[etype].has_key(name):
                        data.append((generator.__name__, etype, name))
    printTabular(data)

def doQuit(cmd, core):
    '''exit program'''
    raise SystemExit, 0

def doUpdate(_, core):
    '''Process pending fs events'''
    core.fam.Service()

def doVersion(_, dummy):
    '''print out code version'''
    print __revision__
        
if __name__ == '__main__':
    Bcfg2.Logging.setup_logging('bcfg2-info', to_syslog=False)
    logger = logging.getLogger('bcfg2-info')
    dispatch = {'build': doBuild, 'bundles': doBundles, 'clients': doClients,
                'generators': doGenerators, 'groups': doGroups,
                'help': doHelp, 'mappings': doMappings, 'quit': doQuit,
                'update': doUpdate, 'version': doVersion}
    if '-c' in sys.argv:
        cfile = sys.argv[-1]
    else:
        cfile = '/etc/bcfg2.conf'
    try:
        bcore = Bcfg2.Server.Core.Core({}, cfile)
    except Bcfg2.Server.Core.CoreInitError, msg:
        print "Core load failed because %s" % msg
        raise SystemExit, 1
    for i in range(25):
        bcore.fam.Service()
        time.sleep(0.5)
    ucmd = getInput()
    while True:
        if ucmd[0] == 'debug':
            break
        else:
            if dispatch.has_key(ucmd[0]):
                try:
                    dispatch[ucmd[0]](ucmd, bcore)
                except SystemExit, code:
                    raise SystemExit, code
                except:
                    logger.error("command failure", exc_info=1)
            else:
                print "Unknown command %s" % ucmd[0]
        ucmd = getInput()