summaryrefslogtreecommitdiff
path: root/morphlib/plugins/fix_ref_plugin.py
blob: 3dba26ea3612574c56a51905130301974b5b0cef (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
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
# Copyright (C) 2016  Codethink Limited
#
# 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, see <http://www.gnu.org/licenses/>.


import cliapp
import contextlib
import glob
import logging
import os
import shutil

import morphlib


class FixRefPlugin(cliapp.Plugin):

    def enable(self):
        self.app.add_subcommand(
            'fix', self.fix, arg_synopsis='FIX YOUR REFS')
        self.app.add_subcommand(
            'refresh', self.refresh, arg_synopsis='FIX YOUR REFS')
        self.app.add_subcommand(
            'unfix', self.unfix, arg_synopsis='UNFIX YOUR REFS')

    def disable(self):
        pass

    def fix(self, args):
        '''Convert all chunk refs in a system branch to be fixed SHA1s.

        This modifies all git commit references in system and stratum
        morphologies, in the current system branch, to be fixed SHA
        commit identifiers, rather than symbolic branch or tag names.
        This is useful for making sure none of the components in a system
        branch change accidentally.

        Consider the following scenario:

        * The `master` system branch refers to `gcc` using the
          `baserock/morph` ref. This is appropriate, since the main line
          of development should use the latest curated code.

        * You create a system branch to prepare for a release, called
          `TROVE_ID/release/2.0`. The reference to `gcc` is still
          `baserock/morph`.

        * You test everything, and make a release. You deploy the release
          images onto devices, which get shipped to your customers.

        * A new version GCC is committed to the `baserock/morph` branch.

        * Your release branch suddenly uses a new compiler, which may
          or may not work for your particular system at that release.

        To avoid this, you need to _petrify_ all git references
        so that they do not change accidentally. If you've tested
        your release with the GCC release that is stored in commit
        `94c50665324a7aeb32f3096393ec54b2e63bfb28`, then you should
        continue to use that version of GCC, regardless of what might
        happen in the master system branch. If, and only if, you decide
        that a new compiler would be good for your release should you
        include it in your release branch. This way, only the things
        that you change intentionally change in your release branch.

        '''

#        if args:
#            raise cliapp.AppException('morph petrify takes no arguments')


        def fix_ref(repo_cache, chunk):
            repo = chunk.get('repo')
            ref = chunk.get('ref')
            if repo is None or ref is None:
                return chunk
            sha1, _ = repo_cache.resolve_ref_to_commit_and_tree(repo, ref)
            if sha1 != chunk['ref']:
                chunk['ref'] = sha1
                chunk['unpetrify-ref'] = ref
            return chunk

        self._modify_refs(args, fix_ref)


    def refresh(self, args):
        def refresh_ref(repo_cache, chunk):
            repo = chunk.get('repo')
            ref = chunk.get('unpetrify-ref')
            if repo is None or ref is None:
                return chunk
            sha1, _ = repo_cache.resolve_ref_to_commit_and_tree(repo, ref)
            chunk['ref'] = sha1
            return chunk

        self._modify_refs(args, refresh_ref)

    def _modify_refs(self, args, modifier):

        definitions_repo = morphlib.definitions_repo.open(
            '.', search_for_root=True, app=self.app)
        loader = definitions_repo.get_morphology_loader()

        repo_cache = morphlib.util.new_repo_cache(self.app)



#        source_pool_context = definitions_repo.source_pool(
#            definitions_repo.HEAD, filename)
#        with source_pool_context as source_pool:
#            bc = morphlib.buildcommand.BuildCommand(self.app)
#            root = bc.resolve_artifacts(source_pool)
#            print root
        def unset_defaults(morphology):
            '''If a field is equal to its default, delete it.

            The morphology is assumed to be valid.

            '''

            kind = morphology['kind']
            defaults = morphlib.morphloader.MorphologyLoader._static_defaults[
                    kind]
            for key in defaults:
                if key in morphology and morphology[key] == defaults[key]:
                    del morphology[key]
            for spec in morphology['chunks']:
                if 'repo' in spec and spec['repo'] == spec['name']:
                    del spec['repo']
                if 'build-mode' in spec and spec['build-mode'] == \
                        morphlib.morphloader.MorphologyLoader._static_defaults[
                                'chunk']['build-mode']:
                    del spec['build-mode']
                if 'prefix' in spec and spec['prefix'] == \
                        morphlib.morphloader.MorphologyLoader._static_defaults[
                                'chunk']['prefix']:
                    del spec['prefix']
                if 'submodules' in spec and spec['submodules'] == \
                        morphlib.morphloader.MorphologyLoader._static_defaults[
                                'chunk']['submodules']:
                    del spec['submodules']


        if len(args) > 0:
            morphs = []
            for arg in args:
                filename = morphlib.util.sanitise_morphology_path(arg)
                filename = definitions_repo.relative_path(filename, cwd='.')
                morph = definitions_repo.load_morphology_file(filename, loader)
                morphs.append(morph)
        else:
            morphs = definitions_repo.load_all_morphologies()

        for morph in morphs:
            if morph['kind'] != 'stratum':
                continue
            print morph['name']
            for chunk in morph['chunks']:
                chunk = modifier(repo_cache, chunk)

            unset_defaults(morph)
            loader.save_to_file(morph.filename, morph)


    def unfix(self, args):
        '''Reverse the process of petrification.

        This undoes the changes `morph petrify` did.

        '''

        if args:
            raise cliapp.AppException('morph petrify takes no arguments')

        ws = morphlib.workspace.open('.')
        sb = morphlib.sysbranchdir.open_from_within('.')
        loader = morphlib.morphloader.MorphologyLoader()

        morphs = self._load_all_sysbranch_morphologies(sb, loader)

        # Restore the ref for each stratum and chunk
        morphs.unpetrify_all()

        # Write morphologies back out again.
        self._save_dirty_morphologies(loader, sb, morphs.morphologies)