aboutsummaryrefslogtreecommitdiff
path: root/ipsutils/config.py
blob: 756d8fa64d951233aab91a5e769ed0c52402b555 (plain) (blame)
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
# This file is part of ipsutils.

# ipsutils 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, either version 3 of the License, or
# (at your option) any later version.
#
# ipsutils 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 ipsutils.  If not, see <http://www.gnu.org/licenses/>.
import shlex
import string
import StringIO
import collections

class Config(object):
    def __init__(self, ipsfile=''):
        """SPEC file parsing engine.
        """
        super(Config, self).__init__()
        
        #Defines possible FMRI section keywords used by IPS
        key_dict = collections.OrderedDict()
        key_dict['name'] = ''
        key_dict['repackage'] = ''
        key_dict['version'] = ''
        key_dict['release'] = ''
        key_dict['group'] = ''
        key_dict['summary'] = ''
        key_dict['license'] = ''
        key_dict['maintainer'] = ''
        key_dict['upstream_url'] = ''
        key_dict['source_url'] = ''
        key_dict['arch'] = ''
        key_dict['classification'] = ''
        key_dict['description'] = ''

        #Define valid build script sections in SPEC file
        script_dict = collections.OrderedDict()
        script_dict['prep'] =  []
        script_dict['build'] = []
        script_dict['install'] =  []
        script_dict['transforms'] =  []
        
        if not ipsfile:
            self.key_dict = key_dict
            self.script_dict = script_dict
            return

        #Initial key_dict population with raw values taken from ips file
        for key in key_dict:
            for line in file(ipsfile).readlines():
                parts = shlex.split(line)
                if key + ":" in parts:
                    key_dict[key] = parts[1]

        #Drop using the original file in favor of a StringIO buffer
        #Because we need room to breathe without rewriting the file
        #every time we run a build... that's a no-go.
        ipsfile_input = StringIO.StringIO()
        for line in file(ipsfile).readlines():
            t = string.Template(line)
            new_line = t.safe_substitute(key_dict)
            ipsfile_input.writelines(new_line)

        #Turn our input buffer into an output buffer then ditch the original.
        ipsfile_output = StringIO.StringIO(ipsfile_input.getvalue()).readlines()
        ipsfile_input.close()

        # Populate key_dict with expanded values
        for key in key_dict:
            for line in ipsfile_output:
                parts = shlex.split(line)
                if key + ":" in parts:
                    key_dict[key] = parts[1]

        #Parse user defined scripts by section and store them in script_dict
        found_data = False
        code_section = ['%build', '%prep', '%install', '%transforms']

        for section in code_section:
            for line in ipsfile_output:
                if not line:
                    continue
                if line.startswith('#'):
                    continue
                parts = shlex.split(line)
                if '%end' in parts:
                    found_data = False
                if section in parts:
                    found_data = True
                    continue
                if found_data:
                    script_dict[section.strip('%')].append(parts)

        #Assign completed dictionaries to global class scope
        self.key_dict = key_dict
        self.script_dict = script_dict
        if not self.check_keywords():
            exit(1)

    def check_keywords(self):
        """Validate SPEC file's FMRI section
        """
        mandatory = ['arch', 'classification', 'description', 'group',
        'license', 'maintainer', 'name', 'release', 'source_url', 'summary',
        'version']
        #Get list of keys without data
        for k, v in sorted(self.key_dict.items()):
            if k in mandatory and not v:
                print("Mandatory keyword \'{}\' has no value".format(k))
                return False
        return True