autopkg.py 11.1 KB
Newer Older
1
# Copyright 2012, 2013 GRNET S.A. All rights reserved.
Vangelis Koukis's avatar
Vangelis Koukis committed
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
#
# Redistribution and use in source and binary forms, with or
# without modification, are permitted provided that the following
# conditions are met:
#
#   1. Redistributions of source code must retain the above
#      copyright notice, this list of conditions and the following
#      disclaimer.
#
#   2. Redistributions in binary form must reproduce the above
#      copyright notice, this list of conditions and the following
#      disclaimer in the documentation and/or other materials
#      provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#
# The views and conclusions contained in the software and
# documentation are those of the authors and should not be
# interpreted as representing official policies, either expressed
# or implied, of GRNET S.A.

import git
import os
import sys
from optparse import OptionParser
38
39
from collections import namedtuple
from sh import mktemp, cd, rm, git_dch, python
Vangelis Koukis's avatar
Vangelis Koukis committed
40

41
from devflow import versioning
Christos Stavrakakis's avatar
Christos Stavrakakis committed
42

Vangelis Koukis's avatar
Vangelis Koukis committed
43
44
45
46
47
48
49
50
51
52
53
try:
    from colors import red, green
except ImportError:
    red = lambda x: x
    green = lambda x: x

print_red = lambda x: sys.stdout.write(red(x) + "\n")
print_green = lambda x: sys.stdout.write(green(x) + "\n")

AVAILABLE_MODES = ["release", "snapshot"]

54
55
56
57
58
59
60
61
branch_type = namedtuple("branch_type", ["default_debian_branch"])
BRANCH_TYPES = {
    "feature": branch_type("debian-develop"),
    "develop": branch_type("debian-develop"),
    "release": branch_type("debian-develop"),
    "master": branch_type("debian"),
    "hotfix": branch_type("debian")}

Christos Stavrakakis's avatar
Christos Stavrakakis committed
62

Christos Stavrakakis's avatar
Christos Stavrakakis committed
63
64
65
66
67
68
DESCRIPTION = """Tool for automatical build of debian packages.

%(prog)s is a helper script for automatic build of debian packages from
repositories that follow the `git flow` development model
<http://nvie.com/posts/a-successful-git-branching-model/>.

69
70
This script must run from inside a clean git repository and will perform the
following steps:
Christos Stavrakakis's avatar
Christos Stavrakakis committed
71
72
73
74
75
76
77
78
    * Clone your repository to a temporary directory
    * Merge the current branch with the corresponding debian branch
    * Compute the version of the new package and update the python
      version files
    * Create a new entry in debian/changelog, using `git-dch`
    * Create the debian packages, using `git-buildpackage`
    * Tag the appropriate branches if in `release` mode

79
80
81
%(prog)s will work with the packages that are declared in `autopkg.conf`
file, which must exist in the toplevel directory of the git repository.

Christos Stavrakakis's avatar
Christos Stavrakakis committed
82
83
84
85
86
87
"""


def print_help(prog):
    print DESCRIPTION % {"prog": prog}

Vangelis Koukis's avatar
Vangelis Koukis committed
88
89
90
91

def main():
    from devflow.version import __version__
    parser = OptionParser(usage="usage: %prog [options] mode",
Christos Stavrakakis's avatar
Christos Stavrakakis committed
92
93
94
95
96
97
                          version="devflow %s" % __version__,
                          add_help_option=False)
    parser.add_option("-h", "--help",
                      action="store_true",
                      default=False,
                      help="show this help message")
Vangelis Koukis's avatar
Vangelis Koukis committed
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
    parser.add_option("-k", "--keep-repo",
                      action="store_true",
                      dest="keep_repo",
                      default=False,
                      help="Do not delete the cloned repository")
    parser.add_option("-b", "--build-dir",
                      dest="build_dir",
                      default=None,
                      help="Directory to store created pacakges")
    parser.add_option("-r", "--repo-dir",
                      dest="repo_dir",
                      default=None,
                      help="Directory to clone repository")
    parser.add_option("-d", "--dirty",
                      dest="force_dirty",
                      default=False,
                      action="store_true",
                      help="Do not check if working directory is dirty")
