build: nicer error message on missing file in waf dist
[samba.git] / buildtools / wafsamba / samba_dist.py
1 # customised version of 'waf dist' for Samba tools
2 # uses git ls-files to get file lists
3
4 import Utils, os, sys, tarfile, stat, Scripting, Logs
5 from samba_utils import *
6
7 dist_dirs = None
8
9 def add_tarfile(tar, fname, abspath):
10     '''add a file to the tarball'''
11     try:
12         tinfo = tar.gettarinfo(name=abspath, arcname=fname)
13     except OSError:
14         Logs.error('Unable to find file %s - missing from git checkout?' % abspath)
15         sys.exit(1)
16     tinfo.uid   = 0
17     tinfo.gid   = 0
18     tinfo.uname = 'root'
19     tinfo.gname = 'root'
20     fh = open(abspath)
21     tar.addfile(tinfo, fileobj=fh)
22     fh.close()
23
24
25 def dist(appname='',version=''):
26     if not isinstance(appname, str) or not appname:
27         # this copes with a mismatch in the calling arguments for dist()
28         appname = Utils.g_module.APPNAME
29         version = Utils.g_module.VERSION
30     if not version:
31         version = Utils.g_module.VERSION
32
33     srcdir = os.path.normpath(os.path.join(os.path.dirname(Utils.g_module.root_path), Utils.g_module.srcdir))
34
35     if not dist_dirs:
36         print('You must use samba_dist.DIST_DIRS() to set which directories to package')
37         sys.exit(1)
38
39     dist_base = '%s-%s' % (appname, version)
40     dist_name = '%s.tar.gz' % (dist_base)
41
42     tar = tarfile.open(dist_name, 'w:gz')
43
44     for dir in dist_dirs.split():
45         if dir.find(':') != -1:
46             destdir=dir.split(':')[1]
47             dir=dir.split(':')[0]
48         else:
49             destdir = '.'
50         absdir = os.path.join(srcdir, dir)
51         git_cmd = [ 'git', 'ls-files', '--full-name', absdir ]
52         try:
53             files = Utils.cmd_output(git_cmd).split()
54         except:
55             print('git command failed: %s' % ' '.join(git_cmd))
56             sys.exit(1)
57         for f in files:
58             abspath = os.path.join(srcdir, f)
59             if dir != '.':
60                 f = f[len(dir)+1:]
61             if destdir != '.':
62                 f = destdir + '/' + f
63             fname = dist_base + '/' + f
64             add_tarfile(tar, fname, abspath)
65
66     tar.close()
67
68     print('Created %s' % dist_name)
69     return dist_name
70
71
72 @conf
73 def DIST_DIRS(dirs):
74     '''set the directories to package, relative to top srcdir'''
75     global dist_dirs
76     if not dist_dirs:
77         dist_dirs = dirs
78
79 Scripting.dist = dist