summaryrefslogtreecommitdiffstats
path: root/modules/sourceball.py
blob: d66277e59e5bf2c56d6049fc2eeb2cbfb2ba2f77 (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
# Fedora Developer Shell
#
# 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 Library 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
#
# Authors: Yaakov M. Nemoy <ynemoy@redhat.com>
#
from __future__ import with_statement

import tarfile

from os import makedirs, getcwd
from os.path import abspath, split, basename, join
from shutil import copytree
from subprocess import Popen, PIPE
from tempfile import mkdtemp

from base.base import log
from base.exceptions import ExecutionException
from base.util import pwd, copy, move

from modules.dirfactory import DirFactory
from modules.package import PackageSource

class SourceBall(PackageSource):
    '''a type of package that is a single sourceball, a spec file, and some patches'''
    _type = 'sourceball'
    def __init__(self, name=None, tarball=''):
        if tarball:
            tmp_dir = mkdtemp()
            with pwd(tmp_dir):
                sourceball_name = copy(tarball, split(tarball)[1])
                sourceball = tarfile.open(sourceball_name)
            extract_dir = min([(x.name, x) for x in sourceball])[0]
            extract_dir = basename(abspath(extract_dir))
            if name and not name == extract_dir:
                raise ExecutionException("tarball is not target directory")
            if not name:
                name = extract_dir
        super(SourceBall, self).__init__(name)
        if tarball:
            with pwd(self.dir):
                sourceball.extractall()
            with pwd(self.branch_dir):
                sourceball.extractall()
                move(self.name, self.orig_dir(self.name))
            with pwd(self.parent):
                move(join(tmp_dir, sourceball_name), sourceball_name)
        self.cfg['sourceball'] = sourceball_name

    def make_dir(self, dir):
        super(PackageSource, self).make_dir(dir)
        with pwd(dir):
            makedirs('.pkg_src')

    @property
    def branch(self):
        return join('.pkg_src', 'branches')

    @property
    def branch_dir(self):
        return join(self.dir, self.branch)

    @property
    def sourceball(self):
        return self.cfg['sourceball']
    
    @property
    def sourceball_loc(self):
        return join(self.parent, self.sourceball)

    def orig_dir(self, dir):
        '''where is the original source kept

        use for making patches against modified source'''
        return dir + '_orig'

    def source(self, *args):
        '''gives source directory

        first parameter, if 'orig' gives the original source, otherwise
        you get the modified source
        '''
        if args[0] == 'orig':
            return join(self.branch, self.orig_dir(self.name)))
        else:
            return self.name

    def source_dir(self, *args):
        return join(self.dir, self.source(*args))


__all__ = ['SourceBall']