116
117
118
    parser.add_option("-c", "--config-file",
                      dest="config_file",
                      help="Override default configuration file")
Vangelis Koukis's avatar
Vangelis Koukis committed
119
120
121

    (options, args) = parser.parse_args()

Christos Stavrakakis's avatar
Christos Stavrakakis committed
122
123
124
125
126
    if options.help:
        print_help(parser.get_prog_name())
        parser.print_help()
        return

127
    # Get build mode
Christos Stavrakakis's avatar
Christos Stavrakakis committed
128
129
130
131
132
    try:
        mode = args[0]
    except IndexError:
        raise ValueError("Mode argument is mandatory. Usage: %s"
                         % parser.usage)
Vangelis Koukis's avatar
Vangelis Koukis committed
133
134
135
136
137
138
    if mode not in AVAILABLE_MODES:
        raise ValueError(red("Invalid argument! Mode must be one: %s"
                         % ", ".join(AVAILABLE_MODES)))

    os.environ["GITFLOW_BUILD_MODE"] = mode

139
    # Load the repository
Vangelis Koukis's avatar
Vangelis Koukis committed
140
141
142
143
144
    try:
        original_repo = git.Repo(".")
    except git.git.InvalidGitRepositoryError:
        raise RuntimeError(red("Current directory is not git repository."))

145
    # Check that repository is clean
Christos Stavrakakis's avatar
Christos Stavrakakis committed
146
    toplevel = original_repo.working_dir
Vangelis Koukis's avatar
Vangelis Koukis committed
147
148
149
    if original_repo.is_dirty() and not options.force_dirty:
        raise RuntimeError(red("Repository %s is dirty." % toplevel))

150
151
152
    # Get packages from configuration file
    config_file = options.config_file or os.path.join(toplevel, "autopkg.conf")
    packages = get_packages_to_build(config_file)
Christos Stavrakakis's avatar
Christos Stavrakakis committed
153
    print_green("Will build the following packages:\n" + "\n".join(packages))
Christos Stavrakakis's avatar
Christos Stavrakakis committed
154

155
    # Clone the repo
Christos Stavrakakis's avatar
Christos Stavrakakis committed
156
    repo_dir = options.repo_dir or create_temp_directory("df-repo")
Vangelis Koukis's avatar
Vangelis Koukis committed
157
    repo = original_repo.clone(repo_dir)
Christos Stavrakakis's avatar
Christos Stavrakakis committed
158
    print_green("Cloned repository to '%s'." % repo_dir)
Vangelis Koukis's avatar
Vangelis Koukis committed
159

Christos Stavrakakis's avatar
Christos Stavrakakis committed
160
    # Get current branch name and type and check if it is a valid one
Vangelis Koukis's avatar
Vangelis Koukis committed
161
    branch = repo.head.reference.name
Christos Stavrakakis's avatar
Christos Stavrakakis committed
162
163
164
165
    branch_type = versioning.get_branch_type(branch)

    if branch_type not in BRANCH_TYPES.keys():
        allowed_branches = ", ".join(BRANCH_TYPES.keys())
166
167
168
        raise ValueError("Malformed branch name '%s', cannot classify as"
                         " one of %s" % (branch, allowed_branches))

Christos Stavrakakis's avatar
Christos Stavrakakis committed
169
170
    # Find the debian branch
    debian_branch = "debian-" + branch
171
172
173
    origin_debian = "origin/" + debian_branch
    if not origin_debian in repo.references:
        # Get default debian branch
Christos Stavrakakis's avatar
Christos Stavrakakis committed
174
175
176
177
        default_debian = BRANCH_TYPES[branch_type].default_debian_branch
        origin_debian = "origin/" + default_debian

    repo.git.branch(debian_branch, origin_debian)
