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
|
#!/usr/bin/python
# Authors:
# Endi S. Dewata <edewata@redhat.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; version 2 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# Copyright (C) 2014 Red Hat, Inc.
# All rights reserved.
#
from __future__ import absolute_import
import os
from lxml import etree as ET
import pki
import pki.server.upgrade
class ReplaceRESTEasyApplicationClass(pki.server.upgrade.PKIServerUpgradeScriptlet):
def __init__(self):
super(ReplaceRESTEasyApplicationClass, self).__init__()
self.message = 'Replace RESTEasy application class'
self.class_names = {
'ca': 'org.dogtagpki.server.ca.rest.CAApplication',
'kra': 'org.dogtagpki.server.kra.rest.KRAApplication',
'ocsp': 'org.dogtagpki.server.ocsp.rest.OCSPApplication',
'tks': 'org.dogtagpki.server.tks.rest.TKSApplication',
'tps': 'org.dogtagpki.server.tps.rest.TPSApplication'
}
def upgrade_subsystem(self, instance, subsystem):
if subsystem.name not in self.class_names:
return
web_xml = os.path.join(
instance.base_dir,
'webapps', subsystem.name,
'WEB-INF', 'web.xml')
self.backup(web_xml)
doc = ET.parse(web_xml)
root = doc.getroot()
servlet = root.find('servlet[servlet-name="Resteasy"]')
if servlet is None:
raise Exception('Resteasy servlet not found in ' + web_xml)
param = servlet.find('init-param[param-name="javax.ws.rs.Application"]')
if param is None:
raise Exception('Resteasy application parameter not found in ' + web_xml)
value = param.find('param-value')
if value is None:
raise Exception('Resteasy application value not found in ' + web_xml)
value.text = self.class_names[subsystem.name]
doc.write(web_xml)
|
