summaryrefslogtreecommitdiff
path: root/distixlib/repo.py
blob: 0f900ec1f57e2219f3c14335f6c923046043962e (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
# Copyright 2014  Lars Wirzenius
#
# 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, either version 3 of the License, or
# (at your option) any later version.
#
# 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/>.
#
# =*= License: GPL-3+ =*=


import os
import sys
import uuid

import distixlib


class Repository(object):

    def __init__(self, dirname):
        self._dirname = dirname

    def invent_new_ticket_id(self):
        return uuid.uuid4().hex

    def create(self, description):
        git = distixlib.Git(self._dirname)
        git.init()
        filenames = self._create_repo_yaml(description)
        self.commit_changes(filenames, 'distix initial commit')

    def _create_repo_yaml(self, description):
        metadata = distixlib.Metadata()
        metadata.add('description', description)

        filename = os.path.join(self._dirname, 'repo.yaml')
        saver = distixlib.MetadataSaver()
        saver.save_to_file(filename, metadata)

        return [filename]

    def require_clean_working_tree(self, stderr=sys.stderr):
        states = {
            '??': 'unknown',
            'M': 'modified',
        }

        git = distixlib.Git(self._dirname)
        dirt = git.find_dirt()
        if dirt:
            stderr.write('ERROR: unclean repository:\n')
            for speck in dirt:
                status = speck[:2].strip()
                filename = speck[3:].strip()
                stderr.write(
                    '  {}: {}\n'.format(states.get(status, status), filename))
            sys.exit(1)

    def get_description(self):
        loader = distixlib.MetadataLoader()
        filename = os.path.join(self._dirname, 'repo.yaml')
        metadata = loader.load_from_file(filename)
        return metadata.get_first_value('description')

    def open_ticket_store(self, relative_dirname):
        joined_pathname = os.path.join(self._dirname, relative_dirname)
        return distixlib.TicketStore(joined_pathname)

    def commit_changes(self, filenames, msg):  # pragma: no cover
        filenames = [
            os.path.relpath(filename, self._dirname)
            for filename in filenames]
        git = distixlib.Git(self._dirname)
        git.add_and_commit(filenames, msg)