178
179
    print_green("Created branch '%s' to track '%s'" % (debian_branch,
                origin_debian))
Vangelis Koukis's avatar
Vangelis Koukis committed
180

181
    # Go to debian branch
Vangelis Koukis's avatar
Vangelis Koukis committed
182
183
184
    repo.git.checkout(debian_branch)
    print_green("Changed to branch '%s'" % debian_branch)

185
    # Merge with starting branch
Vangelis Koukis's avatar
Vangelis Koukis committed
186
    repo.git.merge(branch)
Christos Stavrakakis's avatar
Christos Stavrakakis committed
187
    print_green("Merged branch '%s' into '%s'" % (branch, debian_branch))
Vangelis Koukis's avatar
Vangelis Koukis committed
188

189
    # Compute python and debian version
Vangelis Koukis's avatar
Vangelis Koukis committed
190
    cd(repo_dir)
191
    python_version = versioning.get_python_version()
Christos Stavrakakis's avatar
Christos Stavrakakis committed
192
    debian_version = versioning.\
Christos Stavrakakis's avatar
Christos Stavrakakis committed
193
        debian_version_from_python_version(python_version)
Christos Stavrakakis's avatar
Christos Stavrakakis committed
194
    print_green("The new debian version will be: '%s'" % debian_version)
Vangelis Koukis's avatar
Vangelis Koukis committed
195

Christos Stavrakakis's avatar
Christos Stavrakakis committed
196
197
198
199
    # Tag branch with python version
    branch_tag = python_version
    repo.git.tag(branch_tag, branch)

200
    # Update changelog
Vangelis Koukis's avatar
Vangelis Koukis committed
201
    dch = git_dch("--debian-branch=%s" % debian_branch,
Christos Stavrakakis's avatar
Christos Stavrakakis committed
202
203
204
205
206
                  "--git-author",
                  "--ignore-regex=\".*\"",
                  "--multimaint-merge",
                  "--since=HEAD",
                  "--new-version=%s" % debian_version)
Vangelis Koukis's avatar
Vangelis Koukis committed
207
208
209
    print_green("Successfully ran '%s'" % " ".join(dch.cmd))

    if mode == "release":
210
        call("vim debian/changelog")
211
212
213
214
    else:
        f = open("debian/changelog", 'r+')
        lines = f.readlines()
        lines[0] = lines[0].replace("UNRELEASED", "unstable")
Christos Stavrakakis's avatar
Christos Stavrakakis committed
215
        lines[2] = lines[2].replace("UNRELEASED", "Snapshot build")
216
217
218
        f.seek(0)
        f.writelines(lines)
        f.close()
Christos Stavrakakis's avatar
Christos Stavrakakis committed
219
220
221
222
223
224
225
226

    # Add changelog to INDEX
    repo.git.add("debian/changelog")
    # Commit Changes
    repo.git.commit("-s", "-a", m="Bump version to %s" % debian_version)
    # Tag debian branch
    debian_branch_tag = "debian/" + branch_tag
    repo.git.tag(debian_branch_tag)
Vangelis Koukis's avatar
Vangelis Koukis committed
227

228
229
    # Update the python version files
    # TODO: remove this
Christos Stavrakakis's avatar
Christos Stavrakakis committed
230
    for package in packages:
Vangelis Koukis's avatar
Vangelis Koukis committed
231
232
233
234
235
        # python setup.py should run in its directory
        cd(package)
        package_dir = repo_dir + "/" + package
        res = python(package_dir + "/setup.py", "sdist", _out=sys.stdout)
        print res.stdout
Christos Stavrakakis's avatar
Christos Stavrakakis committed
236
237
        if package != ".":
            cd("../")
Vangelis Koukis's avatar
Vangelis Koukis committed
238
239

    # Add version.py files to repo
240
    call("grep \"__version_vcs\" -r . -l -I | xargs git add -f")
Vangelis Koukis's avatar
Vangelis Koukis committed
241

Christos Stavrakakis's avatar
Christos Stavrakakis committed
242
243
244
    # Create debian packages
    build_dir = options.build_dir or create_temp_directory("df-build")
    print_green("Build directory: '%s'" % build_dir)
Vangelis Koukis's avatar
Vangelis Koukis committed
245

Christos Stavrakakis's avatar
Christos Stavrakakis committed
246
    cd(repo_dir)
247
248
    call("git-buildpackage --git-export-dir=%s --git-upstream-branch=%s"
         " --git-debian-branch=%s --git-export=INDEX --git-ignore-new -sa"
Christos Stavrakakis's avatar
Christos Stavrakakis committed
249
250
         " -i\/version\.py"
         % (build_dir, branch, debian_branch))
Vangelis Koukis's avatar
Vangelis Koukis committed
251

252
    # Remove cloned repo
253
    if mode != 'release' and not options.keep_repo:
Vangelis Koukis's avatar
Vangelis Koukis committed
254
255
256
        print_green("Removing cloned repo '%s'." % repo_dir)
        rm("-r", repo_dir)

Christos Stavrakakis's avatar
Christos Stavrakakis committed
257
258
259
260
261
262
263
264
265
    # Print final info
    info = (("Version", debian_version),
            ("Upstream branch", branch),
            ("Upstream tag", branch_tag),
            ("Debian branch", debian_branch),
            ("Debian tag", debian_branch_tag),
            ("Repository directory", repo_dir),
            ("Packages directory", build_dir))
    print_green("\n".join(["%s: %s" % (name, val) for name, val in info]))
Christos Stavrakakis's avatar
Christos Stavrakakis committed
266

267
    # Print help message
Christos Stavrakakis's avatar
Christos Stavrakakis committed
268
    if mode == "release":
Christos Stavrakakis's avatar
Christos Stavrakakis committed
269
270
271
272
        origin = original_repo.remote().url
        repo.create_remote("original_origin", origin)
        print_green("Created remote 'original_origin' for the repository '%s'"
                    % origin)
Christos Stavrakakis's avatar
Christos Stavrakakis committed
273

Christos Stavrakakis's avatar
Christos Stavrakakis committed
274
275
276
277
278
279
        print_green("To update repositories '%s' and '%s' go to '%s' and run:"
                    % (toplevel, origin, repo_dir))
        for remote in ['origin', 'original_origin']:
            print
            for obj in [debian_branch, branch_tag, debian_branch_tag]:
                print_green("git push %s %s" % (remote, obj))
Vangelis Koukis's avatar
Vangelis Koukis committed
280
281


282
283
284
285
def get_packages_to_build(config_file):
    config_file = os.path.abspath(config_file)
    try:
        f = open(config_file)
286
    except IOError as e:
Christos Stavrakakis's avatar
Christos Stavrakakis committed
287
        raise IOError("Can not access configuration file '%s': %s"
288
                      % (config_file, e.strerror))
289
290
291
292

    lines = [l.strip() for l in f.readlines()]
    l = [l for l in lines if not l.startswith("#")]
    f.close()
Christos Stavrakakis's avatar
Christos Stavrakakis committed
293
294
295
    if not l:
        raise RuntimeError("Configuration file '%s' is empty."
                           " No packages to build." % config_fule)
296
297
298
    return l


299
300
301
302
303
def create_temp_directory(suffix):
    create_dir_cmd = mktemp("-d", "/tmp/" + suffix + "-XXXXX")
    return create_dir_cmd.stdout.strip()


304
305
306
def call(cmd):
    rc = os.system(cmd)
    if rc:
Christos Stavrakakis's avatar
Christos Stavrakakis committed
307
        raise RuntimeError("Command '%s' failed!" % cmd)
308

Vangelis Koukis's avatar
Vangelis Koukis committed
309
310
if __name__ == "__main__":
    sys.exit(main())